BetterDiscordApp-rauenzi/renderer/src/modules/datastore.js

import {Config} from "data";
import Utilities from "./utilities";
import Logger from "common/logger";
const fs = require("fs");
const path = require("path");
const releaseChannel = window?.DiscordNative?.app?.getReleaseChannel?.() ?? "stable";
const discordVersion = window?.DiscordNative?.remoteApp?.getVersion?.() ?? "0.0.309";

// Schema
// =======================
// %appdata%\BetterDiscord
//     -> data
//         -> [releaseChannel]\ (stable/canary/ptb)
//             -> settings.json
//             -> plugins.json
//             -> themes.json

export default new class DataStore {
    constructor() {
        this.data = {misc: {}};
        this.pluginData = {};
        this.cacheData = {};
    }

    initialize() {
        const bdFolderExists = fs.existsSync(Config.dataPath);
        if (!bdFolderExists) fs.mkdirSync(Config.dataPath);

        const pluginFolderExists = fs.existsSync(this.pluginFolder);
        if (!pluginFolderExists) fs.mkdirSync(this.pluginFolder);

        const themeFolderExists = fs.existsSync(this.themeFolder);
        if (!themeFolderExists) fs.mkdirSync(this.themeFolder);

        const newStorageExists = fs.existsSync(this.baseFolder);
        if (!newStorageExists) fs.mkdirSync(this.baseFolder);

        if (!fs.existsSync(this.dataFolder)) fs.mkdirSync(this.dataFolder);
        if (!fs.existsSync(this.customCSS)) fs.writeFileSync(this.customCSS, "");

        const dataFiles = fs.readdirSync(this.dataFolder).filter(f => !fs.statSync(path.resolve(this.dataFolder, f)).isDirectory() && f.endsWith(".json"));
        for (const file of dataFiles) {
            let data = {};
            try {data = __non_webpack_require__(path.resolve(this.dataFolder, file));}
            catch (e) {Logger.stacktrace("DataStore", `Could not load file ${file}`, e);}
            this.data[file.split(".")[0]] = data;
        }

        if (newStorageExists) return;

        try {this.convertOldData();} // Convert old data if it exists (routine checks existence and removes existence)
        catch (e) {Logger.stacktrace("DataStore", `Could not convert old data.`, e);}
    }

    convertOldData() {
        const oldFile = path.join(Config.dataPath, "bdstorage.json");
        if (!fs.existsSync(oldFile)) return;

        const oldData = __non_webpack_require__(oldFile); // got the data
        fs.renameSync(oldFile, `${oldFile}.bak`); // rename file after grabbing data to prevent loop
        const setChannelData = (channel, key, value, ext = "json") => fs.writeFileSync(path.resolve(this.baseFolder, channel, `${key}.${ext}`), JSON.stringify(value, null, 4));
        const channels = ["stable", "canary", "ptb"];
        let customcss = "";
        let favoriteEmotes = {};
        try {customcss = oldData.bdcustomcss ? atob(oldData.bdcustomcss) : "";}
        catch (e) {Logger.stacktrace("DataStore:convertOldData", "Error decoding custom css", e);}
        try {favoriteEmotes = oldData.bdfavemotes ? JSON.parse(atob(oldData.bdfavemotes)) : {};}
        catch (e) {Logger.stacktrace("DataStore:convertOldData", "Error decoding favorite emotes", e);}
        for (const channel of channels) {
            if (!fs.existsSync(path.resolve(this.baseFolder, channel))) fs.mkdirSync(path.resolve(this.baseFolder, channel));
            const channelData = oldData.settings[channel];
            if (!channelData || !channelData.settings) continue;
            const oldSettings = channelData.settings;
            const newSettings = {
                general: {publicServers: oldSettings["bda-gs-1"], voiceDisconnect: oldSettings["bda-dc-0"], showToasts: oldSettings["fork-ps-2"]},
                appearance: {twentyFourHour: oldSettings["bda-gs-6"], minimalMode: oldSettings["bda-gs-2"], coloredText: oldSettings["bda-gs-7"]},
                addons: {addonErrors: oldSettings["fork-ps-1"], autoReload: oldSettings["fork-ps-5"]},
                developer: {debuggerHotkey: oldSettings["bda-gs-8"], reactDevTools: oldSettings.reactDevTools}
            };

            const newEmotes = {
                general: {download: oldSettings["fork-es-3"], emoteMenu: oldSettings["bda-es-0"], hideEmojiMenu: !oldSettings["bda-es-9"], modifiers: oldSettings["bda-es-8"], animateOnHover: oldSettings["fork-es-2"]},
                categories: {twitchglobal: oldSettings["bda-es-7"], twitchsubscriber: oldSettings["bda-es-7"], frankerfacez: oldSettings["bda-es-1"], bttv: oldSettings["bda-es-2"]}
            };

            setChannelData(channel, "settings", newSettings); // settingsCookie
            setChannelData(channel, "emotes", newEmotes); // emotes (from settingsCookie)
            setChannelData(channel, "plugins", channelData.plugins || {}); // pluginCookie
            setChannelData(channel, "themes", channelData.themes || {}); // themeCookie
            setChannelData(channel, "misc", {favoriteEmotes}); // favorite emotes
            fs.writeFileSync(path.resolve(this.baseFolder, channel, "custom.css"), customcss); // customcss
        }

        this.initialize(); // Reinitialize data store with the converted data
    }

    get injectionPath() {
        if (this._injectionPath) return this._injectionPath;
        const base = Config.appPath;
        const roamingBase = Config.userData;
        const roamingLocation = path.resolve(roamingBase, discordVersion, "modules", "discord_desktop_core", "injector");
        const location = path.resolve(base, "..", "app");
        const realLocation = fs.existsSync(location) ? location : fs.existsSync(roamingLocation) ? roamingLocation : null;
        if (!realLocation) return this._injectionPath = null;
        return this._injectionPath = realLocation;
    }

    get pluginFolder() {return this._pluginFolder || (this._pluginFolder = path.resolve(Config.dataPath, "plugins"));}
    get themeFolder() {return this._themeFolder || (this._themeFolder = path.resolve(Config.dataPath, "themes"));}
    get customCSS() {return this._customCSS || (this._customCSS = path.resolve(this.dataFolder, "custom.css"));}
    get baseFolder() {return this._baseFolder || (this._baseFolder = path.resolve(Config.dataPath, "data"));}
    get dataFolder() {return this._dataFolder || (this._dataFolder = path.resolve(this.baseFolder, `${releaseChannel}`));}
    getPluginFile(pluginName) {return path.resolve(Config.dataPath, "plugins", pluginName + ".config.json");}


    _getFile(key) {
        if (key == "settings" || key == "plugins" || key == "themes" || key == "window") return path.resolve(this.dataFolder, `${key}.json`);
        return path.resolve(this.dataFolder, `misc.json`);
    }

    getBDData(key) {
        return this.data.misc[key] || "";
    }

    setBDData(key, value) {
        this.data.misc[key] = value;
        fs.writeFileSync(path.resolve(this.dataFolder, `misc.json`), JSON.stringify(this.data.misc, null, 4));
    }

    getLocale(locale) {
        const file = path.resolve(this.localeFolder, `${locale}.json`);
        if (!fs.existsSync(file)) return null;
        return Utilities.testJSON(fs.readFileSync(file).toString());
    }

    saveLocale(locale, strings) {
        fs.writeFileSync(path.resolve(this.localeFolder, `${locale}.json`), JSON.stringify(strings, null, 4));
    }

    getData(key) {
        return this.data[key] || "";
    }

    setData(key, value) {
        this.data[key] = value;
        fs.writeFileSync(path.resolve(this.dataFolder, `${key}.json`), JSON.stringify(value, null, 4));
    }

    loadCustomCSS() {
        return fs.readFileSync(this.customCSS).toString();
    }

    saveCustomCSS(css) {
        return fs.writeFileSync(this.customCSS, css);
    }

    ensurePluginData(pluginName) {
        if (typeof(this.pluginData[pluginName]) !== "undefined") return; // Already have data cached

        // Setup blank data if config doesn't exist
        if (!fs.existsSync(this.getPluginFile(pluginName))) return this.pluginData[pluginName] = {};

        // Getting here means not cached, read from disk
        this.pluginData[pluginName] = JSON.parse(fs.readFileSync(this.getPluginFile(pluginName)));
    }

    getPluginData(pluginName, key) {
        this.ensurePluginData(pluginName); // Ensure plugin data, if any, is cached
        return this.pluginData[pluginName][key]; // Return blindly to allow falsey values
    }

    setPluginData(pluginName, key, value) {
        if (value === undefined) return; // Can't set undefined, use deletePluginData
        this.ensurePluginData(pluginName); // Ensure plugin data, if any, is cached

        this.pluginData[pluginName][key] = value; // Set the value blindly to allow falsey values
        fs.writeFileSync(this.getPluginFile(pluginName), JSON.stringify(this.pluginData[pluginName], null, 4)); // Save to disk
    }

    deletePluginData(pluginName, key) {
        this.ensurePluginData(pluginName); // Ensure plugin data, if any, is cached
        delete this.pluginData[pluginName][key];
        fs.writeFileSync(this.getPluginFile(pluginName), JSON.stringify(this.pluginData[pluginName], null, 4));
    }
};