scrum-skills 0.3.2 → 0.3.3

package/tools/diagram-import.mjs

@@ -0,0 +1,238 @@
+#!/usr/bin/env node
+/**
+ * 图例源文件摘要（零依赖，node >= 18）
+ *
+ * 顶替上游三个 Python 提取脚本（drawio / mermaid / excalidraw）。它**不画图**，
+ * 只把「又长又啰嗦的源文件」压成节点 + 连线的摘要，喂给模型重新按 diagram-design 的规范画。
+ *
+ * 为什么值得有这一层：一个 3000 行的 .excalidraw 直接塞给模型，上下文一半没了，
+ * 剩下的钱全花在读坐标上——而坐标对重画毫无用处，文字和连线才有用。
+ *
+ * 用法:
+ *   node tools/diagram-import.mjs <file.mmd|.drawio|.excalidraw> [--json] [--max 40]
+ * 退出码: 0 成功, 1 失败
+ */
+import { readFileSync } from 'node:fs';
+import { extname } from 'node:path';
+
+const args = process.argv.slice(2);
+const files = [];
+let json = false;
+let max = 40;
+for (let i = 0; i < args.length; i++) {
+  const a = args[i];
+  if (a === '--json') json = true;
+  else if (a === '--max') max = Number.parseInt(args[++i] ?? '40', 10) || 40;
+  else if (a.startsWith('--')) { console.error(`✘ 未知参数: ${a}`); process.exit(1); }
+  else files.push(a);
+}
+if (files.length !== 1) {
+  console.error('用法: node tools/diagram-import.mjs <file.mmd|.drawio|.excalidraw> [--json] [--max 40]');
+  process.exit(1);
+}
+
+const decode = s => String(s)
+  .replace(/&lt;/g, '<').replace(/&gt;/g, '>').replace(/&quot;/g, '"')
+  .replace(/&#39;/g, "'").replace(/&nbsp;/g, ' ').replace(/&amp;/g, '&');
+
+const clean = s => decode(s ?? '').replace(/<[^>]*>/g, '').replace(/\s+/g, ' ').trim();
+
+const unquote = s => {
+  const t = String(s ?? '').trim();
+  if ((t.startsWith('"') && t.endsWith('"')) || (t.startsWith("'") && t.endsWith("'"))) return t.slice(1, -1);
+  return t;
+};
+
+// ---------------- mermaid ----------------
+
+const MERMAID_KINDS = [
+  [/^\s*(?:flowchart|graph)\b/i, 'flowchart'],
+  [/^\s*sequenceDiagram\b/i, 'sequence'],
+  [/^\s*erDiagram\b/i, 'er'],
+  [/^\s*classDiagram\b/i, 'class'],
+  [/^\s*stateDiagram(?:-v2)?\b/i, 'state'],
+];
+
+function mermaid(text) {
+  const lines = text.split('\n').filter(l => !/^\s*%%/.test(l));
+  const kind = MERMAID_KINDS.find(([re]) => re.test(text))?.[1];
+  if (!kind) throw new Error('无法识别：mermaid 里没找到 flowchart / sequenceDiagram / erDiagram / classDiagram 之一');
+
+  const nodes = new Map();
+  const edges = [];
+  const addNode = (id, label = '') => {
+    if (!id) return;
+    const prev = nodes.get(id);
+    if (!prev || (!prev.label && label)) nodes.set(id, { id, label: label || prev?.label || '' });
+  };
+
+  if (kind === 'flowchart') {
+    // 形状：A[文字] / A(文字) / A{文字} / A[(库)] / A((圆)) / A[[子流程]]
+    const shape = /([A-Za-z_][\w-]*)\s*(\(\(([^)]*)\)\)|\[\(([^)]*)\)\]|\[\[([^\]]*)\]\]|\{\{([^}]*)\}\}|\[([^\]]*)\]|\(([^)]*)\)|\{([^}]*)\})/g;
+    for (const m of text.matchAll(shape)) {
+      addNode(m[1], clean(unquote(m.slice(3).find(Boolean) ?? '')));
+    }
+    // 边：允许两端各带形状（`A[浏览器] -->|提交账号| B[网关]` 是最常见写法）
+    const SHAPE = String.raw`(?:\(\([^)]*\)\)|\[\([^)]*\)\]|\[\[[^\]]*\]\]|\{\{[^}]*\}\}|\[[^\]]*\]|\([^)]*\)|\{[^}]*\})`;
+    const edgeRe = new RegExp(
+      String.raw`([A-Za-z_]\w*)\s*${SHAPE}?\s*(?:-\.->|=>+>|-{2,}[>ox]?|--[ox]|<--+)\s*(?:\|([^|]*)\|)?\s*([A-Za-z_]\w*)`,
+      'g',
+    );
+    for (const line of lines) {
+      for (const m of line.matchAll(edgeRe)) {
+        addNode(m[1]);
+        addNode(m[3]);
+        edges.push({ from: m[1], to: m[3], label: clean(unquote(m[2] ?? '')) });
+      }
+    }
+  } else if (kind === 'sequence') {
+    const parts = new Map();
+    for (const line of lines) {
+      const p = /^\s*(?:participant|actor)\s+([A-Za-z_]\w*)(?:\s+as\s+(.+))?$/i.exec(line);
+      if (p) parts.set(p[1], clean(unquote(p[2] ?? p[1])));
+    }
+    // 箭头必须显式匹配，id 里不许夹带 `-`：`A-->>U` 若放开连字符，会把 from 吃成 "A-"
+    const msgRe = /([A-Za-z_]\w*)\s*(-->>|->>|-->|->|--x|-x|--\)|-\))\s*([+-])?\s*([A-Za-z_]\w*)\s*:\s*(.+)$/;
+    for (const line of lines) {
+      if (/^\s*(?:participant|actor|note|loop|alt|else|end|opt|par|critical|break|rect|activate|deactivate|autonumber)\b/i.test(line)) continue;
+      const m = msgRe.exec(line);
+      if (!m) continue;
+      for (const id of [m[1], m[4]]) addNode(id, parts.get(id) ?? '');
+      edges.push({ from: m[1], to: m[4], label: clean(m[5]) });
+    }
+    for (const [id, label] of parts) addNode(id, label);
+  } else {
+    // er / class / state：都按「关系行」处理，够看就行
+    const rel = /([A-Za-z_][\w-]*)\s*([|}{o]{1,2}--[|}{o]{1,2}|<\|--|--\|>|-->|<--|\.\.>)\s*([A-Za-z_][\w-]*)\s*:\s*(\S+)/g;
+    for (const m of text.matchAll(rel)) {
+      addNode(m[1]);
+      addNode(m[3]);
+      edges.push({ from: m[1], to: m[3], label: clean(unquote(m[4])) });
+    }
+    for (const m of text.matchAll(/^\s*(?:class\s+)?([A-Z][\w-]*)\s*\{/gm)) addNode(m[1]);
+    if (!nodes.size) throw new Error(`${kind} 里没有解析出实体或关系`);
+  }
+
+  return { format: 'mermaid', kind, nodes: [...nodes.values()], edges };
+}
+
+// ---------------- drawio ----------------
+
+function drawio(text) {
+  const nodes = new Map();
+  const edges = [];
+  const cells = [...text.matchAll(/<mxCell\b([^>]*)>/g)].map(m => {
+    const attrs = {};
+    for (const a of m[1].matchAll(/([\w:-]+)="([^"]*)"/g)) attrs[a[1]] = a[2];
+    return attrs;
+  });
+  if (!cells.length) throw new Error('无法识别：没找到 <mxCell>，确认这是 drawio 的 XML');
+
+  for (const c of cells) {
+    const label = clean(c.value ?? '');
+    if (c.vertex === '1' && label) nodes.set(c.id, { id: c.id, label });
+    if (c.edge === '1' && c.source && c.target) {
+      edges.push({ from: c.source, to: c.target, label });
+    }
+  }
+  for (const e of edges) {
+    if (!nodes.has(e.from)) nodes.set(e.from, { id: e.from, label: '' });
+    if (!nodes.has(e.to)) nodes.set(e.to, { id: e.to, label: '' });
+  }
+  return { format: 'drawio', kind: 'graph', nodes: [...nodes.values()], edges };
+}
+
+// ---------------- excalidraw ----------------
+
+const SHAPES = new Set(['rectangle', 'diamond', 'ellipse', 'image', 'frame']);
+
+function excalidraw(text) {
+  let scene;
+  try { scene = JSON.parse(text); } catch (e) { throw new Error(`不是合法 JSON：${e.message}`); }
+  const all = (scene.elements ?? []).filter(e => !e?.isDeleted);
+  if (!Array.isArray(scene.elements)) throw new Error('无法识别：JSON 里没有 elements');
+
+  const textById = new Map();
+  const textOfContainer = new Map();
+  for (const e of all) {
+    if (e.type !== 'text') continue;
+    textById.set(e.id, e.text ?? '');
+    if (e.containerId) textOfContainer.set(e.containerId, e.text ?? '');
+  }
+
+  const nodes = new Map();
+  for (const e of all) {
+    if (!SHAPES.has(e.type)) continue;
+    const label = clean(textOfContainer.get(e.id) ?? '');
+    nodes.set(e.id, { id: e.id, label });
+  }
+
+  const edges = [];
+  for (const e of all) {
+    if (e.type !== 'arrow' && e.type !== 'line') continue;
+    const from = e.startBinding?.elementId;
+    const to = e.endBinding?.elementId;
+    if (!from || !to) continue;
+    edges.push({ from, to, label: clean(textOfContainer.get(e.id) ?? '') });
+  }
+  return { format: 'excalidraw', kind: 'whiteboard', nodes: [...nodes.values()], edges };
+}
+
+// ---------------- 输出 ----------------
+
+const KIND_CN = {
+  flowchart: '流程图', sequence: '时序图', er: 'ER 关系图', class: '类图',
+  state: '状态图', graph: '结构图', whiteboard: '白板草图',
+};
+
+function summarize(d) {
+  const nodes = d.nodes.slice(0, max);
+  const edges = d.edges.slice(0, max);
+  const restN = d.nodes.length - nodes.length;
+  const restE = d.edges.length - edges.length;
+  const nameOf = id => nodes.find(n => n.id === id)?.label || id;
+  const out = [];
+  out.push(`# ${d.format} · ${KIND_CN[d.kind] ?? d.kind}（${d.kind}）— ${d.nodes.length} 个节点 / ${d.edges.length} 条边`);
+  out.push('');
+  if (nodes.length) {
+    out.push('节点：');
+    for (const n of nodes) out.push(`- ${n.label || n.id}`);
+    if (restN > 0) out.push(`- …其余 ${restN} 个节点省略（--max 调大）`);
+    out.push('');
+  }
+  if (edges.length) {
+    out.push('连线：');
+    for (const e of edges) out.push(`- ${nameOf(e.from)} → ${nameOf(e.to)}${e.label ? `（${e.label}）` : ''}`);
+    if (restE > 0) out.push(`- …其余 ${restE} 条连线省略`);
+  }
+  return out.join('\n');
+}
+
+const file = files[0];
+const ext = extname(file).toLowerCase();
+const text = readFileSync(file, 'utf8');
+let data;
+try {
+  if (ext === '.mmd' || ext === '.mermaid' || ext === '.md') data = mermaid(text);
+  else if (ext === '.drawio' || ext === '.xml') data = drawio(text);
+  else if (ext === '.excalidraw' || ext === '.json') data = excalidraw(text);
+  else throw new Error(`不支持的格式 ${ext || '（无扩展名）'}：只认 .mmd / .drawio / .excalidraw`);
+} catch (e) {
+  console.error(`✘ ${e.message}`);
+  process.exit(1);
+}
+
+if (json) {
+  const nodes = data.nodes.slice(0, max);
+  const edges = data.edges.slice(0, max);
+  console.log(JSON.stringify({
+    format: data.format,
+    kind: data.kind,
+    stats: { nodes: data.nodes.length, edges: data.edges.length },
+    nodes,
+    edges,
+    truncated: nodes.length < data.nodes.length || edges.length < data.edges.length,
+  }, null, 2));
+} else {
+  console.log(summarize(data));
+}