@milian/linker-cli 0.0.0-stage → 0.17.3

package/src/commands/mcp.js

@@ -0,0 +1,485 @@
+'use strict';
+
+const { loadCredentials } = require('../config');
+const { authedRequest } = require('../api');
+const { resolvePlatformRecord } = require('./platforms');
+const { renderTable, truncate } = require('../ui');
+
+/**
+ * MCP 工具来源类型（业务方定义）
+ * 1 = 连接器资源（api）  2 = 连接流资源（flow）
+ */
+const SOURCE_TYPE_NAMES = { 1: '连接器资源', 2: '连接流资源' };
+
+/** 来源查询接口的 pageNum/pageSize 由平台约定写死 */
+const SOURCE_PAGE_NUM = 1;
+const SOURCE_PAGE_SIZE = 1000;
+
+// 表格显示上限：来源可能两三百个，全量倒给 AI 既慢又费上下文
+const DISPLAY_LIMIT = 20;
+const NAME_MAX_WIDTH = 40;
+const GROUP_MAX_WIDTH = 16;
+
+const SOURCES_USAGE = `用法: wb mcp sources [选项]
+
+查询可创建 MCP 工具的来源列表（连接器资源 / 连接流资源）。
+
+可选:
+  --type, -t <1|2>        1=连接器资源（默认） 2=连接流资源；也可写名称：api / flow / 连接器 / 连接流
+  --name, -n <关键词>     工具名，模糊查询
+  --platform, -p <id|名称> 限定某个连接器应用（等价于传该连接器的 groupId）
+                          仅 --type 1 有意义；不给则查全部连接器
+  --all                   打印全部结果（默认只显示前 ${DISPLAY_LIMIT} 条）
+  --json                  输出接口原始 JSON
+
+示例:
+  wb mcp sources                                  # 全部连接器资源来源
+  wb mcp sources --platform 泛微E10 --name 应收
+  wb mcp sources --type 2                         # 连接流资源来源`;
+
+const CREATE_USAGE = `用法: wb mcp create --source <sourceId|工具名> [选项]
+
+把来源封装成 MCP 工具（支持批量）。已创建过（exists=true）的来源会被拦下。
+
+必填:
+  --source, -s <id|名称>  来源。可给 sourceId，也可给工具名（模糊唯一匹配）。
+                          可重复，或逗号分隔多个
+
+可选:
+  --type, -t <1|2>        1=连接器资源（默认） 2=连接流资源
+  --platform, -p <id|名称> 限定某个连接器应用（缩小来源检索范围）
+  --skip-exists           批量中遇到"已创建过"的自动跳过，只创建其余（默认整体中止）
+  --dry-run               只打印将提交的 JSON，不发请求
+  --json                  输出接口原始 JSON
+
+示例:
+  wb mcp create --source 2034441472166580229 --dry-run
+  wb mcp create --platform E10 --source "E10-取得应收单查询列表资料"
+  wb mcp create --type 2 --source 数组套数组出来
+  wb mcp create --source a,b,c --skip-exists`;
+
+function parseCommon(argv) {
+  const opts = { type: '', name: '', platform: '', all: false, json: false, help: false };
+  for (let i = 0; i < argv.length; i += 1) {
+    const a = argv[i];
+    const next = () => {
+      const v = argv[i + 1];
+      i += 1;
+      return v === undefined ? '' : v;
+    };
+    if (a === '--type' || a === '-t') opts.type = next();
+    else if (a === '--name' || a === '-n') opts.name = next();
+    else if (a === '--platform' || a === '-p') opts.platform = next();
+    // --source 的取值不能被当成位置参数（create 时单独解析，这里只是吃掉）
+    else if (a === '--source' || a === '-s') next();
+    else if (a === '--all') opts.all = true;
+    else if (a === '--json') opts.json = true;
+    else if (a === '--help' || a === '-h') opts.help = true;
+    else if (!a.startsWith('-')) {
+      if (!opts.name) opts.name = a;
+    }
+  }
+  return opts;
+}
+
+/** 解析来源类型：支持数字 1/2，也支持名称 */
+function parseSourceType(raw) {
+  const v = String(raw === undefined || raw === null ? '' : raw).trim();
+  if (!v) return 1;
+  if (/^[12]$/.test(v)) return Number(v);
+
+  const s = v.toUpperCase();
+  if (['API', 'CONNECTOR', '连接器', '连接器资源'].includes(s)) return 1;
+  if (['FLOW', 'PROCESS', '连接流', '连接流资源'].includes(s)) return 2;
+
+  throw new Error(
+    `--type "${v}" 不合法：可选 1=连接器资源 / 2=连接流资源（也可写 api / flow / 连接器 / 连接流）`
+  );
+}
+
+function sourceTypeLabel(n) {
+  const v = Number(n);
+  return SOURCE_TYPE_NAMES[v] ? `${v} ${SOURCE_TYPE_NAMES[v]}` : String(n);
+}
+
+/**
+ * 查询 MCP 工具来源
+ * POST /index/sys/api/v1/mcp/getMcpSourceList
+ * 入参 pageNum/pageSize 平台约定写死 1 / 1000
+ */
+async function fetchMcpSources(cfg, cred, opts) {
+  const o = Object.assign({ name: '', groupId: null, sourceType: 1 }, opts || {});
+  const body = {
+    name: String(o.name || ''),
+    groupId: o.groupId === undefined ? null : o.groupId,
+    sourceType: Number(o.sourceType) || 1,
+    pageNum: SOURCE_PAGE_NUM,
+    pageSize: SOURCE_PAGE_SIZE
+  };
+
+  const res = await authedRequest(cfg, cred, 'POST', '/sys/api/v1/mcp/getMcpSourceList', body);
+  const json = res.json;
+
+  if (!json || typeof json !== 'object') {
+    throw new Error(`查询 MCP 来源失败（HTTP ${res.status}）：${String(res.text).slice(0, 200)}`);
+  }
+  if (json.code !== 200) {
+    throw new Error(`查询 MCP 来源失败：code=${json.code} msg=${json.msg || '未知错误'}`);
+  }
+
+  const data = json.data || {};
+  const records = Array.isArray(data.records) ? data.records : [];
+  return {
+    total: typeof data.total === 'number' ? data.total : records.length,
+    records,
+    raw: json
+  };
+}
+
+/**
+ * 把用户给的 --source 解析成来源记录（纯函数）
+ * - 纯数字 → 按 sourceId 精确匹配
+ * - 其他   → 按 name 模糊匹配，命中多条不猜
+ *
+ * @returns {{picked:object[], missing:object[], ambiguous:object[]}}
+ */
+function pickSources(records, inputs) {
+  const picked = [];
+  const missing = [];
+  const ambiguous = [];
+
+  for (const raw of inputs || []) {
+    const key = String(raw === null || raw === undefined ? '' : raw).trim();
+    if (!key) continue;
+
+    let hit = null;
+    if (/^\d+$/.test(key)) {
+      hit = (records || []).find((r) => String(r.sourceId) === key) || null;
+      if (!hit) {
+        missing.push({ input: key, reason: 'sourceId 不在来源列表中（可能类型不对，或超出 1000 条）' });
+        continue;
+      }
+    } else {
+      const matched = (records || []).filter((r) => String(r.name || '').includes(key));
+      if (matched.length === 0) {
+        missing.push({ input: key, reason: '工具名无匹配' });
+        continue;
+      }
+      if (matched.length > 1) {
+        ambiguous.push({ input: key, matched });
+        continue;
+      }
+      hit = matched[0];
+    }
+
+    if (!picked.some((p) => String(p.sourceId) === String(hit.sourceId))) picked.push(hit);
+  }
+
+  return { picked, missing, ambiguous };
+}
+
+/** 统计来源可用情况（纯函数） */
+function summarizeSources(records) {
+  const list = records || [];
+  const created = list.filter((r) => r.exists === true);
+  const creatable = list.filter((r) => r.exists !== true);
+  return { total: list.length, created, creatable };
+}
+
+/**
+ * 组装批量创建请求体（纯函数）
+ * 四个字段一律**原样透传**来源接口的返回值：
+ *   - sourceType=1 时 groupId 是字符串，sourceType=2 时是字符串 "0"、groupName 是"连接流"
+ *   自己拼 groupId 会错，所以不转换。
+ */
+function buildAddBody(records) {
+  return {
+    list: (records || []).map((r) => ({
+      sourceType: r.sourceType === undefined ? null : r.sourceType,
+      sourceId: r.sourceId === undefined ? null : r.sourceId,
+      groupId: r.groupId === undefined ? null : r.groupId,
+      groupName: r.groupName === undefined ? null : r.groupName
+    }))
+  };
+}
+
+/** 把 --platform 解析成该连接器的 groupId（连接器资源来源检索用） */
+async function resolvePlatformGroupId(cfg, cred, platformInput, out) {
+  const target = await resolvePlatformRecord(cfg, cred, platformInput, out);
+  const groupId = target.raw && target.raw.groupId !== undefined ? target.raw.groupId : null;
+  if (groupId === null || groupId === undefined || groupId === '') {
+    throw new Error(
+      `连接器 ${target.platformName || target.platformId} 没有返回 groupId，无法按连接器过滤来源。\n` +
+        `        可去掉 --platform 查全部连接器的来源。`
+    );
+  }
+  return {
+    platformId: target.platformId,
+    platformName: target.platformName,
+    groupId,
+    groupName: target.raw && target.raw.groupName ? String(target.raw.groupName) : ''
+  };
+}
+
+/** mcp sources */
+async function commandMcpSources(cfg, argv) {
+  const opts = parseCommon(argv);
+  if (opts.help) {
+    process.stdout.write(SOURCES_USAGE + '\n');
+    process.exitCode = 0;
+    return;
+  }
+
+  const cred = loadCredentials();
+  if (!cred || !cred.tokenValue) {
+    throw new Error('未登录或授权已失效，请先执行 auth login');
+  }
+
+  let sourceType = parseSourceType(opts.type);
+  let groupId = null;
+  let platLabel = '';
+
+  if (opts.platform) {
+    if (sourceType !== 1) {
+      process.stderr.write('[warn] --platform 只对连接器资源（--type 1）有效，已忽略\n');
+    } else {
+      const p = await resolvePlatformGroupId(cfg, cred, opts.platform, process.stdout);
+      groupId = p.groupId;
+      platLabel = p.platformName || p.platformId;
+    }
+  }
+
+  const { total, records, raw } = await fetchMcpSources(cfg, cred, {
+    name: opts.name,
+    groupId,
+    sourceType
+  });
+
+  if (opts.json) {
+    process.stdout.write(JSON.stringify(raw, null, 2) + '\n');
+    process.exitCode = 0;
+    return;
+  }
+
+  process.stdout.write(`来源类型: ${sourceTypeLabel(sourceType)}\n`);
+  process.stdout.write(`连接器  : ${platLabel ? `${platLabel}（groupId=${groupId}）` : '（全部）'}\n`);
+  process.stdout.write(`查询条件: name=${opts.name ? `"${opts.name}"` : '（全部）'}  pageNum=${SOURCE_PAGE_NUM} pageSize=${SOURCE_PAGE_SIZE}\n`);
+
+  const sum = summarizeSources(records);
+  process.stdout.write(`共 ${total} 个来源，本次返回 ${records.length} 个；可创建 ${sum.creatable.length} 个、已创建 ${sum.created.length} 个\n\n`);
+
+  if (!records.length) {
+    process.stdout.write('（无匹配结果）\n');
+    if (opts.name) process.stdout.write('换个关键词试试，或去掉 --name\n');
+    else process.stdout.write(`当前没有可用的${SOURCE_TYPE_NAMES[sourceType] || ''}来源\n`);
+    process.exitCode = 0;
+    return;
+  }
+
+  const shown = opts.all ? records : records.slice(0, DISPLAY_LIMIT);
+  const rows = shown.map((r, i) => [
+    String(i + 1),
+    truncate(r.name || '', NAME_MAX_WIDTH),
+    String(r.sourceId || ''),
+    truncate(r.groupName || '', GROUP_MAX_WIDTH),
+    r.exists === true ? '已创建' : '未创建'
+  ]);
+  process.stdout.write(renderTable(['#', 'name', 'sourceId', 'groupName', 'created'], rows) + '\n');
+
+  if (sum.created.length) {
+    process.stdout.write(`\n⚠ 其中 ${sum.created.length} 个已创建过 MCP 工具，不能再创建（created=已创建）\n`);
+  }
+  process.stdout.write(`\n创建：wb mcp create --type ${sourceType} --source <sourceId>\n`);
+
+  if (shown.length < records.length) {
+    process.stdout.write(
+      `\n仅显示前 ${shown.length} 条（共 ${records.length} 条）。` +
+        `用 --name / --platform 缩小范围，或加 --all 查看全部（名称列已按 ${NAME_MAX_WIDTH} 列截断，完整值用 --json）\n`
+    );
+  } else if (total > records.length) {
+    process.stdout.write(
+      `\n注意：只有 ${total - records.length} 个结果没显示（接口 pageSize 固定 ${SOURCE_PAGE_SIZE}）。\n` +
+        `        请用 --name 或 --platform 缩小范围后再查。\n`
+    );
+  }
+  process.exitCode = 0;
+}
+
+/** mcp create */
+async function commandMcpCreate(cfg, argv) {
+  const opts = parseCommon(argv);
+  opts.sources = [];
+  opts.skipExists = false;
+  opts.dryRun = false;
+
+  for (let i = 0; i < argv.length; i += 1) {
+    const a = argv[i];
+    const next = () => {
+      const v = argv[i + 1];
+      i += 1;
+      return v === undefined ? '' : v;
+    };
+    if (a === '--source' || a === '-s') {
+      for (const part of String(next()).split(',')) {
+        const s = part.trim();
+        if (s) opts.sources.push(s);
+      }
+    } else if (a === '--skip-exists') opts.skipExists = true;
+    else if (a === '--dry-run') opts.dryRun = true;
+  }
+
+  if (opts.help) {
+    process.stdout.write(CREATE_USAGE + '\n');
+    process.exitCode = 0;
+    return;
+  }
+
+  const cred = loadCredentials();
+  if (!cred || !cred.tokenValue) {
+    throw new Error('未登录或授权已失效，请先执行 auth login');
+  }
+
+  const sourceType = parseSourceType(opts.type);
+
+  if (!opts.sources.length) {
+    throw new Error(
+      `缺少 --source（来源 sourceId 或工具名）。\n` +
+        `        先查可用来源：wb mcp sources --type ${sourceType}\n\n${CREATE_USAGE}`
+    );
+  }
+
+  // 步骤 1/4：确定连接器（可选，用于缩小来源范围）
+  let groupId = null;
+  let platLabel = '';
+  if (opts.platform && sourceType === 1) {
+    const p = await resolvePlatformGroupId(cfg, cred, opts.platform, process.stdout);
+    groupId = p.groupId;
+    platLabel = `${p.platformName || p.platformId}（groupId=${groupId}）`;
+  } else if (opts.platform && sourceType === 2) {
+    process.stderr.write('[warn] --platform 只对连接器资源（--type 1）有效，已忽略\n');
+  }
+  process.stdout.write(`[1/4] 来源类型: ${sourceTypeLabel(sourceType)}   连接器: ${platLabel || '（全部）'}\n`);
+
+  // 步骤 2/4：拉来源列表并定位
+  const { total, records } = await fetchMcpSources(cfg, cred, { name: '', groupId, sourceType });
+  process.stdout.write(`[2/4] 拉取来源 ${records.length} 个（共 ${total} 个）\n`);
+  if (total > records.length) {
+    process.stderr.write(
+      `[warn] 接口只返回前 ${SOURCE_PAGE_SIZE} 个来源，可能漏掉目标。建议加 --platform 或改用 --name 缩小范围\n`
+    );
+  }
+
+  const { picked, missing, ambiguous } = pickSources(records, opts.sources);
+
+  if (ambiguous.length) {
+    for (const amb of ambiguous) {
+      const rows = amb.matched.map((r, i) => [
+        String(i + 1),
+        String(r.name || ''),
+        String(r.sourceId || ''),
+        String(r.groupName || ''),
+        r.exists === true ? '已创建' : '未创建'
+      ]);
+      process.stdout.write(`名称 "${amb.input}" 命中 ${amb.matched.length} 个来源，请指定其一：\n\n`);
+      process.stdout.write(renderTable(['#', 'name', 'sourceId', 'groupName', 'created'], rows) + '\n\n');
+    }
+    throw new Error('请用 --source <sourceId> 指定具体来源');
+  }
+
+  if (missing.length) {
+    const detail = missing.map((m) => `        ${m.input}：${m.reason}`).join('\n');
+    throw new Error(
+      `以下来源没找到：\n${detail}\n` +
+        `        查看可用来源：wb mcp sources --type ${sourceType}`
+    );
+  }
+
+  if (!picked.length) {
+    throw new Error(`没有解析出任何可用来源\n\n${CREATE_USAGE}`);
+  }
+
+  // 步骤 3/4：exists=true 的不能创建
+  const created = picked.filter((r) => r.exists === true);
+  const toCreate = picked.filter((r) => r.exists !== true);
+
+  if (created.length) {
+    const rows = created.map((r, i) => [
+      String(i + 1),
+      String(r.name || ''),
+      String(r.sourceId || ''),
+      String(r.groupName || '')
+    ]);
+    process.stdout.write(`\n以下来源已经创建过 MCP 工具，不能重复创建：\n\n`);
+    process.stdout.write(renderTable(['#', 'name', 'sourceId', 'groupName'], rows) + '\n\n');
+
+    if (!opts.skipExists) {
+      throw new Error(
+        `有 ${created.length} 个来源已创建过，已中止（未提交任何数据）。\n` +
+          `        去掉这些 --source 重试，或加 --skip-exists 只创建其余 ${toCreate.length} 个`
+      );
+    }
+    process.stdout.write(`--skip-exists：跳过 ${created.length} 个，只创建其余 ${toCreate.length} 个\n`);
+  }
+
+  if (!toCreate.length) {
+    process.stdout.write('[3/4] 没有需要创建的工具\n');
+    process.exitCode = 0;
+    return;
+  }
+
+  const body = buildAddBody(toCreate);
+
+  process.stdout.write(`[3/4] 待创建 ${toCreate.length} 个：\n`);
+  for (const r of toCreate) {
+    process.stdout.write(
+      `      ${r.name}   sourceId=${r.sourceId}  sourceType=${r.sourceType}  groupId=${r.groupId}  groupName=${r.groupName}\n`
+    );
+  }
+
+  if (opts.dryRun) {
+    process.stdout.write(`[4/4] --dry-run，未提交。将发送：\n`);
+    process.stdout.write(`POST {baseUrl}/index/sys/api/v1/mcp/batch/add\n`);
+    process.stdout.write(JSON.stringify(body, null, 2) + '\n');
+    process.exitCode = 0;
+    return;
+  }
+
+  // 步骤 4/4：批量创建
+  const res = await authedRequest(cfg, cred, 'POST', '/sys/api/v1/mcp/batch/add', body);
+  const json = res.json;
+
+  if (!json || typeof json !== 'object') {
+    throw new Error(`创建失败：接口返回异常（HTTP ${res.status}）${String(res.text).slice(0, 200)}`);
+  }
+  if (json.code !== 200) {
+    if (opts.json) process.stdout.write(JSON.stringify(json, null, 2) + '\n');
+    throw new Error(`创建失败：code=${json.code} msg=${json.msg || '未知错误'}`);
+  }
+
+  process.stdout.write(`[4/4] 创建成功，共 ${toCreate.length} 个 MCP 工具\n`);
+  for (const r of toCreate) {
+    process.stdout.write(`      ${r.name}（sourceId=${r.sourceId}）\n`);
+  }
+  if (opts.json) process.stdout.write(JSON.stringify(json, null, 2) + '\n');
+
+  process.stdout.write(`\n复核：wb mcp sources --type ${sourceType} --name <工具名>\n`);
+  process.exitCode = 0;
+}
+
+module.exports = {
+  commandMcpSources,
+  commandMcpCreate,
+  fetchMcpSources,
+  resolvePlatformGroupId,
+  pickSources,
+  summarizeSources,
+  buildAddBody,
+  parseSourceType,
+  sourceTypeLabel,
+  parseCommon,
+  SOURCE_TYPE_NAMES,
+  SOURCE_PAGE_NUM,
+  SOURCE_PAGE_SIZE,
+  DISPLAY_LIMIT,
+  NAME_MAX_WIDTH
+};