hungry-ghost-hive 0.48.0 → 0.49.0

package/src/db/queries/heartbeat.ts

@@ -1,7 +1,6 @@
 // Licensed under the Hungry Ghost Hive License. See LICENSE.
 
-import type { Database } from 'sql.js';
-import { run } from '../client.js';
+import type { DatabaseProvider } from '../provider.js';
 
 /** Default timeout in seconds for considering an agent stale */
 const DEFAULT_STALE_TIMEOUT_SECONDS = 15;
@@ -9,23 +8,28 @@ const DEFAULT_STALE_TIMEOUT_SECONDS = 15;
 /**
  * Update agent's last_seen timestamp (heartbeat)
  */
-export function updateAgentHeartbeat(db: Database, agentId: string): void {
-  run(db, 'UPDATE agents SET last_seen = CURRENT_TIMESTAMP WHERE id = ?', [agentId]);
+export async function updateAgentHeartbeat(
+  provider: DatabaseProvider,
+  agentId: string
+): Promise<void> {
+  await provider.run('UPDATE agents SET last_seen = CURRENT_TIMESTAMP WHERE id = ?', [agentId]);
 }
 
 /**
  * Get agents that haven't sent a heartbeat within the specified timeout (in seconds)
  */
-export function getStaleAgents(
-  db: Database,
+export async function getStaleAgents(
+  provider: DatabaseProvider,
   timeoutSeconds: number = DEFAULT_STALE_TIMEOUT_SECONDS
-): Array<{
-  id: string;
-  type: string;
-  status: string;
-  last_seen: string | null;
-  seconds_since_heartbeat: number;
-}> {
+): Promise<
+  Array<{
+    id: string;
+    type: string;
+    status: string;
+    last_seen: string | null;
+    seconds_since_heartbeat: number;
+  }>
+> {
   const query = `
     SELECT
       id,
@@ -42,40 +46,23 @@ export function getStaleAgents(
       )
   `;
 
-  const stmt = db.prepare(query);
-  stmt.bind([timeoutSeconds, timeoutSeconds]);
-
-  const results: Array<{
+  return await provider.queryAll<{
     id: string;
     type: string;
     status: string;
     last_seen: string | null;
     seconds_since_heartbeat: number;
-  }> = [];
-
-  while (stmt.step()) {
-    const row = stmt.getAsObject() as {
-      id: string;
-      type: string;
-      status: string;
-      last_seen: string | null;
-      seconds_since_heartbeat: number;
-    };
-    results.push(row);
-  }
-  stmt.free();
-
-  return results;
+  }>(query, [timeoutSeconds, timeoutSeconds]);
 }
 
 /**
  * Check if agent's heartbeat is current (within timeout)
  */
-export function isAgentHeartbeatCurrent(
-  db: Database,
+export async function isAgentHeartbeatCurrent(
+  provider: DatabaseProvider,
   agentId: string,
   timeoutSeconds: number = DEFAULT_STALE_TIMEOUT_SECONDS
-): boolean {
+): Promise<boolean> {
   const query = `
     SELECT
       CASE
@@ -87,15 +74,6 @@ export function isAgentHeartbeatCurrent(
     WHERE id = ?
   `;
 
-  const stmt = db.prepare(query);
-  stmt.bind([timeoutSeconds, agentId]);
-
-  let isCurrent = false;
-  if (stmt.step()) {
-    const row = stmt.getAsObject() as { is_current: number };
-    isCurrent = row.is_current === 1;
-  }
-  stmt.free();
-
-  return isCurrent;
+  const row = await provider.queryOne<{ is_current: number }>(query, [timeoutSeconds, agentId]);
+  return row?.is_current === 1;
 }

package/src/db/queries/integration-sync.ts

@@ -1,8 +1,7 @@
 // Licensed under the Hungry Ghost Hive License. See LICENSE.
 
 import { nanoid } from 'nanoid';
-import type { Database } from 'sql.js';
-import { queryAll, queryOne, run } from '../client.js';
+import type { DatabaseProvider } from '../provider.js';
 
 export type SyncEntityType = 'story' | 'requirement' | 'pull_request';
 export type SyncProvider = 'jira' | 'github' | 'confluence';
@@ -28,12 +27,14 @@ export interface CreateSyncRecordInput {
   externalId: string;
 }
 
-export function createSyncRecord(db: Database, input: CreateSyncRecordInput): IntegrationSyncRow {
+export async function createSyncRecord(
+  db: DatabaseProvider,
+  input: CreateSyncRecordInput
+): Promise<IntegrationSyncRow> {
   const id = `SYNC-${nanoid(8).toUpperCase()}`;
   const now = new Date().toISOString();
 
-  run(
-    db,
+  await db.run(
     `
     INSERT INTO integration_sync (id, entity_type, entity_id, provider, external_id, last_synced_at, sync_status, created_at, updated_at)
     VALUES (?, ?, ?, ?, ?, ?, 'synced', ?, ?)
@@ -41,35 +42,36 @@ export function createSyncRecord(db: Database, input: CreateSyncRecordInput): In
     [id, input.entityType, input.entityId, input.provider, input.externalId, now, now, now]
   );
 
-  return getSyncRecordById(db, id)!;
+  return (await getSyncRecordById(db, id))!;
 }
 
-export function getSyncRecordById(db: Database, id: string): IntegrationSyncRow | undefined {
-  return queryOne<IntegrationSyncRow>(db, 'SELECT * FROM integration_sync WHERE id = ?', [id]);
+export async function getSyncRecordById(
+  db: DatabaseProvider,
+  id: string
+): Promise<IntegrationSyncRow | undefined> {
+  return await db.queryOne<IntegrationSyncRow>('SELECT * FROM integration_sync WHERE id = ?', [id]);
 }
 
-export function getSyncRecordByEntity(
-  db: Database,
+export async function getSyncRecordByEntity(
+  db: DatabaseProvider,
   entityType: SyncEntityType,
   entityId: string,
   provider: SyncProvider
-): IntegrationSyncRow | undefined {
-  return queryOne<IntegrationSyncRow>(
-    db,
+): Promise<IntegrationSyncRow | undefined> {
+  return await db.queryOne<IntegrationSyncRow>(
     'SELECT * FROM integration_sync WHERE entity_type = ? AND entity_id = ? AND provider = ?',
     [entityType, entityId, provider]
   );
 }
 
-export function updateSyncStatus(
-  db: Database,
+export async function updateSyncStatus(
+  db: DatabaseProvider,
   id: string,
   status: SyncStatus,
   errorMessage?: string | null
-): void {
+): Promise<void> {
   const now = new Date().toISOString();
-  run(
-    db,
+  await db.run(
     `
     UPDATE integration_sync
     SET sync_status = ?, error_message = ?, last_synced_at = ?, updated_at = ?
@@ -79,20 +81,18 @@ export function updateSyncStatus(
   );
 }
 
-export function getSyncRecordsByProvider(
-  db: Database,
+export async function getSyncRecordsByProvider(
+  db: DatabaseProvider,
   provider: SyncProvider
-): IntegrationSyncRow[] {
-  return queryAll<IntegrationSyncRow>(
-    db,
+): Promise<IntegrationSyncRow[]> {
+  return await db.queryAll<IntegrationSyncRow>(
     'SELECT * FROM integration_sync WHERE provider = ? ORDER BY created_at DESC',
     [provider]
   );
 }
 
-export function getFailedSyncRecords(db: Database): IntegrationSyncRow[] {
-  return queryAll<IntegrationSyncRow>(
-    db,
+export async function getFailedSyncRecords(db: DatabaseProvider): Promise<IntegrationSyncRow[]> {
+  return await db.queryAll<IntegrationSyncRow>(
     "SELECT * FROM integration_sync WHERE sync_status = 'failed' ORDER BY created_at DESC"
   );
 }