larkcc 0.1.0

package/dist/agent.js

@@ -0,0 +1,177 @@
+import { query } from "@anthropic-ai/claude-agent-sdk";
+import os from "os";
+import { execSync } from "child_process";
+import { replyFinalCard, sendToolCard, updateToolCard, } from "./feishu.js";
+import { getSession, setSession } from "./session.js";
+import { logger } from "./logger.js";
+const TOOL_LABELS = {
+    Read: "📂 读取文件",
+    Write: "✏️  写入文件",
+    Edit: "✏️  编辑文件",
+    Bash: "⚡ 执行命令",
+    Glob: "🔍 查找文件",
+    Grep: "🔎 搜索内容",
+    LS: "📁 列出目录",
+    ExitPlanMode: "📋 退出计划模式",
+    AskUserQuestion: "💬 提问",
+};
+const SILENT_TOOLS = new Set(["ExitPlanMode", "TodoWrite", "TodoRead"]);
+function formatInput(name, input) {
+    if (["Read", "Write", "Edit"].includes(name))
+        return String(input.file_path ?? input.path ?? "");
+    if (name === "Bash")
+        return String(input.command ?? "").slice(0, 100);
+    if (name === "Grep")
+        return `${input.pattern} in ${input.path ?? "."}`;
+    if (name === "LS")
+        return String(input.path ?? ".");
+    if (name === "Glob")
+        return String(input.pattern ?? "");
+    return JSON.stringify(input).slice(0, 100);
+}
+function truncate(str, len) {
+    return str.length > len ? str.slice(0, len) + "..." : str;
+}
+export async function runAgent(prompt, cwd, config, client, chatId, rootMsgId, images, abortSignal, // 可选中断信号
+profile // 机器人配置名
+) {
+    const sessionId = getSession();
+    let textBuffer = "";
+    const toolMsgMap = new Map();
+    // 构建 prompt content（支持图片）
+    const promptContent = [];
+    if (images && images.length > 0) {
+        for (const img of images) {
+            promptContent.push({
+                type: "image",
+                source: { type: "base64", media_type: img.mediaType, data: img.base64 },
+            });
+        }
+    }
+    promptContent.push({ type: "text", text: prompt });
+    // 如果有图片，使用 AsyncIterable<SDKUserMessage> 格式
+    // 否则使用普通字符串
+    const hasImages = images && images.length > 0;
+    async function* messageGenerator() {
+        yield {
+            type: 'user',
+            message: {
+                role: 'user',
+                content: promptContent,
+            },
+            parent_tool_use_id: null,
+            session_id: sessionId || crypto.randomUUID(),
+        };
+    }
+    const queryPrompt = hasImages ? messageGenerator() : prompt;
+    try {
+        for await (const event of query({
+            prompt: queryPrompt,
+            options: {
+                cwd,
+                resume: sessionId,
+                permissionMode: config.claude.permission_mode,
+                allowedTools: config.claude.allowed_tools,
+                abortSignal,
+            },
+        })) {
+            // 检查是否已中断
+            if (abortSignal?.aborted) {
+                logger.info("Agent aborted by user");
+                await replyFinalCard(client, chatId, rootMsgId, "⏹ 任务已中断", {
+                    profile: profile ?? "default",
+                    cwd,
+                    sessionId: getSession() ?? "",
+                    overflow: config.overflow,
+                    chatId,
+                    rootMsgId,
+                    appId: config.feishu.app_id,
+                    appSecret: config.feishu.app_secret,
+                });
+                break;
+            }
+            if (event.type === "assistant") {
+                const blocks = event.message.content;
+                for (const block of blocks) {
+                    if (block.type === "text" && block.text) {
+                        textBuffer += block.text;
+                    }
+                    if (block.type === "tool_use" && block.id && block.name) {
+                        if (SILENT_TOOLS.has(block.name))
+                            break;
+                        const label = TOOL_LABELS[block.name] ?? `🔧 ${block.name}`;
+                        if (block.name === "AskUserQuestion") {
+                            const input = block.input;
+                            const questions = input.questions?.map(q => q.question).join("\n") ?? "";
+                            if (questions)
+                                await replyFinalCard(client, chatId, rootMsgId, questions, {
+                                    profile: profile ?? "default",
+                                    cwd,
+                                    sessionId: getSession() ?? "",
+                                    overflow: config.overflow,
+                                    chatId,
+                                    rootMsgId,
+                                    appId: config.feishu.app_id,
+                                    appSecret: config.feishu.app_secret,
+                                });
+                            break;
+                        }
+                        const detail = formatInput(block.name, block.input ?? {});
+                        logger.tool(block.name, detail);
+                        const msgId = await sendToolCard(client, chatId, rootMsgId, label, detail, "running");
+                        toolMsgMap.set(block.id, { msgId, label, detail });
+                    }
+                }
+            }
+            if (event.type === "user") {
+                const blocks = event.message.content;
+                for (const block of blocks) {
+                    if (block.type === "tool_result" && block.tool_use_id) {
+                        const toolInfo = toolMsgMap.get(block.tool_use_id);
+                        if (toolInfo) {
+                            const raw = typeof block.content === "string"
+                                ? block.content
+                                : JSON.stringify(block.content ?? "");
+                            await updateToolCard(client, toolInfo.msgId, toolInfo.label, toolInfo.detail, truncate(raw, 500));
+                        }
+                    }
+                }
+            }
+            if (event.type === "result") {
+                const resultEvent = event;
+                if (resultEvent.session_id) {
+                    setSession(resultEvent.session_id);
+                    logger.dim(`session saved: ${resultEvent.session_id}`);
+                }
+                if (textBuffer) {
+                    await replyFinalCard(client, chatId, rootMsgId, textBuffer, {
+                        profile: profile ?? "default",
+                        cwd,
+                        sessionId: getSession() ?? "",
+                        overflow: config.overflow,
+                        chatId,
+                        rootMsgId,
+                        appId: config.feishu.app_id,
+                        appSecret: config.feishu.app_secret,
+                    });
+                }
+                logger.reply(chatId);
+            }
+        }
+    }
+    catch (err) {
+        console.error(`[query error]:`, err);
+        throw err;
+    }
+}
+export function ensureEnv() {
+    if (!process.env.HOME)
+        process.env.HOME = os.homedir();
+    try {
+        const shellPath = execSync("bash -lc 'echo $PATH' 2>/dev/null", { timeout: 3000 }).toString().trim();
+        if (shellPath)
+            process.env.PATH = shellPath;
+    }
+    catch { }
+}
+//# sourceMappingURL=agent.js.map

package/dist/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,gCAAgC,CAAC;AACvD,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAEzC,OAAO,EACL,cAAc,EACd,YAAY,EACZ,cAAc,GACf,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AACtD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAGrC,MAAM,WAAW,GAA2B;IAC1C,IAAI,EAAa,SAAS;IAC1B,KAAK,EAAY,UAAU;IAC3B,IAAI,EAAa,UAAU;IAC3B,IAAI,EAAa,QAAQ;IACzB,IAAI,EAAa,SAAS;IAC1B,IAAI,EAAa,SAAS;IAC1B,EAAE,EAAe,SAAS;IAC1B,YAAY,EAAK,WAAW;IAC5B,eAAe,EAAE,OAAO;CACzB,CAAC;AAEF,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,CAAC,cAAc,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC,CAAC;AAExE,SAAS,WAAW,CAAC,IAAY,EAAE,KAA8B;IAC/D,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;QAC1C,OAAO,MAAM,CAAC,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IACrD,IAAI,IAAI,KAAK,MAAM;QAAE,OAAO,MAAM,CAAC,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IACtE,IAAI,IAAI,KAAK,MAAM;QAAE,OAAO,GAAG,KAAK,CAAC,OAAO,OAAO,KAAK,CAAC,IAAI,IAAI,GAAG,EAAE,CAAC;IACvE,IAAI,IAAI,KAAK,IAAI;QAAI,OAAO,MAAM,CAAC,KAAK,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC;IACtD,IAAI,IAAI,KAAK,MAAM;QAAE,OAAO,MAAM,CAAC,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;IACxD,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;AAC7C,CAAC;AAED,SAAS,QAAQ,CAAC,GAAW,EAAE,GAAW;IACxC,OAAO,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC;AAC5D,CAAC;AAOD,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,MAAc,EACd,GAAW,EACX,MAAoB,EACpB,MAAmB,EACnB,MAAc,EACd,SAAiB,EACjB,MAAqB,EACrB,WAAyB,EAAI,SAAS;AACtC,OAAgB,CAAc,SAAS;;IAEvC,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;IAE/B,IAAI,UAAU,GAAG,EAAE,CAAC;IACpB,MAAM,UAAU,GAAG,IAAI,GAAG,EAA4D,CAAC;IAEvF,0BAA0B;IAC1B,MAAM,aAAa,GAAU,EAAE,CAAC;IAChC,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChC,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,OAAO;gBACb,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,GAAG,CAAC,SAAS,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,EAAE;aACxE,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IACD,aAAa,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;IAEnD,4CAA4C;IAC5C,YAAY;IACZ,MAAM,SAAS,GAAG,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;IAE9C,KAAK,SAAS,CAAC,CAAC,gBAAgB;QAC9B,MAAM;YACJ,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE;gBACP,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,aAAa;aACvB;YACD,kBAAkB,EAAE,IAAI;YACxB,UAAU,EAAE,SAAS,IAAI,MAAM,CAAC,UAAU,EAAE;SAC7C,CAAC;IACJ,CAAC;IAED,MAAM,WAAW,GAAG,SAAS,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;IAE5D,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,KAAK,CAAC;YAC9B,MAAM,EAAE,WAAW;YACnB,OAAO,EAAE;gBACP,GAAG;gBACH,MAAM,EAAE,SAAS;gBACjB,cAAc,EAAE,MAAM,CAAC,MAAM,CAAC,eAAgC;gBAC9D,YAAY,EAAE,MAAM,CAAC,MAAM,CAAC,aAAa;gBACzC,WAAW;aACZ;SACK,CAAC,EAAE,CAAC;YACZ,UAAU;YACV,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;gBACzB,MAAM,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;gBACrC,MAAM,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE;oBACzD,OAAO,EAAE,OAAO,IAAI,SAAS;oBAC7B,GAAG;oBACH,SAAS,EAAE,UAAU,EAAE,IAAI,EAAE;oBAC7B,QAAQ,EAAE,MAAM,CAAC,QAAS;oBAC1B,MAAM;oBACN,SAAS;oBACT,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;oBAC3B,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,UAAU;iBACpC,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;YAED,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC/B,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,OAM3B,CAAC;gBAEH,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;wBACxC,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC;oBAC3B,CAAC;oBAED,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;wBACxD,IAAI,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;4BAAE,MAAM;wBACxC,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;wBAC5D,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;4BACrC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAoD,CAAC;4BACzE,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;4BACzE,IAAI,SAAS;gCAAE,MAAM,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE;oCACxE,OAAO,EAAE,OAAO,IAAI,SAAS;oCAC7B,GAAG;oCACH,SAAS,EAAE,UAAU,EAAE,IAAI,EAAE;oCAC7B,QAAQ,EAAE,MAAM,CAAC,QAAS;oCAC1B,MAAM;oCACN,SAAS;oCACT,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;oCAC3B,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,UAAU;iCACpC,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBACD,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;wBAC1D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;wBAChC,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;wBACtF,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;oBACrD,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,OAI3B,CAAC;gBACH,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;wBACtD,MAAM,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;wBACnD,IAAI,QAAQ,EAAE,CAAC;4BACb,MAAM,GAAG,GAAG,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;gCAC3C,CAAC,CAAC,KAAK,CAAC,OAAO;gCACf,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;4BACxC,MAAM,cAAc,CAAC,MAAM,EAAE,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;wBACpG,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC5B,MAAM,WAAW,GAAG,KAAgC,CAAC;gBACrD,IAAI,WAAW,CAAC,UAAU,EAAE,CAAC;oBAC3B,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;oBACnC,MAAM,CAAC,GAAG,CAAC,kBAAkB,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;gBACzD,CAAC;gBACD,IAAI,UAAU,EAAE,CAAC;oBACf,MAAM,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE;wBAC1D,OAAO,EAAE,OAAO,IAAI,SAAS;wBAC7B,GAAG;wBACH,SAAS,EAAE,UAAU,EAAE,IAAI,EAAE;wBAC7B,QAAQ,EAAE,MAAM,CAAC,QAAS;wBAC1B,MAAM;wBACN,SAAS;wBACT,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;wBAC3B,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,UAAU;qBACpC,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;IACD,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;QACrC,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC;AAED,MAAM,UAAU,SAAS;IACvB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI;QAAE,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;IACvD,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,QAAQ,CAAC,mCAAmC,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;QACrG,IAAI,SAAS;YAAE,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,SAAS,CAAC;IAC9C,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;AACZ,CAAC"}

package/dist/app.d.ts

@@ -0,0 +1,14 @@
+import { LarkccConfig } from "./config.js";
+export interface RunningProcess {
+    profile: string;
+    pid: number;
+    cwd: string;
+    startedAt: string;
+    isContinue: boolean;
+    alive: boolean;
+}
+export declare function listRunningProcesses(): {
+    processes: RunningProcess[];
+    cleaned: string[];
+};
+export declare function startApp(cwd: string, config: LarkccConfig, profile: string | undefined, continueSession?: boolean): Promise<void>;