/** * Copyright (c) 2021 OpenLens Authors * * Permission is hereby granted, free of charge, to any person obtaining a copy of * this software and associated documentation files (the "Software"), to deal in * the Software without restriction, including without limitation the rights to * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of * the Software, and to permit persons to whom the Software is furnished to do so, * subject to the following conditions: * * The above copyright notice and this permission notice shall be included in all * copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS * FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER * IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ import path from "path"; import Config from "conf"; import type { Options as ConfOptions } from "conf/dist/source/types"; import { ipcMain, ipcRenderer } from "electron"; import { IReactionOptions, makeObservable, reaction, runInAction } from "mobx"; import { getAppVersion, Singleton, toJS, Disposer } from "./utils"; import logger from "../main/logger"; import { broadcastMessage, ipcMainOn, ipcRendererOn } from "./ipc"; import isEqual from "lodash/isEqual"; import { isTestEnv } from "./vars"; import { kebabCase } from "lodash"; import { getPath } from "./utils/getPath"; export interface BaseStoreParams extends ConfOptions { syncOptions?: IReactionOptions; } /** * Note: T should only contain base JSON serializable types. */ export abstract class BaseStore extends Singleton { protected storeConfig?: Config; protected syncDisposers: Disposer[] = []; protected constructor(protected params: BaseStoreParams) { super(); makeObservable(this); if (ipcRenderer) { params.migrations = undefined; // don't run migrations on renderer } } /** * This must be called after the last child's constructor is finished (or just before it finishes) */ load() { this.storeConfig = new Config({ ...this.params, projectName: "lens", projectVersion: getAppVersion(), cwd: this.cwd(), }); const res: any = this.fromStore(this.storeConfig.store); if (res instanceof Promise || (typeof res === "object" && res && typeof res.then === "function")) { console.error(`${this.constructor.name} extends BaseStore's fromStore method returns a Promise or promise-like object. This is an error and must be fixed.`); } this.enableSync(); if (!isTestEnv) { logger.info(`[${kebabCase(this.constructor.name).toUpperCase()}]: LOADED from ${this.path}`); } } get name() { return path.basename(this.path); } protected get syncRendererChannel() { return `store-sync-renderer:${this.path}`; } protected get syncMainChannel() { return `store-sync-main:${this.path}`; } get path() { return this.storeConfig?.path || ""; } protected cwd() { return getPath("userData"); } protected async saveToFile(model: T) { logger.info(`[STORE]: SAVING ${this.path}`); // todo: update when fixed https://github.com/sindresorhus/conf/issues/114 if (this.storeConfig) { for (const [key, value] of Object.entries(model)) { this.storeConfig.set(key, value); } } } enableSync() { this.syncDisposers.push( reaction( () => toJS(this.toJSON()), // unwrap possible observables and react to everything model => this.onModelChange(model), this.params.syncOptions, ), ); if (ipcMain) { this.syncDisposers.push(ipcMainOn(this.syncMainChannel, (event, model: T) => { logger.silly(`[STORE]: SYNC ${this.name} from renderer`, { model }); this.onSync(model); })); } if (ipcRenderer) { this.syncDisposers.push(ipcRendererOn(this.syncRendererChannel, (event, model: T) => { logger.silly(`[STORE]: SYNC ${this.name} from main`, { model }); this.onSyncFromMain(model); })); } } protected onSyncFromMain(model: T) { this.applyWithoutSync(() => { this.onSync(model); }); } unregisterIpcListener() { ipcRenderer?.removeAllListeners(this.syncMainChannel); ipcRenderer?.removeAllListeners(this.syncRendererChannel); } disableSync() { this.syncDisposers.forEach(dispose => dispose()); this.syncDisposers.length = 0; } protected applyWithoutSync(callback: () => void) { this.disableSync(); runInAction(callback); this.enableSync(); } protected onSync(model: T) { // todo: use "resourceVersion" if merge required (to avoid equality checks => better performance) if (!isEqual(this.toJSON(), model)) { this.fromStore(model); } } protected async onModelChange(model: T) { if (ipcMain) { this.saveToFile(model); // save config file broadcastMessage(this.syncRendererChannel, model); } else { broadcastMessage(this.syncMainChannel, model); } } /** * fromStore is called internally when a child class syncs with the file * system. * * Note: This function **must** be synchronous. * * @param data the parsed information read from the stored JSON file */ protected abstract fromStore(data: T): void; /** * toJSON is called when syncing the store to the filesystem. It should * produce a JSON serializable object representation of the current state. * * It is recommended that a round trip is valid. Namely, calling * `this.fromStore(this.toJSON())` shouldn't change the state. */ abstract toJSON(): T; }