oh-my-harness 0.7.0 → 0.9.0

package/dist/catalog/blocks/compact-context.d.ts

@@ -0,0 +1,2 @@
+import type { BuildingBlock } from "../types.js";
+export declare const compactContext: BuildingBlock;

package/dist/catalog/blocks/compact-context.js

@@ -0,0 +1,27 @@
+export const compactContext = {
+    id: "compact-context",
+    name: "Compact Context",
+    description: "Re-injects project context after context compaction",
+    category: "automation",
+    event: "SessionStart",
+    matcher: "compact",
+    canBlock: false,
+    params: [
+        {
+            name: "contextFile",
+            type: "string",
+            description: "Path to the context file to re-inject",
+            default: "CLAUDE.md",
+            required: false,
+        },
+    ],
+    tags: ["context", "compaction", "session", "automation"],
+    template: `#!/bin/bash
+set -euo pipefail
+INPUT=$(cat)
+CONTEXT_FILE="{{{contextFile}}}"
+if [[ -f "$CONTEXT_FILE" ]]; then
+  cat "$CONTEXT_FILE"
+fi
+exit 0`,
+};

package/dist/catalog/blocks/config-audit.d.ts

@@ -0,0 +1,2 @@
+import type { BuildingBlock } from "../types.js";
+export declare const configAudit: BuildingBlock;

package/dist/catalog/blocks/config-audit.js

@@ -0,0 +1,29 @@
+export const configAudit = {
+    id: "config-audit",
+    name: "Config Audit",
+    description: "Logs configuration changes for audit trail",
+    category: "audit",
+    event: "ConfigChange",
+    matcher: "",
+    canBlock: false,
+    params: [
+        {
+            name: "logFile",
+            type: "string",
+            description: "Path to the audit log file",
+            default: ".claude/hooks/.state/config-audit.log",
+            required: false,
+        },
+    ],
+    tags: ["audit", "config", "logging"],
+    template: `#!/bin/bash
+set -euo pipefail
+INPUT=$(cat)
+LOG_FILE="{{{logFile}}}"
+mkdir -p "$(dirname "$LOG_FILE")" 2>/dev/null || true
+SOURCE=$(echo "$INPUT" | jq -r '.source // "unknown"' 2>/dev/null)
+FILE=$(echo "$INPUT" | jq -r '.file_path // "unknown"' 2>/dev/null)
+TS=$(date -u +%Y-%m-%dT%H:%M:%SZ)
+echo "{\\"ts\\":\\"$TS\\",\\"source\\":\\"$SOURCE\\",\\"file\\":\\"$FILE\\"}" >> "$LOG_FILE"
+exit 0`,
+};

package/dist/catalog/blocks/desktop-notify.d.ts

@@ -0,0 +1,2 @@
+import type { BuildingBlock } from "../types.js";
+export declare const desktopNotify: BuildingBlock;

package/dist/catalog/blocks/desktop-notify.js

@@ -0,0 +1,21 @@
+export const desktopNotify = {
+    id: "desktop-notify",
+    name: "Desktop Notify",
+    description: "Sends desktop notification when Claude needs attention",
+    category: "notification",
+    event: "Notification",
+    matcher: "",
+    canBlock: false,
+    params: [],
+    tags: ["notification", "desktop", "alert"],
+    template: `#!/bin/bash
+set -euo pipefail
+INPUT=$(cat)
+MESSAGE=$(echo "$INPUT" | jq -r '.message // "Claude Code needs your attention"' 2>/dev/null)
+if [[ "$(uname)" == "Darwin" ]]; then
+  osascript -e "display notification \\"$MESSAGE\\" with title \\"Claude Code\\"" 2>/dev/null || true
+elif command -v notify-send &>/dev/null; then
+  notify-send "Claude Code" "$MESSAGE" 2>/dev/null || true
+fi
+exit 0`,
+};

package/dist/catalog/blocks/index.js

@@ -9,6 +9,12 @@ import { formatOnSave } from "./format-on-save.js";
 import { autoPr } from "./auto-pr.js";
 import { secretFileGuard } from "./secret-file-guard.js";
 import { tddGuard } from "./tdd-guard.js";
+import { sqlGuard } from "./sql-guard.js";
+import { testOnSave } from "./test-on-save.js";
+import { desktopNotify } from "./desktop-notify.js";
+import { configAudit } from "./config-audit.js";
+import { compactContext } from "./compact-context.js";
+import { worktreeSetup } from "./worktree-setup.js";
 export const builtinBlocks = [
     branchGuard,
     commitTestGate,
@@ -21,4 +27,10 @@ export const builtinBlocks = [
     autoPr,
     secretFileGuard,
     tddGuard,
+    sqlGuard,
+    testOnSave,
+    desktopNotify,
+    configAudit,
+    compactContext,
+    worktreeSetup,
 ];

package/dist/catalog/blocks/sql-guard.d.ts

@@ -0,0 +1,2 @@
+import type { BuildingBlock } from "../types.js";
+export declare const sqlGuard: BuildingBlock;

package/dist/catalog/blocks/sql-guard.js

@@ -0,0 +1,34 @@
+export const sqlGuard = {
+    id: "sql-guard",
+    name: "SQL Guard",
+    description: "Blocks dangerous SQL commands in shell",
+    category: "security",
+    event: "PreToolUse",
+    matcher: "Bash",
+    canBlock: true,
+    params: [
+        {
+            name: "patterns",
+            type: "string[]",
+            description: "Dangerous SQL patterns to block",
+            default: ["DROP TABLE", "DROP DATABASE", "TRUNCATE TABLE", "DELETE FROM", "DROP COLUMN", "DROP INDEX"],
+            required: true,
+        },
+    ],
+    tags: ["security", "sql", "guard", "database"],
+    template: `#!/bin/bash
+set -euo pipefail
+INPUT=$(cat)
+COMMAND=$(echo "$INPUT" | jq -r '.tool_input.command // empty' 2>/dev/null)
+[[ -z "$COMMAND" ]] && exit 0
+COMMAND_LOWER=$(echo "$COMMAND" | tr '[:upper:]' '[:lower:]')
+PATTERNS=({{#each patterns}}"{{{this}}}" {{/each}})
+for PATTERN in "\${PATTERNS[@]}"; do
+  PATTERN_LOWER=$(echo "$PATTERN" | tr '[:upper:]' '[:lower:]')
+  if echo "$COMMAND_LOWER" | grep -qF -- "$PATTERN_LOWER"; then
+    echo "{\\"decision\\": \\"block\\", \\"reason\\": \\"oh-my-harness: SQL command matches blocked pattern: $PATTERN\\"}"
+    exit 0
+  fi
+done
+exit 0`,
+};

package/dist/catalog/blocks/tdd-guard.js

@@ -72,8 +72,11 @@ fi
 
 # edit-history에서 테스트 파일 검색
 if jq -e --arg b "\$BASENAME" '.edits[] | select(contains($b) and (contains(".test.") or contains(".spec.") or contains("test_")))' "\$HISTORY_FILE" >/dev/null 2>&1; then
-  # 테스트 먼저 수정됨 → 소스 기록 + 통과
-  UPDATED=$(jq --arg f "\$FILE_PATH" '.edits += [$f] | .edits |= unique' "\$HISTORY_FILE" 2>/dev/null) || true
+  # 테스트 먼저 수정됨 → 매칭 테스트 기록 소비(제거) + 소스 기록 + 통과
+  UPDATED=$(jq --arg b "\$BASENAME" --arg f "\$FILE_PATH" '
+    .edits |= [.[] | select((contains($b) and (contains(".test.") or contains(".spec.") or contains("test_"))) | not)]
+    | .edits += [$f] | .edits |= unique
+  ' "\$HISTORY_FILE" 2>/dev/null) || true
   if [[ -n "\$UPDATED" ]]; then
     echo "\$UPDATED" > "\$HISTORY_FILE"
   fi

package/dist/catalog/blocks/test-on-save.d.ts

@@ -0,0 +1,2 @@
+import type { BuildingBlock } from "../types.js";
+export declare const testOnSave: BuildingBlock;

package/dist/catalog/blocks/test-on-save.js

@@ -0,0 +1,37 @@
+export const testOnSave = {
+    id: "test-on-save",
+    name: "Test on Save",
+    description: "Runs related tests after source file edits",
+    category: "quality",
+    event: "PostToolUse",
+    matcher: "Edit|Write",
+    canBlock: false,
+    params: [
+        {
+            name: "testCommand",
+            type: "string",
+            description: "Command to run tests (e.g. npx vitest run)",
+            required: true,
+        },
+        {
+            name: "filePattern",
+            type: "string",
+            description: "Regex pattern to match source files",
+            default: "\\.(ts|tsx|js|jsx|py)$",
+            required: false,
+        },
+    ],
+    tags: ["quality", "testing", "auto-run", "save"],
+    template: `#!/bin/bash
+set -euo pipefail
+INPUT=$(cat)
+FILE_PATH=$(echo "$INPUT" | jq -r '.tool_input.file_path // .tool_input.path // empty' 2>/dev/null)
+[[ -z "$FILE_PATH" ]] && exit 0
+FILE_RE='{{{filePattern}}}'
+FILE_RE="\${FILE_RE//\\\\\\\\/\\\\}"
+if [[ "$FILE_PATH" =~ $FILE_RE ]]; then
+  echo "oh-my-harness: Running tests after edit..." >&2
+  {{{testCommand}}} >&2 2>&1 || true
+fi
+exit 0`,
+};

package/dist/catalog/blocks/worktree-setup.d.ts

@@ -0,0 +1,2 @@
+import type { BuildingBlock } from "../types.js";
+export declare const worktreeSetup: BuildingBlock;

package/dist/catalog/blocks/worktree-setup.js

@@ -0,0 +1,91 @@
+export const worktreeSetup = {
+    id: "worktree-setup",
+    name: "Worktree Setup",
+    description: "Initializes new git worktrees with symlinked dependencies and copied config files",
+    category: "automation",
+    event: "WorktreeCreate",
+    matcher: "",
+    canBlock: false,
+    params: [
+        {
+            name: "symlinkPaths",
+            type: "string[]",
+            description: "Paths to symlink from main worktree (heavy dependencies)",
+            required: false,
+            default: ["node_modules", ".venv", "vendor", "target", ".build"],
+        },
+        {
+            name: "copyPaths",
+            type: "string[]",
+            description: "Paths to hard-copy from main worktree (config files)",
+            required: false,
+            default: [".env", ".env.local"],
+        },
+        {
+            name: "installCommand",
+            type: "string",
+            description: "Package install command to run if symlink source missing",
+            required: false,
+            default: "",
+        },
+    ],
+    tags: ["worktree", "git", "dependencies", "symlink", "automation"],
+    template: `#!/bin/bash
+set -euo pipefail
+INPUT=$(cat)
+
+# Parse worktree info from stdin
+WORKTREE_PATH=$(echo "$INPUT" | jq -r '.worktree_path // empty' 2>/dev/null)
+if [[ -z "$WORKTREE_PATH" ]]; then
+  echo "oh-my-harness: worktree-setup — no worktree_path in input" >&2
+  exit 0
+fi
+
+# Find the main worktree (parent repo)
+MAIN_WORKTREE=$(git -C "$WORKTREE_PATH" worktree list --porcelain 2>/dev/null | head -1 | sed 's/^worktree //')
+if [[ -z "$MAIN_WORKTREE" ]]; then
+  echo "oh-my-harness: worktree-setup — cannot determine main worktree" >&2
+  exit 0
+fi
+
+echo "oh-my-harness: worktree-setup — setting up $WORKTREE_PATH"
+
+# Symlink heavy directories from main worktree
+SYMLINKS=({{#each symlinkPaths}}"{{{this}}}" {{/each}})
+for rel in "\${SYMLINKS[@]}"; do
+  [[ -z "$rel" ]] && continue
+  src="$MAIN_WORKTREE/$rel"
+  dst="$WORKTREE_PATH/$rel"
+  if [[ -e "$src" && ! -e "$dst" ]]; then
+    mkdir -p "$(dirname "$dst")"
+    ln -s "$src" "$dst"
+    echo "  symlinked: $rel"
+  fi
+done
+
+# Hard-copy config files from main worktree
+COPIES=({{#each copyPaths}}"{{{this}}}" {{/each}})
+for rel in "\${COPIES[@]}"; do
+  [[ -z "$rel" ]] && continue
+  src="$MAIN_WORKTREE/$rel"
+  dst="$WORKTREE_PATH/$rel"
+  if [[ -f "$src" && ! -f "$dst" ]]; then
+    mkdir -p "$(dirname "$dst")"
+    cp "$src" "$dst"
+    echo "  copied: $rel"
+  fi
+done
+
+# Run install command if provided and no symlinked deps exist
+INSTALL_CMD="{{{installCommand}}}"
+if [[ -n "$INSTALL_CMD" ]]; then
+  first_symlink="\${SYMLINKS[0]}"
+  if [[ -n "$first_symlink" && ! -e "$WORKTREE_PATH/$first_symlink" ]]; then
+    echo "  running: $INSTALL_CMD"
+    cd "$WORKTREE_PATH" && eval "$INSTALL_CMD"
+  fi
+fi
+
+echo "oh-my-harness: worktree-setup complete"
+exit 0`,
+};

package/dist/catalog/converter.js

@@ -4,7 +4,7 @@ export async function convertHookEntries(entries, registry, projectDir) {
     const hooksConfig = {};
     const scripts = new Map();
     const errors = [];
-    const seenBlockIds = new Set();
+    const blockInstanceCount = new Map();
     for (const entry of entries) {
         const block = registry.get(entry.block);
         if (!block) {
@@ -17,11 +17,6 @@ export async function convertHookEntries(entries, registry, projectDir) {
             errors.push(...paramErrors);
             continue;
         }
-        if (seenBlockIds.has(entry.block)) {
-            errors.push(`Duplicate block id skipped: "${entry.block}"`);
-            continue;
-        }
-        seenBlockIds.add(entry.block);
         let scriptContent;
         try {
             scriptContent = renderTemplate(block.template, resolvedParams);
@@ -30,7 +25,10 @@ export async function convertHookEntries(entries, registry, projectDir) {
             errors.push(`Failed to render block "${entry.block}": ${err.message}`);
             continue;
         }
-        const scriptName = `${entry.block}.sh`;
+        // Support multiple instances of the same block with different params
+        const count = blockInstanceCount.get(entry.block) ?? 0;
+        blockInstanceCount.set(entry.block, count + 1);
+        const scriptName = count === 0 ? `${entry.block}.sh` : `${entry.block}-${count}.sh`;
         const scriptPath = path.join(projectDir, ".claude", "hooks", scriptName);
         scripts.set(scriptPath, scriptContent);
         const hookEntry = {

package/dist/catalog/types.d.ts

@@ -1,6 +1,6 @@
 import { z } from "zod";
-export type HookEvent = "PreToolUse" | "PostToolUse" | "PreCompact" | "PostCompact" | "Notification" | "Stop" | "SubagentStop" | "PreBash" | "PostBash" | "PreEdit" | "PostEdit" | "PreRead" | "PostRead" | "PreWrite" | "PostWrite" | "SessionStart" | "SessionEnd" | "PreToolResult" | "PostToolResult" | "UserPromptSubmit";
-export type BuildingBlockCategory = "git" | "quality" | "security" | "notification" | "formatting" | "custom" | "auto-fix" | "automation" | "file-protection";
+export type HookEvent = "PreToolUse" | "PostToolUse" | "PreCompact" | "PostCompact" | "Notification" | "Stop" | "SubagentStop" | "PreBash" | "PostBash" | "PreEdit" | "PostEdit" | "PreRead" | "PostRead" | "PreWrite" | "PostWrite" | "SessionStart" | "SessionEnd" | "PreToolResult" | "PostToolResult" | "UserPromptSubmit" | "ConfigChange" | "WorktreeCreate" | "WorktreeRemove";
+export type BuildingBlockCategory = "git" | "quality" | "security" | "notification" | "formatting" | "custom" | "auto-fix" | "automation" | "file-protection" | "audit";
 export interface ParamDefinition {
     name: string;
     type: "string" | "boolean" | "number" | "string[]";
@@ -47,8 +47,8 @@ export declare const BuildingBlockSchema: z.ZodObject<{
     id: z.ZodString;
     name: z.ZodString;
     description: z.ZodString;
-    category: z.ZodEnum<["git", "quality", "security", "notification", "formatting", "custom", "auto-fix", "automation", "file-protection"]>;
-    event: z.ZodEnum<["PreToolUse", "PostToolUse", "PreCompact", "PostCompact", "Notification", "Stop", "SubagentStop", "PreBash", "PostBash", "PreEdit", "PostEdit", "PreRead", "PostRead", "PreWrite", "PostWrite", "SessionStart", "SessionEnd", "PreToolResult", "PostToolResult", "UserPromptSubmit"]>;
+    category: z.ZodEnum<["git", "quality", "security", "notification", "formatting", "custom", "auto-fix", "automation", "file-protection", "audit"]>;
+    event: z.ZodEnum<["PreToolUse", "PostToolUse", "PreCompact", "PostCompact", "Notification", "Stop", "SubagentStop", "PreBash", "PostBash", "PreEdit", "PostEdit", "PreRead", "PostRead", "PreWrite", "PostWrite", "SessionStart", "SessionEnd", "PreToolResult", "PostToolResult", "UserPromptSubmit", "ConfigChange", "WorktreeCreate", "WorktreeRemove"]>;
     matcher: z.ZodOptional<z.ZodString>;
     canBlock: z.ZodBoolean;
     params: z.ZodArray<z.ZodObject<{
@@ -85,8 +85,8 @@ export declare const BuildingBlockSchema: z.ZodObject<{
     template: string;
     name: string;
     tags: string[];
-    category: "custom" | "git" | "quality" | "security" | "notification" | "formatting" | "auto-fix" | "automation" | "file-protection";
-    event: "PreToolUse" | "PostToolUse" | "PreCompact" | "PostCompact" | "Notification" | "Stop" | "SubagentStop" | "PreBash" | "PostBash" | "PreEdit" | "PostEdit" | "PreRead" | "PostRead" | "PreWrite" | "PostWrite" | "SessionStart" | "SessionEnd" | "PreToolResult" | "PostToolResult" | "UserPromptSubmit";
+    event: "PreToolUse" | "PostToolUse" | "SessionStart" | "Notification" | "ConfigChange" | "WorktreeCreate" | "PreCompact" | "PostCompact" | "Stop" | "SubagentStop" | "PreBash" | "PostBash" | "PreEdit" | "PostEdit" | "PreRead" | "PostRead" | "PreWrite" | "PostWrite" | "SessionEnd" | "PreToolResult" | "PostToolResult" | "UserPromptSubmit" | "WorktreeRemove";
+    category: "notification" | "custom" | "git" | "quality" | "security" | "formatting" | "auto-fix" | "automation" | "file-protection" | "audit";
     canBlock: boolean;
     matcher?: string | undefined;
 }, {
@@ -102,8 +102,8 @@ export declare const BuildingBlockSchema: z.ZodObject<{
     template: string;
     name: string;
     tags: string[];
-    category: "custom" | "git" | "quality" | "security" | "notification" | "formatting" | "auto-fix" | "automation" | "file-protection";
-    event: "PreToolUse" | "PostToolUse" | "PreCompact" | "PostCompact" | "Notification" | "Stop" | "SubagentStop" | "PreBash" | "PostBash" | "PreEdit" | "PostEdit" | "PreRead" | "PostRead" | "PreWrite" | "PostWrite" | "SessionStart" | "SessionEnd" | "PreToolResult" | "PostToolResult" | "UserPromptSubmit";
+    event: "PreToolUse" | "PostToolUse" | "SessionStart" | "Notification" | "ConfigChange" | "WorktreeCreate" | "PreCompact" | "PostCompact" | "Stop" | "SubagentStop" | "PreBash" | "PostBash" | "PreEdit" | "PostEdit" | "PreRead" | "PostRead" | "PreWrite" | "PostWrite" | "SessionEnd" | "PreToolResult" | "PostToolResult" | "UserPromptSubmit" | "WorktreeRemove";
+    category: "notification" | "custom" | "git" | "quality" | "security" | "formatting" | "auto-fix" | "automation" | "file-protection" | "audit";
     canBlock: boolean;
     matcher?: string | undefined;
 }>;

package/dist/catalog/types.js

@@ -10,8 +10,8 @@ export const BuildingBlockSchema = z.object({
     id: z.string(),
     name: z.string(),
     description: z.string(),
-    category: z.enum(["git", "quality", "security", "notification", "formatting", "custom", "auto-fix", "automation", "file-protection"]),
-    event: z.enum(["PreToolUse", "PostToolUse", "PreCompact", "PostCompact", "Notification", "Stop", "SubagentStop", "PreBash", "PostBash", "PreEdit", "PostEdit", "PreRead", "PostRead", "PreWrite", "PostWrite", "SessionStart", "SessionEnd", "PreToolResult", "PostToolResult", "UserPromptSubmit"]),
+    category: z.enum(["git", "quality", "security", "notification", "formatting", "custom", "auto-fix", "automation", "file-protection", "audit"]),
+    event: z.enum(["PreToolUse", "PostToolUse", "PreCompact", "PostCompact", "Notification", "Stop", "SubagentStop", "PreBash", "PostBash", "PreEdit", "PostEdit", "PreRead", "PostRead", "PreWrite", "PostWrite", "SessionStart", "SessionEnd", "PreToolResult", "PostToolResult", "UserPromptSubmit", "ConfigChange", "WorktreeCreate", "WorktreeRemove"]),
     matcher: z.string().optional(),
     canBlock: z.boolean(),
     params: z.array(ParamDefinitionSchema),

package/dist/cli/commands/catalog.js

@@ -11,6 +11,7 @@ const CATEGORY_EMOJI = {
     notification: "🔔",
     prompt: "💬",
     permission: "🔐",
+    audit: "📝",
 };
 function categoryLabel(category) {
     const emoji = CATEGORY_EMOJI[category] ?? "•";

package/dist/cli/commands/doctor.d.ts

@@ -3,6 +3,7 @@ export interface DoctorOptions {
 }
 export interface DoctorResult {
     healthy: boolean;
+    exitCode: number;
     checks: {
         stateFile: boolean;
         claudeMd: boolean;

package/dist/cli/commands/doctor.js

@@ -71,6 +71,7 @@ export async function doctorCommand(options = {}) {
         checks.hooksExecutable = true;
     }
     const healthy = Object.values(checks).every(Boolean);
+    const exitCode = healthy ? 0 : 1;
     if (healthy) {
         console.log("oh-my-harness: all checks passed");
     }
@@ -80,5 +81,5 @@ export async function doctorCommand(options = {}) {
             console.log(`  ${msg}`);
         }
     }
-    return { healthy, checks, messages };
+    return { healthy, exitCode, checks, messages };
 }

package/dist/cli/commands/init.js

@@ -59,7 +59,17 @@ export async function initCommand(presetNames, options = {}) {
     const presetsDir = options.presetsDir ?? getDefaultPresetsDir();
     // If --preset flag is used, go through preset flow
     if (options.preset && options.preset.length > 0) {
-        await initWithPresets(options.preset, projectDir, presetsDir, options);
+        try {
+            await initWithPresets(options.preset, projectDir, presetsDir, options);
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            console.error(`oh-my-harness: ${message}`);
+            if (!message.includes("Preset not found")) {
+                console.error("Run `oh-my-harness init --help` for usage information.");
+            }
+            process.exitCode = 1;
+        }
         return;
     }
     // If preset names are provided as positional args, use legacy flow

package/dist/cli/index.js

@@ -36,7 +36,10 @@ export function createCli() {
         .description("Validate harness configuration health")
         .action(async () => {
         const { doctorCommand } = await import("./commands/doctor.js");
-        await doctorCommand();
+        const result = await doctorCommand();
+        if (result.exitCode !== 0) {
+            process.exitCode = result.exitCode;
+        }
     });
     program
         .command("test")

package/dist/cli/stats/index.js

@@ -3,6 +3,10 @@ import React from "react";
 import { loadStatsData } from "./data.js";
 import { App } from "./App.js";
 export async function statsCommand(options = {}) {
+    if (!process.stdin.isTTY || !process.stdout.isTTY) {
+        console.error("oh-my-harness: stats requires an interactive terminal (TTY). Run this command directly in your terminal.");
+        return;
+    }
     const projectDir = options.projectDir ?? process.cwd();
     const data = await loadStatsData(projectDir);
     const { waitUntilExit } = render(React.createElement(App, { initialData: data, projectDir }));

package/dist/cli/tui/init-flow.js

@@ -9,7 +9,9 @@ import { PresetRegistry } from "../../core/preset-registry.js";
 import { loadAndMergePresets, writeHarnessState } from "../commands/init.js";
 import { mergePresets } from "../../core/config-merger.js";
 import { generate } from "../../core/generator.js";
-import { generateHarnessConfig } from "../../nl/parse-intent.js";
+import { generateHarnessConfig, createDefaultRunner } from "../../nl/parse-intent.js";
+import { hasProviderConfig } from "../../nl/config-store.js";
+import { runProviderSetup } from "./provider-setup.js";
 import { mergeEnforcementAndHooks } from "../../core/harness-converter.js";
 import { HarnessConfigSchema } from "../../core/harness-schema.js";
 import { detectProject } from "../../detector/project-detector.js";
@@ -187,7 +189,17 @@ export async function runInitTUI(options) {
     let harnessConfig;
     let presetNames;
     if (mode === "nl") {
-        // Step 4a: NL Mode
+        // Step 4a: NL Mode — check provider config
+        const hasConfig = await hasProviderConfig();
+        if (!hasConfig) {
+            p.log.info("No AI provider configured yet. Let's set one up.");
+            const providerConfig = await runProviderSetup();
+            if (!providerConfig) {
+                p.cancel("Provider setup cancelled. Use preset mode instead.");
+                process.exit(0);
+            }
+        }
+        const runner = await createDefaultRunner();
         const description = await p.text({
             message: "Describe your project:",
             placeholder: "e.g., Next.js e-commerce app with Stripe and Tailwind",
@@ -208,7 +220,7 @@ export async function runInitTUI(options) {
                 description: b.description,
                 params: b.params.map((pp) => ({ name: pp.name, required: pp.required, default: pp.default, description: pp.description })),
             }));
-            harnessConfig = await generateHarnessConfig(description, undefined, catalogBlocks, projectFacts);
+            harnessConfig = await generateHarnessConfig(description, runner, catalogBlocks, projectFacts);
             genSpinner.stop("Configuration generated");
         }
         catch (err) {

package/dist/cli/tui/provider-setup.d.ts

@@ -0,0 +1,2 @@
+import { type ProviderConfig } from "../../nl/config-store.js";
+export declare function runProviderSetup(): Promise<ProviderConfig | undefined>;

package/dist/cli/tui/provider-setup.js

@@ -0,0 +1,85 @@
+import * as p from "@clack/prompts";
+import { getAvailableProviders, getProviderDefinition, } from "../../nl/provider-registry.js";
+import { saveProviderConfig, } from "../../nl/config-store.js";
+export async function runProviderSetup() {
+    p.intro("AI Provider Setup");
+    const providers = getAvailableProviders();
+    // Step 1: Select provider
+    const providerName = await p.select({
+        message: "Select AI provider for natural language mode:",
+        options: providers.map((prov) => ({
+            value: prov.name,
+            label: prov.displayName,
+        })),
+    });
+    if (p.isCancel(providerName)) {
+        p.cancel("Provider setup cancelled.");
+        return undefined;
+    }
+    const def = getProviderDefinition(providerName);
+    // Step 2: Select method (CLI or API)
+    let method;
+    if (def.supportsCli && def.supportsApi) {
+        const selected = await p.select({
+            message: "How would you like to connect?",
+            options: [
+                { value: "cli", label: `CLI tool (${def.cliCommand ?? def.name})` },
+                { value: "api", label: "API Key" },
+            ],
+        });
+        if (p.isCancel(selected)) {
+            p.cancel("Provider setup cancelled.");
+            return undefined;
+        }
+        method = selected;
+    }
+    else if (def.supportsCli) {
+        method = "cli";
+    }
+    else {
+        method = "api";
+    }
+    const config = {
+        provider: providerName,
+        method,
+    };
+    // Step 3: Get API key if needed
+    if (method === "api") {
+        const apiKey = await p.text({
+            message: `Enter your ${def.displayName} API key:`,
+            placeholder: "sk-...",
+            validate: (value) => {
+                if (!value || !value.trim())
+                    return "API key is required";
+                return undefined;
+            },
+        });
+        if (p.isCancel(apiKey)) {
+            p.cancel("Provider setup cancelled.");
+            return undefined;
+        }
+        config.apiKey = apiKey;
+        // Select model from available list
+        const selectedModel = await p.select({
+            message: "Select model:",
+            options: def.availableModels.map((m) => ({
+                value: m.id,
+                label: m.label,
+                hint: m.id === def.defaultModel ? "default" : undefined,
+            })),
+            initialValue: def.defaultModel,
+        });
+        if (p.isCancel(selectedModel)) {
+            p.cancel("Provider setup cancelled.");
+            return undefined;
+        }
+        config.model = selectedModel;
+    }
+    else {
+        config.cliCommand = def.cliCommand ?? def.name;
+    }
+    // Save config
+    await saveProviderConfig(config);
+    p.log.success(`Provider saved: ${def.displayName} (${method})`);
+    return config;
+}