src/utils/download.ts

/*
* This file is part of SudoBot.
*
* Copyright (C) 2021-2023 OSN Developers.
*
* SudoBot is free software; you can redistribute it and/or modify it
* under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* SudoBot is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
*/

import axios from "axios";
import { createWriteStream } from "fs";
import { basename, join } from "path";
import stream from "stream";
import { promisify } from "util";
import { logInfo } from "./logger";
import { sudoPrefix } from "./utils";

export async function downloadFile({ url, path, name }: DownloadFileOptions) {
    logInfo("Attempting to download file: " + url);

    const finished = promisify(stream.finished);
    const storagePath = path ?? sudoPrefix("storage");
    const filePath = join(storagePath, name ?? basename(url));
    const writer = createWriteStream(filePath);
    const response = await axios.get(url, {
        responseType: "stream"
    });

    if (response.status < 200 || response.status >= 300) {
        throw new Error("HTTP error: Non 2xx response received: code " + response.status);
    }

    response.data.pipe(writer);

    await finished(writer);

    if (!writer.closed) await promisify(writer.close.bind(writer))();

    logInfo("Saved downloaded file to: " + filePath);

    return {
        filePath,
        storagePath
    };
}

interface DownloadFileOptions {
    url: string;
    path?: string;
    name?: string;
}
1	rakinar2	577	/*
2			* This file is part of SudoBot.
3			*
4			* Copyright (C) 2021-2023 OSN Developers.
5			*
6			* SudoBot is free software; you can redistribute it and/or modify it
7			* under the terms of the GNU Affero General Public License as published by
8			* the Free Software Foundation, either version 3 of the License, or
9			* (at your option) any later version.
10			*
11			* SudoBot is distributed in the hope that it will be useful, but
12			* WITHOUT ANY WARRANTY; without even the implied warranty of
13			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			* GNU Affero General Public License for more details.
15			*
16			* You should have received a copy of the GNU Affero General Public License
17			* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
18			*/
19
20			import axios from "axios";
21			import { createWriteStream } from "fs";
22			import { basename, join } from "path";
23			import stream from "stream";
24			import { promisify } from "util";
25			import { logInfo } from "./logger";
26			import { sudoPrefix } from "./utils";
27
28			export async function downloadFile({ url, path, name }: DownloadFileOptions) {
29			logInfo("Attempting to download file: " + url);
30
31			const finished = promisify(stream.finished);
32			const storagePath = path ?? sudoPrefix("storage");
33			const filePath = join(storagePath, name ?? basename(url));
34			const writer = createWriteStream(filePath);
35			const response = await axios.get(url, {
36			responseType: "stream"
37			});
38
39			if (response.status < 200 \|\| response.status >= 300) {
40			throw new Error("HTTP error: Non 2xx response received: code " + response.status);
41			}
42
43			response.data.pipe(writer);
44
45			await finished(writer);
46
47			if (!writer.closed) await promisify(writer.close.bind(writer))();
48
49			logInfo("Saved downloaded file to: " + filePath);
50
51			return {
52			filePath,
53			storagePath
54			};
55			}
56
57			interface DownloadFileOptions {
58			url: string;
59			path?: string;
60			name?: string;
61			}