httpUtil.js 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155
  1. "use strict";
  2. /**
  3. * @license
  4. * Copyright 2023 Google Inc.
  5. * SPDX-License-Identifier: Apache-2.0
  6. */
  7. var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
  8. if (k2 === undefined) k2 = k;
  9. var desc = Object.getOwnPropertyDescriptor(m, k);
  10. if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
  11. desc = { enumerable: true, get: function() { return m[k]; } };
  12. }
  13. Object.defineProperty(o, k2, desc);
  14. }) : (function(o, m, k, k2) {
  15. if (k2 === undefined) k2 = k;
  16. o[k2] = m[k];
  17. }));
  18. var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
  19. Object.defineProperty(o, "default", { enumerable: true, value: v });
  20. }) : function(o, v) {
  21. o["default"] = v;
  22. });
  23. var __importStar = (this && this.__importStar) || function (mod) {
  24. if (mod && mod.__esModule) return mod;
  25. var result = {};
  26. if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
  27. __setModuleDefault(result, mod);
  28. return result;
  29. };
  30. Object.defineProperty(exports, "__esModule", { value: true });
  31. exports.getText = exports.getJSON = exports.downloadFile = exports.httpRequest = exports.headHttpRequest = void 0;
  32. const fs_1 = require("fs");
  33. const http = __importStar(require("http"));
  34. const https = __importStar(require("https"));
  35. const url_1 = require("url");
  36. const proxy_agent_1 = require("proxy-agent");
  37. function headHttpRequest(url) {
  38. return new Promise(resolve => {
  39. const request = httpRequest(url, 'HEAD', response => {
  40. // consume response data free node process
  41. response.resume();
  42. resolve(response.statusCode === 200);
  43. }, false);
  44. request.on('error', () => {
  45. resolve(false);
  46. });
  47. });
  48. }
  49. exports.headHttpRequest = headHttpRequest;
  50. function httpRequest(url, method, response, keepAlive = true) {
  51. const options = {
  52. protocol: url.protocol,
  53. hostname: url.hostname,
  54. port: url.port,
  55. path: url.pathname + url.search,
  56. method,
  57. headers: keepAlive ? { Connection: 'keep-alive' } : undefined,
  58. auth: (0, url_1.urlToHttpOptions)(url).auth,
  59. agent: new proxy_agent_1.ProxyAgent(),
  60. };
  61. const requestCallback = (res) => {
  62. if (res.statusCode &&
  63. res.statusCode >= 300 &&
  64. res.statusCode < 400 &&
  65. res.headers.location) {
  66. httpRequest(new url_1.URL(res.headers.location), method, response);
  67. // consume response data to free up memory
  68. // And prevents the connection from being kept alive
  69. res.resume();
  70. }
  71. else {
  72. response(res);
  73. }
  74. };
  75. const request = options.protocol === 'https:'
  76. ? https.request(options, requestCallback)
  77. : http.request(options, requestCallback);
  78. request.end();
  79. return request;
  80. }
  81. exports.httpRequest = httpRequest;
  82. /**
  83. * @internal
  84. */
  85. function downloadFile(url, destinationPath, progressCallback) {
  86. return new Promise((resolve, reject) => {
  87. let downloadedBytes = 0;
  88. let totalBytes = 0;
  89. function onData(chunk) {
  90. downloadedBytes += chunk.length;
  91. progressCallback(downloadedBytes, totalBytes);
  92. }
  93. const request = httpRequest(url, 'GET', response => {
  94. if (response.statusCode !== 200) {
  95. const error = new Error(`Download failed: server returned code ${response.statusCode}. URL: ${url}`);
  96. // consume response data to free up memory
  97. response.resume();
  98. reject(error);
  99. return;
  100. }
  101. const file = (0, fs_1.createWriteStream)(destinationPath);
  102. file.on('finish', () => {
  103. return resolve();
  104. });
  105. file.on('error', error => {
  106. return reject(error);
  107. });
  108. response.pipe(file);
  109. totalBytes = parseInt(response.headers['content-length'], 10);
  110. if (progressCallback) {
  111. response.on('data', onData);
  112. }
  113. });
  114. request.on('error', error => {
  115. return reject(error);
  116. });
  117. });
  118. }
  119. exports.downloadFile = downloadFile;
  120. async function getJSON(url) {
  121. const text = await getText(url);
  122. try {
  123. return JSON.parse(text);
  124. }
  125. catch {
  126. throw new Error('Could not parse JSON from ' + url.toString());
  127. }
  128. }
  129. exports.getJSON = getJSON;
  130. function getText(url) {
  131. return new Promise((resolve, reject) => {
  132. const request = httpRequest(url, 'GET', response => {
  133. let data = '';
  134. if (response.statusCode && response.statusCode >= 400) {
  135. return reject(new Error(`Got status code ${response.statusCode}`));
  136. }
  137. response.on('data', chunk => {
  138. data += chunk;
  139. });
  140. response.on('end', () => {
  141. try {
  142. return resolve(String(data));
  143. }
  144. catch {
  145. return reject(new Error('Chrome version not found'));
  146. }
  147. });
  148. }, false);
  149. request.on('error', err => {
  150. reject(err);
  151. });
  152. });
  153. }
  154. exports.getText = getText;
  155. //# sourceMappingURL=httpUtil.js.map