From 332e5cec2992fefb302251962a3ceca38437a110 Mon Sep 17 00:00:00 2001 From: Clawd Date: Sat, 28 Feb 2026 07:26:43 -0800 Subject: Phase 2: Claude integration layer - Add @anthropic-ai/claude-agent-sdk dependency - Implement src/main/claude/phases.ts with phase configs (research/plan/implement) - Implement src/main/claude/index.ts with SDK wrapper - query() integration with session management - Session resume support - Artifact read/write utilities - Phase advancement logic --- src/main/claude/index.ts | 142 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 142 insertions(+) create mode 100644 src/main/claude/index.ts (limited to 'src/main/claude/index.ts') diff --git a/src/main/claude/index.ts b/src/main/claude/index.ts new file mode 100644 index 0000000..34a914e --- /dev/null +++ b/src/main/claude/index.ts @@ -0,0 +1,142 @@ +import { query, type SDKMessage, type Query } from "@anthropic-ai/claude-agent-sdk"; +import type { Session } from "../db/sessions"; +import { getPhaseConfig, getNextPhase, getArtifactFilename } from "./phases"; +import type { Phase, UserPermissionMode } from "./phases"; +import { getProject } from "../db/projects"; +import { updateSession } from "../db/sessions"; +import fs from "node:fs"; +import path from "node:path"; + +// Track active queries by session ID +const activeQueries = new Map(); + +function ensureArtifactDir(projectPath: string): void { + const dir = path.join(projectPath, ".claude-flow"); + if (!fs.existsSync(dir)) { + fs.mkdirSync(dir, { recursive: true }); + } +} + +export interface SendMessageOptions { + session: Session; + message: string; + onMessage: (msg: SDKMessage) => void; +} + +export async function sendMessage({ + session, + message, + onMessage, +}: SendMessageOptions): Promise { + const project = getProject(session.project_id); + if (!project) throw new Error("Project not found"); + + ensureArtifactDir(project.path); + + const phaseConfig = getPhaseConfig( + session.phase as Phase, + session.permission_mode as UserPermissionMode + ); + + const q = query({ + prompt: message, + options: { + cwd: project.path, + resume: session.claude_session_id ?? undefined, + tools: phaseConfig.tools, + permissionMode: phaseConfig.permissionMode, + // Add system prompt via extraArgs since there's no direct option + extraArgs: { + "system-prompt": phaseConfig.systemPrompt, + }, + }, + }); + + activeQueries.set(session.id, q); + + try { + for await (const msg of q) { + // Capture session ID from init message + if (msg.type === "system" && msg.subtype === "init") { + if (!session.claude_session_id) { + updateSession(session.id, { claude_session_id: msg.session_id }); + } + } + onMessage(msg); + } + } finally { + activeQueries.delete(session.id); + } +} + +export function interruptSession(sessionId: string): void { + const q = activeQueries.get(sessionId); + if (q) { + q.close(); + activeQueries.delete(sessionId); + } +} + +/** + * Trigger a review: Claude reads the document and addresses user annotations + */ +export async function triggerReview( + session: Session, + onMessage: (msg: SDKMessage) => void +): Promise { + const docName = getArtifactFilename(session.phase as Phase); + const message = `I've updated .claude-flow/${docName} with annotations. Read the file, find all my inline notes (marked with // REVIEW:, // NOTE:, TODO:, or similar), address each one, and update the document accordingly. Do not implement anything yet.`; + + await sendMessage({ session, message, onMessage }); +} + +/** + * Advance to the next phase + */ +export function advancePhase(session: Session): Phase | null { + const nextPhase = getNextPhase(session.phase as Phase); + if (nextPhase) { + updateSession(session.id, { phase: nextPhase }); + } + return nextPhase; +} + +/** + * Read an artifact file from the project's .claude-flow directory + */ +export function readArtifact( + projectPath: string, + filename: string +): string | null { + const filePath = path.join(projectPath, ".claude-flow", filename); + if (fs.existsSync(filePath)) { + return fs.readFileSync(filePath, "utf-8"); + } + return null; +} + +/** + * Write an artifact file to the project's .claude-flow directory + */ +export function writeArtifact( + projectPath: string, + filename: string, + content: string +): void { + const dir = path.join(projectPath, ".claude-flow"); + if (!fs.existsSync(dir)) { + fs.mkdirSync(dir, { recursive: true }); + } + fs.writeFileSync(path.join(dir, filename), content, "utf-8"); +} + +/** + * Get the initial message for a phase + */ +export function getPhaseInitialMessage(phase: Phase): string { + return getPhaseConfig(phase).initialMessage; +} + +// Re-export types +export type { SDKMessage } from "@anthropic-ai/claude-agent-sdk"; +export type { Phase, UserPermissionMode } from "./phases"; -- cgit v1.2.3