hungry-ghost-hive 0.48.0 → 0.49.0

package/src/cli/commands/my-stories.ts

@@ -3,7 +3,7 @@
 import chalk from 'chalk';
 import { Command } from 'commander';
 import { syncStatusForStory } from '../../connectors/project-management/operations.js';
-import { queryAll, queryOne, run, type StoryRow } from '../../db/client.js';
+import type { StoryRow } from '../../db/client.js';
 import { createLog } from '../../db/queries/logs.js';
 import { createStory, getStoryDependencies, updateStory } from '../../db/queries/stories.js';
 import { requireAgentBySession, requireStory } from '../../utils/cli-helpers.js';
@@ -17,8 +17,9 @@ export const myStoriesCommand = new Command('my-stories')
     await withReadOnlyHiveContext(async ({ db }) => {
       if (!session) {
         // Show all in-progress stories
-        const stories = queryAll<StoryRow & { tmux_session?: string; target_branch?: string }>(
-          db.db,
+        const stories = await db.provider.queryAll<
+          StoryRow & { tmux_session?: string; target_branch?: string }
+        >(
           `
           SELECT s.*, a.tmux_session, r.target_branch
           FROM stories s
@@ -42,8 +43,7 @@ export const myStoriesCommand = new Command('my-stories')
       }
 
       // Find agent by tmux session
-      const agent = queryOne<{ id: string; team_id: string }>(
-        db.db,
+      const agent = await db.provider.queryOne<{ id: string; team_id: string }>(
         "SELECT id, team_id FROM agents WHERE tmux_session = ? AND status != 'terminated'",
         [session]
       );
@@ -51,8 +51,7 @@ export const myStoriesCommand = new Command('my-stories')
       if (!agent) {
         console.error(chalk.red(`No agent found with session: ${session}`));
         console.log(chalk.gray('Available sessions:'));
-        const agents = queryAll<{ tmux_session: string }>(
-          db.db,
+        const agents = await db.provider.queryAll<{ tmux_session: string }>(
           "SELECT tmux_session FROM agents WHERE tmux_session IS NOT NULL AND status != 'terminated'"
         );
         for (const a of agents) {
@@ -64,8 +63,7 @@ export const myStoriesCommand = new Command('my-stories')
       let stories: (StoryRow & { target_branch?: string })[];
       if (options.all && agent.team_id) {
         // Show all team stories
-        stories = queryAll<StoryRow & { target_branch?: string }>(
-          db.db,
+        stories = await db.provider.queryAll<StoryRow & { target_branch?: string }>(
           `
           SELECT s.*, r.target_branch
           FROM stories s
@@ -85,8 +83,7 @@ export const myStoriesCommand = new Command('my-stories')
         );
       } else {
         // Show only assigned active stories (exclude merged/terminal states)
-        stories = queryAll<StoryRow & { target_branch?: string }>(
-          db.db,
+        stories = await db.provider.queryAll<StoryRow & { target_branch?: string }>(
           `
           SELECT s.*, r.target_branch
           FROM stories s
@@ -124,10 +121,10 @@ myStoriesCommand
   .action(async (storyId: string, options: { session: string }) => {
     await withHiveContext(async ({ root, db }) => {
       // Find agent by session
-      const agent = requireAgentBySession(db.db, options.session);
+      const agent = await requireAgentBySession(db.provider, options.session);
 
       // Check story exists and is available
-      const story = requireStory(db.db, storyId);
+      const story = await requireStory(db.provider, storyId);
 
       if (story.assigned_agent_id && story.assigned_agent_id !== agent.id) {
         console.error(chalk.red(`Story already assigned to another agent.`));
@@ -135,7 +132,7 @@ myStoriesCommand
       }
 
       // Check if all dependencies are resolved (merged)
-      const dependencies = getStoryDependencies(db.db, storyId);
+      const dependencies = await getStoryDependencies(db.provider, storyId);
       const unresolvedDeps = dependencies.filter(dep => dep.status !== 'merged');
       if (unresolvedDeps.length > 0) {
         console.error(chalk.red(`Cannot claim story: unresolved dependencies`));
@@ -149,8 +146,7 @@ myStoriesCommand
       }
 
       // Claim the story
-      run(
-        db.db,
+      await db.provider.run(
         `
         UPDATE stories
         SET assigned_agent_id = ?, status = 'in_progress', updated_at = datetime('now')
@@ -161,7 +157,7 @@ myStoriesCommand
       db.save();
 
       // Sync status change to Jira
-      await syncStatusForStory(root, db.db, storyId, 'in_progress');
+      await syncStatusForStory(root, db.provider, storyId, 'in_progress');
 
       console.log(chalk.green(`Claimed story: ${storyId}`));
       console.log(chalk.gray(`Title: ${story.title}`));
@@ -173,10 +169,9 @@ myStoriesCommand
   .description('Mark a story as complete (ready for review)')
   .action(async (storyId: string) => {
     await withHiveContext(async ({ root, db }) => {
-      requireStory(db.db, storyId);
+      await requireStory(db.provider, storyId);
 
-      run(
-        db.db,
+      await db.provider.run(
         `
         UPDATE stories
         SET status = 'review', updated_at = datetime('now')
@@ -187,7 +182,7 @@ myStoriesCommand
       db.save();
 
       // Sync status change to Jira
-      await syncStatusForStory(root, db.db, storyId, 'review');
+      await syncStatusForStory(root, db.provider, storyId, 'review');
 
       console.log(chalk.green(`Story ${storyId} marked as ready for review.`));
     });
@@ -229,7 +224,7 @@ myStoriesCommand
       }
 
       await withHiveContext(async ({ paths, db }) => {
-        const agent = requireAgentBySession(db.db, options.session);
+        const agent = await requireAgentBySession(db.provider, options.session);
 
         if (!agent.team_id) {
           console.error(
@@ -245,8 +240,8 @@ myStoriesCommand
           ? trimmedTitle
           : `Refactor: ${trimmedTitle}`;
 
-        const story = createStory(
-          db.db,
+        const story = await createStory(
+          db.provider,
           {
             teamId: agent.team_id,
             title: normalizedTitle,
@@ -257,8 +252,8 @@ myStoriesCommand
           paths.storiesDir
         );
 
-        const updatedStory = updateStory(
-          db.db,
+        const updatedStory = await updateStory(
+          db.provider,
           story.id,
           {
             complexityScore: points,
@@ -268,7 +263,7 @@ myStoriesCommand
           paths.storiesDir
         );
 
-        createLog(db.db, {
+        await createLog(db.provider, {
           agentId: agent.id,
           storyId: story.id,
           eventType: 'STORY_CREATED',

package/src/cli/commands/nuke.test.ts

@@ -19,7 +19,7 @@ vi.mock('../../git/worktree.js', () => ({
 
 vi.mock('../../utils/with-hive-context.js', () => ({
   withHiveContext: vi.fn(callback =>
-    callback({ db: { db: {}, save: vi.fn() }, root: '/root', paths: {} })
+    callback({ db: { db: {}, provider: {}, save: vi.fn() }, root: '/root', paths: {} })
   ),
 }));
 

package/src/cli/commands/pr.test.ts

@@ -89,9 +89,13 @@ vi.mock('../../utils/story-id.js', () => ({
 
 vi.mock('../../utils/with-hive-context.js', () => ({
   withHiveContext: vi.fn(callback =>
-    callback({ db: { db: {}, save: vi.fn() }, root: '/tmp', paths: { hiveDir: '/tmp/.hive' } })
+    callback({
+      db: { db: {}, provider: {}, save: vi.fn() },
+      root: '/tmp',
+      paths: { hiveDir: '/tmp/.hive' },
+    })
   ),
-  withReadOnlyHiveContext: vi.fn(callback => callback({ db: { db: {} } })),
+  withReadOnlyHiveContext: vi.fn(callback => callback({ db: { db: {}, provider: {} } })),
 }));
 
 import { prCommand } from './pr.js';
@@ -114,7 +118,7 @@ describe('pr command', () => {
     vi.clearAllMocks();
     resetCommandOptions(prCommand);
 
-    vi.mocked(getPullRequestById).mockReturnValue({
+    vi.mocked(getPullRequestById).mockResolvedValue({
       id: 'pr-1',
       story_id: 'TEST-1',
       team_id: 'team-1',
@@ -216,7 +220,7 @@ describe('pr command', () => {
     });
 
     it('should auto-close existing PRs with different github_pr_number', async () => {
-      vi.mocked(getOpenPullRequestsByStory).mockReturnValue([
+      vi.mocked(getOpenPullRequestsByStory).mockResolvedValue([
         {
           id: 'old-pr-1',
           story_id: 'TEST-1',
@@ -267,7 +271,7 @@ describe('pr command', () => {
     });
 
     it('should not close GitHub PR when auto-closing duplicate internal PR records', async () => {
-      vi.mocked(getOpenPullRequestsByStory).mockReturnValue([
+      vi.mocked(getOpenPullRequestsByStory).mockResolvedValue([
         {
           id: 'old-pr-1',
           story_id: 'TEST-1',
@@ -310,7 +314,7 @@ describe('pr command', () => {
     });
 
     it('should skip auto-close when resubmitting same github PR number', async () => {
-      vi.mocked(getOpenPullRequestsByStory).mockReturnValue([
+      vi.mocked(getOpenPullRequestsByStory).mockResolvedValue([
         {
           id: 'existing-pr-1',
           story_id: 'TEST-1',

package/src/cli/commands/pr.ts

@@ -56,20 +56,20 @@ prCommand
 
         // Get team from story
         let teamId = options.team || null;
-        const story = requireStory(db.db, storyId);
+        const story = await requireStory(db.provider, storyId);
 
         teamId = story.team_id;
 
         // Auto-close any existing open PRs for this story
         const incomingPrNumber = options.prNumber ? parseInt(options.prNumber, 10) : null;
-        const existingPRs = getOpenPullRequestsByStory(db.db, storyId);
+        const existingPRs = await getOpenPullRequestsByStory(db.provider, storyId);
         for (const existingPR of existingPRs) {
           // Skip auto-close if this is a resubmit of the same GitHub PR
           if (incomingPrNumber !== null && existingPR.github_pr_number === incomingPrNumber) {
             continue;
           }
-          updatePullRequest(db.db, existingPR.id, { status: 'closed' });
-          createLog(db.db, {
+          await updatePullRequest(db.provider, existingPR.id, { status: 'closed' });
+          await createLog(db.provider, {
             agentId: options.from || 'system',
             storyId,
             eventType: 'PR_CLOSED',
@@ -79,12 +79,12 @@ prCommand
         }
 
         // Update story status
-        updateStory(db.db, storyId, { status: 'pr_submitted' });
+        await updateStory(db.provider, storyId, { status: 'pr_submitted' });
 
         // Sync status change to Jira
-        await syncStatusForStory(root, db.db, storyId, 'pr_submitted');
+        await syncStatusForStory(root, db.provider, storyId, 'pr_submitted');
 
-        const pr = createPullRequest(db.db, {
+        const pr = await createPullRequest(db.provider, {
           storyId,
           teamId,
           branchName: options.branch,
@@ -95,7 +95,7 @@ prCommand
 
         db.save();
 
-        const position = getQueuePosition(db.db, pr.id);
+        const position = await getQueuePosition(db.provider, pr.id);
 
         console.log(chalk.green(`PR submitted to merge queue`));
         console.log(chalk.gray(`  ID: ${pr.id}`));
@@ -106,7 +106,7 @@ prCommand
         }
 
         if (options.from) {
-          createLog(db.db, {
+          await createLog(db.provider, {
             agentId: options.from,
             storyId: storyId || undefined,
             eventType: 'PR_SUBMITTED',
@@ -119,7 +119,7 @@ prCommand
         // Post Jira comment for PR created event
         try {
           const config = loadConfig(paths.hiveDir);
-          await postLifecycleComment(db.db, paths.hiveDir, config, storyId, 'pr_created', {
+          await postLifecycleComment(db.provider, paths.hiveDir, config, storyId, 'pr_created', {
             agentName: options.from,
             prUrl: pr.github_pr_url || undefined,
           });
@@ -135,7 +135,7 @@ prCommand
             : null;
 
           if (!config.cluster.enabled || clusterStatus?.is_leader) {
-            const scheduler = new Scheduler(db.db, {
+            const scheduler = new Scheduler(db.provider, {
               scaling: config.scaling,
               models: config.models,
               qa: config.qa,
@@ -162,7 +162,7 @@ prCommand
   .option('--json', 'Output as JSON')
   .action(async (options: { team?: string; json?: boolean }) => {
     await withReadOnlyHiveContext(async ({ db }) => {
-      const queue = getMergeQueue(db.db, options.team);
+      const queue = await getMergeQueue(db.provider, options.team);
 
       if (options.json) {
         console.log(JSON.stringify(queue, null, 2));
@@ -204,14 +204,14 @@ prCommand
   .option('--from <session>', 'QA agent session')
   .action(async (options: { team?: string; from?: string }) => {
     await withHiveContext(async ({ db }) => {
-      const pr = getNextInQueue(db.db, options.team);
+      const pr = await getNextInQueue(db.provider, options.team);
 
       if (!pr) {
         console.log(chalk.yellow('No PRs waiting for review.'));
         return;
       }
 
-      updatePullRequest(db.db, pr.id, {
+      await updatePullRequest(db.provider, pr.id, {
         status: 'reviewing',
         reviewedBy: options.from || null,
       });
@@ -230,7 +230,7 @@ prCommand
       console.log(chalk.gray(`  hive pr reject ${pr.id} --reason "..."`));
 
       if (options.from) {
-        createLog(db.db, {
+        await createLog(db.provider, {
           agentId: options.from,
           storyId: pr.story_id || undefined,
           eventType: 'PR_REVIEW_STARTED',
@@ -248,7 +248,7 @@ prCommand
   .description('View details of a PR')
   .action(async (prId: string) => {
     await withReadOnlyHiveContext(async ({ db }) => {
-      const pr = requirePullRequest(db.db, prId);
+      const pr = await requirePullRequest(db.provider, prId);
 
       console.log(chalk.bold(`\nPull Request: ${pr.id}\n`));
       console.log(chalk.gray(`Branch:       ${pr.branch_name}`));
@@ -267,7 +267,7 @@ prCommand
         console.log(pr.review_notes);
       }
 
-      const position = getQueuePosition(db.db, pr.id);
+      const position = await getQueuePosition(db.provider, pr.id);
       if (position > 0) {
         console.log(chalk.cyan(`\nQueue Position: ${position}`));
       }
@@ -284,7 +284,7 @@ prCommand
   .option('--no-merge', 'Approve without merging (manual merge needed)')
   .action(async (prId: string, options: { notes?: string; from?: string; merge?: boolean }) => {
     await withHiveContext(async ({ root, db }) => {
-      const pr = requirePullRequest(db.db, prId);
+      const pr = await requirePullRequest(db.provider, prId);
 
       if (pr.status === 'merged') {
         console.log(chalk.yellow('PR already merged.'));
@@ -305,7 +305,7 @@ prCommand
         // Use the team's repo path as cwd so gh knows which repo to operate on
         let repoCwd = root;
         if (pr.team_id) {
-          const team = getTeamById(db.db, pr.team_id);
+          const team = await getTeamById(db.provider, pr.team_id);
           if (team?.repo_path) {
             repoCwd = join(root, team.repo_path);
           }
@@ -347,21 +347,21 @@ prCommand
         ? markManualMergeRequired(options.notes)
         : (options.notes ?? null);
 
-      updatePullRequest(db.db, prId, {
+      await updatePullRequest(db.provider, prId, {
         status: newStatus,
         reviewedBy: options.from || pr.reviewed_by,
         reviewNotes,
       });
 
       if (storyId && newStatus === 'merged') {
-        updateStory(db.db, storyId, { status: 'merged' });
+        await updateStory(db.provider, storyId, { status: 'merged' });
       }
 
       db.save();
 
       // Sync status change to Jira
       if (storyId && newStatus === 'merged') {
-        await syncStatusForStory(root, db.db, storyId, 'merged');
+        await syncStatusForStory(root, db.provider, storyId, 'merged');
       }
 
       // Immediately attempt to auto-merge approved PRs instead of waiting for manager daemon cycle
@@ -386,7 +386,7 @@ prCommand
       }
 
       if (options.from) {
-        createLog(db.db, {
+        await createLog(db.provider, {
           agentId: options.from,
           storyId: storyId || undefined,
           eventType: newStatus === 'merged' ? 'PR_MERGED' : 'PR_APPROVED',
@@ -406,9 +406,9 @@ prCommand
   .option('--from <session>', 'QA agent session')
   .action(async (prId: string, options: { reason: string; from?: string }) => {
     await withHiveContext(async ({ root, db }) => {
-      const pr = requirePullRequest(db.db, prId);
+      const pr = await requirePullRequest(db.provider, prId);
 
-      updatePullRequest(db.db, prId, {
+      await updatePullRequest(db.provider, prId, {
         status: 'rejected',
         reviewedBy: options.from || pr.reviewed_by,
         reviewNotes: options.reason,
@@ -420,14 +420,14 @@ prCommand
         storyId = extractStoryIdFromBranch(pr.branch_name);
       }
       if (storyId) {
-        updateStory(db.db, storyId, { status: 'qa_failed' });
+        await updateStory(db.provider, storyId, { status: 'qa_failed' });
       }
 
       db.save();
 
       // Sync status change to Jira
       if (storyId) {
-        await syncStatusForStory(root, db.db, storyId, 'qa_failed');
+        await syncStatusForStory(root, db.provider, storyId, 'qa_failed');
       }
 
       console.log(chalk.yellow(`PR ${prId} rejected.`));
@@ -461,7 +461,7 @@ prCommand
       }
 
       if (options.from) {
-        createLog(db.db, {
+        await createLog(db.provider, {
           agentId: options.from,
           storyId: storyId || undefined,
           eventType: 'PR_REJECTED',
@@ -480,7 +480,10 @@ prCommand
   .option('-r, --repo <path>', 'Repository path (relative to repos/)')
   .action(async (options: { repo?: string }) => {
     await withHiveContext(async ({ root, paths, db }) => {
-      const { existingBranches, existingPrNumbers } = getExistingPRIdentifiers(db.db, false);
+      const { existingBranches, existingPrNumbers } = await getExistingPRIdentifiers(
+        db.provider,
+        false
+      );
 
       // Find repo directory
       const repoDir = options.repo ? `${root}/repos/${options.repo}` : process.cwd();
@@ -489,7 +492,13 @@ prCommand
 
       let result;
       try {
-        result = await syncOpenGitHubPRs(db.db, repoDir, null, existingBranches, existingPrNumbers);
+        result = await syncOpenGitHubPRs(
+          db.provider,
+          repoDir,
+          null,
+          existingBranches,
+          existingPrNumbers
+        );
       } catch (err) {
         console.error(chalk.red('Failed to list GitHub PRs. Is gh CLI authenticated?'), err);
         process.exit(1);
@@ -512,7 +521,7 @@ prCommand
             : null;
 
           if (!config.cluster.enabled || clusterStatus?.is_leader) {
-            const scheduler = new Scheduler(db.db, {
+            const scheduler = new Scheduler(db.provider, {
               scaling: config.scaling,
               models: config.models,
               rootDir: root,
@@ -541,7 +550,7 @@ prCommand
   .action(async (options: { limit?: string; json?: boolean }) => {
     await withReadOnlyHiveContext(async ({ db }) => {
       const limit = parseInt(options.limit ?? '20', 10);
-      const logs = getLogsByEventType(db.db, 'PR_CLOSED', limit);
+      const logs = await getLogsByEventType(db.provider, 'PR_CLOSED', limit);
 
       if (options.json) {
         console.log(JSON.stringify(logs, null, 2));

package/src/cli/commands/progress.test.ts

@@ -44,7 +44,7 @@ vi.mock('../../integrations/jira/comments.js', () => ({
 
 vi.mock('../../utils/with-hive-context.js', () => ({
   withHiveContext: vi.fn(callback =>
-    callback({ root: '/tmp', db: { db: {} }, paths: { hiveDir: '/tmp/.hive' } })
+    callback({ root: '/tmp', db: { db: {}, provider: {} }, paths: { hiveDir: '/tmp/.hive' } })
   ),
 }));
 

package/src/cli/commands/progress.ts

@@ -7,7 +7,6 @@ import {
   postProgressUpdate,
   transitionSubtaskStatus,
 } from '../../connectors/project-management/operations.js';
-import { queryOne } from '../../db/client.js';
 import { createLog } from '../../db/queries/logs.js';
 import { requireStory } from '../../utils/cli-helpers.js';
 import { withHiveContext } from '../../utils/with-hive-context.js';
@@ -24,7 +23,7 @@ export const progressCommand = new Command('progress')
       const pmProvider = config.integrations?.project_management?.provider || 'none';
 
       if (pmProvider === 'none') {
-        createLog(db.db, {
+        await createLog(db.provider, {
           agentId: options.from || 'manager',
           storyId,
           eventType: 'STORY_PROGRESS_UPDATE',
@@ -43,7 +42,7 @@ export const progressCommand = new Command('progress')
         return;
       }
 
-      const story = requireStory(db.db, storyId);
+      const story = await requireStory(db.provider, storyId);
 
       if (!story.external_subtask_key) {
         console.error(
@@ -60,8 +59,7 @@ export const progressCommand = new Command('progress')
       // Resolve agent name from session if provided
       let agentName = options.from;
       if (options.from) {
-        const agent = queryOne<{ id: string; tmux_session: string }>(
-          db.db,
+        const agent = await db.provider.queryOne<{ id: string; tmux_session: string }>(
           "SELECT id, tmux_session FROM agents WHERE tmux_session = ? AND status != 'terminated'",
           [options.from]
         );
@@ -70,7 +68,14 @@ export const progressCommand = new Command('progress')
         }
       }
 
-      await postProgressUpdate(db.db, paths.hiveDir, config, storyId, options.message, agentName);
+      await postProgressUpdate(
+        db.provider,
+        paths.hiveDir,
+        config,
+        storyId,
+        options.message,
+        agentName
+      );
 
       console.log(chalk.green(`Posted progress update to subtask ${story.external_subtask_key}`));