/*
 * 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);
    }
}