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