packages/storage/src/sqlite-artifact-metadata.ts

import { createHash } from 'node:crypto';
import { resolve } from 'node:path';
import type { ArtifactRecord } from '@maka/core';
import type { ArtifactMetadataRepository } from './artifact-metadata-repository.js';
import { decodeArtifactRecordJsons } from './artifact-metadata-codec.js';
import {
  acquireOperationalStateDatabase,
  type OperationalStateDatabaseLease,
} from './operational-state-store.js';

export function createSqliteArtifactMetadataRepository(
  workspaceRoot: string,
): ArtifactMetadataRepository {
  return new SqliteArtifactMetadataRepository(workspaceRoot);
}

class SqliteArtifactMetadataRepository implements ArtifactMetadataRepository {
  readonly #lease: OperationalStateDatabaseLease;
  #closed = false;

  constructor(workspaceRoot: string) {
    this.#lease = acquireOperationalStateDatabase(resolve(workspaceRoot));
  }

  ready(): Promise<void> {
    this.assertOpen();
    return Promise.resolve();
  }

  readAll(): ArtifactRecord[] {
    this.assertOpen();
    const rows = this.#lease.database
      .prepare(`
        SELECT record_json
        FROM artifact_records
        ORDER BY created_at, storage_key
      `)
      .all() as Array<{ record_json: string }>;
    return decodeRows(rows);
  }

  replaceAll(records: readonly ArtifactRecord[]): void {
    this.assertOpen();
    this.#lease.transaction('write', () => {
      this.#lease.database.prepare('DELETE FROM artifact_records').run();
      const insert = this.#lease.database.prepare(`
        INSERT INTO artifact_records(
          storage_key,
          artifact_id,
          session_id,
          created_at,
          status,
          relative_path,
          record_json
        ) VALUES (?, ?, ?, ?, ?, ?, ?)
      `);
      for (const record of records) {
        insert.run(
          artifactIdentityKey(record.id),
          record.id,
          record.sessionId,
          record.createdAt,
          record.status,
          record.relativePath,
          JSON.stringify(record),
        );
      }
    });
  }

  close(): void {
    if (this.#closed) return;
    this.#closed = true;
    this.#lease.close();
  }

  private assertOpen(): void {
    if (this.#closed) throw new Error('Artifact metadata repository is closed');
  }
}

function decodeRows(rows: readonly { record_json: string }[]): ArtifactRecord[] {
  return decodeArtifactRecordJsons(rows.map((row) => row.record_json));
}

function artifactIdentityKey(id: string): string {
  return createHash('sha256').update(JSON.stringify(id)).digest('hex');
}