"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.sendMessage = sendMessage;
exports.interruptSession = interruptSession;
exports.triggerReview = triggerReview;
exports.advancePhase = advancePhase;
exports.readArtifact = readArtifact;
exports.writeArtifact = writeArtifact;
exports.getPhaseInitialMessage = getPhaseInitialMessage;
const claude_agent_sdk_1 = require("@anthropic-ai/claude-agent-sdk");
const phases_1 = require("./phases");
const projects_1 = require("../db/projects");
const sessions_1 = require("../db/sessions");
const node_fs_1 = __importDefault(require("node:fs"));
const node_path_1 = __importDefault(require("node:path"));
// Track active queries by session ID
const activeQueries = new Map();
function ensureArtifactDir(projectPath) {
    const dir = node_path_1.default.join(projectPath, ".claude-flow");
    if (!node_fs_1.default.existsSync(dir)) {
        node_fs_1.default.mkdirSync(dir, { recursive: true });
    }
}
async function sendMessage({ session, message, onMessage, }) {
    const project = (0, projects_1.getProject)(session.project_id);
    if (!project)
        throw new Error("Project not found");
    ensureArtifactDir(project.path);
    const phaseConfig = (0, phases_1.getPhaseConfig)(session.phase, session.permission_mode);
    const q = (0, claude_agent_sdk_1.query)({
        prompt: message,
        options: {
            cwd: project.path,
            resume: session.claude_session_id ?? undefined,
            tools: phaseConfig.tools,
            permissionMode: phaseConfig.permissionMode,
            systemPrompt: 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) {
                    (0, sessions_1.updateSession)(session.id, { claude_session_id: msg.session_id });
                }
            }
            onMessage(msg);
        }
    }
    finally {
        activeQueries.delete(session.id);
    }
}
function interruptSession(sessionId) {
    const q = activeQueries.get(sessionId);
    if (q) {
        q.close();
        activeQueries.delete(sessionId);
    }
}
/**
 * Trigger a review: Claude reads the document and addresses user annotations
 */
async function triggerReview(session, onMessage) {
    const docName = (0, phases_1.getArtifactFilename)(session.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
 */
function advancePhase(session) {
    const nextPhase = (0, phases_1.getNextPhase)(session.phase);
    if (nextPhase) {
        (0, sessions_1.updateSession)(session.id, { phase: nextPhase });
    }
    return nextPhase;
}
/**
 * Read an artifact file from the project's .claude-flow directory
 */
function readArtifact(projectPath, filename) {
    const filePath = node_path_1.default.join(projectPath, ".claude-flow", filename);
    if (node_fs_1.default.existsSync(filePath)) {
        return node_fs_1.default.readFileSync(filePath, "utf-8");
    }
    return null;
}
/**
 * Write an artifact file to the project's .claude-flow directory
 */
function writeArtifact(projectPath, filename, content) {
    const dir = node_path_1.default.join(projectPath, ".claude-flow");
    if (!node_fs_1.default.existsSync(dir)) {
        node_fs_1.default.mkdirSync(dir, { recursive: true });
    }
    node_fs_1.default.writeFileSync(node_path_1.default.join(dir, filename), content, "utf-8");
}
/**
 * Get the initial message for a phase
 */
function getPhaseInitialMessage(phase) {
    return (0, phases_1.getPhaseConfig)(phase).initialMessage;
}