@sixb/sqlite 0.1.19 → 0.1.21

package/src/rules-storage.ts

@@ -11,7 +11,6 @@ import type {
   StoredRuleResolvedEvent,
   StoredRuleTriggeredEvent,
 } from "@sixb/core/storage"
-import { installFreshSqliteSchema } from "./migrations"
 import {
   closeSqliteStoreConnection,
   openSqliteStoreConnection,
@@ -33,10 +32,6 @@ export class SqliteRulesStorage implements RulesStorage {
   constructor(options: SqliteRulesStorageOptions = {}) {
     this.connection = openSqliteStoreConnection(options)
     this.db = this.connection.db
-
-    if (this.connection.installFreshSchema) {
-      installFreshSqliteSchema(this.db)
-    }
   }
 
   async getActive(params: {

package/src/share-grant-storage.ts

@@ -9,6 +9,7 @@ import {
 import type {
   CreateShareGrantInput,
   GetShareGrantByIdInput,
+  GetShareGrantsByIdsInput,
   ListShareGrantsInput,
   ListShareGrantsResult,
   RevokeShareGrantInput,
@@ -16,7 +17,6 @@ import type {
   ShareGrantStorage,
 } from "@sixb/core/storage"
 import { ShareGrantStorageError } from "@sixb/core/storage"
-import { installFreshSqliteSchema } from "./migrations"
 import { isUniqueConstraintError } from "./storage-errors"
 import {
   closeSqliteStoreConnection,
@@ -37,7 +37,6 @@ export class SqliteShareGrantStorage implements ShareGrantStorage {
   constructor(options: SqliteShareGrantStorageOptions = {}) {
     this.connection = openSqliteStoreConnection(options)
     this.db = this.connection.db
-    if (this.connection.installFreshSchema) installFreshSqliteSchema(this.db)
   }
 
   async create(input: CreateShareGrantInput): Promise<ShareGrantRecord> {
@@ -87,6 +86,30 @@ export class SqliteShareGrantStorage implements ShareGrantStorage {
     return this.getRow(normalized.projectId, normalized.id)
   }
 
+  async getByIds(input: GetShareGrantsByIdsInput): Promise<readonly ShareGrantRecord[]> {
+    const ids = [...new Set(input.ids)]
+    if (ids.length === 0) {
+      return []
+    }
+
+    const rows = this.db
+      .query(
+        `
+        SELECT *
+        FROM share_grants
+        WHERE project_id = ?
+          AND id IN (SELECT value FROM json_each(?))
+      `
+      )
+      .all(input.projectId, JSON.stringify(ids)) as SqliteShareGrantRow[]
+
+    const byId = new Map(rows.map((row) => [row.id, row]))
+    return ids.flatMap((id) => {
+      const row = byId.get(id)
+      return row ? [rowToRecord(row)] : []
+    })
+  }
+
   async list(input: ListShareGrantsInput): Promise<ListShareGrantsResult> {
     const normalized = normalizeListShareGrantsInput(input)
     const where = ["project_id = ?"]

package/src/share-session-storage.ts

@@ -15,7 +15,6 @@ import type {
   ShareSessionStorage,
 } from "@sixb/core/storage"
 import { ShareSessionStorageError } from "@sixb/core/storage"
-import { installFreshSqliteSchema } from "./migrations"
 import { isForeignKeyConstraintError, isUniqueConstraintError } from "./storage-errors"
 import {
   closeSqliteStoreConnection,
@@ -36,7 +35,6 @@ export class SqliteShareSessionStorage implements ShareSessionStorage {
   constructor(options: SqliteShareSessionStorageOptions = {}) {
     this.connection = openSqliteStoreConnection(options)
     this.db = this.connection.db
-    if (this.connection.installFreshSchema) installFreshSqliteSchema(this.db)
   }
 
   async create(input: CreateShareSessionInput): Promise<ShareSessionRecord> {

package/src/sync-run-storage.ts

@@ -17,7 +17,6 @@ import type {
 } from "@sixb/core/storage"
 import { SYNC_RUN_FAILURE_CODES, SyncRunError } from "@sixb/core/storage"
 import { queryLatestRunsByOwnerId } from "./latest-run-query"
-import { installFreshSqliteSchema } from "./migrations"
 import {
   appendRunListFilters,
   hasEmptyStatuses,
@@ -48,10 +47,6 @@ export class SqliteSyncRunStorage implements SyncRunStorage {
     this.connection = openSqliteStoreConnection(options)
     this.db = this.connection.db
     this.executions = options.executions
-
-    if (this.connection.installFreshSchema) {
-      installFreshSqliteSchema(this.db)
-    }
   }
 
   async queue(input: QueueSyncRunInput): Promise<SyncRunRecord> {

package/src/timeseries-storage.ts

@@ -5,7 +5,6 @@ import type {
   TimeseriesPoint,
   TimeseriesStorage,
 } from "@sixb/core/storage"
-import { installFreshSqliteSchema } from "./migrations"
 import {
   closeSqliteStoreConnection,
   openSqliteStoreConnection,
@@ -31,10 +30,6 @@ export class SqliteTimeseriesStorage implements TimeseriesStorage {
   constructor(options: SqliteTimeseriesStorageOptions = {}) {
     this.connection = openSqliteStoreConnection(options)
     this.db = this.connection.db
-
-    if (this.connection.installFreshSchema) {
-      installFreshSqliteSchema(this.db)
-    }
   }
 
   async getHistory(params: {

package/src/transactions.ts

@@ -2,11 +2,11 @@ import type { Database } from "bun:sqlite"
 import { Database as SqliteDatabase } from "bun:sqlite"
 import { mkdirSync } from "node:fs"
 import { dirname } from "node:path"
+import { freshSqliteDatabase } from "./migrations"
 
 export interface SqliteStoreConnection {
   readonly db: Database
   readonly ownsConnection: boolean
-  readonly installFreshSchema: boolean
 }
 
 export interface SqliteStoreConnectionOptions {
@@ -22,20 +22,20 @@ export function openSqliteStoreConnection(
     return {
       db: options.connection.db,
       ownsConnection: false,
-      installFreshSchema: false,
     }
   }
 
   const path = options.path ?? ":memory:"
   if (path !== ":memory:" && !options.readonly) mkdirSync(dirname(path), { recursive: true })
-  const db = new SqliteDatabase(path, options.readonly ? { readonly: true } : undefined)
+  // A file is brought up to date by its migrators; a writable in-memory database starts with the
+  // whole schema.
+  const db =
+    path === ":memory:" && !options.readonly
+      ? freshSqliteDatabase()
+      : new SqliteDatabase(path, options.readonly ? { readonly: true } : undefined)
   db.run("PRAGMA foreign_keys = ON")
   db.run("PRAGMA busy_timeout = 5000")
-  return {
-    db,
-    ownsConnection: true,
-    installFreshSchema: path === ":memory:" && !options.readonly,
-  }
+  return { db, ownsConnection: true }
 }
 
 export function closeSqliteStoreConnection(connection: SqliteStoreConnection): void {

package/src/webhook-run-storage.ts

@@ -13,7 +13,6 @@ import type {
   WebhookRunStorage,
 } from "@sixb/core/storage"
 import { canRetryWebhookRun, WEBHOOK_RUN_FAILURE_CODES, WebhookRunError } from "@sixb/core/storage"
-import { installFreshSqliteSchema } from "./migrations"
 import {
   appendRunListFilters,
   hasEmptyStatuses,
@@ -45,8 +44,6 @@ export class SqliteWebhookRunStorage implements WebhookRunStorage {
     this.connection = openSqliteStoreConnection(options)
     this.db = this.connection.db
     this.executions = options.executions
-
-    if (this.connection.installFreshSchema) installFreshSqliteSchema(this.db)
   }
 
   async start(input: StartWebhookRunInput): Promise<WebhookRunRecord> {

package/src/workflow-intervention-storage.ts

@@ -12,7 +12,6 @@ import type {
   WorkflowInterventionStorage,
 } from "@sixb/core/storage"
 import { WorkflowInterventionError } from "@sixb/core/storage"
-import { installFreshSqliteSchema } from "./migrations"
 import { isUniqueConstraintError } from "./storage-errors"
 import {
   closeSqliteStoreConnection,
@@ -34,10 +33,6 @@ export class SqliteWorkflowInterventionStorage implements WorkflowInterventionSt
   constructor(options: SqliteWorkflowInterventionStorageOptions = {}) {
     this.connection = openSqliteStoreConnection(options)
     this.db = this.connection.db
-
-    if (this.connection.installFreshSchema) {
-      installFreshSqliteSchema(this.db)
-    }
   }
 
   async create(input: CreateWorkflowInterventionInput): Promise<WorkflowInterventionRecord> {

package/src/workflow-run-storage.ts

@@ -44,7 +44,6 @@ import {
   WorkflowRunError,
 } from "@sixb/core/storage"
 import { queryLatestRunsByOwnerId } from "./latest-run-query"
-import { installFreshSqliteSchema } from "./migrations"
 import {
   appendRunListFilters,
   hasEmptyStatuses,
@@ -80,10 +79,6 @@ export class SqliteWorkflowRunStorage implements WorkflowRunStorage {
     this.db = this.connection.db
     this.executions = options.executions
 
-    if (this.connection.installFreshSchema) {
-      installFreshSqliteSchema(this.db)
-    }
-
     this.nodes = new SqliteWorkflowNodeRunStorage(this.db)
     this.agentNodes = new SqliteWorkflowAgentNodeRunStorage(this.db, this.executions)
   }
@@ -363,6 +358,33 @@ export class SqliteWorkflowRunStorage implements WorkflowRunStorage {
     return row ? rowToWorkflowRunRecord(row) : null
   }
 
+  async getByExecutionIds(params: {
+    readonly projectId: string
+    readonly executionIds: readonly string[]
+  }): Promise<readonly WorkflowRunRecord[]> {
+    const executionIds = [...new Set(params.executionIds)]
+    if (executionIds.length === 0) {
+      return []
+    }
+
+    const rows = this.db
+      .query(
+        `
+        SELECT *
+        FROM workflow_runs
+        WHERE project_id = ?
+          AND execution_id IN (SELECT value FROM json_each(?))
+      `
+      )
+      .all(params.projectId, JSON.stringify(executionIds)) as WorkflowRunDatabaseRow[]
+
+    const byExecution = new Map(rows.map((row) => [row.execution_id, row]))
+    return executionIds.flatMap((executionId) => {
+      const row = byExecution.get(executionId)
+      return row ? [rowToWorkflowRunRecord(row)] : []
+    })
+  }
+
   async list(input: ListWorkflowRunsInput): Promise<ListWorkflowRunsResult> {
     if (hasEmptyStatuses(input) || input.workflowIds?.length === 0) {
       return {