chatccc 0.2.252 → 0.2.254

package/dist/src/im-skills.js

@@ -0,0 +1,121 @@
+import { readdir, readFile, mkdir, writeFile } from "node:fs/promises";
+import { join, dirname } from "node:path";
+import { PROJECT_ROOT } from "./config.js";
+export const DEFAULT_IM_SKILLS_DIR = join(PROJECT_ROOT, "im-skills");
+function renderTemplate(template, variables) {
+    return template.replace(/\{\{\s*([a-zA-Z0-9_]+)\s*\}\}/g, (_match, key) => {
+        return variables[key] ?? "";
+    });
+}
+async function loadSkillDocs(skillsDir) {
+    let entries;
+    try {
+        entries = await readdir(skillsDir, { withFileTypes: true });
+    }
+    catch {
+        return [];
+    }
+    const docs = await Promise.all(entries
+        .filter((entry) => entry.isDirectory())
+        .sort((a, b) => a.name.localeCompare(b.name))
+        .map(async (entry) => {
+        const skillPath = join(skillsDir, entry.name, "skill.md");
+        try {
+            const content = await readFile(skillPath, "utf-8");
+            return { name: entry.name, content };
+        }
+        catch {
+            return null;
+        }
+    }));
+    return docs.filter((doc) => doc !== null);
+}
+export async function buildImSkillsPrompt(input) {
+    const skillsDir = input.skillsDir ?? DEFAULT_IM_SKILLS_DIR;
+    const enabledSkillNames = input.enabledSkillNames
+        ? new Set(input.enabledSkillNames)
+        : null;
+    const docs = await loadSkillDocs(skillsDir);
+    return docs
+        .filter((doc) => !enabledSkillNames || enabledSkillNames.has(doc.name))
+        .map((doc) => {
+        const rendered = renderTemplate(doc.content, input.variables).trim();
+        return [
+            `[ChatCCC IM skill: ${doc.name}]`,
+            rendered,
+            `[/ChatCCC IM skill: ${doc.name}]`,
+        ].join("\n");
+    })
+        .join("\n\n");
+}
+// ---------------------------------------------------------------------------
+// 会话级缓存：相同 session_id + cwd 的渲染结果完全相同，避免每轮重复读文件+渲染
+// ---------------------------------------------------------------------------
+const promptCache = new Map();
+function promptCacheKey(input) {
+    const names = input.enabledSkillNames
+        ? [...input.enabledSkillNames].sort().join(",")
+        : "*";
+    const { session_id, cwd } = input.variables;
+    return `${input.skillsDir ?? DEFAULT_IM_SKILLS_DIR}|${names}|${session_id}|${cwd}`;
+}
+/** 带会话级缓存的 buildImSkillsPrompt。同 session + 同 cwd 时直接返回缓存的渲染结果。 */
+export async function buildImSkillsPromptCached(input) {
+    const key = promptCacheKey(input);
+    const cached = promptCache.get(key);
+    if (cached !== undefined)
+        return cached;
+    const result = await buildImSkillsPrompt(input);
+    promptCache.set(key, result);
+    return result;
+}
+/** 清除指定会话的缓存（如 /cd 后 cwd 变了，旧 key 不再需要） */
+export function clearImSkillsPromptCache(sessionId) {
+    for (const key of promptCache.keys()) {
+        if (key.includes(`|${sessionId}|`))
+            promptCache.delete(key);
+    }
+}
+/**
+ * 渲染技能目录下的子文档（skill.md 除外）并写入 outputDir。
+ * 返回写入的文件路径列表。通常在会话初始化时调用，写入的文档供 Agent 按需读取。
+ */
+export async function exportSkillSubDocs(input, outputDir) {
+    const skillsDir = input.skillsDir ?? DEFAULT_IM_SKILLS_DIR;
+    const enabledSkillNames = input.enabledSkillNames
+        ? new Set(input.enabledSkillNames)
+        : null;
+    const files = [];
+    let entries;
+    try {
+        entries = await readdir(skillsDir, { withFileTypes: true });
+    }
+    catch {
+        return files;
+    }
+    for (const entry of entries) {
+        if (!entry.isDirectory())
+            continue;
+        if (enabledSkillNames && !enabledSkillNames.has(entry.name))
+            continue;
+        const skillPath = join(skillsDir, entry.name);
+        let subFiles;
+        try {
+            subFiles = await readdir(skillPath);
+        }
+        catch {
+            continue;
+        }
+        for (const sf of subFiles) {
+            if (sf === "skill.md" || !sf.endsWith(".md"))
+                continue;
+            const content = await readFile(join(skillPath, sf), "utf-8");
+            const rendered = renderTemplate(content, input.variables);
+            const outPath = join(outputDir, entry.name, sf);
+            await mkdir(dirname(outPath), { recursive: true });
+            await writeFile(outPath, rendered, "utf-8");
+            files.push(outPath);
+        }
+    }
+    return files;
+}