Enable opening Grist documents anywhere directly

ext/app/electron/DocRegistry.ts

@@ -0,0 +1,69 @@
+import * as log from "app/server/lib/log";
+import * as path from "path";
+import { HomeDBManager } from "app/gen-server/lib/HomeDBManager";
+
+export class DocRegistry {
+
+  private idToPathMap: Map<string, string>;
+  private pathToIdMap: Map<string, string>;
+  private db: HomeDBManager;
+
+  private constructor(dbManager: HomeDBManager) {
+    this.db = dbManager;
+  }
+
+  public static async create(dbManager: HomeDBManager) {
+    // Allocate space.
+    const dr = new DocRegistry(dbManager);
+    dr.idToPathMap = new Map<string, string>;
+    dr.pathToIdMap = new Map<string, string>;
+
+    // Go over all documents we know about.
+    for (const doc of await dr.db.getAllDocs()) {
+      // All documents are supposed to have externalId set.
+      if (doc.options?.externalId) {
+        // Cache the two-way mapping docID <-> path.
+        dr.idToPathMap.set(doc.id, doc.options?.externalId);
+        dr.pathToIdMap.set(doc.options?.externalId, doc.id);
+        log.info(`Document ${doc.id} has path ${doc.options?.externalId}`);
+      } else {
+        // Remove this document - it should not appear in a DB for Grist Desktop.
+        // TODO: do we need a transaction to do this?
+        await dr.db.connection.transaction(async manager => {
+          manager.remove(doc);
+        });
+      }
+    }
+    return dr;
+  }
+
+  public lookupById(docId: string): string | null {
+    return this.idToPathMap.get(docId) ?? null;
+  }
+
+  public lookupByPath(docPath: string): string | null {
+    return this.pathToIdMap.get(docPath) ?? null;
+  }
+
+  public async registerDoc(docPath: string) {
+    const user = await this.db.getUserByLogin(process.env.GRIST_DEFAULT_EMAIL as string);
+    if (!user) { throw new Error('cannot find default user'); }
+    const wss = this.db.unwrapQueryResult(await this.db.getOrgWorkspaces({userId: user.id}, 0));
+    for (const doc of wss[0].docs) {
+      if (doc.options?.externalId === docPath) {
+        // We might be able to do better.
+        throw Error("DocRegistry cache incoherent. Please try restarting the app.");
+      }
+    }
+    const docId = this.db.unwrapQueryResult(await this.db.addDocument({
+      userId: user.id,
+    }, wss[0].id, {
+      name: path.basename(docPath, '.grist'),
+      options: { externalId: docPath },
+    }));
+    this.pathToIdMap.set(docPath, docId);
+    this.idToPathMap.set(docId, docPath);
+    return docId;
+  }
+
+}