fix(electron): export and import (#9767)

packages/frontend/apps/electron/src/helper/nbstore/handlers.ts

@@ -9,12 +9,17 @@ import { getSpaceDBPath } from '../workspace/meta';
 
 const POOL = new DocStoragePool();
 
+export function getDocStoragePool() {
+  return POOL;
+}
+
 export const nbstoreHandlers: NativeDBApis = {
   connect: async (universalId: string) => {
     const { peer, type, id } = parseUniversalId(universalId);
     const dbPath = await getSpaceDBPath(peer, type, id);
     await fs.ensureDir(path.dirname(dbPath));
     await POOL.connect(universalId, dbPath);
+    await POOL.setSpaceId(universalId, id);
   },
   disconnect: POOL.disconnect.bind(POOL),
   pushUpdate: POOL.pushUpdate.bind(POOL),

packages/frontend/apps/electron/src/helper/nbstore/index.ts

@@ -1,3 +1,4 @@
 export { nbstoreHandlers } from './handlers';
+export { getDocStoragePool } from './handlers';
 export { dbEvents as dbEventsV1, dbHandlers as dbHandlersV1 } from './v1';
 export { universalId } from '@affine/nbstore';

packages/frontend/apps/electron/src/helper/nbstore/v1/db-adapter.ts

@@ -255,4 +255,12 @@ export class SQLiteAdapter {
       await this.db.clearSyncMetadata();
     },
   };
+
+  async getDocTimestamps() {
+    if (!this.db) {
+      logger.warn(`${this.path} is not connected`);
+      return [];
+    }
+    return await this.db.getDocTimestamps();
+  }
 }

packages/frontend/apps/electron/src/helper/nbstore/v1/ensure-db.ts

@@ -1,6 +1,9 @@
+import { existsSync } from 'node:fs';
+
 import type { SpaceType } from '@affine/nbstore';
 
 import { logger } from '../../logger';
+import { getWorkspaceMeta } from '../../workspace/meta';
 import type { WorkspaceSQLiteDB } from './workspace-db-adapter';
 import { openWorkspaceDatabase } from './workspace-db-adapter';
 
@@ -35,10 +38,32 @@ async function getWorkspaceDB(spaceType: SpaceType, id: string) {
 
     process.on('beforeExit', cleanup);
   }
-  // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+
+  // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion
   return db!;
 }
 
-export function ensureSQLiteDB(spaceType: SpaceType, id: string) {
+export async function ensureSQLiteDB(
+  spaceType: SpaceType,
+  id: string
+): Promise<WorkspaceSQLiteDB | null> {
+  const meta = await getWorkspaceMeta(spaceType, id);
+
+  // do not auto create v1 db anymore
+  if (!existsSync(meta.mainDBPath)) {
+    return null;
+  }
+
   return getWorkspaceDB(spaceType, id);
 }
+
+export async function ensureSQLiteDisconnected(
+  spaceType: SpaceType,
+  id: string
+) {
+  const db = await ensureSQLiteDB(spaceType, id);
+
+  if (db) {
+    await db.destroy();
+  }
+}

packages/frontend/apps/electron/src/helper/nbstore/v1/index.ts

@@ -1,6 +1,5 @@
 import type { SpaceType } from '@affine/nbstore';
 
-import { mainRPC } from '../../main-rpc';
 import type { MainEventRegister } from '../../type';
 import { ensureSQLiteDB } from './ensure-db';
 
@@ -13,119 +12,41 @@ export const dbHandlers = {
     subdocId: string
   ) => {
     const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
+
+    if (!spaceDB) {
+      // means empty update in yjs
+      return new Uint8Array([0, 0]);
+    }
+
     return spaceDB.getDocAsUpdates(subdocId);
   },
-  applyDocUpdate: async (
-    spaceType: SpaceType,
-    workspaceId: string,
-    update: Uint8Array,
-    subdocId: string
-  ) => {
+  getDocTimestamps: async (spaceType: SpaceType, workspaceId: string) => {
     const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.addUpdateToSQLite(update, subdocId);
-  },
-  deleteDoc: async (
-    spaceType: SpaceType,
-    workspaceId: string,
-    subdocId: string
-  ) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.deleteUpdate(subdocId);
-  },
-  addBlob: async (
-    spaceType: SpaceType,
-    workspaceId: string,
-    key: string,
-    data: Uint8Array
-  ) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.addBlob(key, data);
+
+    if (!spaceDB) {
+      return [];
+    }
+
+    return spaceDB.getDocTimestamps();
   },
   getBlob: async (spaceType: SpaceType, workspaceId: string, key: string) => {
     const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
+
+    if (!spaceDB) {
+      return null;
+    }
+
     return spaceDB.getBlob(key);
   },
-  deleteBlob: async (
-    spaceType: SpaceType,
-    workspaceId: string,
-    key: string
-  ) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.deleteBlob(key);
-  },
   getBlobKeys: async (spaceType: SpaceType, workspaceId: string) => {
     const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
+
+    if (!spaceDB) {
+      return [];
+    }
+
     return spaceDB.getBlobKeys();
   },
-  getDefaultStorageLocation: async () => {
-    return await mainRPC.getPath('sessionData');
-  },
-  getServerClock: async (
-    spaceType: SpaceType,
-    workspaceId: string,
-    key: string
-  ) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.adapter.serverClock.get(key);
-  },
-  setServerClock: async (
-    spaceType: SpaceType,
-    workspaceId: string,
-    key: string,
-    data: Uint8Array
-  ) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.adapter.serverClock.set(key, data);
-  },
-  getServerClockKeys: async (spaceType: SpaceType, workspaceId: string) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.adapter.serverClock.keys();
-  },
-  clearServerClock: async (spaceType: SpaceType, workspaceId: string) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.adapter.serverClock.clear();
-  },
-  delServerClock: async (
-    spaceType: SpaceType,
-    workspaceId: string,
-    key: string
-  ) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.adapter.serverClock.del(key);
-  },
-  getSyncMetadata: async (
-    spaceType: SpaceType,
-    workspaceId: string,
-    key: string
-  ) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.adapter.syncMetadata.get(key);
-  },
-  setSyncMetadata: async (
-    spaceType: SpaceType,
-    workspaceId: string,
-    key: string,
-    data: Uint8Array
-  ) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.adapter.syncMetadata.set(key, data);
-  },
-  getSyncMetadataKeys: async (spaceType: SpaceType, workspaceId: string) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.adapter.syncMetadata.keys();
-  },
-  clearSyncMetadata: async (spaceType: SpaceType, workspaceId: string) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.adapter.syncMetadata.clear();
-  },
-  delSyncMetadata: async (
-    spaceType: SpaceType,
-    workspaceId: string,
-    key: string
-  ) => {
-    const spaceDB = await ensureSQLiteDB(spaceType, workspaceId);
-    return spaceDB.adapter.syncMetadata.del(key);
-  },
 };
 
 export const dbEvents = {} satisfies Record<string, MainEventRegister>;

packages/frontend/apps/electron/src/helper/nbstore/v1/workspace-db-adapter.ts

@@ -67,6 +67,10 @@ export class WorkspaceSQLiteDB {
     }
   };
 
+  async getDocTimestamps() {
+    return this.adapter.getDocTimestamps();
+  }
+
   async addBlob(key: string, value: Uint8Array) {
     this.update$.next();
     const res = await this.adapter.addBlob(key, value);