src/services/SnippetManager.ts

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