blob: d67dc09cfe35ac58d408f85b8d994a569fe46634 [file] [log] [blame]
/**
* 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 { HttpProxyAgent } from 'http-proxy-agent';
import { HttpsProxyAgent } from 'https-proxy-agent';
import { getProxyForUrl } from 'proxy-from-env';
import { SocksProxyAgent } from 'socks-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 proxy = getProxyForUrl(url.toString());
let agent;
if (proxy) {
const proxyUrl = new URL(proxy);
if (proxyUrl.protocol === 'http:') {
agent = new HttpProxyAgent(proxyUrl);
}
else if (proxyUrl.protocol === 'https:') {
agent = new HttpsProxyAgent(proxyUrl);
}
else if (proxyUrl.protocol.startsWith('socks')) {
agent = new SocksProxyAgent(proxyUrl);
}
}
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,
};
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