foliko 1.1.67 → 1.1.69

package/src/core/framework.js

@@ -12,10 +12,9 @@ const { Agent } = require('./agent');
 const { Subagent } = require('./subagent');
 const { SubAgentConfigManager } = require('./sub-agent-config');
 const { ContextManager } = require('./context-manager');
-const { SessionContext } = require('./session-context');
-const { SessionStorageAdapter, getDefaultAdapter } = require('./session-storage');
+const { SessionManager } = require('./session-manager');
 const { CoordinatorManager } = require('./coordinator-manager');
-const { logEmitter, Logger, LOG_LEVELS } = require('../utils/logger');
+const { Logger, LOG_LEVELS } = require('../utils/logger');
 // 创建一个连接到 Framework 的 logger
 function createFrameworkLogger(framework) {
   const logger = new Logger({ namespace: 'foliko' });
@@ -124,13 +123,11 @@ class Framework extends EventEmitter {
     this._contextManager = new ContextManager(this);
 
     // SessionContext 缓存（轻量管理，不依赖 ContextManager）
-    this._sessionContexts = new Map(); // sessionId → SessionContext
+    this._sessionContexts = new Map(); // sessionId → SessionManager
 
-    // Session 存储适配器（默认使用文件存储）
-    this._sessionStorage = null;
-    this._sessionStorageConfig = config.sessionStorage || {
-      type: 'file',
-      baseDir: '.agent/sessions',
+    // Session 存储配置（JSONL 格式）
+    this._sessionStorageConfig = {
+      baseDir: '.foliko/sessions',
     };
 
     // 注册 logger 到 framework
@@ -245,6 +242,70 @@ class Framework extends EventEmitter {
     return this;
   }
 
+  /**
+   * 注册 LLM Provider
+   * @param {string} name - Provider 名称
+   * @param {Object} config - Provider 配置
+   */
+  registerProvider(name, config) {
+    const { registerProvider } = require('./provider');
+    registerProvider(name, config);
+    return this;
+  }
+
+  /**
+   * 获取 Provider Registry
+   */
+  getProviderRegistry() {
+    const { getProviderRegistry } = require('./provider');
+    return getProviderRegistry();
+  }
+
+  /**
+   * 注册 CLI 命令
+   * @param {string} name - 命令名称
+   * @param {Function} handler - 命令处理函数
+   * @param {string} description - 命令描述
+   * @param {Function} [getArgumentCompletions] - 自动补全函数
+   */
+  registerCommand(name, handler, description, getArgumentCompletions) {
+    const { getCommandRegistry } = require('./command-registry');
+    const registry = getCommandRegistry();
+    registry.registerCommand(name, handler, description, getArgumentCompletions);
+    return this;
+  }
+
+  /**
+   * 获取 CommandRegistry
+   */
+  getCommandRegistry() {
+    const { getCommandRegistry } = require('./command-registry');
+    return getCommandRegistry();
+  }
+
+  /**
+   * 创建 UI 扩展上下文
+   * @param {string} pluginName - 插件名称
+   * @param {Object} chatUI - ChatUI 实例
+   * @returns {ExtensionUIContext}
+   */
+  createUIContext(pluginName, chatUI) {
+    const { ExtensionUIContext } = require('./ui-extension-context');
+    if (!this._uiExtensions) {
+      this._uiExtensions = new Map();
+    }
+    const ctx = new ExtensionUIContext(chatUI, pluginName);
+    this._uiExtensions.set(pluginName, ctx);
+    return ctx;
+  }
+
+  /**
+   * 获取 UI 扩展上下文
+   */
+  getUIContext(pluginName) {
+    return this._uiExtensions?.get(pluginName) || null;
+  }
+
   /**
    * 加载并启动插件）
    * @param {Plugin|Object} plugin - 插件
@@ -324,6 +385,55 @@ class Framework extends EventEmitter {
     return this.toolRegistry.getAll();
   }
 
+  // ============================================================================
+  // 公开 API — 减少外部直接访问私有属性
+  // ============================================================================
+
+  /**
+   * 获取主 Agent
+   * @returns {Agent|null}
+   */
+  getMainAgent() {
+    return this._mainAgent;
+  }
+
+  /**
+   * 获取主 Agent 的 System Prompt
+   * @returns {string}
+   */
+  getSystemPrompt() {
+    if (this._mainAgent) {
+      return this._mainAgent._buildSystemPrompt();
+    }
+    return '';
+  }
+
+  /**
+   * 获取插件实例（安全访问）
+   * @param {string} name - 插件名称
+   * @returns {Object|null}
+   */
+  getPluginInstance(name) {
+    const entry = this.pluginManager.get(name);
+    return entry || null;
+  }
+
+  /**
+   * 获取所有 Agent 列表
+   * @returns {Array<Agent>}
+   */
+  getAgents() {
+    return [...this._agents];
+  }
+
+  /**
+   * 获取所有 SessionContext 的 ID 列表
+   * @returns {string[]}
+   */
+  listSessionContexts() {
+    return Array.from(this._sessionContexts.keys());
+  }
+
   // ============================================================================
   // 事件描述注册（供 Ambient Agent 使用）
   // ============================================================================
@@ -398,10 +508,10 @@ class Framework extends EventEmitter {
   // ============================================================================
 
   /**
-   * 获取或创建 SessionContext
+   * 获取或创建 SessionManager
    * @param {string} sessionId - 会话 ID
    * @param {Object} options - 配置选项
-   * @returns {SessionContext}
+   * @returns {SessionManager}
    */
   getOrCreateSessionContext(sessionId, options = {}) {
     if (!sessionId) {
@@ -409,32 +519,42 @@ class Framework extends EventEmitter {
     }
 
     // 先检查内存缓存
-    let ctx = this._sessionContexts.get(sessionId);
-    if (ctx) {
-      return ctx;
+    let manager = this._sessionContexts.get(sessionId);
+    if (manager) {
+      return manager;
     }
 
-    // 尝试从存储同步加载（仅文件存储支持）
-    const storage = this._getSessionStorage();
-    const loadedCtx = SessionContext.loadSync(sessionId, this, { ...options, storage });
-    if (loadedCtx) {
-      this._sessionContexts.set(sessionId, loadedCtx);
-      this.emit('session:context-created', loadedCtx);
-      return loadedCtx;
+    // 获取工作目录
+    const cwd = options.cwd || process.cwd();
+    const sessionDir = path.join(cwd, '.foliko', 'sessions');
+
+    // 获取或创建 session file path
+    const sessionFile = options.sessionFile || path.join(sessionDir, `${sessionId}.jsonl`);
+
+    // 尝试打开已存在的 session
+    try {
+      if (require('fs').existsSync(sessionFile)) {
+        manager = SessionManager.open(sessionFile, sessionDir, cwd);
+      }
+    } catch (err) {
+      // 文件损坏或不存在，创建新的
     }
 
-    // 创建新的 SessionContext
-    ctx = new SessionContext(sessionId, this, { ...options, storage });
-    this._sessionContexts.set(sessionId, ctx);
-    this.emit('session:context-created', ctx);
-    return ctx;
+    if (!manager) {
+      // 创建新的 SessionManager
+      manager = new SessionManager(cwd, sessionDir, sessionFile, true);
+    }
+
+    this._sessionContexts.set(sessionId, manager);
+    this.emit('session:context-created', { sessionId, manager });
+    return manager;
   }
 
   /**
-   * 异步加载 SessionContext（如果存在）
+   * 异步加载 SessionManager（如果存在）
    * @param {string} sessionId
    * @param {Object} options
-   * @returns {Promise<SessionContext|null>}
+   * @returns {Promise<SessionManager|null>}
    */
   async loadSessionContext(sessionId, options = {}) {
     if (!sessionId) {
@@ -442,55 +562,49 @@ class Framework extends EventEmitter {
     }
 
     // 先检查内存缓存
-    let ctx = this._sessionContexts.get(sessionId);
-    if (ctx) {
-      return ctx;
+    let manager = this._sessionContexts.get(sessionId);
+    if (manager) {
+      return manager;
     }
 
-    const storage = this._getSessionStorage();
-    ctx = await SessionContext.load(sessionId, this, { ...options, storage });
-    if (ctx) {
-      this._sessionContexts.set(sessionId, ctx);
-      this.emit('session:context-created', ctx);
-    }
-    return ctx;
-  }
+    const cwd = options.cwd || process.cwd();
+    const sessionDir = path.join(cwd, '.foliko', 'sessions');
+    const sessionFile = path.join(sessionDir, `${sessionId}.jsonl`);
 
-  /**
-   * 获取 Session 存储适配器
-   * @returns {SessionStorageAdapter}
-   * @private
-   */
-  _getSessionStorage() {
-    if (!this._sessionStorage) {
-      this._sessionStorage = new SessionStorageAdapter(this._sessionStorageConfig);
+    try {
+      if (require('fs').existsSync(sessionFile)) {
+        manager = SessionManager.open(sessionFile, sessionDir, cwd);
+        this._sessionContexts.set(sessionId, manager);
+        this.emit('session:context-created', { sessionId, manager });
+      }
+    } catch (err) {
+      // 文件不存在或损坏
     }
-    return this._sessionStorage;
+
+    return manager || null;
   }
 
   /**
-   * 配置 Session 存储类型
+   * 配置 Session 存储目录
    * @param {Object} config - 存储配置
-   * @param {string} [config.type='memory'] - 存储类型: 'memory', 'file'
-   * @param {string} [config.baseDir='.agent/sessions'] - 文件存储目录
+   * @param {string} [config.baseDir='.foliko/sessions'] - 文件存储目录
    */
   configureSessionStorage(config) {
     this._sessionStorageConfig = config;
-    this._sessionStorage = new SessionStorageAdapter(config);
   }
 
   /**
-   * 获取当前 Session 存储适配器
-   * @returns {SessionStorageAdapter}
+   * 获取当前 Session 存储配置
+   * @returns {Object}
    */
   getSessionStorage() {
-    return this._getSessionStorage();
+    return this._sessionStorageConfig;
   }
 
   /**
-   * 获取 SessionContext
+   * 获取 SessionManager
    * @param {string} sessionId - 会话 ID
-   * @returns {SessionContext|null}
+   * @returns {SessionManager|null}
    */
   getSessionContext(sessionId) {
     return this._sessionContexts.get(sessionId) || null;
@@ -498,11 +612,13 @@ class Framework extends EventEmitter {
 
   /**
    * 获取当前 Session 上下文（从 AsyncLocalStorage 获取）
-   * @returns {SessionContext|null}
+   * @returns {SessionManager|null}
    */
   getCurrentSessionContext() {
     const store = sessionStorage.getStore();
-    return store?.sessionContext || null;
+    if (!store) return null;
+    const sessionId = store.sessionId;
+    return sessionId ? this._sessionContexts.get(sessionId) || null : null;
   }
 
   /**
@@ -527,27 +643,82 @@ class Framework extends EventEmitter {
       return fn();
     }
 
-    const sessionCtx = this.getOrCreateSessionContext(sessionId, options);
-    const context = { sessionContext: sessionCtx, sessionId };
+    const manager = this.getOrCreateSessionContext(sessionId, options);
+    const context = { sessionContext: manager, sessionId };
     return sessionStorage.run(context, fn);
   }
 
   /**
-   * 销毁 Session Context
+   * 销毁 Session Manager
    * @param {string} sessionId
    */
   destroySessionContext(sessionId) {
     if (this._sessionContexts.has(sessionId)) {
-      const ctx = this._sessionContexts.get(sessionId);
-      ctx.destroy();
+      const manager = this._sessionContexts.get(sessionId);
       this._sessionContexts.delete(sessionId);
+      this.emit('session:context-destroyed', { sessionId });
+    }
+  }
 
-      // 从存储中删除
-      if (this._sessionStorage) {
-        this._sessionStorage.delete(sessionId).catch(() => {});
-      }
+  // Session TTL 清理配置
+  _sessionTTL = 30 * 60 * 1000; // 默认 30 分钟
+  _sessionCleanupInterval = null;
+
+  /**
+   * 配置 Session TTL
+   * @param {number} ttlMs - TTL 毫秒数
+   */
+  setSessionTTL(ttlMs) {
+    this._sessionTTL = ttlMs;
+    return this;
+  }
 
-      this.emit('session:context-destroyed', ctx);
+  /**
+   * 启动 Session 自动清理
+   * @param {number} [intervalMs=60000] - 检查间隔
+   */
+  startSessionCleanup(intervalMs = 60000) {
+    if (this._sessionCleanupInterval) {
+      clearInterval(this._sessionCleanupInterval);
+    }
+    this._sessionCleanupInterval = setInterval(() => {
+      this._cleanupStaleSessions();
+    }, intervalMs);
+    if (this._sessionCleanupInterval.unref) {
+      this._sessionCleanupInterval.unref();
+    }
+    return this;
+  }
+
+  /**
+   * 停止 Session 自动清理
+   */
+  stopSessionCleanup() {
+    if (this._sessionCleanupInterval) {
+      clearInterval(this._sessionCleanupInterval);
+      this._sessionCleanupInterval = null;
+    }
+    return this;
+  }
+
+  /**
+   * 清理过期的 Session
+   * @private
+   */
+  _cleanupStaleSessions() {
+    const now = Date.now();
+    const staleIds = [];
+    for (const [sessionId, manager] of this._sessionContexts) {
+      const metadata = manager.getMetadata?.();
+      if (metadata && metadata.lastActive && (now - metadata.lastActive) > this._sessionTTL) {
+        staleIds.push(sessionId);
+      }
+    }
+    for (const id of staleIds) {
+      this.destroySessionContext(id);
+    }
+    if (staleIds.length > 0) {
+      this.logger.debug(`Cleaned up ${staleIds.length} stale sessions`);
     }
   }
 
@@ -737,9 +908,9 @@ class Framework extends EventEmitter {
 
   /**
    * 同步主Agent的提示词部分到所有子Agent
-   * @private
+   * @public — 插件可能需要调用
    */
-  _syncPromptPartsToSubagents() {
+  syncPromptPartsToSubagents() {
     if (!this._mainAgent || !this._mainAgent._systemPromptBuilder) return;
     const mainParts = this._mainAgent._systemPromptBuilder._parts;
     if (!mainParts) return;
@@ -784,8 +955,9 @@ class Framework extends EventEmitter {
     lines.push('');
 
     // 2. 主Agent的系统提示词（基础部分）
-    if (this._mainAgent) {
-      const mainPrompt = this._mainAgent._originalPrompt;
+    const mainAgent = this.getMainAgent();
+    if (mainAgent) {
+      const mainPrompt = mainAgent.getOriginalPrompt();
       if (mainPrompt) {
         lines.push('## 主Agent的系统提示词');
         lines.push(mainPrompt);
@@ -833,9 +1005,9 @@ class Framework extends EventEmitter {
 
   /**
    * Bootstrap - 使用默认配置启动框架
-   * 自动加载 .agent/ 目录下的配置和所有默认插件）
+   * 自动加载 .foliko/ 目录下的配置和所有默认插件）
    * @param {Object} config - 配置
-   * @param {string} [config.agentDir='.agent'] - Agent 配置目录
+   * @param {string} [config.agentDir='.foliko'] - Agent 配置目录
    * @param {Object} [config.aiConfig] - AI 配置（可选，覆盖文件配置）
    * @returns {Promise<Framework>}
    */
@@ -848,7 +1020,7 @@ class Framework extends EventEmitter {
 
     // 先加载默认插件配置
     const defaultsPlugin = new DefaultPlugins({
-      agentDir: config.agentDir || '.agent',
+      agentDir: config.agentDir || '.foliko',
     });
 
     await this.loadPlugin(defaultsPlugin);
@@ -956,12 +1128,6 @@ class Framework extends EventEmitter {
       this._coordinatorManager = null;
     }
 
-    // 刷新所有 Session 存储
-    if (this._sessionStorage) {
-      await this._sessionStorage.flush();
-      await this._sessionStorage.destroy();
-    }
-
     // 销毁所有 agents
     for (const agent of this._agents) {
       // 移除 agent 的事件转发监听器
@@ -988,10 +1154,20 @@ class Framework extends EventEmitter {
       this._toolRegistryListeners = [];
     }
 
-    // 卸载所有插件
+    // 停止 Session 清理
+    this.stopSessionCleanup();
+
+    // 清空 session contexts
+    this._sessionContexts.clear();
+
+    // 卸载所有插件（逐个保护，防止一个失败阻断后续）
     const plugins = this.pluginManager.getAll();
     for (const { name } of plugins) {
-      await this.pluginManager.unload(name);
+      try {
+        await this.pluginManager.unload(name);
+      } catch (err) {
+        this.logger.warn(`Failed to unload plugin '${name}': ${err.message}`);
+      }
     }
 
     // 清空工具