175 lines
8.0 KiB
JavaScript
175 lines
8.0 KiB
JavaScript
import {Config} from "data";
|
|
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 = "";
|
|
try {customcss = oldData.bdcustomcss ? atob(oldData.bdcustomcss) : "";}
|
|
catch (e) {Logger.stacktrace("DataStore:convertOldData", "Error decoding custom css", 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}
|
|
};
|
|
|
|
setChannelData(channel, "settings", newSettings); // settingsCookie
|
|
setChannelData(channel, "plugins", channelData.plugins || {}); // pluginCookie
|
|
setChannelData(channel, "themes", channelData.themes || {}); // themeCookie
|
|
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;
|
|
try {return JSON.parse(fs.readFileSync(file).toString());}
|
|
catch {return false;}
|
|
}
|
|
|
|
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));
|
|
}
|
|
}; |