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