42 lines
1.3 KiB
TypeScript
42 lines
1.3 KiB
TypeScript
|
import { createWriteStream } from "fs";
|
||
|
|
import type { IncomingMessage } from "http";
|
||
|
|
import { RequestOptions, get } from "https";
|
||
|
|
import { finished } from "stream/promises";
|
||
|
|
|
||
|
|
export async function downloadFile(url: string, file: string, options: RequestOptions = {}) {
|
||
|
|
const res = await simpleReq(url, options);
|
||
|
|
await finished(
|
||
|
|
res.pipe(createWriteStream(file, {
|
||
|
|
autoClose: true
|
||
|
|
}))
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
export function simpleReq(url: string, options: RequestOptions = {}) {
|
||
|
|
return new Promise<IncomingMessage>((resolve, reject) => {
|
||
|
|
get(url, options, res => {
|
||
|
|
const { statusCode, statusMessage, headers } = res;
|
||
|
|
if (statusCode! >= 400)
|
||
|
|
return void reject(`${statusCode}: ${statusMessage} - ${url}`);
|
||
|
|
if (statusCode! >= 300)
|
||
|
|
return simpleReq(headers.location!, options)
|
||
|
|
.then(resolve)
|
||
|
|
.catch(reject);
|
||
|
|
|
||
|
|
resolve(res);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
export async function simpleGet(url: string, options: RequestOptions = {}) {
|
||
|
|
const res = await simpleReq(url, options);
|
||
|
|
|
||
|
|
return new Promise<Buffer>((resolve, reject) => {
|
||
|
|
const chunks = [] as Buffer[];
|
||
|
|
|
||
|
|
res.once("error", reject);
|
||
|
|
res.on("data", chunk => chunks.push(chunk));
|
||
|
|
res.once("end", () => resolve(Buffer.concat(chunks)));
|
||
|
|
});
|
||
|
|
}
|