@ai-setting/roy-agent-core 1.3.10 → 1.3.14

package/src/config/env-source.ts

@@ -1,301 +0,0 @@
-import {
-  ConfigSource,
-  ConfigSourceType,
-  ConfigChangeEvent,
-} from "./config-component";
-import { toEnvKey, fromEnvKey } from "./env-key";
-
-/**
- * EnvSource 配置
- */
-export interface EnvSourceOptions {
-  /** 环境变量前缀 */
-  prefix?: string;
-  /** 值转换函数 */
-  transform?: (value: string, key: string) => unknown;
-  /** 轮询间隔（ms），用于检测环境变量变化 */
-  pollInterval?: number;
-  /** 优先级 */
-  priority?: number;
-  /** 是否启用监听（轮询），默认 true */
-  watch?: boolean;
-}
-
-/**
- * 环境变量配置源
- * 
- * 从 process.env 读取配置，支持：
- * - 前缀过滤
- * - 值转换
- * - 环境变量监听（轮询）
- */
-export class EnvSource implements ConfigSource {
-  readonly name: ConfigSourceType = "env";
-  readonly priority = 20;
-
-  private prefix: string;
-  private transform?: (value: string, key: string) => unknown;
-  private pollInterval?: number;
-  private watchers = new Set<(event: ConfigChangeEvent) => void>();
-  private pollTimer?: NodeJS.Timeout;
-  private lastValues = new Map<string, string>();
-  private watchEnabled: boolean = true;
-
-  constructor(options: EnvSourceOptions = {}) {
-    this.prefix = options.prefix ?? "";
-    this.transform = options.transform;
-    this.pollInterval = options.pollInterval;
-    this.watchEnabled = options.watch ?? true;
-  }
-
-  /**
-   * 获取环境变量名（带前缀）
-   * 将点号路径转换为下划线分隔的大写形式
-   * 例如: "llm.provider" + prefix "LLM" -> "LLM_LLM_PROVIDER"
-   * 例如: "DEBUG" + prefix "MY_APP_" -> "MY_APP_DEBUG"
-   * 例如: "llm.defaultModel" + prefix "LLM" -> "LLM_LLM_DEFAULT_MODEL"
-   */
-  private getEnvKey(key: string): string {
-    return toEnvKey(key, this.prefix);
-  }
-
-  /**
-   * 移除前缀获取内部 key
-   * 注意：保持原始格式（大小写），仅移除前缀
-   */
-  private getInternalKey(envKey: string): string {
-    return fromEnvKey(envKey, this.prefix);
-  }
-
-  /**
-   * 读取环境变量
-   */
-  read(key: string): unknown {
-    const envKey = this.getEnvKey(key);
-    const value = process.env[envKey];
-    
-    if (value === undefined) {
-      return undefined;
-    }
-
-    if (this.transform) {
-      return this.transform(value, key);
-    }
-
-    return value;
-  }
-
-  /**
-   * 写入环境变量
-   */
-  write(key: string, value: unknown): boolean {
-    const envKey = this.getEnvKey(key);
-    const oldValue = process.env[envKey];
-    const stringValue = String(value);
-
-    // 设置环境变量
-    process.env[envKey] = stringValue;
-
-    // 通知监听器
-    const event: ConfigChangeEvent = {
-      type: oldValue === undefined ? "add" : "change",
-      key,
-      oldValue: oldValue !== undefined ? this.transformValue(oldValue, key) : undefined,
-      newValue: this.transformValue(stringValue, key),
-      source: this.name,
-      timestamp: Date.now(),
-    };
-    this.notifyWatchers(event);
-
-    return true;
-  }
-
-  /**
-   * 删除环境变量
-   */
-  delete(key: string): boolean {
-    const envKey = this.getEnvKey(key);
-    const oldValue = process.env[envKey];
-
-    if (oldValue === undefined) {
-      return false;
-    }
-
-    delete process.env[envKey];
-
-    // 通知监听器
-    const event: ConfigChangeEvent = {
-      type: "delete",
-      key,
-      oldValue: this.transformValue(oldValue, key),
-      newValue: undefined,
-      source: this.name,
-      timestamp: Date.now(),
-    };
-    this.notifyWatchers(event);
-
-    return true;
-  }
-
-  /**
-   * 列出所有环境变量
-   */
-  list(): Array<{ key: string; value: unknown }> {
-    const result: Array<{ key: string; value: unknown }> = [];
-    const prefix = this.prefix.toUpperCase();
-
-    for (const envKey of Object.keys(process.env)) {
-      // 检查前缀匹配
-      if (prefix && !envKey.startsWith(prefix)) {
-        continue;
-      }
-
-      const key = this.getInternalKey(envKey);
-      const value = process.env[envKey];
-
-      if (value !== undefined) {
-        result.push({
-          key,
-          value: this.transformValue(value, key),
-        });
-      }
-    }
-
-    return result;
-  }
-
-  /**
-   * 监听环境变量变化
-   */
-  watch(callback: (event: ConfigChangeEvent) => void): () => void {
-    this.watchers.add(callback);
-
-    // 如果 watch 被禁用，直接返回
-    if (!this.watchEnabled) {
-      return () => {
-        this.watchers.delete(callback);
-      };
-    }
-
-    // 启动轮询
-    this.ensurePolling();
-
-    // 记录初始值
-    this.recordCurrentValues();
-
-    return () => {
-      this.watchers.delete(callback);
-
-      // 停止轮询如果没有监听器
-      if (this.watchers.size === 0) {
-        this.stopPolling();
-      }
-    };
-  }
-
-  private ensurePolling(): void {
-    if (this.pollTimer !== undefined) {
-      return;
-    }
-
-    const interval = this.pollInterval ?? 1000;
-    this.pollTimer = setInterval(() => {
-      this.checkForChanges();
-    }, interval);
-  }
-
-  private stopPolling(): void {
-    if (this.pollTimer) {
-      clearInterval(this.pollTimer);
-      this.pollTimer = undefined;
-    }
-  }
-
-  private recordCurrentValues(): void {
-    this.lastValues.clear();
-    const entries = this.list();
-    for (const entry of entries) {
-      const envKey = this.getEnvKey(entry.key);
-      const value = process.env[envKey];
-      if (value !== undefined) {
-        this.lastValues.set(envKey, value);
-      }
-    }
-  }
-
-  private checkForChanges(): void {
-    const currentEntries = this.list();
-    const currentValues = new Map<string, string>();
-
-    for (const entry of currentEntries) {
-      const envKey = this.getEnvKey(entry.key);
-      const value = process.env[envKey];
-      if (value !== undefined) {
-        currentValues.set(envKey, value);
-      }
-    }
-
-    // 检测删除
-    for (const [envKey, oldValue] of this.lastValues.entries()) {
-      if (!currentValues.has(envKey)) {
-        const key = this.getInternalKey(envKey);
-        this.notifyWatchers({
-          type: "delete",
-          key,
-          oldValue: this.transformValue(oldValue, key),
-          newValue: undefined,
-          source: this.name,
-          timestamp: Date.now(),
-        });
-      }
-    }
-
-    // 检测添加和变更
-    for (const [envKey, newValue] of currentValues.entries()) {
-      const key = this.getInternalKey(envKey);
-      const oldValue = this.lastValues.get(envKey);
-
-      if (oldValue === undefined) {
-        this.notifyWatchers({
-          type: "add",
-          key,
-          oldValue: undefined,
-          newValue: this.transformValue(newValue, key),
-          source: this.name,
-          timestamp: Date.now(),
-        });
-      } else if (oldValue !== newValue) {
-        this.notifyWatchers({
-          type: "change",
-          key,
-          oldValue: this.transformValue(oldValue, key),
-          newValue: this.transformValue(newValue, key),
-          source: this.name,
-          timestamp: Date.now(),
-        });
-      }
-    }
-
-    // 更新记录
-    this.lastValues = currentValues;
-  }
-
-  private transformValue(value: string, key: string): unknown {
-    if (this.transform) {
-      return this.transform(value, key);
-    }
-    return value;
-  }
-
-  private notifyWatchers(event: ConfigChangeEvent): void {
-    this.watchers.forEach((cb) => cb(event));
-  }
-
-  /**
-   * 关闭并清理资源
-   */
-  close(): void {
-    this.stopPolling();
-    this.watchers.clear();
-  }
-}

package/src/config/file-source.test.ts

@@ -1,357 +0,0 @@
-import { describe, test, expect, beforeEach, afterEach, vi } from "vitest";
-import { promises as fs } from "fs";
-import path from "path";
-import { FileSource } from "./file-source";
-
-describe("FileSource", () => {
-  const testDir = path.join(__dirname, ".test-file-source");
-  const testFile = path.join(testDir, "config.json");
-
-  beforeEach(async () => {
-    // 创建测试目录
-    await fs.mkdir(testDir, { recursive: true });
-  });
-
-  afterEach(async () => {
-    // 清理测试目录
-    try {
-      await fs.rm(testDir, { recursive: true, force: true });
-    } catch {
-      // ignore
-    }
-  });
-
-  test("read returns undefined for non-existent key", async () => {
-    const source = new FileSource(testFile);
-    const value = source.read("nonExistent");
-    expect(value).toBeUndefined();
-  });
-
-  test("read loads value from JSON file", async () => {
-    // 初始化文件
-    await fs.writeFile(testFile, JSON.stringify({ name: "roy", age: 30 }), "utf-8");
-
-    const source = new FileSource(testFile);
-    expect(source.read("name")).toBe("roy");
-    expect(source.read("age")).toBe(30);
-  });
-
-  test("read returns nested value using dot notation", async () => {
-    await fs.writeFile(
-      testFile,
-      JSON.stringify({ user: { profile: { name: "alice" } } }),
-      "utf-8"
-    );
-
-    const source = new FileSource(testFile);
-    expect(source.read("user.profile.name")).toBe("alice");
-  });
-
-  test("write saves value to JSON file", async () => {
-    const source = new FileSource(testFile);
-    const result = source.write("name", "roy");
-    expect(result).toBe(true);
-
-    // 等待防抖延迟完成
-    await new Promise((r) => setTimeout(r, 150));
-
-    const content = await fs.readFile(testFile, "utf-8");
-    const data = JSON.parse(content);
-    expect(data.name).toBe("roy");
-  });
-
-  test("write updates existing value", async () => {
-    await fs.writeFile(testFile, JSON.stringify({ name: "old" }), "utf-8");
-
-    const source = new FileSource(testFile);
-    source.write("name", "new");
-
-    // 等待防抖延迟完成
-    await new Promise((r) => setTimeout(r, 150));
-
-    const content = await fs.readFile(testFile, "utf-8");
-    const data = JSON.parse(content);
-    expect(data.name).toBe("new");
-  });
-
-  test("write creates nested structure for dot notation keys", async () => {
-    const source = new FileSource(testFile);
-    source.write("user.profile.name", "bob");
-
-    // 等待防抖延迟完成
-    await new Promise((r) => setTimeout(r, 150));
-
-    const content = await fs.readFile(testFile, "utf-8");
-    const data = JSON.parse(content);
-    expect(data.user.profile.name).toBe("bob");
-  });
-
-  test("list returns all key-value pairs", async () => {
-    await fs.writeFile(testFile, JSON.stringify({ name: "roy", age: 30, active: true }), "utf-8");
-
-    const source = new FileSource(testFile);
-    const entries = source.list();
-
-    expect(entries).toContainEqual({ key: "name", value: "roy" });
-    expect(entries).toContainEqual({ key: "age", value: 30 });
-    expect(entries).toContainEqual({ key: "active", value: true });
-  });
-
-  test("list returns flattened nested structure", async () => {
-    await fs.writeFile(
-      testFile,
-      JSON.stringify({ user: { name: "roy", age: 30 } }),
-      "utf-8"
-    );
-
-    const source = new FileSource(testFile);
-    const entries = source.list();
-
-    expect(entries).toContainEqual({ key: "user.name", value: "roy" });
-    expect(entries).toContainEqual({ key: "user.age", value: 30 });
-  });
-
-  test("watch monitors file changes", async () => {
-    await fs.writeFile(testFile, JSON.stringify({ name: "roy" }), "utf-8");
-
-    const source = new FileSource(testFile);
-    const events: any[] = [];
-
-    const unwatch = source.watch((event) => {
-      events.push(event);
-    });
-
-    // 等待文件系统稳定
-    await new Promise((r) => setTimeout(r, 100));
-
-    // 修改文件
-    await fs.writeFile(testFile, JSON.stringify({ name: "alice" }), "utf-8");
-
-    // 等待事件触发
-    await new Promise((r) => setTimeout(r, 200));
-
-    unwatch();
-
-    expect(events.length).toBeGreaterThan(0);
-  });
-
-  test("watch returns unsubscribe function", async () => {
-    const source = new FileSource(testFile);
-    const callback = vi.fn();
-
-    const unwatch = source.watch(callback);
-    unwatch();
-
-    await fs.writeFile(testFile, JSON.stringify({ name: "changed" }), "utf-8");
-    await new Promise((r) => setTimeout(r, 200));
-
-    expect(callback).not.toHaveBeenCalled();
-  });
-
-  test("has correct name and priority", () => {
-    const source = new FileSource(testFile);
-    expect(source.name).toBe("file");
-    expect(source.priority).toBe(10);
-  });
-
-  test("delete removes key from file", async () => {
-    await fs.writeFile(testFile, JSON.stringify({ name: "roy", age: 30 }), "utf-8");
-
-    const source = new FileSource(testFile);
-    const result = source.delete("name");
-    expect(result).toBe(true);
-
-    // 等待防抖延迟完成
-    await new Promise((r) => setTimeout(r, 150));
-
-    const content = await fs.readFile(testFile, "utf-8");
-    const data = JSON.parse(content);
-    expect(data.name).toBeUndefined();
-    expect(data.age).toBe(30);
-  });
-
-  test("delete returns false for non-existent key", async () => {
-    await fs.writeFile(testFile, JSON.stringify({ name: "roy" }), "utf-8");
-
-    const source = new FileSource(testFile);
-    const result = source.delete("nonExistent");
-    expect(result).toBe(false);
-  });
-
-  test("should parse JSONC file with single-line comments", async () => {
-    const jsoncFile = path.join(testDir, "config.jsonc");
-    const content = `{
-      // API Configuration
-      "apiKey": "test-key",
-      "model": "gpt-4o"
-    }`;
-    await fs.writeFile(jsoncFile, content, "utf-8");
-
-    const source = new FileSource(jsoncFile);
-    expect(source.read("apiKey")).toBe("test-key");
-    expect(source.read("model")).toBe("gpt-4o");
-  });
-
-  test("should parse JSONC file with multi-line comments", async () => {
-    const jsoncFile = path.join(testDir, "config.jsonc");
-    const content = `{
-      /* This is a
-         multi-line comment */
-      "name": "test"
-    }`;
-    await fs.writeFile(jsoncFile, content, "utf-8");
-
-    const source = new FileSource(jsoncFile);
-    expect(source.read("name")).toBe("test");
-  });
-
-  test("should substitute env vars in config values", async () => {
-    process.env.TEST_API_KEY = "my-secret-key";
-    const envFile = path.join(testDir, "env-config.json");
-    await fs.writeFile(envFile, JSON.stringify({ apiKey: "${TEST_API_KEY}" }), "utf-8");
-
-    const source = new FileSource(envFile);
-    expect(source.read("apiKey")).toBe("my-secret-key");
-
-    delete process.env.TEST_API_KEY;
-  });
-
-  test("should substitute env vars with defaults", async () => {
-    delete process.env.NONEXISTENT_KEY;
-    const envFile = path.join(testDir, "env-defaults.json");
-    await fs.writeFile(envFile, JSON.stringify({ key: "${NONEXISTENT_KEY:-default-value}" }), "utf-8");
-
-    const source = new FileSource(envFile);
-    expect(source.read("key")).toBe("default-value");
-  });
-
-  test("should handle combined JSONC and env vars", async () => {
-    process.env.CFG_HOST = "localhost";
-    process.env.CFG_PORT = "8080";
-    const combinedFile = path.join(testDir, "combined.jsonc");
-    const content = `{
-      // Server configuration
-      "host": "\${CFG_HOST}",  // from env
-      "port": "\${CFG_PORT}",
-      "name": "app"
-    }`;
-    await fs.writeFile(combinedFile, content, "utf-8");
-
-    const source = new FileSource(combinedFile);
-    expect(source.read("host")).toBe("localhost");
-    expect(source.read("port")).toBe("8080");
-    expect(source.read("name")).toBe("app");
-
-    delete process.env.CFG_HOST;
-    delete process.env.CFG_PORT;
-  });
-
-  test("should preserve non-string values during env substitution", async () => {
-    const mixedFile = path.join(testDir, "mixed.json");
-    await fs.writeFile(mixedFile, JSON.stringify({ 
-      count: "${UNDEFINED_VAR:-42}",
-      enabled: true,
-      rate: 0.95
-    }), "utf-8");
-
-    const source = new FileSource(mixedFile);
-    expect(source.read("count")).toBe("42");  // env var resolved to string
-    expect(source.read("enabled")).toBe(true);
-    expect(source.read("rate")).toBe(0.95);
-  });
-
-  test("should resolve protocol references when resolver is provided", async () => {
-    // 创建被引用的文件
-    const authFile = path.join(testDir, "auth.json");
-    await fs.writeFile(authFile, JSON.stringify({
-      api: { key: "secret-key-from-file", url: "https://api.example.com" }
-    }), "utf-8");
-
-    // 创建主配置文件
-    const mainFile = path.join(testDir, "main.json");
-    await fs.writeFile(mainFile, JSON.stringify({
-      apiKey: "${file://" + authFile + "#api.key}",
-      apiUrl: "${file://" + authFile + "#api.url}"
-    }), "utf-8");
-
-    // 使用绝对路径的 file:// 协议
-    const { ProtocolResolver } = require("./protocol-resolver");
-    const resolver = new ProtocolResolver();
-    const source = new FileSource(mainFile, resolver);
-
-    expect(source.read("apiKey")).toBe("secret-key-from-file");
-    expect(source.read("apiUrl")).toBe("https://api.example.com");
-  });
-
-  test("should resolve file:// protocol references", async () => {
-    // 创建引用文件
-    const refFile = path.join(testDir, "ref.json");
-    await fs.writeFile(refFile, JSON.stringify({
-      token: "file-token-xxx",
-      secret: "file-secret"
-    }), "utf-8");
-
-    // 创建主配置文件
-    const mainFile = path.join(testDir, "app.json");
-    await fs.writeFile(mainFile, JSON.stringify({
-      token: "${file://ref.json#token}",
-      secret: "${file://ref.json#secret}"
-    }), "utf-8");
-
-    // 使用 file:// 协议
-    const { ProtocolResolver } = require("./protocol-resolver");
-    const resolver = new ProtocolResolver({
-      xdgDataHome: testDir,
-    });
-    const source = new FileSource(mainFile, resolver);
-
-    expect(source.read("token")).toBe("file-token-xxx");
-    expect(source.read("secret")).toBe("file-secret");
-  });
-
-  test("should resolve env:// protocol references", async () => {
-    process.env.PROTOCOL_TEST_VAR = "protocol-env-value";
-
-    const envFile = path.join(testDir, "env.json");
-    await fs.writeFile(envFile, JSON.stringify({
-      value: "${env://PROTOCOL_TEST_VAR}",
-      other: "static"
-    }), "utf-8");
-
-    const { ProtocolResolver } = require("./protocol-resolver");
-    const resolver = new ProtocolResolver();
-    const source = new FileSource(envFile, resolver);
-
-    expect(source.read("value")).toBe("protocol-env-value");
-    expect(source.read("other")).toBe("static");
-
-    delete process.env.PROTOCOL_TEST_VAR;
-  });
-
-  test("should handle combined env and protocol references", async () => {
-    process.env.DEFAULT_DB_HOST = "db.example.com";
-
-    // 创建引用文件
-    const refFile = path.join(testDir, "ref.json");
-    await fs.writeFile(refFile, JSON.stringify({
-      apiKey: "secret-api-key"
-    }), "utf-8");
-
-    const combinedFile = path.join(testDir, "combined.json");
-    await fs.writeFile(combinedFile, JSON.stringify({
-      host: "${env://DEFAULT_DB_HOST:-localhost}",
-      apiKey: "${file://ref.json#apiKey:-no-key}"
-    }), "utf-8");
-
-    const { ProtocolResolver } = require("./protocol-resolver");
-    const resolver = new ProtocolResolver({
-      xdgDataHome: testDir,
-    });
-    const source = new FileSource(combinedFile, resolver);
-
-    expect(source.read("host")).toBe("db.example.com");
-    expect(source.read("apiKey")).toBe("secret-api-key");
-
-    delete process.env.DEFAULT_DB_HOST;
-  });
-});