upstream-affine/packages/common/nbstore/src/impls/idb/indexer-sync.ts

import { share } from '../../connection';
import {
  type DocIndexedClock,
  IndexerSyncStorageBase,
} from '../../storage/indexer-sync';
import { IDBConnection, type IDBConnectionOptions } from './db';

export class IndexedDBIndexerSyncStorage extends IndexerSyncStorageBase {
  static readonly identifier = 'IndexedDBIndexerSyncStorage';

  readonly connection = share(new IDBConnection(this.options));

  constructor(private readonly options: IDBConnectionOptions) {
    super();
  }

  async getDocIndexedClock(docId: string): Promise<DocIndexedClock | null> {
    const tx = this.connection.inner.db.transaction('indexerSync', 'readonly');
    const store = tx.store;
    const result = await store.get(docId);
    return result
      ? {
          docId: result.docId,
          timestamp: result.indexedClock,
          indexerVersion: result.indexerVersion ?? 0,
        }
      : null;
  }

  async setDocIndexedClock(docClock: DocIndexedClock): Promise<void> {
    const tx = this.connection.inner.db.transaction('indexerSync', 'readwrite');
    const store = tx.store;
    await store.put({
      docId: docClock.docId,
      indexedClock: docClock.timestamp,
      indexerVersion: docClock.indexerVersion,
    });
  }

  async clearDocIndexedClock(docId: string): Promise<void> {
    const tx = this.connection.inner.db.transaction('indexerSync', 'readwrite');
    const store = tx.store;
    await store.delete(docId);
  }
}