src/services/SnippetManager.ts

/**
* This file is part of SudoBot.
* 
* Copyright (C) 2021-2022 OSN Inc.
*
* 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 { readFile, writeFile } from "fs";
import path from "path";
import DiscordClient from "../client/Client";
import MessageEmbed from "../client/MessageEmbed";
import Service from "../utils/structures/Service";
import { deleteFile, parseEmbedsInString } from "../utils/util";

export type Snippet = {
    name: string;
    content: string;
    files: string[],
    embeds?: MessageEmbed[]
}; 

export type SnippetContainer = {
    [guildID: string]: Snippet[];
}; 

export default class SnippetManager extends Service {
    snippets: SnippetContainer = {};
    filePath: string;

    constructor(client: DiscordClient) {
        super(client);
        this.filePath = path.resolve(process.env.SUDO_PREFIX ?? path.join(__dirname, '../..'), 'config/snippets.json');
        this.load();
    }

    load() {
        readFile(this.filePath, (err, data) => {
            if (err) {
                console.log(err);                
            }

            this.snippets = JSON.parse(data.toString());
        });
    }

    write() {
        writeFile(this.filePath, JSON.stringify(this.snippets), () => null);
    }

    set(guildID: string, name: string, content: string, files: string[] = []): void {
        this.snippets[guildID].push({
            name,
            content,
            files
        });
    }

    get(guildID: string, name: string): Snippet | null {
        if (!this.snippets[guildID]) {
            return null;
        }
        
        for (const s of this.snippets[guildID]) {
            if (s.name === name) {
                return s;
            }
        }

        return null;
    }
    
    getParsed(guildID: string, name: string): Snippet | null {
        const snippet = this.get(guildID, name);

        if (!snippet) {
            return null;
        }

        return this.parseEmbeds(snippet) ?? snippet;
    }

    parseEmbeds(snippet: Snippet) {        
        try {
            const { embeds, content } = parseEmbedsInString(snippet.content);

            console.log(content);            

            return <Snippet> {
                ...snippet,
                content,
                embeds,
            };
        }
        catch (e) {
            console.log(e);
        }
    }

    async delete(guildID: string, name: string): Promise<void> {
        for (const i in this.snippets[guildID]) {
            if (this.snippets[guildID][i].name === name) {
                for await (const file of this.snippets[guildID][i].files) {
                    try {
                        await deleteFile(path.resolve(process.env.SUDO_PREFIX ?? path.join(__dirname, '../..'), 'storage', file));
                    }
                    catch (e) {
                        console.log(e);                
                    }
                }

                await this.snippets[guildID].splice(parseInt(i), 1);

                return;
            }
        }
    }

    async deleteData(guildID: string, name: string): Promise<void> {
        for (const i in this.snippets[guildID]) {
            if (this.snippets[guildID][i].name === name) {
                await this.snippets[guildID].splice(parseInt(i), 1);
                return;
            }
        }
    }
}
1	/**
2	* This file is part of SudoBot.
3	*
4	* Copyright (C) 2021-2022 OSN Inc.
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 { readFile, writeFile } from "fs";
21	import path from "path";
22	import DiscordClient from "../client/Client";
23	import MessageEmbed from "../client/MessageEmbed";
24	import Service from "../utils/structures/Service";
25	import { deleteFile, parseEmbedsInString } from "../utils/util";
26
27	export type Snippet = {
28	name: string;
29	content: string;
30	files: string[],
31	embeds?: MessageEmbed[]
32	};
33
34	export type SnippetContainer = {
35	[guildID: string]: Snippet[];
36	};
37
38	export default class SnippetManager extends Service {
39	snippets: SnippetContainer = {};
40	filePath: string;
41
42	constructor(client: DiscordClient) {
43	super(client);
44	this.filePath = path.resolve(process.env.SUDO_PREFIX ?? path.join(__dirname, '../..'), 'config/snippets.json');
45	this.load();
46	}
47
48	load() {
49	readFile(this.filePath, (err, data) => {
50	if (err) {
51	console.log(err);
52	}
53
54	this.snippets = JSON.parse(data.toString());
55	});
56	}
57
58	write() {
59	writeFile(this.filePath, JSON.stringify(this.snippets), () => null);
60	}
61
62	set(guildID: string, name: string, content: string, files: string[] = []): void {
63	this.snippets[guildID].push({
64	name,
65	content,
66	files
67	});
68	}
69
70	get(guildID: string, name: string): Snippet \| null {
71	if (!this.snippets[guildID]) {
72	return null;
73	}
74
75	for (const s of this.snippets[guildID]) {
76	if (s.name === name) {
77	return s;
78	}
79	}
80
81	return null;
82	}
83
84	getParsed(guildID: string, name: string): Snippet \| null {
85	const snippet = this.get(guildID, name);
86
87	if (!snippet) {
88	return null;
89	}
90
91	return this.parseEmbeds(snippet) ?? snippet;
92	}
93
94	parseEmbeds(snippet: Snippet) {
95	try {
96	const { embeds, content } = parseEmbedsInString(snippet.content);
97
98	console.log(content);
99
100	return <Snippet> {
101	...snippet,
102	content,
103	embeds,
104	};
105	}
106	catch (e) {
107	console.log(e);
108	}
109	}
110
111	async delete(guildID: string, name: string): Promise<void> {
112	for (const i in this.snippets[guildID]) {
113	if (this.snippets[guildID][i].name === name) {
114	for await (const file of this.snippets[guildID][i].files) {
115	try {
116	await deleteFile(path.resolve(process.env.SUDO_PREFIX ?? path.join(__dirname, '../..'), 'storage', file));
117	}
118	catch (e) {
119	console.log(e);
120	}
121	}
122
123	await this.snippets[guildID].splice(parseInt(i), 1);
124
125	return;
126	}
127	}
128	}
129
130	async deleteData(guildID: string, name: string): Promise<void> {
131	for (const i in this.snippets[guildID]) {
132	if (this.snippets[guildID][i].name === name) {
133	await this.snippets[guildID].splice(parseInt(i), 1);
134	return;
135	}
136	}
137	}
138	}