52 lines
1.8 KiB
TypeScript
52 lines
1.8 KiB
TypeScript
/*
|
|
* SPDX-License-Identifier: GPL-3.0
|
|
* Vesktop, a desktop app aiming to give you a snappier Discord Experience
|
|
* Copyright (c) 2023 Vendicated and Vencord contributors
|
|
*/
|
|
|
|
import { createWriteStream } from "fs";
|
|
import type { IncomingMessage } from "http";
|
|
import { get, RequestOptions } 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 = {}): Promise<IncomingMessage> {
|
|
return new Promise<IncomingMessage>((resolve, reject) => {
|
|
get(url, options, res => {
|
|
const { statusCode, statusMessage, headers } = res;
|
|
if (statusCode && statusCode >= 400) return reject(new Error(`${statusCode}: ${statusMessage} - ${url}`));
|
|
if (statusCode && statusCode >= 300 && headers.location) {
|
|
return simpleReq(headers.location, options).then(resolve).catch(reject);
|
|
}
|
|
resolve(res);
|
|
}).on("error", err => {
|
|
reject(new Error(`Network error: ${err.message}`));
|
|
});
|
|
});
|
|
}
|
|
|
|
export async function simpleGet(url: string, options: RequestOptions = {}) {
|
|
try {
|
|
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)));
|
|
});
|
|
} catch (e) {
|
|
console.log(e);
|
|
}
|
|
}
|