refactor(electron): use sqlite to store server clock & sync meta (#6957)

After this PR, IDB should not be used in desktop any longer.

packages/frontend/electron/src/helper/db/db-adapter.ts

@@ -1,6 +1,7 @@
 import type { InsertRow } from '@affine/native';
 import { SqliteConnection, ValidationResult } from '@affine/native';
 import { WorkspaceVersion } from '@toeverything/infra/blocksuite';
+import type { ByteKVBehavior } from '@toeverything/infra/storage';
 
 import { logger } from '../logger';
 import { applyGuidCompatibilityFix, migrateToLatest } from './migration';
@@ -175,4 +176,82 @@ export class SQLiteAdapter {
       logger.error('replaceUpdates', error);
     }
   }
+
+  serverClock: ByteKVBehavior = {
+    get: async key => {
+      if (!this.db) {
+        logger.warn(`${this.path} is not connected`);
+        return null;
+      }
+      const blob = await this.db.getServerClock(key);
+      return blob?.data ?? null;
+    },
+    set: async (key, data) => {
+      if (!this.db) {
+        logger.warn(`${this.path} is not connected`);
+        return;
+      }
+      await this.db.setServerClock(key, data);
+    },
+    keys: async () => {
+      if (!this.db) {
+        logger.warn(`${this.path} is not connected`);
+        return [];
+      }
+      return await this.db.getServerClockKeys();
+    },
+    del: async key => {
+      if (!this.db) {
+        logger.warn(`${this.path} is not connected`);
+        return;
+      }
+      await this.db.delServerClock(key);
+    },
+    clear: async () => {
+      if (!this.db) {
+        logger.warn(`${this.path} is not connected`);
+        return;
+      }
+      await this.db.clearServerClock();
+    },
+  };
+
+  syncMetadata: ByteKVBehavior = {
+    get: async key => {
+      if (!this.db) {
+        logger.warn(`${this.path} is not connected`);
+        return null;
+      }
+      const blob = await this.db.getSyncMetadata(key);
+      return blob?.data ?? null;
+    },
+    set: async (key, data) => {
+      if (!this.db) {
+        logger.warn(`${this.path} is not connected`);
+        return;
+      }
+      await this.db.setSyncMetadata(key, data);
+    },
+    keys: async () => {
+      if (!this.db) {
+        logger.warn(`${this.path} is not connected`);
+        return [];
+      }
+      return await this.db.getSyncMetadataKeys();
+    },
+    del: async key => {
+      if (!this.db) {
+        logger.warn(`${this.path} is not connected`);
+        return;
+      }
+      await this.db.delSyncMetadata(key);
+    },
+    clear: async () => {
+      if (!this.db) {
+        logger.warn(`${this.path} is not connected`);
+        return;
+      }
+      await this.db.clearSyncMetadata();
+    },
+  };
 }

packages/frontend/electron/src/helper/db/index.ts

@@ -40,6 +40,54 @@ export const dbHandlers = {
   getDefaultStorageLocation: async () => {
     return await mainRPC.getPath('sessionData');
   },
+  getServerClock: async (workspaceId: string, key: string) => {
+    const workspaceDB = await ensureSQLiteDB(workspaceId);
+    return workspaceDB.adapter.serverClock.get(key);
+  },
+  setServerClock: async (
+    workspaceId: string,
+    key: string,
+    data: Uint8Array
+  ) => {
+    const workspaceDB = await ensureSQLiteDB(workspaceId);
+    return workspaceDB.adapter.serverClock.set(key, data);
+  },
+  getServerClockKeys: async (workspaceId: string) => {
+    const workspaceDB = await ensureSQLiteDB(workspaceId);
+    return workspaceDB.adapter.serverClock.keys();
+  },
+  clearServerClock: async (workspaceId: string) => {
+    const workspaceDB = await ensureSQLiteDB(workspaceId);
+    return workspaceDB.adapter.serverClock.clear();
+  },
+  delServerClock: async (workspaceId: string, key: string) => {
+    const workspaceDB = await ensureSQLiteDB(workspaceId);
+    return workspaceDB.adapter.serverClock.del(key);
+  },
+  getSyncMetadata: async (workspaceId: string, key: string) => {
+    const workspaceDB = await ensureSQLiteDB(workspaceId);
+    return workspaceDB.adapter.syncMetadata.get(key);
+  },
+  setSyncMetadata: async (
+    workspaceId: string,
+    key: string,
+    data: Uint8Array
+  ) => {
+    const workspaceDB = await ensureSQLiteDB(workspaceId);
+    return workspaceDB.adapter.syncMetadata.set(key, data);
+  },
+  getSyncMetadataKeys: async (workspaceId: string) => {
+    const workspaceDB = await ensureSQLiteDB(workspaceId);
+    return workspaceDB.adapter.syncMetadata.keys();
+  },
+  clearSyncMetadata: async (workspaceId: string) => {
+    const workspaceDB = await ensureSQLiteDB(workspaceId);
+    return workspaceDB.adapter.syncMetadata.clear();
+  },
+  delSyncMetadata: async (workspaceId: string, key: string) => {
+    const workspaceDB = await ensureSQLiteDB(workspaceId);
+    return workspaceDB.adapter.syncMetadata.del(key);
+  },
 };
 
 export const dbEvents = {} satisfies Record<string, MainEventRegister>;

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

@@ -31,7 +31,7 @@ export class WorkspaceSQLiteDB {
     this.update$.complete();
   }
 
-  toDBDocId = (docId: string) => {
+  private readonly toDBDocId = (docId: string) => {
     return this.workspaceId === docId ? undefined : docId;
   };