foliko 1.0.0

package/plugins/audit-plugin.js

@@ -0,0 +1,187 @@
+/**
+ * Audit 审计日志插件
+ * 记录所有操作日志，支持日志查询和统计
+ */
+
+const { Plugin } = require('../src/core/plugin-base')
+const { z } = require('zod')
+
+class AuditPlugin extends Plugin {
+  constructor(config = {}) {
+    super()
+    this.name = 'audit'
+    this.version = '1.0.0'
+    this.description = '审计日志插件，记录工具调用、技能执行、错误信息等操作历史'
+    this.priority = 20
+
+    this.config = {
+      maxLogs: config.maxLogs || 1000,
+      retentionDays: config.retentionDays || 30,
+      logDir: config.logDir || '.agent/logs'
+    }
+
+    this._framework = null
+    this._logs = []
+    this._logIdCounter = 0
+  }
+
+  install(framework) {
+    this._framework = framework
+
+    // 监听框架事件
+    framework.on('tool:call', (data) => this._log('tool_call', data))
+    framework.on('tool:result', (data) => this._log('tool_result', data))
+    framework.on('tool:error', (data) => this._log('tool_error', data))
+    framework.on('agent:message', (data) => this._log('agent_message', data))
+
+    return this
+  }
+
+  start(framework) {
+    // 注册审计查询工具
+    framework.registerTool({
+      name: 'audit_query',
+      description: '查询审计日志',
+      inputSchema: z.object({
+        type: z.string().optional().describe('日志类型: tool_call, tool_result, tool_error, agent_message'),
+        limit: z.number().optional().describe('返回数量限制，默认 50'),
+        offset: z.number().optional().describe('偏移量，默认 0')
+      }),
+      execute: async (args) => {
+        let logs = this._logs
+
+        if (args.type) {
+          logs = logs.filter(log => log.type === args.type)
+        }
+
+        const offset = args.offset || 0
+        const limit = args.limit || 50
+
+        return {
+          success: true,
+          logs: logs.slice(-limit - offset, logs.length - offset),
+          total: logs.length,
+          offset,
+          limit
+        }
+      }
+    })
+
+    framework.registerTool({
+      name: 'audit_stats',
+      description: '获取审计统计信息',
+      inputSchema: z.object({}),
+      execute: async () => {
+        const stats = this.getStats()
+        return {
+          success: true,
+          stats
+        }
+      }
+    })
+
+    framework.registerTool({
+      name: 'audit_export',
+      description: '导出审计日志',
+      inputSchema: z.object({
+        format: z.string().optional().describe('导出格式: json (默认), csv'),
+        type: z.string().optional().describe('日志类型过滤'),
+        startDate: z.string().optional().describe('开始日期 ISO 格式'),
+        endDate: z.string().optional().describe('结束日期 ISO 格式')
+      }),
+      execute: async (args) => {
+        const result = this.exportLogs(args)
+        return {
+          success: true,
+          count: result.logs.length,
+          data: result.logs
+        }
+      }
+    })
+
+    return this
+  }
+
+  /**
+   * 记录日志
+   * @private
+   */
+  _log(type, data) {
+    const log = {
+      id: ++this._logIdCounter,
+      type,
+      data,
+      timestamp: new Date()
+    }
+
+    this._logs.push(log)
+
+    // 限制日志数量
+    if (this._logs.length > this.config.maxLogs) {
+      this._logs = this._logs.slice(-this.config.maxLogs)
+    }
+
+    return log
+  }
+
+  /**
+   * 获取统计信息
+   */
+  getStats() {
+    const counts = {}
+    for (const log of this._logs) {
+      counts[log.type] = (counts[log.type] || 0) + 1
+    }
+
+    return {
+      total: this._logs.length,
+      maxLogs: this.config.maxLogs,
+      counts,
+      oldest: this._logs.length > 0 ? this._logs[0].timestamp : null,
+      newest: this._logs.length > 0 ? this._logs[this._logs.length - 1].timestamp : null
+    }
+  }
+
+  /**
+   * 导出日志
+   */
+  exportLogs(options = {}) {
+    let logs = [...this._logs]
+
+    // 按类型过滤
+    if (options.type) {
+      logs = logs.filter(log => log.type === options.type)
+    }
+
+    // 按日期过滤
+    if (options.startDate) {
+      const start = new Date(options.startDate)
+      logs = logs.filter(log => log.timestamp >= start)
+    }
+
+    if (options.endDate) {
+      const end = new Date(options.endDate)
+      logs = logs.filter(log => log.timestamp <= end)
+    }
+
+    return { logs }
+  }
+
+  /**
+   * 手动记录日志
+   */
+  log(type, data) {
+    return this._log(type, data)
+  }
+
+  reload(framework) {
+    this._framework = framework
+  }
+
+  uninstall(framework) {
+    this._logs = []
+    this._framework = null
+  }
+}
+
+module.exports = { AuditPlugin }

package/plugins/default-plugins.js

@@ -0,0 +1,412 @@
+/**
+ * 默认插件配置加载器
+ * 检测 .agent/ 目录下的配置，提供给 bootstrap() 使用
+ */
+
+const fs = require('fs')
+const path = require('path')
+const { Plugin } = require('../src/core/plugin-base')
+
+/**
+ * 加载 .agent 目录下的配置
+ * 返回配置对象，可用于手动加载插件
+ */
+function loadAgentConfig(agentDir = '.agent') {
+  const config = {
+    agentDir,
+    ai: {},
+    mcpServers: {},
+    plugins: [],
+    skillsDirs: []
+  }
+
+  const resolvedDir = path.resolve(process.cwd(), agentDir)
+
+  if (!fs.existsSync(resolvedDir)) {
+    console.log(`[AgentConfig] .agent directory not found: ${resolvedDir}`)
+    return config
+  }
+
+  console.log(`[AgentConfig] Loading config from: ${resolvedDir}`)
+
+  // 加载 config 文件（key=value 格式）
+  const configFile = path.join(resolvedDir, 'config')
+  if (fs.existsSync(configFile)) {
+    try {
+      const lines = fs.readFileSync(configFile, 'utf-8').split('\n')
+      for (const line of lines) {
+        const trimmed = line.trim()
+        if (!trimmed || trimmed.startsWith('#')) continue
+
+        const colonIndex = trimmed.indexOf(':')
+        if (colonIndex === -1) continue
+
+        const key = trimmed.substring(0, colonIndex).trim()
+        const value = trimmed.substring(colonIndex + 1).trim()
+
+        if (key === 'ai_key' || key === 'api_key') {
+          config.ai.apiKey = value
+        } else if (key === 'ai_model' || key === 'model') {
+          config.ai.model = value
+        } else if (key === 'ai_provider' || key === 'provider') {
+          config.ai.provider = value
+        } else if (key === 'ai_base_url' || key === 'baseURL') {
+          config.ai.baseURL = value
+        }
+      }
+    } catch (err) {
+      console.error(`[AgentConfig] Failed to load config:`, err.message)
+    }
+  }
+
+  // 加载 ai.json
+  const aiFile = path.join(resolvedDir, 'ai.json')
+  if (fs.existsSync(aiFile)) {
+    try {
+      const content = fs.readFileSync(aiFile, 'utf-8')
+      const aiConfig = JSON.parse(content)
+      config.ai = { ...config.ai, ...aiConfig }
+    } catch (err) {
+      console.error(`[AgentConfig] Failed to load ai.json:`, err.message)
+    }
+  }
+
+  // 加载 plugins.json
+  const pluginsFile = path.join(resolvedDir, 'plugins.json')
+  if (fs.existsSync(pluginsFile)) {
+    try {
+      const content = fs.readFileSync(pluginsFile, 'utf-8')
+      config.plugins = JSON.parse(content)
+    } catch (err) {
+      console.error(`[AgentConfig] Failed to load plugins.json:`, err.message)
+    }
+  }
+
+  // 加载 mcp_config.json
+  const mcpFile = path.join(resolvedDir, 'mcp_config.json')
+  if (fs.existsSync(mcpFile)) {
+    try {
+      const content = fs.readFileSync(mcpFile, 'utf-8')
+      const mcpConfig = JSON.parse(content)
+      config.mcpServers = mcpConfig.mcpServers || {}
+    } catch (err) {
+      console.error(`[AgentConfig] Failed to load mcp_config.json:`, err.message)
+    }
+  }
+
+  // 添加 .agent/skills 目录（不存在则创建）
+  const skillsDir = path.join(resolvedDir, 'skills')
+  if (fs.existsSync(resolvedDir) && !fs.existsSync(skillsDir)) {
+    fs.mkdirSync(skillsDir, { recursive: true })
+    console.log(`[AgentConfig] Created skills directory: ${skillsDir}`)
+  }
+  if (fs.existsSync(skillsDir)) {
+    config.skillsDirs.push(skillsDir)
+  }
+
+  // 添加 agentDir 父目录的 skills/ 目录
+  const parentDir = path.dirname(__dirname)
+  const rootSkillsDir = path.join(parentDir, 'skills')
+  if (fs.existsSync(rootSkillsDir)) {
+    config.skillsDirs.push(rootSkillsDir)
+  }
+
+  // const defaultSkillsDir = path.join(process.cwd(), 'defaultSkills')
+  // if (fs.existsSync(defaultSkillsDir)) {
+  //   config.skillsDirs.push(defaultSkillsDir)
+  // }
+
+  return config
+}
+
+/**
+ * DefaultPlugins - 默认插件配置加载器
+ */
+class DefaultPlugins extends Plugin {
+  constructor(config = {}) {
+    super()
+    this.name = 'defaults'
+    this.version = '1.0.0'
+    this.description = '默认插件配置加载器'
+    this.priority = 0  // 最先加载
+
+    this._framework = null
+    this._agentDir = config.agentDir || '.agent'
+    this._config = null
+  }
+
+  install(framework) {
+    this._framework = framework
+    return this
+  }
+
+  start(framework) {
+    // 加载配置
+    this._config = loadAgentConfig(this._agentDir)
+    return this
+  }
+
+  /**
+   * 获取加载的配置
+   */
+  getConfig() {
+    return this._config || loadAgentConfig(this._agentDir)
+  }
+
+  reload(framework) {
+    this._framework = framework
+    this._config = loadAgentConfig(this._agentDir)
+  }
+
+  uninstall(framework) {
+    this._framework = null
+    this._config = null
+  }
+}
+
+/**
+ * Bootstrap 辅助函数
+ * 根据配置加载所有默认插件
+ */
+async function bootstrapDefaults(framework, config = {}) {
+  // 如果已经有配置，使用现有配置；否则加载
+  const agentConfig = config._config
+  if (!agentConfig) {
+    console.error('[Bootstrap] No config provided, skipping plugin loading')
+    return
+  }
+
+  // 设置 bootstrap 模式，避免重复启动
+  framework.pluginManager.setBootstrapping(true)
+
+  // 合并 AI 配置
+  const aiConfig = { ...agentConfig.ai, ...config.aiConfig }
+
+  // 0. Install 工具插件（最先加载，让其他插件能用到它的 node_modules）
+  if (!framework.pluginManager.has('install')) {
+    const { InstallPlugin } = require('./install-plugin')
+    await framework.loadPlugin(new InstallPlugin({ agentDir: agentConfig.agentDir }))
+    console.log('[Bootstrap] Install Plugin loaded')
+  }
+
+  // 合并 skillsDirs 配置（bootstrap 传入的优先）
+  const skillsDirs = [
+    ...(config.skillsDirs || []),
+    ...(agentConfig.skillsDirs || [])
+  ]
+
+  console.log('[Bootstrap] Loading default plugins...')
+
+  // 检查是否已经加载
+  if (framework.pluginManager.has('ai')) {
+    console.log('[Bootstrap] AI Plugin already loaded, skipping')
+  } else if (aiConfig.provider || aiConfig.model || aiConfig.apiKey) {
+    const { AIPlugin } = require('./ai-plugin')
+    const aiPlugin = new AIPlugin({
+      provider: aiConfig.provider || 'deepseek',
+      model: aiConfig.model || 'deepseek-chat',
+      apiKey: aiConfig.apiKey || process.env.DEEPSEEK_API_KEY,
+      baseURL: aiConfig.baseURL
+    })
+    await framework.loadPlugin(aiPlugin)
+    console.log('[Bootstrap] AI Plugin loaded')
+  }
+
+  // 2. Storage 存储插件
+  if (!framework.pluginManager.has('storage')) {
+    const { StoragePlugin } = require('./storage-plugin')
+    await framework.loadPlugin(new StoragePlugin())
+    console.log('[Bootstrap] Storage Plugin loaded')
+  } else {
+    console.log('[Bootstrap] Storage Plugin already loaded, skipping')
+  }
+
+  // 3. 内置工具插件
+  if (!framework.pluginManager.has('tools')) {
+    const { ToolsPlugin } = require('./tools-plugin')
+    await framework.loadPlugin(new ToolsPlugin())
+    console.log('[Bootstrap] Tools Plugin loaded')
+  } else {
+    console.log('[Bootstrap] Tools Plugin already loaded, skipping')
+  }
+
+  // 4. 工作流插件
+  if (!framework.pluginManager.has('workflow')) {
+    const { WorkflowPlugin } = require('../src/capabilities/workflow-engine')
+    await framework.loadPlugin(new WorkflowPlugin())
+    console.log('[Bootstrap] Workflow Plugin loaded')
+  } else {
+    console.log('[Bootstrap] Workflow Plugin already loaded, skipping')
+  }
+
+  // 5. Skill 管理器插件
+  if (!framework.pluginManager.has('skill-manager')) {
+    if (skillsDirs.length > 0) {
+      const { SkillManagerPlugin } = require('../src/capabilities/skill-manager')
+      // 传递所有 skills 目录
+      await framework.loadPlugin(new SkillManagerPlugin({ skillsDirs }))
+      console.log('[Bootstrap] Skill Manager loaded')
+    }
+  } else {
+    console.log('[Bootstrap] Skill Manager already loaded, skipping')
+  }
+
+  // 6. MCP 执行器插件（始终加载，确保 mcp_reload 工具可用）
+  if (!framework.pluginManager.has('mcp-executor')) {
+    const mcpServers = Object.entries(agentConfig.mcpServers || {})
+    const servers = mcpServers.map(([name, cfg]) => ({
+      ...cfg,
+      name,
+      command: cfg.command,
+      args: cfg.args || [],
+      env: cfg.env || {},
+      url: cfg.url,
+      headers: cfg.headers
+    }))
+    const { MCPExecutorPlugin } = require('../src/executors/mcp-executor')
+    await framework.loadPlugin(new MCPExecutorPlugin({ servers }))
+    console.log(`[Bootstrap] MCP Executor loaded${servers.length > 0 ? ` (${servers.length} servers)` : ' (no servers)'}`)
+  } else {
+    console.log('[Bootstrap] MCP Executor already loaded, skipping')
+  }
+
+  // 7. Shell 执行器插件
+  if (!framework.pluginManager.has('shell-executor')) {
+    const { ShellExecutorPlugin } = require('./shell-executor-plugin')
+    await framework.loadPlugin(new ShellExecutorPlugin())
+    console.log('[Bootstrap] Shell Executor loaded')
+  } else {
+    console.log('[Bootstrap] Shell Executor already loaded, skipping')
+  }
+
+  // 8. Python 执行器插件
+  if (!framework.pluginManager.has('python-executor')) {
+    const { PythonExecutorPlugin } = require('./python-executor-plugin')
+    await framework.loadPlugin(new PythonExecutorPlugin())
+    console.log('[Bootstrap] Python Executor loaded')
+  } else {
+    console.log('[Bootstrap] Python Executor already loaded, skipping')
+  }
+
+  // 9. Session 会话管理插件
+  if (!framework.pluginManager.has('session')) {
+    const { SessionPlugin } = require('./session-plugin')
+    await framework.loadPlugin(new SessionPlugin())
+    console.log('[Bootstrap] Session Plugin loaded')
+  } else {
+    console.log('[Bootstrap] Session Plugin already loaded, skipping')
+  }
+
+  // 10. Audit 审计日志插件
+  if (!framework.pluginManager.has('audit')) {
+    const { AuditPlugin } = require('./audit-plugin')
+    await framework.loadPlugin(new AuditPlugin())
+    console.log('[Bootstrap] Audit Plugin loaded')
+  } else {
+    console.log('[Bootstrap] Audit Plugin already loaded, skipping')
+  }
+
+  // 10. Rules 规则引擎插件
+  if (!framework.pluginManager.has('rules')) {
+    const { RulesPlugin } = require('./rules-plugin')
+    await framework.loadPlugin(new RulesPlugin())
+    console.log('[Bootstrap] Rules Plugin loaded')
+  } else {
+    console.log('[Bootstrap] Rules Plugin already loaded, skipping')
+  }
+
+  // 11. Scheduler 定时任务插件
+  if (!framework.pluginManager.has('scheduler')) {
+    const { SchedulerPlugin } = require('./scheduler-plugin')
+    await framework.loadPlugin(new SchedulerPlugin())
+    console.log('[Bootstrap] Scheduler Plugin loaded')
+  } else {
+    console.log('[Bootstrap] Scheduler Plugin already loaded, skipping')
+  }
+
+  // 11. FileSystem 文件系统插件
+  if (!framework.pluginManager.has('file-system')) {
+    const { FileSystemPlugin } = require('./file-system-plugin')
+    await framework.loadPlugin(new FileSystemPlugin())
+    console.log('[Bootstrap] FileSystem Plugin loaded')
+  } else {
+    console.log('[Bootstrap] FileSystem Plugin already loaded, skipping')
+  }
+
+  // 12. Think 主动思考插件
+  if (!framework.pluginManager.has('think')) {
+    const { ThinkPlugin } = require('./think-plugin')
+    await framework.loadPlugin(new ThinkPlugin())
+    console.log('[Bootstrap] Think Plugin loaded')
+  } else {
+    console.log('[Bootstrap] Think Plugin already loaded, skipping')
+  }
+
+  // 13. 加载自定义插件
+  await loadCustomPlugins(framework, agentConfig)
+
+  console.log('[Bootstrap] All plugins loaded')
+
+  // 统一启动所有插件（避免重复启动）
+  await framework.pluginManager.startAll()
+
+  // 清除 bootstrap 模式
+  framework.pluginManager.setBootstrapping(false)
+
+  console.log('[Bootstrap] Plugins:', framework.pluginManager.getAll().map(p => p.name))
+  console.log('[Bootstrap] Tools:', framework.getTools().map(t => t.name))
+}
+
+async function loadCustomPlugins(framework, agentConfig) {
+  // 加载 .agent/plugins 目录下的自定义插件
+  const pluginsDir = path.resolve(process.cwd(), '.agent', 'plugins')
+
+  if (!fs.existsSync(pluginsDir)) {
+    return
+  }
+
+  const files = fs.readdirSync(pluginsDir).filter(f => f.endsWith('.js'))
+
+  for (const file of files) {
+    try {
+      const pluginPath = path.join(pluginsDir, file)
+
+      // 清除缓存
+      delete require.cache[require.resolve(pluginPath)]
+      const pluginModule = require(pluginPath)
+
+      let plugin
+      if (typeof pluginModule === 'function') {
+        // 免引入写法：直接传递函数，让 loadPlugin 内部处理
+        plugin = pluginModule
+      } else if (pluginModule.default) {
+        plugin = pluginModule.default
+      } else {
+        plugin = pluginModule
+      }
+
+      // 获取插件名称
+      const tempPlugin = typeof plugin === 'function' && plugin.prototype?.name
+        ? new plugin() : plugin
+      const pluginName = tempPlugin.name || file.replace('.js', '')
+
+      // 如果插件已加载且已启动，跳过
+      if (framework.pluginManager.has(pluginName) &&
+          framework.pluginManager.get(pluginName)?._started) {
+        continue
+      }
+
+      console.log(`[Bootstrap] Loading custom plugin: ${file}`)
+      await framework.loadPlugin(plugin)
+    } catch (err) {
+      console.error(`[Bootstrap] Failed to load plugin ${file}:`, err.message)
+    }
+  }
+}
+
+module.exports = {
+  DefaultPlugins,
+  loadAgentConfig,
+  bootstrapDefaults,
+  loadCustomPlugins
+}