import {Config} from "data"; import Utilities from "./utilities"; const fs = require("fs"); const path = require("path"); const releaseChannel = DiscordNative.globals ? DiscordNative.globals.releaseChannel : DiscordNative.app ? DiscordNative.app.getReleaseChannel() : "stable"; // 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() { if (!fs.existsSync(this.baseFolder)) fs.mkdirSync(this.baseFolder); if (!fs.existsSync(this.dataFolder)) fs.mkdirSync(this.dataFolder); if (!fs.existsSync(this.localeFolder)) fs.mkdirSync(this.localeFolder); if (!fs.existsSync(this.emoteFolder)) fs.mkdirSync(this.emoteFolder); if (!fs.existsSync(this.cacheFile)) fs.writeFileSync(this.cacheFile, JSON.stringify({})); 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) { this.data[file.split(".")[0]] = __non_webpack_require__(path.resolve(this.dataFolder, file)); } this.cacheData = Utilities.testJSON(fs.readFileSync(this.cacheFile).toString()) || {}; } get injectionPath() { if (this._injectionPath) return this._injectionPath; const electron = require("electron").remote.app; const base = electron.getAppPath(); const roamingBase = electron.getPath("userData"); const roamingLocation = path.resolve(roamingBase, electron.getVersion(), "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 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}`));} get localeFolder() {return this._localeFolder || (this._localeFolder = path.resolve(this.baseFolder, `locales`));} get emoteFolder() {return this._emoteFolder || (this._emoteFolder = path.resolve(this.baseFolder, `emotes`));} get cacheFile() {return this._cacheFile || (this._cacheFile = path.resolve(this.baseFolder, `.cache`));} 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)); } getCacheHash(category, key) { if (!this.cacheData[category]) return ""; if (!fs.existsSync(path.resolve(this.baseFolder, category, `${key}.json`))) return ""; return this.cacheData[category][key] || ""; } setCacheHash(category, key, hash) { if (!this.cacheData[category]) this.cacheData[category] = {}; this.cacheData[category][key] = hash; fs.writeFileSync(this.cacheFile, JSON.stringify(this.cacheData)); } invalidateCache(category, key) { if (!this.cacheData[category]) return; delete this.cacheData[category][key]; fs.writeFileSync(this.cacheFile, JSON.stringify(this.cacheData)); } emotesExist(category) { return fs.existsSync(path.resolve(this.emoteFolder, `${category}.json`)); } getEmoteData(category) { const file = path.resolve(this.emoteFolder, `${category}.json`); if (!fs.existsSync(file)) return null; return Utilities.testJSON(fs.readFileSync(file).toString()); } saveEmoteData(category, data) { fs.writeFileSync(path.resolve(this.emoteFolder, `${category}.json`), JSON.stringify(data)); } 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); } getPluginData(pluginName, key) { if (this.pluginData[pluginName] !== undefined) return this.pluginData[pluginName][key] || undefined; if (!fs.existsSync(this.getPluginFile(pluginName))) return undefined; this.pluginData[pluginName] = JSON.parse(fs.readFileSync(this.getPluginFile(pluginName))); return this.pluginData[pluginName][key] || undefined; } setPluginData(pluginName, key, value) { if (value === undefined) return; if (this.pluginData[pluginName] === undefined) this.pluginData[pluginName] = {}; this.pluginData[pluginName][key] = value; fs.writeFileSync(this.getPluginFile(pluginName), JSON.stringify(this.pluginData[pluginName], null, 4)); } deletePluginData(pluginName, key) { if (this.pluginData[pluginName] === undefined) this.pluginData[pluginName] = {}; delete this.pluginData[pluginName][key]; fs.writeFileSync(this.getPluginFile(pluginName), JSON.stringify(this.pluginData[pluginName], null, 4)); } };