123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129 |
- /**
- * Copyright 2023 Google Inc. All rights reserved.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- import { createWriteStream } from 'fs';
- import * as http from 'http';
- import * as https from 'https';
- import { URL, urlToHttpOptions } from 'url';
- import { ProxyAgent } from 'proxy-agent';
- export function headHttpRequest(url) {
- return new Promise(resolve => {
- const request = httpRequest(url, 'HEAD', response => {
- resolve(response.statusCode === 200);
- }, false);
- request.on('error', () => {
- resolve(false);
- });
- });
- }
- export function httpRequest(url, method, response, keepAlive = true) {
- const options = {
- protocol: url.protocol,
- hostname: url.hostname,
- port: url.port,
- path: url.pathname + url.search,
- method,
- headers: keepAlive ? { Connection: 'keep-alive' } : undefined,
- auth: urlToHttpOptions(url).auth,
- agent: new ProxyAgent(),
- };
- const requestCallback = (res) => {
- if (res.statusCode &&
- res.statusCode >= 300 &&
- res.statusCode < 400 &&
- res.headers.location) {
- httpRequest(new URL(res.headers.location), method, response);
- }
- else {
- response(res);
- }
- };
- const request = options.protocol === 'https:'
- ? https.request(options, requestCallback)
- : http.request(options, requestCallback);
- request.end();
- return request;
- }
- /**
- * @internal
- */
- export function downloadFile(url, destinationPath, progressCallback) {
- return new Promise((resolve, reject) => {
- let downloadedBytes = 0;
- let totalBytes = 0;
- function onData(chunk) {
- downloadedBytes += chunk.length;
- progressCallback(downloadedBytes, totalBytes);
- }
- const request = httpRequest(url, 'GET', response => {
- if (response.statusCode !== 200) {
- const error = new Error(`Download failed: server returned code ${response.statusCode}. URL: ${url}`);
- // consume response data to free up memory
- response.resume();
- reject(error);
- return;
- }
- const file = createWriteStream(destinationPath);
- file.on('finish', () => {
- return resolve();
- });
- file.on('error', error => {
- return reject(error);
- });
- response.pipe(file);
- totalBytes = parseInt(response.headers['content-length'], 10);
- if (progressCallback) {
- response.on('data', onData);
- }
- });
- request.on('error', error => {
- return reject(error);
- });
- });
- }
- export async function getJSON(url) {
- const text = await getText(url);
- try {
- return JSON.parse(text);
- }
- catch {
- throw new Error('Could not parse JSON from ' + url.toString());
- }
- }
- export function getText(url) {
- return new Promise((resolve, reject) => {
- const request = httpRequest(url, 'GET', response => {
- let data = '';
- if (response.statusCode && response.statusCode >= 400) {
- return reject(new Error(`Got status code ${response.statusCode}`));
- }
- response.on('data', chunk => {
- data += chunk;
- });
- response.on('end', () => {
- try {
- return resolve(String(data));
- }
- catch {
- return reject(new Error('Chrome version not found'));
- }
- });
- }, false);
- request.on('error', err => {
- reject(err);
- });
- });
- }
- //# sourceMappingURL=httpUtil.js.map
|