foliko 2.0.11 → 2.0.13

package/plugins/core/skill-manager/index.js

@@ -263,6 +263,59 @@ function buildSchemaFromType(type, desc, defaultValue) {
   return schema;
 }
 
+/**
+ * 从 options 构建 JSON Schema（供 AI SDK 直接使用，不走 Zod）
+ * 用于替换旧的 CLI 字符串模式，让 LLM 能用结构化参数调用
+ *
+ * type 可选值：'string' | 'number' | 'boolean' | 'array' | 'object'
+ * （对应 JS 原生类型 String, Boolean, Number, Array, Object）
+ */
+function buildInputSchemaFromOptions(options) {
+  if (!Array.isArray(options) || options.length === 0) {
+    return { type: 'object', properties: {}, required: [] };
+  }
+
+  const properties = {};
+  const required = [];
+
+  for (const opt of options) {
+    const name = extractOptionName(opt.flags);
+    if (!name) continue;
+
+    const hasValue = optionHasValue(opt.flags);
+    const isRequired = opt.required === true && hasValue;
+    const desc = opt.description || '';
+    const defaultVal = opt.defaultValue;
+
+    let jsonType = 'string';
+    if (opt.schema && typeof opt.schema.parse === 'function') {
+      // 从 Zod 实例猜测类型
+      const s = opt.schema;
+      if (s._def.typeName === 'ZodNumber' || s._def.typeName === 'ZodInt') jsonType = 'number';
+      else if (s._def.typeName === 'ZodBoolean') jsonType = 'boolean';
+      else if (s._def.typeName === 'ZodArray') jsonType = 'array';
+      else if (s._def.typeName === 'ZodObject' || s._def.typeName === 'ZodRecord') jsonType = 'object';
+    } else if (opt.type) {
+      const L = opt.type.toLowerCase();
+      if (L === 'number' || L === 'int' || L === 'integer') jsonType = 'number';
+      else if (L === 'boolean' || L === 'bool') jsonType = 'boolean';
+      else if (L === 'array') jsonType = 'array';
+      else if (L === 'object') jsonType = 'object';
+    } else {
+      jsonType = hasValue ? 'string' : 'boolean';
+    }
+
+    const prop = { type: jsonType, description: desc };
+    if (defaultVal !== undefined && defaultVal !== null) {
+      prop.default = defaultVal;
+    }
+    properties[name] = prop;
+    if (isRequired) required.push(name);
+  }
+
+  return { type: 'object', properties, required };
+}
+
 /**
  * 生成等效的 JSON schema（用于 _rawSchema，让 ext_skill 渲染更清晰）
  */
@@ -329,6 +382,50 @@ function buildExampleFromOptions(options) {
   return parts.join(' ');
 }
 
+/**
+ * 解析命令行参数字符串为结构化对象（供 messaging 插件 backward compatible 使用）
+ * 支持：--name value, -n value, --flag, -f
+ * @param {string} str - 参数字符串
+ * @param {Array} options - 选项定义数组
+ * @returns {Object} 解析后的参数对象
+ */
+function parseCommandArgs(str, options) {
+  if (typeof str !== 'string' || !str.trim()) return {};
+  const result = {};
+  const tokens = parseShellArgs(str);
+  let i = 0;
+  while (i < tokens.length) {
+    const token = tokens[i];
+    const optMatch = token.match(/^(-[a-zA-Z]|-{1,2}[a-zA-Z][a-zA-Z0-9-]*)$/);
+    if (optMatch) {
+      const optKey = optMatch[1];
+      const opt = options.find(o => {
+        const flags = o.flags || '';
+        return flags.includes(optKey) || flags.includes(optKey.replace(/^--?/, '--'));
+      });
+      if (opt) {
+        const name = extractOptionName(opt.flags);
+        const hasValue = optionHasValue(opt.flags);
+        if (hasValue && i + 1 < tokens.length) {
+          const next = tokens[i + 1];
+          if (!next.match(/^-/)) {
+            result[name] = next;
+            i += 2;
+            continue;
+          }
+        } else if (!hasValue) {
+          result[name] = true;
+          i += 1;
+          continue;
+        }
+      }
+    }
+    i += 1;
+  }
+  if (i < tokens.length) result._ = tokens.slice(i);
+  return result;
+}
+
 /**
  * Skill 类
  */
@@ -356,6 +453,17 @@ class Skill {
     return [];
   }
 
+  /**
+   * 简单命令行字符串解析器（backward compatible，供 messaging 插件使用）
+   * 支持：--name value, -n value, --flag, -f, positional args
+   * @param {string} str - 参数字符串
+   * @param {Array} options - 选项定义数组
+   * @returns {Object} 解析后的参数对象
+   */
+  _parseCommandString(str, options) {
+    return parseCommandArgs(str, options);
+  }
+
   /**
    * 注册 skill 的命令到 extension 系统
    * 异步版本：用动态 import() 替代 require()，支持 ESM 包
@@ -415,87 +523,13 @@ class Skill {
         // 命令名格式: {skill_name}:{cmd_name}
         const fullName = `${skillName}:${cmd.name}`;
 
-        // 获取参数解析器（优先使用 argumentParser，否则用 options 构建）
-        let argumentParser = cmd.argumentParser;
-        if (!argumentParser && Array.isArray(cmd.options) && cmd.options.length > 0) {
-          argumentParser = (rawArgs) => {
-            try {
-              const program = new Command();
-              program.exitOverride(); // 不自动退出
-              program.configureOutput({ writeErr: () => {} }); // 隐藏错误输出
-              for (const opt of cmd.options) {
-                program.option(opt.flags, opt.description, opt.defaultValue);
-              }
-              program.arguments('[args...]');
-              program.parse(['node', 'cmd', ...parseShellArgs(rawArgs)]);
-              return { ...program.opts(), args: program.args };
-            } catch (err) {
-              log.warn(`[SkillManager] Command parse error: ${err.message}`);
-              // Commander 解析失败时，用简单方式从 rawArgs 提取选项值
-              const result = {};
-              const tokens = parseShellArgs(rawArgs);
-              let i = 0;
-              while (i < tokens.length) {
-                const token = tokens[i];
-                // 匹配选项（-x 或 --xxx）
-                const optMatch = token.match(/^(-[a-zA-Z]|-{1,2}[a-zA-Z][a-zA-Z0-9-]*)$/);
-                if (optMatch) {
-                  const optKey = optMatch[1];
-                  // 找对应的 option
-                  const opt = cmd.options.find(o => {
-                    const flags = o.flags || '';
-                    return flags.includes(optKey) || flags.includes(optKey.replace(/^--?/, '--'));
-                  });
-                  if (opt) {
-                    const name = extractOptionName(opt.flags);
-                    // 判断是否有值（下一个 token 不是选项）
-                    const hasValue = /<[^>]+>/.test(opt.flags);
-                    if (hasValue && i + 1 < tokens.length) {
-                      const next = tokens[i + 1];
-                      if (!next.match(/^-/)) {
-                        result[name] = next;
-                        i += 2;
-                        continue;
-                      } else {
-                        // 下个 token 是另一个选项，但当前选项需要值，用默认值
-                        result[name] = opt.defaultValue ?? undefined;
-                        i += 1;
-                        continue;
-                      }
-                    } else if (!hasValue) {
-                      result[name] = true;
-                      i += 1;
-                      continue;
-                    } else if (hasValue && i + 1 >= tokens.length) {
-                      // 当前选项需要值但没有更多 token，用默认值
-                      result[name] = opt.defaultValue ?? undefined;
-                      i += 1;
-                      continue;
-                    }
-                  }
-                }
-                i += 1;
-              }
-              // 剩余的 token 当 args
-              result.args = tokens.slice(i);
-              return result;
-            }
-          };
-        }
-
-        // 直接使用 command 字符串的 schema，describe 说明命令行格式
-        const exampleArgs = cmd.options && cmd.options.length > 0
-          ? buildExampleFromOptions(cmd.options)
-          : '';
+        // 新模式：结构化参数 schema，让 LLM 直接用 JSON 对象调用
         const toolDef = {
           name: cmd.name,
           description: cmd.description || 'Skill command',
-          inputSchema: z.object({
-            command: z.string().optional().describe(exampleArgs || '命令行字符串，如：-n value1 -v'),
-          }),
+          // 直接用 JSON Schema，不走 Zod（让 LLM 看到清晰的结构化参数）
+          inputSchema: buildInputSchemaFromOptions(cmd.options),
           _options: cmd.options || null,
-          // 把原始 options 数组也保存下来，便于 ext_skill 显示
-          _rawSchema: buildRawJsonSchema(cmd.options),
           execute: async (toolArgs, framework) => {
             const ctx = {
               sessionId: framework?._currentSessionId || 'unknown',
@@ -503,13 +537,13 @@ class Skill {
               framework: framework,
               cwd: framework?.getCwd?.() ?? process.cwd(),
             };
-            // 直接使用 command 字符串
-            const commandString = toolArgs?.command || '';
-            let parsedArgs = commandString;
-            if (typeof argumentParser === 'function') {
-              parsedArgs = argumentParser(commandString);
+            // backward compatible：如果 args.command 是字符串（messaging 插件传来），
+            // 用 options 做简单解析；否则直接传递结构化参数（AI SDK / ext_call 传来）
+            let args = toolArgs || {};
+            if (typeof args.command === 'string' && Array.isArray(cmd.options) && cmd.options.length > 0) {
+              args = parseCommandArgs(args.command, cmd.options);
             }
-            return await handler(parsedArgs, ctx);
+            return await handler(args, ctx);
           },
         };
 
@@ -1283,72 +1317,12 @@ class SkillManagerPlugin extends Plugin {
 
         const fullName = `${name}:${cmd.name}`;
 
-        // 获取参数解析器
-        let argumentParser = cmd.argumentParser;
-        if (!argumentParser && Array.isArray(cmd.options) && cmd.options.length > 0) {
-          argumentParser = (rawArgs) => {
-            try {
-              const program = new Command();
-              program.exitOverride();
-              program.configureOutput({ writeErr: () => {} });
-              for (const opt of cmd.options) {
-                program.option(opt.flags, opt.description, opt.defaultValue);
-              }
-              program.arguments('[args...]');
-              program.parse(['node', 'cmd', ...parseShellArgs(rawArgs)]);
-              return { ...program.opts(), args: program.args };
-            } catch (err) {
-              log.warn(`[SkillManager] Command parse error: ${err.message}`);
-              // Commander 解析失败时，用简单方式从 rawArgs 提取选项值
-              const result = {};
-              const tokens = parseShellArgs(rawArgs);
-              let i = 0;
-              while (i < tokens.length) {
-                const token = tokens[i];
-                const optMatch = token.match(/^(-[a-zA-Z]|-{1,2}[a-zA-Z][a-zA-Z0-9-]*)$/);
-                if (optMatch) {
-                  const optKey = optMatch[1];
-                  const opt = cmd.options.find(o => {
-                    const flags = o.flags || '';
-                    return flags.includes(optKey) || flags.includes(optKey.replace(/^--?/, '--'));
-                  });
-                  if (opt) {
-                    const name = extractOptionName(opt.flags);
-                    const hasValue = /<[^>]+>/.test(opt.flags);
-                    if (hasValue && i + 1 < tokens.length) {
-                      const next = tokens[i + 1];
-                      if (!next.match(/^-/)) {
-                        result[name] = next;
-                        i += 2;
-                        continue;
-                      }
-                    } else if (!hasValue) {
-                      result[name] = true;
-                      i += 1;
-                      continue;
-                    }
-                  }
-                }
-                i += 1;
-              }
-              result.args = tokens.slice(i);
-              return result;
-            }
-          };
-        }
-
-        // 转为 Zod schema（统一风格）
-        const exampleArgs2 = cmd.options && cmd.options.length > 0
-          ? buildExampleFromOptions(cmd.options)
-          : '';
+        // 新模式：结构化参数 schema
         const toolDef = {
-          name: cmd.name,  // 只用命令名
+          name: cmd.name,
           description: cmd.description || 'Skill command',
-          inputSchema: z.object({
-            command: z.string().optional().describe(exampleArgs2 || '命令行字符串'),
-          }),
+          inputSchema: buildInputSchemaFromOptions(cmd.options),
           _options: cmd.options || null,
-          _rawSchema: buildRawJsonSchema(cmd.options),
           execute: async (toolArgs, framework) => {
             const ctx = {
               sessionId: framework?._currentSessionId || 'unknown',
@@ -1356,12 +1330,12 @@ class SkillManagerPlugin extends Plugin {
               framework: framework,
               cwd: framework?.getCwd?.() ?? process.cwd(),
             };
-            const commandString = toolArgs?.command || '';
-            let parsedArgs = commandString;
-            if (typeof argumentParser === 'function') {
-              parsedArgs = argumentParser(commandString);
+            // backward compatible：如果 args.command 是字符串，解析为结构化对象
+            let args = toolArgs || {};
+            if (typeof args.command === 'string' && Array.isArray(cmd.options) && cmd.options.length > 0) {
+              args = parseCommandArgs(args.command, cmd.options);
             }
-            return await handler(parsedArgs, ctx);
+            return await handler(args, ctx);
           },
         };