astrocode-workflow 0.4.4 → 0.4.6

package/src/tools/status.ts

@@ -1,8 +1,10 @@
 import { tool, type ToolDefinition } from "@opencode-ai/plugin/tool";
 import type { AstrocodeConfig } from "../config/schema";
-import type { SqliteDb } from "../state/db";
+import type { RuntimeState } from "../state/types";
 import { decideNextAction, getActiveRun, getStageRuns, getStory } from "../workflow/state-machine";
-
+import { openSqlite, configurePragmas, ensureSchema } from "../state/db";
+import { getAstroPaths } from "../shared/paths";
+import fs from "node:fs";
 
 function statusIcon(status: string): string {
   switch (status) {
@@ -36,8 +38,8 @@ function stageIcon(status: string): string {
   }
 }
 
-export function createAstroStatusTool(opts: { ctx: any; config: AstrocodeConfig; db?: SqliteDb | null }): ToolDefinition {
-  const { ctx, config, db } = opts;
+export function createAstroStatusTool(opts: { ctx: any; config: AstrocodeConfig; runtime: RuntimeState }): ToolDefinition {
+  const { ctx, config, runtime } = opts;
 
   return tool({
     description: "Show a compact Astrocode status dashboard: active run/stage, pipeline, story board counts, and next action.",
@@ -46,18 +48,39 @@ export function createAstroStatusTool(opts: { ctx: any; config: AstrocodeConfig;
       include_recent_events: tool.schema.boolean().default(false),
     },
     execute: async ({ include_board, include_recent_events }) => {
+      // Lazy initialization: if DB is missing but file exists, try to connect
+      if (!runtime.db) {
+        const repoRoot = (ctx as any).directory || process.cwd();
+        const paths = getAstroPaths(repoRoot, config.db.path);
+        if (fs.existsSync(paths.dbPath)) {
+          try {
+            const db = openSqlite(paths.dbPath, { busyTimeoutMs: config.db.busy_timeout_ms });
+            configurePragmas(db, config.db.pragmas);
+            ensureSchema(db, { allowAutoMigrate: config.db.allow_auto_migrate, silent: true });
+            runtime.db = db;
+            runtime.limitedMode = false;
+            runtime.limitedModeReason = null;
+          } catch {
+            // Ignore lazy init failures, will fall through to "not initialized" message
+          }
+        }
+      }
+
+      const { db } = runtime;
+
       if (!db) {
         return [
           `⚠️ Astrocode not initialized.`,
           ``,
           `- Reason: Database not available`,
           ``,
-          `Next: run **astro_init**, then restart the agent/runtime, then run /astro-status.`,
+          `Next: run **astro_init**, then run /astro-status again.`,
         ].join("\n");
       }
 
       try {
         const active = getActiveRun(db);
+        // ... rest of existing logic ...
 
         const lines: string[] = [];
         lines.push(`# Astrocode Status`);

package/src/tools/story.ts

@@ -1,6 +1,6 @@
 import { tool, type ToolDefinition } from "@opencode-ai/plugin/tool";
 import type { AstrocodeConfig } from "../config/schema";
-import type { SqliteDb } from "../state/db";
+import type { RuntimeState } from "../state/types";
 import { withTx } from "../state/db";
 import { nowISO } from "../shared/time";
 import type { StoryState } from "../state/types";
@@ -9,8 +9,8 @@ import { insertStory } from "../workflow/story-helpers";
 
 
 
-export function createAstroStoryQueueTool(opts: { ctx: any; config: AstrocodeConfig; db: SqliteDb }): ToolDefinition {
-  const { db } = opts;
+export function createAstroStoryQueueTool(opts: { ctx: any; config: AstrocodeConfig; runtime: RuntimeState }): ToolDefinition {
+  const { runtime } = opts;
 
   return tool({
     description: "Create a queued story (ticket) in Astrocode. Returns story_key.",
@@ -21,6 +21,9 @@ export function createAstroStoryQueueTool(opts: { ctx: any; config: AstrocodeCon
       priority: tool.schema.number().int().default(0),
     },
     execute: async ({ title, body_md, epic_key, priority }) => {
+      const { db } = runtime;
+      if (!db) return "⚠️ Astrocode not initialized. Run **astro_init** first.";
+
       const story_key = withTx(db, () => {
         const key = insertStory(db, { title, body_md, epic_key: epic_key ?? null, priority: priority ?? 0, state: 'queued' });
         return key;
@@ -31,8 +34,8 @@ export function createAstroStoryQueueTool(opts: { ctx: any; config: AstrocodeCon
   });
 }
 
-export function createAstroStoryApproveTool(opts: { ctx: any; config: AstrocodeConfig; db: SqliteDb }): ToolDefinition {
-  const { db } = opts;
+export function createAstroStoryApproveTool(opts: { ctx: any; config: AstrocodeConfig; runtime: RuntimeState }): ToolDefinition {
+  const { runtime } = opts;
 
   return tool({
     description: "Approve a story so it becomes eligible to run.",
@@ -40,6 +43,9 @@ export function createAstroStoryApproveTool(opts: { ctx: any; config: AstrocodeC
       story_key: tool.schema.string().min(1),
     },
     execute: async ({ story_key }) => {
+      const { db } = runtime;
+      if (!db) return "⚠️ Astrocode not initialized. Run **astro_init** first.";
+
       const now = nowISO();
       const row = db.prepare("SELECT story_key, state, title FROM stories WHERE story_key=?").get(story_key) as any;
       if (!row) throw new Error(`Story not found: ${story_key}`);
@@ -52,8 +58,8 @@ export function createAstroStoryApproveTool(opts: { ctx: any; config: AstrocodeC
   });
 }
 
-export function createAstroStoryBoardTool(opts: { ctx: any; config: AstrocodeConfig; db: SqliteDb }): ToolDefinition {
-  const { db } = opts;
+export function createAstroStoryBoardTool(opts: { ctx: any; config: AstrocodeConfig; runtime: RuntimeState }): ToolDefinition {
+  const { runtime } = opts;
 
   return tool({
     description: "Show stories grouped by state (a compact board).",
@@ -61,6 +67,9 @@ export function createAstroStoryBoardTool(opts: { ctx: any; config: AstrocodeCon
       limit_per_state: tool.schema.number().int().positive().default(20),
     },
     execute: async ({ limit_per_state }) => {
+      const { db } = runtime;
+      if (!db) return "⚠️ Astrocode not initialized. Run **astro_init** first.";
+
       const states: StoryState[] = ["queued", "approved", "in_progress", "blocked", "done", "archived"];
       const lines: string[] = [];
       lines.push("# Story board");
@@ -80,8 +89,8 @@ export function createAstroStoryBoardTool(opts: { ctx: any; config: AstrocodeCon
   });
 }
 
-export function createAstroStorySetStateTool(opts: { ctx: any; config: AstrocodeConfig; db: SqliteDb }): ToolDefinition {
-  const { db } = opts;
+export function createAstroStorySetStateTool(opts: { ctx: any; config: AstrocodeConfig; runtime: RuntimeState }): ToolDefinition {
+  const { runtime } = opts;
 
   return tool({
     description:
@@ -92,6 +101,9 @@ export function createAstroStorySetStateTool(opts: { ctx: any; config: Astrocode
       note: tool.schema.string().default(""),
     },
     execute: async ({ story_key, state, note }) => {
+      const { db } = runtime;
+      if (!db) return "⚠️ Astrocode not initialized. Run **astro_init** first.";
+
       const now = nowISO();
       const row = db.prepare("SELECT story_key, title, state FROM stories WHERE story_key=?").get(story_key) as any;
       if (!row) throw new Error(`Story not found: ${story_key}`);

package/src/tools/workflow.ts

@@ -2,7 +2,7 @@
 import { tool, type ToolDefinition } from "@opencode-ai/plugin/tool";
 import path from "node:path";
 import type { AstrocodeConfig } from "../config/schema";
-import type { SqliteDb } from "../state/db";
+import type { RuntimeState } from "../state/types";
 import { withTx } from "../state/db";
 import type { StageKey } from "../state/types";
 import type { UiEmitEvent } from "../workflow/state-machine";
@@ -171,8 +171,8 @@ function buildUiMessage(e: UiEmitEvent): { title: string; message: string; varia
   }
 }
 
-export function createAstroWorkflowProceedTool(opts: { ctx: any; config: AstrocodeConfig; db: SqliteDb; agents?: Record<string, AgentConfig> }): ToolDefinition {
-  const { ctx, config, db, agents } = opts;
+export function createAstroWorkflowProceedTool(opts: { ctx: any; config: AstrocodeConfig; runtime: RuntimeState; agents?: Record<string, AgentConfig> }): ToolDefinition {
+  const { ctx, config, runtime, agents } = opts;
   const toasts = createToastManager({ ctx, throttleMs: config.ui.toasts.throttle_ms });
 
   return tool({
@@ -183,6 +183,11 @@ export function createAstroWorkflowProceedTool(opts: { ctx: any; config: Astroco
       max_steps: tool.schema.number().int().positive().default(config.workflow.default_max_steps),
     },
     execute: async ({ mode, max_steps }) => {
+      const { db } = runtime;
+      if (!db) {
+        return "⚠️ Cannot proceed: Astrocode is not initialized. Run **astro_init** first.";
+      }
+
       const sessionId = (ctx as any).sessionID as string | undefined;
       const steps = Math.min(max_steps, config.workflow.loop_max_steps_hard_cap);