Equicord/src/equicordplugins/holyNotes/NoteHandler.ts

/*
 * Vencord, a Discord client mod
 * Copyright (c) 2024 Vendicated and contributors
 * SPDX-License-Identifier: GPL-3.0-or-later
 */

import { Channel, Message } from "@vencord/discord-types";
import { proxyLazyWebpack } from "@webpack";
import { ChannelStore, Toasts, UserStore } from "@webpack/common";

import { AllNotebooks, Note, Notebook } from "./types";
import { deleteCacheFromDataStore, DeleteEntireStore, saveCacheToDataStore } from "./utils";

export const noteHandlerCache = new Map<string, Notebook>();

function showToast(message: string, type: typeof Toasts.Type[keyof typeof Toasts.Type] = Toasts.Type.SUCCESS) {
    Toasts.show({ id: Toasts.genId(), message, type });
}

function formatNote(channel: Channel, message: Message): Note {
    return {
        id: message.id,
        channel_id: message.channel_id,
        guild_id: channel.guild_id,
        content: message.content,
        author: {
            id: message.author.id,
            avatar: message.author.avatar,
            discriminator: message.author.discriminator,
            username: message.author.username,
        },
        flags: message.flags,
        timestamp: message.timestamp.toString(),
        attachments: message.attachments,
        embeds: message.embeds,
        reactions: message.reactions,
        stickerItems: message.stickerItems,
    };
}

export const noteHandler = proxyLazyWebpack(() => ({
    getNotes(notebook: string): Notebook | undefined {
        return noteHandlerCache.get(notebook);
    },

    getAllNotes(): AllNotebooks {
        return Object.fromEntries(noteHandlerCache.entries());
    },

    addNote(message: Message, notebook: string) {
        const notes = this.getNotes(notebook) ?? {};
        const channel = ChannelStore.getChannel(message.channel_id);
        const newNotes = { [message.id]: formatNote(channel, message), ...notes };

        noteHandlerCache.set(notebook, newNotes);
        saveCacheToDataStore(notebook, newNotes);
        showToast(`Added note to ${notebook}`);
    },

    deleteNote(noteId: string, notebook: string) {
        const notes = this.getNotes(notebook);
        if (!notes) return;

        const { [noteId]: _, ...rest } = notes;
        noteHandlerCache.set(notebook, rest);
        saveCacheToDataStore(notebook, rest);
        showToast(`Deleted note from ${notebook}`);
    },

    moveNote(note: Note, from: string, to: string) {
        const fromNotes = this.getNotes(from);
        const toNotes = this.getNotes(to);
        if (!fromNotes || !toNotes) return;

        const { [note.id]: _, ...restFrom } = fromNotes;
        const newTo = { [note.id]: note, ...toNotes };

        noteHandlerCache.set(from, restFrom);
        noteHandlerCache.set(to, newTo);
        saveCacheToDataStore(from, restFrom);
        saveCacheToDataStore(to, newTo);
        showToast(`Moved note from ${from} to ${to}`);
    },

    newNoteBook(name: string, silent?: boolean) {
        if (noteHandlerCache.has(name)) {
            showToast(`Notebook ${name} already exists`, Toasts.Type.FAILURE);
            return;
        }

        noteHandlerCache.set(name, {});
        saveCacheToDataStore(name, {});
        if (!silent) showToast(`Created notebook ${name}`);
    },

    deleteNotebook(name: string) {
        noteHandlerCache.delete(name);
        deleteCacheFromDataStore(name);
        showToast(`Deleted notebook ${name}`);
    },

    refreshAvatars() {
        const notebooks = this.getAllNotes();

        for (const notebookName in notebooks) {
            const notebook = notebooks[notebookName];
            for (const noteId in notebook) {
                const note = notebook[noteId];
                const user = UserStore.getUser(note.author.id);
                if (user) {
                    note.author.avatar = user.avatar;
                    note.author.discriminator = user.discriminator;
                    note.author.username = user.username;
                }
            }
            noteHandlerCache.set(notebookName, notebook);
            saveCacheToDataStore(notebookName, notebook);
        }

        showToast("Refreshed avatars");
    },

    async deleteEverything() {
        noteHandlerCache.clear();
        await DeleteEntireStore();
        showToast("Deleted all notes");
    },

    exportNotes(): AllNotebooks {
        return this.getAllNotes();
    },

    importNotes(jsonString: string) {
        let parsed: AllNotebooks;
        try {
            parsed = JSON.parse(jsonString);
        } catch {
            showToast("Invalid JSON", Toasts.Type.FAILURE);
            return;
        }

        for (const notebook in parsed) {
            noteHandlerCache.set(notebook, parsed[notebook]);
            saveCacheToDataStore(notebook, parsed[notebook]);
        }

        showToast("Imported notes");
    },
}));