refactor: create a unified storage plugin (context + module)

src/index.ts

@@ -5,20 +5,31 @@ import nodered from "node-red";
 import { LocalSettings } from "@node-red/runtime";
 import { dirname, resolve } from "path";
 import { fileURLToPath } from "url";
-import { storageModule } from "./storage.js";
-import { createStorage } from "unstorage";
 import mongodbDriver from "unstorage/drivers/mongodb";
-import { contextModule } from "./context.js";
+import storage from "./storage/index.js";
 
 dotenv.config();
 
-const storage = createStorage({
-  //@ts-ignore
-  driver: mongodbDriver({
-    connectionString: process.env.MONGO_DB_URL,
-    databaseName: "nodeRed",
-    collectionName: "unstorage",
-  }),
+const { contextStore, storageModule } = storage({
+  app: "app0",
+
+  storageOptions: {
+    //@ts-ignore
+    driver: mongodbDriver({
+      connectionString: process.env.MONGO_DB_URL,
+      databaseName: "nodeRed",
+      collectionName: "storage",
+    }),
+  },
+
+  contextOptions: {
+    //@ts-ignore
+    driver: mongodbDriver({
+      connectionString: process.env.MONGO_DB_URL,
+      databaseName: "nodeRed",
+      collectionName: "context",
+    }),
+  },
 });
 
 const cwd = dirname(fileURLToPath(import.meta.url));
@@ -36,9 +47,11 @@ const settings: LocalSettings = {
   //@ts-ignore
   storageModule: storageModule,
 
-  storageSettings: {
-    storage: storage,
-    appName: process.env.APP_NAME || "default",
+  contextStorage: {
+    unstorage: {
+      //@ts-ignore
+      module: contextStore,
+    },
   },
 
   adminAuth: {
@@ -54,17 +67,6 @@ const settings: LocalSettings = {
       },
     ],
   },
-
-  contextStorage: {
-    unstorage: {
-      //@ts-ignore
-      module: contextModule,
-      config: {
-        storage: storage,
-        appName: process.env.APP_NAME || "default",
-      },
-    },
-  },
 };
 
 const app = express();

src/context.ts → src/storage/context.ts

@@ -1,9 +1,17 @@
-import { prefixStorage, Storage } from "unstorage";
+import { prefixStorage, Storage, createStorage } from "unstorage";
+import type { CreateStorageOptions } from "unstorage";
 
 var appStorage: Storage;
 
-var ContextStore = function (storageSettings) {
-  appStorage = prefixStorage(storageSettings.storage, storageSettings.appName);
+var ContextStore = function (args: {
+  app: string;
+  options: CreateStorageOptions;
+}) {
+  let appStorage: Storage;
+
+  const storage = createStorage(args.options);
+
+  appStorage = prefixStorage(storage, args.app);
 };
 
 ContextStore.prototype.open = () => {};
@@ -49,6 +57,6 @@ ContextStore.prototype.delete = (scope: string) =>
 
 ContextStore.prototype.clean = (activeNodes) => {};
 
-export function contextModule(config) {
-  return new ContextStore(config);
-}
+export default (args: { app: string; options: CreateStorageOptions }) => {
+  return new ContextStore(args);
+};

src/storage/index.ts

@@ -0,0 +1,17 @@
+import type { CreateStorageOptions } from "unstorage";
+import contextStore from "./context.js";
+import storageModule from "./module.js";
+
+export default function (args: {
+  app: string;
+  storageOptions: CreateStorageOptions;
+  contextOptions: CreateStorageOptions;
+}) {
+  const module = storageModule({ app: args.app, options: args.storageOptions });
+
+  return {
+    storageModule: module,
+    contextStore: () =>
+      contextStore({ app: args.app, options: args.contextOptions }),
+  };
+}

src/storage/module.ts

@@ -0,0 +1,99 @@
+//@ts-nocheck
+import { prefixStorage, Storage, createStorage } from "unstorage";
+import type { CreateStorageOptions } from "unstorage";
+
+var appStorage: Storage;
+var libraryStorage: Storage;
+
+export default (args: { app: string; options: CreateStorageOptions }) => {
+  return {
+    init: () => {
+      const storage = createStorage(args.options);
+
+      appStorage = prefixStorage(storage, args.app);
+
+      libraryStorage = prefixStorage(appStorage, "library");
+    },
+
+    getFlows: () => appStorage.getItem("flows").then((flows) => flows || []),
+
+    saveFlows: (flows) => appStorage.setItem("flows", flows),
+
+    getCredentials: () =>
+      appStorage.getItem("credentials").then((creds) => creds || {}),
+
+    saveCredentials: (credentials) =>
+      appStorage.setItem("credentials", credentials),
+
+    getSettings: () =>
+      appStorage.getItem("settings").then((settings) => settings || {}),
+
+    saveSettings: (settings) => appStorage.setItem("settings", settings),
+
+    getSessions: () =>
+      appStorage.getItem("sessions").then((sessions) => sessions || {}),
+
+    saveSessions: (sessions) => appStorage.setItem("sessions", sessions),
+
+    getLibraryEntry: async function (type, name) {
+      if (name == "") {
+        name = "/";
+      } else if (!name.startsWith("/")) {
+        name = "/" + name;
+      }
+
+      const library = await libraryStorage.getItem(name + ":" + type);
+
+      if (library) {
+        const body = JSON.parse(library.body);
+        return body;
+      }
+
+      let libraryKeys = await libraryStorage.getKeys(name);
+
+      if (!libraryKeys) {
+        return [];
+      }
+
+      libraryKeys = libraryKeys.filter((el) => el.endsWith(type));
+
+      var dirs = [];
+      var files = [];
+
+      for (var i = 0; i < libraryKeys.length; i++) {
+        const library = await libraryStorage.getItem(libraryKeys[i]);
+
+        var n = library.name;
+        n = n.replace(name, "");
+        if (n.indexOf("/") == -1) {
+          var f = library.meta;
+          f.fn = n;
+          files.push(f);
+        } else {
+          n = n.substr(0, n.indexOf("/"));
+          dirs.push(n);
+        }
+      }
+
+      dirs = dirs.concat(files);
+
+      return dirs;
+    },
+
+    saveLibraryEntry: function (type, name, meta, body) {
+      var p = name.split("/"); // strip multiple slash
+      p = p.filter(Boolean);
+      name = p.slice(0, p.length).join("/");
+      if (name != "" && !name.startsWith("/")) {
+        name = "/" + name;
+      }
+
+      return libraryStorage.setItem(name + ":" + type, {
+        name,
+        meta,
+        body,
+        type,
+      });
+    },
+  };
+};