@ai-setting/roy-agent-cli 1.3.15 → 1.3.16

package/dist/core/src/env/prompt/prompt-component.test.js

@@ -0,0 +1,397 @@
+/**
+ * @fileoverview PromptComponent Tests
+ *
+ * TDD: Write tests first, then implement
+ */
+import { describe, test, expect, beforeEach, afterEach } from "bun:test";
+import { PromptComponent } from "./prompt-component.js";
+import { ConfigComponent } from "../../config/config-component.js";
+import { writeFile, rm, mkdir } from "fs/promises";
+import { join } from "path";
+import { vi } from "bun:test";
+// Mock Environment
+const createMockEnv = (configComponent) => ({
+    name: "mock-env",
+    version: "1.0.0",
+    getConfig: () => ({ name: "mock-env", version: "1.0.0", enabled: true, env: null }),
+    registerComponent: () => { },
+    unregisterComponent: () => { },
+    getComponent: (name) => name === "config" ? configComponent : undefined,
+    hasComponent: (name) => name === "config" && !!configComponent,
+    listComponents: () => [],
+    handle_query: async () => "",
+    handle_action: async () => ({ success: true, output: "" }),
+    subscribe: () => () => { },
+    subscribeTo: () => () => { },
+    subscribeAll: () => () => { },
+    pushEnvEvent: () => { },
+    logger: vi.fn(),
+    trace: vi.fn(),
+});
+// 创建临时目录用于测试
+const testDir = join(process.cwd(), "test-prompts-temp");
+async function setupTestDir() {
+    await mkdir(testDir, { recursive: true });
+    await mkdir(join(testDir, "subdir"), { recursive: true });
+}
+async function cleanupTestDir() {
+    try {
+        await rm(testDir, { recursive: true, force: true });
+    }
+    catch { }
+}
+describe("PromptComponent", () => {
+    let component;
+    let mockEnv;
+    let configComponent;
+    beforeEach(async () => {
+        component = new PromptComponent();
+        configComponent = new ConfigComponent();
+        mockEnv = createMockEnv(configComponent);
+    });
+    afterEach(async () => {
+        if (component) {
+            await component.stop();
+        }
+    });
+    // ============================================================================
+    // Component basics
+    // ============================================================================
+    describe("Component basics", () => {
+        test("should have correct name", () => {
+            expect(component.name).toBe("prompt");
+        });
+        test("should have version", () => {
+            expect(component.version).toBe("1.0.0");
+        });
+        test("should start with created status", () => {
+            expect(component.getStatus()).toBe("created");
+        });
+    });
+    // ============================================================================
+    // init
+    // ============================================================================
+    describe("init", () => {
+        test("should initialize with defaults", async () => {
+            await component.init({ name: "prompt", version: "1.0.0", enabled: true, env: mockEnv, options: { configComponent } });
+            expect(component.getStatus()).toBe("running");
+        });
+        test("should initialize with config options", async () => {
+            await component.init({
+                name: "prompt",
+                version: "1.0.0",
+                enabled: true,
+                env: mockEnv,
+                options: {
+                    configComponent,
+                    defaultName: "custom-default",
+                    variablePrefix: "[[",
+                    variableSuffix: "]]",
+                    strictMode: false,
+                },
+            });
+            expect(component.getStatus()).toBe("running");
+        });
+        test("should load built-in prompts if prompts dir exists", async () => {
+            await component.init({ name: "prompt", version: "1.0.0", enabled: true, env: mockEnv, options: { configComponent } });
+            // 应该加载内置的 default prompt
+            expect(component.has("default")).toBe(true);
+        });
+        test("should throw error if ConfigComponent is missing", async () => {
+            const badEnv = createMockEnv(undefined);
+            const badComponent = new PromptComponent();
+            await expect(badComponent.init({ name: "prompt", version: "1.0.0", enabled: true, env: badEnv, options: {} })).rejects.toThrow("ConfigComponent is required");
+        });
+    });
+    // ============================================================================
+    // add
+    // ============================================================================
+    describe("add", () => {
+        beforeEach(async () => {
+            await component.init({ name: "prompt", version: "1.0.0", enabled: true, env: mockEnv, options: { configComponent } });
+        });
+        test("should add a prompt with name and content", () => {
+            component.add("test", "Hello {{name}}!");
+            expect(component.has("test")).toBe(true);
+            expect(component.get("test")).toBe("Hello {{name}}!");
+        });
+        test("should add multiple prompts", () => {
+            component.add("prompt1", "Content 1");
+            component.add("prompt2", "Content 2");
+            component.add("prompt3", "Content 3");
+            expect(component.size()).toBeGreaterThanOrEqual(3);
+            expect(component.list()).toContain("prompt1");
+            expect(component.list()).toContain("prompt2");
+            expect(component.list()).toContain("prompt3");
+        });
+        test("should update existing prompt content", () => {
+            component.add("test", "Original");
+            component.add("test", "Updated");
+            expect(component.get("test")).toBe("Updated");
+        });
+    });
+    // ============================================================================
+    // get
+    // ============================================================================
+    describe("get", () => {
+        beforeEach(async () => {
+            await component.init({ name: "prompt", version: "1.0.0", enabled: true, env: mockEnv, options: { configComponent } });
+        });
+        test("should return undefined for non-existent prompt", () => {
+            expect(component.get("non-existent")).toBeUndefined();
+        });
+        test("should return content for existing prompt", () => {
+            component.add("test", "Test content");
+            expect(component.get("test")).toBe("Test content");
+        });
+    });
+    // ============================================================================
+    // has / delete
+    // ============================================================================
+    describe("has / delete", () => {
+        beforeEach(async () => {
+            await component.init({ name: "prompt", version: "1.0.0", enabled: true, env: mockEnv, options: { configComponent } });
+        });
+        test("should return true for existing prompt", () => {
+            component.add("test", "Content");
+            expect(component.has("test")).toBe(true);
+        });
+        test("should return false for non-existent prompt", () => {
+            expect(component.has("non-existent")).toBe(false);
+        });
+        test("should delete existing prompt", () => {
+            component.add("test", "Content");
+            expect(component.delete("test")).toBe(true);
+            expect(component.has("test")).toBe(false);
+        });
+        test("should return false when deleting non-existent prompt", () => {
+            expect(component.delete("non-existent")).toBe(false);
+        });
+    });
+    // ============================================================================
+    // list / size
+    // ============================================================================
+    describe("list / size", () => {
+        beforeEach(async () => {
+            await component.init({ name: "prompt", version: "1.0.0", enabled: true, env: mockEnv, options: { configComponent } });
+        });
+        test("should list all prompt names", () => {
+            component.add("a", "Content A");
+            component.add("b", "Content B");
+            const list = component.list();
+            expect(list).toContain("a");
+            expect(list).toContain("b");
+        });
+        test("should return correct size", () => {
+            const initialSize = component.size();
+            component.add("new1", "Content");
+            component.add("new2", "Content");
+            expect(component.size()).toBe(initialSize + 2);
+        });
+    });
+    // ============================================================================
+    // getPrompt (render)
+    // ============================================================================
+    describe("getPrompt", () => {
+        beforeEach(async () => {
+            await component.init({ name: "prompt", version: "1.0.0", enabled: true, env: mockEnv, options: { configComponent } });
+        });
+        test("should get prompt by name", async () => {
+            component.add("test", "Hello World");
+            const content = await component.getPrompt("test");
+            expect(content).toBe("Hello World");
+        });
+        test("should render variables in prompt", async () => {
+            component.add("greeting", "Hello {{name}}!");
+            const content = await component.getPrompt("greeting", { name: "Alice" });
+            expect(content).toBe("Hello Alice!");
+        });
+        test("should render multiple variables", async () => {
+            component.add("template", "{{greeting}}, {{name}}! Today is {{date}}.");
+            const content = await component.getPrompt("template", {
+                greeting: "Hello",
+                name: "Bob",
+                date: "2026-03-27",
+            });
+            expect(content).toBe("Hello, Bob! Today is 2026-03-27.");
+        });
+        test("should keep undefined variables as-is", async () => {
+            component.add("template", "Hello {{name}}, your email is {{email}}");
+            const content = await component.getPrompt("template", { name: "Alice" });
+            expect(content).toBe("Hello Alice, your email is {{email}}");
+        });
+        test("should fallback to default when prompt not found", async () => {
+            // 默认的 "default" prompt 已加载，所以会返回 default 的内容
+            // 这个测试验证当请求不存在的 prompt 时，会使用 default
+            const defaultContent = await component.getPrompt("default");
+            const fallbackContent = await component.getPrompt("non-existent");
+            // 由于 default 已加载，fallback 应该返回 default 的内容
+            expect(fallbackContent).toBe(defaultContent);
+        });
+    });
+    // ============================================================================
+    // render
+    // ============================================================================
+    describe("render", () => {
+        beforeEach(async () => {
+            await component.init({ name: "prompt", version: "1.0.0", enabled: true, env: mockEnv, options: { configComponent } });
+        });
+        test("should render template with variables", async () => {
+            const result = await component.render("Hello {{name}}!", { name: "World" });
+            expect(result).toBe("Hello World!");
+        });
+        test("should handle empty variables", async () => {
+            const result = await component.render("Hello {{name}}!", {});
+            expect(result).toBe("Hello {{name}}!");
+        });
+        test("should handle custom variable delimiters", async () => {
+            await component.init({
+                name: "prompt",
+                version: "1.0.0",
+                enabled: true,
+                env: mockEnv,
+                options: {
+                    configComponent,
+                    variablePrefix: "[[",
+                    variableSuffix: "]]",
+                },
+            });
+            const result = await component.render("Hello [[name]]!", { name: "World" });
+            expect(result).toBe("Hello World!");
+        });
+    });
+    // ============================================================================
+    // loadFromFile
+    // ============================================================================
+    describe("loadFromFile", () => {
+        beforeEach(async () => {
+            await setupTestDir();
+            await component.init({ name: "prompt", version: "1.0.0", enabled: true, env: mockEnv, options: { configComponent } });
+        });
+        afterEach(async () => {
+            await cleanupTestDir();
+        });
+        test("should load prompt from file", async () => {
+            const filePath = join(testDir, "test.md");
+            await writeFile(filePath, "Content from file");
+            const result = await component.loadFromFile(filePath);
+            expect(result).toBe(true);
+            expect(component.has("test")).toBe(true);
+            expect(component.get("test")).toBe("Content from file");
+        });
+        test("should use filename as prompt name by default", async () => {
+            const filePath = join(testDir, "my-prompt.md");
+            await writeFile(filePath, "My prompt content");
+            await component.loadFromFile(filePath);
+            expect(component.has("my-prompt")).toBe(true);
+        });
+        test("should use custom name when provided", async () => {
+            const filePath = join(testDir, "test.md");
+            await writeFile(filePath, "Content");
+            await component.loadFromFile(filePath, "custom-name");
+            expect(component.has("custom-name")).toBe(true);
+        });
+        test("should return false when file not found", async () => {
+            const result = await component.loadFromFile("/non/existent/file.md");
+            expect(result).toBe(false);
+        });
+    });
+    // ============================================================================
+    // loadFromDirectory
+    // ============================================================================
+    describe("loadFromDirectory", () => {
+        beforeEach(async () => {
+            await setupTestDir();
+            await component.init({ name: "prompt", version: "1.0.0", enabled: true, env: mockEnv, options: { configComponent } });
+        });
+        afterEach(async () => {
+            await cleanupTestDir();
+        });
+        test("should load all .md files from directory", async () => {
+            await writeFile(join(testDir, "prompt1.md"), "Content 1");
+            await writeFile(join(testDir, "prompt2.md"), "Content 2");
+            const count = await component.loadFromDirectory(testDir);
+            expect(count).toBeGreaterThanOrEqual(2);
+            expect(component.has("prompt1")).toBe(true);
+            expect(component.has("prompt2")).toBe(true);
+        });
+        test("should load from subdirectory with recursive", async () => {
+            await writeFile(join(testDir, "subdir", "nested.md"), "Nested content");
+            const count = await component.loadFromDirectory(testDir, ".md", true);
+            expect(component.has("subdir-nested")).toBe(true);
+        });
+        test("should not load from subdirectory without recursive", async () => {
+            await writeFile(join(testDir, "subdir", "nested.md"), "Nested content");
+            await component.loadFromDirectory(testDir, ".md", false);
+            expect(component.has("nested")).toBe(false);
+        });
+        test("should filter by extension", async () => {
+            await writeFile(join(testDir, "valid.txt"), "Should not load");
+            await writeFile(join(testDir, "valid.md"), "Should load");
+            const count = await component.loadFromDirectory(testDir, ".md", false);
+            expect(component.has("valid")).toBe(true);
+        });
+    });
+    // ============================================================================
+    // Hooks
+    // ============================================================================
+    describe("Hooks", () => {
+        beforeEach(async () => {
+            await component.init({ name: "prompt", version: "1.0.0", enabled: true, env: mockEnv, options: { configComponent } });
+        });
+        test("should execute before.render hook", async () => {
+            let hookCalled = false;
+            component.addHook("prompt.before-render", "test-before", async (ctx) => {
+                hookCalled = true;
+                const data = ctx.data;
+                expect(data.name).toBe("test");
+                expect(data.originalContent).toBe("Original");
+            });
+            component.add("test", "Original");
+            await component.getPrompt("test");
+            expect(hookCalled).toBe(true);
+        });
+        test("should execute after.render hook", async () => {
+            let hookCalled = false;
+            component.addHook("prompt.after-render", "test-after", async (ctx) => {
+                hookCalled = true;
+                const data = ctx.data;
+                expect(data.name).toBe("test");
+            });
+            component.add("test", "Original");
+            await component.getPrompt("test", { name: "Alice" });
+            expect(hookCalled).toBe(true);
+        });
+        test("should allow hook to modify rendered content", async () => {
+            component.addHook("prompt.after-render", "modifier", async (ctx) => {
+                const data = ctx.data;
+                data.renderedContent = data.renderedContent + " [modified]";
+            });
+            component.add("test", "Hello");
+            const content = await component.getPrompt("test");
+            expect(content).toBe("Hello [modified]");
+        });
+    });
+    // ============================================================================
+    // extractVariables
+    // ============================================================================
+    describe("extractVariables", () => {
+        beforeEach(async () => {
+            await component.init({ name: "prompt", version: "1.0.0", enabled: true, env: mockEnv, options: { configComponent } });
+        });
+        test("should extract variable names from template", () => {
+            const vars = component.extractVariables("Hello {{name}}, today is {{date}}!");
+            expect(vars).toContain("name");
+            expect(vars).toContain("date");
+        });
+        test("should return unique variable names", () => {
+            const vars = component.extractVariables("{{a}} {{b}} {{a}}");
+            expect(vars).toEqual(["a", "b"]);
+        });
+        test("should return empty array for template without variables", () => {
+            const vars = component.extractVariables("No variables here");
+            expect(vars).toEqual([]);
+        });
+    });
+});
+//# sourceMappingURL=prompt-component.test.js.map

package/dist/core/src/env/prompt/prompt-config-registration.d.ts

@@ -0,0 +1,19 @@
+/**
+ * @fileoverview Prompt Component 配置注册
+ *
+ * 使用 ConfigComponent.registerComponent() 机制统一管理配置
+ */
+import type { ComponentRegistration } from "../../config/config-component";
+/**
+ * Prompt Component 默认配置值
+ *
+ * 注意：键使用完整格式（带 "prompt." 前缀），在 registerConfig 中手动设置
+ */
+export declare const PROMPT_DEFAULTS: Record<string, unknown>;
+/**
+ * Prompt Component 配置注册常量
+ *
+ * 定义了所有 Prompt 相关的配置项和来源
+ */
+export declare const PROMPT_CONFIG_REGISTRATION: ComponentRegistration;
+//# sourceMappingURL=prompt-config-registration.d.ts.map

package/dist/core/src/env/prompt/prompt-config-registration.js

@@ -0,0 +1,34 @@
+/**
+ * @fileoverview Prompt Component 配置注册
+ *
+ * 使用 ConfigComponent.registerComponent() 机制统一管理配置
+ */
+/**
+ * Prompt Component 默认配置值
+ *
+ * 注意：键使用完整格式（带 "prompt." 前缀），在 registerConfig 中手动设置
+ */
+export const PROMPT_DEFAULTS = {
+    "prompt.defaultName": "default",
+    "prompt.variablePrefix": "{{",
+    "prompt.variableSuffix": "}}",
+};
+/**
+ * Prompt Component 配置注册常量
+ *
+ * 定义了所有 Prompt 相关的配置项和来源
+ */
+export const PROMPT_CONFIG_REGISTRATION = {
+    name: "prompt",
+    sources: [
+        { type: "env", envPrefix: "PROMPT", priority: 20, watch: false },
+    ],
+    keys: [
+        { key: "prompt.promptPaths", sources: ["env", "file"] },
+        { key: "prompt.defaultName", sources: ["env", "file"] },
+        { key: "prompt.variablePrefix", sources: ["env", "file"] },
+        { key: "prompt.variableSuffix", sources: ["env", "file"] },
+    ],
+    // 注意：不使用 defaults 字段，默认值在 registerConfig 中手动 set
+};
+//# sourceMappingURL=prompt-config-registration.js.map

package/dist/core/src/env/prompt/prompt-config-registration.test.d.ts

@@ -0,0 +1,7 @@
+/**
+ * @fileoverview Prompt Component 配置注册测试
+ *
+ * TDD: 先写测试，验证配置注册机制
+ */
+export {};
+//# sourceMappingURL=prompt-config-registration.test.d.ts.map

package/dist/core/src/env/prompt/prompt-config-registration.test.js

@@ -0,0 +1,172 @@
+/**
+ * @fileoverview Prompt Component 配置注册测试
+ *
+ * TDD: 先写测试，验证配置注册机制
+ */
+import { describe, it, expect, beforeEach } from "vitest";
+import { ConfigComponent } from "../../config/config-component.js";
+import { PromptComponent } from "./prompt-component.js";
+import { PROMPT_CONFIG_REGISTRATION, PROMPT_DEFAULTS } from "./prompt-config-registration.js";
+describe("Prompt Config Registration", () => {
+    let configComponent;
+    let component;
+    beforeEach(() => {
+        configComponent = new ConfigComponent();
+        component = new PromptComponent();
+    });
+    describe("PROMPT_CONFIG_REGISTRATION", () => {
+        it("应该包含正确的 name", () => {
+            expect(PROMPT_CONFIG_REGISTRATION.name).toBe("prompt");
+        });
+        it("应该包含 env source 配置", () => {
+            const envSource = PROMPT_CONFIG_REGISTRATION.sources.find((s) => s.type === "env");
+            expect(envSource).toBeDefined();
+            expect(envSource?.envPrefix).toBe("PROMPT");
+            expect(envSource?.priority).toBe(20);
+        });
+        it("应该包含所有配置 keys", () => {
+            const keys = PROMPT_CONFIG_REGISTRATION.keys;
+            expect(keys).toContainEqual(expect.objectContaining({ key: "prompt.promptPaths" }));
+            expect(keys).toContainEqual(expect.objectContaining({ key: "prompt.defaultName" }));
+            expect(keys).toContainEqual(expect.objectContaining({ key: "prompt.variablePrefix" }));
+            expect(keys).toContainEqual(expect.objectContaining({ key: "prompt.variableSuffix" }));
+        });
+    });
+    describe("PROMPT_DEFAULTS", () => {
+        it("应该包含 defaultName 默认值（完整格式）", () => {
+            expect(PROMPT_DEFAULTS["prompt.defaultName"]).toBe("default");
+        });
+        it("应该包含 variablePrefix 默认值（完整格式）", () => {
+            expect(PROMPT_DEFAULTS["prompt.variablePrefix"]).toBe("{{");
+        });
+        it("应该包含 variableSuffix 默认值（完整格式）", () => {
+            expect(PROMPT_DEFAULTS["prompt.variableSuffix"]).toBe("}}");
+        });
+    });
+    describe("配置注册到 ConfigComponent", () => {
+        it("应该能够注册 PROMPT_CONFIG_REGISTRATION", () => {
+            configComponent.registerComponent(PROMPT_CONFIG_REGISTRATION);
+            expect(() => configComponent.registerComponent(PROMPT_CONFIG_REGISTRATION)).not.toThrow();
+        });
+    });
+    describe("Component 配置加载", () => {
+        it("应该需要 ConfigComponent 初始化", async () => {
+            await expect(component.init({
+                options: {},
+            })).rejects.toThrow("ConfigComponent is required");
+        });
+        it("应该能够使用 ConfigComponent 初始化", async () => {
+            await component.init({
+                options: {
+                    configComponent,
+                },
+            });
+            expect(component.name).toBe("prompt");
+            expect(component.version).toBe("1.0.0");
+        });
+        it("应该能够通过 config 对象传入配置", async () => {
+            await component.init({
+                options: {
+                    configComponent,
+                    config: {
+                        defaultName: "custom-prompt",
+                        variablePrefix: "[[",
+                        variableSuffix: "]]",
+                    },
+                },
+            });
+            expect(component.name).toBe("prompt");
+        });
+    });
+    describe("环境变量覆盖", () => {
+        it("应该能够通过环境变量覆盖配置", async () => {
+            // 设置环境变量
+            const originalValue = process.env.PROMPT_PROMPT_DEFAULT_NAME;
+            process.env.PROMPT_PROMPT_DEFAULT_NAME = "env-prompt";
+            try {
+                await component.init({
+                    options: {
+                        configComponent,
+                    },
+                });
+                expect(component.name).toBe("prompt");
+            }
+            finally {
+                // 恢复环境变量
+                if (originalValue !== undefined) {
+                    process.env.PROMPT_PROMPT_DEFAULT_NAME = originalValue;
+                }
+                else {
+                    delete process.env.PROMPT_PROMPT_DEFAULT_NAME;
+                }
+            }
+        });
+    });
+    describe("registerConfig 使用 registerComponent + load 模式", () => {
+        it("应该使用 registerComponent 注册配置", async () => {
+            // 注册组件
+            configComponent.registerComponent(PROMPT_CONFIG_REGISTRATION);
+            // 加载配置
+            await configComponent.load("prompt");
+            // 验证注册成功
+            const componentRegs = configComponent.componentRegistrations;
+            expect(componentRegs.has("prompt")).toBe(true);
+        });
+        it("应该通过 load 加载 prompt 配置到内存", async () => {
+            // 注册组件
+            configComponent.registerComponent(PROMPT_CONFIG_REGISTRATION);
+            // 加载配置
+            await configComponent.load("prompt");
+            // 验证配置已注册（组件注册成功）
+            const componentRegs = configComponent.componentRegistrations;
+            expect(componentRegs.has("prompt")).toBe(true);
+        });
+        it("应该从环境变量覆盖默认值", async () => {
+            // 保存原始值
+            const originalValue = process.env.PROMPT_PROMPT_DEFAULT_NAME;
+            try {
+                // 设置环境变量
+                process.env.PROMPT_PROMPT_DEFAULT_NAME = "env-default-name";
+                // 注册组件并加载
+                configComponent.registerComponent(PROMPT_CONFIG_REGISTRATION);
+                await configComponent.load("prompt");
+                // 验证环境变量优先
+                const defaultName = configComponent.get("prompt.defaultName");
+                expect(defaultName).toBe("env-default-name");
+            }
+            finally {
+                // 恢复环境变量
+                if (originalValue !== undefined) {
+                    process.env.PROMPT_PROMPT_DEFAULT_NAME = originalValue;
+                }
+                else {
+                    delete process.env.PROMPT_PROMPT_DEFAULT_NAME;
+                }
+            }
+        });
+        it("应该允许通过 config 对象传入最高优先级配置", async () => {
+            // 注册组件并加载
+            configComponent.registerComponent(PROMPT_CONFIG_REGISTRATION);
+            await configComponent.load("prompt");
+            // 设置内存中的配置（模拟 config 对象的行为）
+            configComponent.set("prompt.defaultName", "config-object-name");
+            // 验证 config 对象优先
+            const defaultName = configComponent.get("prompt.defaultName");
+            expect(defaultName).toBe("config-object-name");
+        });
+        it("PromptComponent.init 应该自动完成配置注册和加载", async () => {
+            await component.init({
+                options: {
+                    configComponent,
+                },
+            });
+            // 验证组件注册
+            const componentRegs = configComponent.componentRegistrations;
+            expect(componentRegs.has("prompt")).toBe(true);
+            // 验证配置已加载
+            const defaultName = configComponent.get("prompt.defaultName");
+            expect(defaultName).toBe("default");
+        });
+    });
+});
+//# sourceMappingURL=prompt-config-registration.test.js.map