86 lines
2.2 KiB
TypeScript
86 lines
2.2 KiB
TypeScript
import * as fs from "fs";
|
|
|
|
import { ipcMain } from "electron";
|
|
|
|
import { AbstractStorageService } from "@bitwarden/common/abstractions/storage.service";
|
|
import { NodeUtils } from "@bitwarden/common/misc/nodeUtils";
|
|
|
|
// See: https://github.com/sindresorhus/electron-store/blob/main/index.d.ts
|
|
interface ElectronStoreOptions {
|
|
defaults: unknown;
|
|
name: string;
|
|
}
|
|
|
|
type ElectronStoreConstructor = new (options: ElectronStoreOptions) => ElectronStore;
|
|
|
|
// eslint-disable-next-line
|
|
const Store: ElectronStoreConstructor = require("electron-store");
|
|
|
|
interface ElectronStore {
|
|
get: (key: string) => unknown;
|
|
set: (key: string, obj: unknown) => void;
|
|
delete: (key: string) => void;
|
|
}
|
|
|
|
interface BaseOptions<T extends string> {
|
|
action: T;
|
|
key: string;
|
|
}
|
|
|
|
interface SaveOptions extends BaseOptions<"save"> {
|
|
obj: unknown;
|
|
}
|
|
|
|
type Options = BaseOptions<"get"> | BaseOptions<"has"> | SaveOptions | BaseOptions<"remove">;
|
|
|
|
export class ElectronStorageService implements AbstractStorageService {
|
|
private store: ElectronStore;
|
|
|
|
constructor(dir: string, defaults = {}) {
|
|
if (!fs.existsSync(dir)) {
|
|
NodeUtils.mkdirpSync(dir, "700");
|
|
}
|
|
const storeConfig: ElectronStoreOptions = {
|
|
defaults: defaults,
|
|
name: "data",
|
|
};
|
|
this.store = new Store(storeConfig);
|
|
|
|
ipcMain.handle("storageService", (event, options: Options) => {
|
|
switch (options.action) {
|
|
case "get":
|
|
return this.get(options.key);
|
|
case "has":
|
|
return this.has(options.key);
|
|
case "save":
|
|
return this.save(options.key, options.obj);
|
|
case "remove":
|
|
return this.remove(options.key);
|
|
}
|
|
});
|
|
}
|
|
|
|
get<T>(key: string): Promise<T> {
|
|
const val = this.store.get(key) as T;
|
|
return Promise.resolve(val != null ? val : null);
|
|
}
|
|
|
|
has(key: string): Promise<boolean> {
|
|
const val = this.store.get(key);
|
|
return Promise.resolve(val != null);
|
|
}
|
|
|
|
save(key: string, obj: unknown): Promise<void> {
|
|
if (obj instanceof Set) {
|
|
obj = Array.from(obj);
|
|
}
|
|
this.store.set(key, obj);
|
|
return Promise.resolve();
|
|
}
|
|
|
|
remove(key: string): Promise<void> {
|
|
this.store.delete(key);
|
|
return Promise.resolve();
|
|
}
|
|
}
|