原始内容
name: "code-to-diagram" description: "分析源代码逻辑,生成 Mermaid 流程图或 SVG 架构图并渲染为 PNG 图片。官方 mermaid 渲染(与 Markdown 预览一致),支持 16 种主题、5 种视觉风格、语义图形体系和 40+ 产品图标。" version: "3.0.0" tags: ["diagram", "mermaid", "svg", "flowchart", "visualization", "code-analysis", "architecture", "mermaid-cli"] bindShells: ["ClaudeCode"]
Code-to-Diagram Skill
分析指定目录或文件的源代码,提取控制流 / 数据流逻辑,输出文件:
- Markdown 文档(
.md):包含图表源码和代码逻辑解释文字 - 渲染后的 PNG 图片(高清,用户不需要图片时可跳过)
Mermaid 路径中用于渲染的
.mmd中间文件会在渲染完成后自动删除,最终只保留.md和.png。
支持两种渲染引擎:
| 引擎 | 输入 | 渲染方式 | 适用场景 |
|---|---|---|---|
| Mermaid(默认) | .mmd |
官方 mermaid(mmdc)→ PNG | 流程图、时序图、类图、状态图、ER 图、XY 图,以及甘特图、思维导图、饼图等全部 mermaid 图表类型 |
| SVG | .svg |
rsvg-convert → PNG | 架构图、AI 系统图、需要品牌图标或定制风格的图表 |
Mermaid 引擎使用官方 mermaid 渲染器(mmdc / Mermaid CLI),PNG 与 GitHub、VS Code 等 Markdown 预览效果一致。 16 个内置主题通过 mermaid
themeVariables应用。
第一步 —— 读取并分析代码
使用 Read、Glob、Grep 工具理解代码结构,识别主要逻辑模式。
分析目标
从代码中提取以下信息,作为图表生成的素材:
| 提取内容 | 分析方法 | 对应图表 |
|---|---|---|
| 函数调用链 / 执行路径 | Grep 函数名,追踪调用关系 | 流程图、时序图 |
| 类 / 接口 / 继承关系 | 搜索 class、implements、extends、interface |
类图 |
| 状态字段 / 状态机 | 搜索 state、status、switch/case、枚举定义 |
状态图 |
| HTTP / RPC 调用 | 搜索 fetch、axios、http、grpc、request |
时序图 |
| 数据库表 / ORM 模型 | 搜索 CREATE TABLE、@Entity、schema、Model |
ER 图 |
| 模块划分 / 目录结构 | Glob 扫描目录,识别子系统边界 | 流程图(subgraph)、架构图 |
| API 路由 / 端点 | 搜索路由定义(router、@Get、@Post、app.get) |
时序图、流程图 |
| 配置 / 依赖关系 | 搜索 import、require、dependency |
流程图、类图 |
分析策略
- 单文件:从入口函数开始,沿调用链向下追踪
- 多文件 / 多模块:先 Glob 扫描目录结构,识别模块边界和入口文件,再逐模块分析核心逻辑
- 大型代码库:优先分析入口文件(
main、index、app),再按依赖方向向外扩展 - 不确定图表类型时:先提取代码特征,再对照图表类型表选择
多文件 / 多模块处理
当分析涉及多个文件或模块时:
- 先画架构全景:用
flowchart TB+subgraph或 SVG 架构图展示模块间关系,再对核心模块单独画详细图 - 模块间用时序图:跨模块调用链适合
sequenceDiagram,参与者对应模块/服务 - 模块内用流程图:单个模块的内部逻辑适合
flowchart - 合并策略:如果多个文件逻辑紧密耦合,合并为一张图;如果松散耦合,分别生成独立图表
- 命名规范:多张图时用
模块名-图表类型命名,如auth-sequence.png、order-flow.png
第二步 —— 选择引擎
使用 Mermaid 引擎
- 标准流程图、时序图、类图、状态图、ER 图、XY 图表
- 用户未要求特定视觉风格
- 快速、美观的主题化图表
使用 SVG 引擎
- 架构图需要产品图标(OpenAI、AWS、PostgreSQL 等)
- AI/Agent 系统图需要语义图形(六边形 Agent、圆柱向量库等)
- 用户要求特定风格(dark-terminal、blueprint、glassmorphism 等)
- 需要精细布局控制
第三步 A —— Mermaid 路径
生成 Mermaid 源码
根据分析结果编写 Mermaid 图表源码。确定图表类型后,仅需了解对应语法即可,不需要加载所有类型。
| 代码特征 | 图表类型 | Mermaid 关键字 | 典型场景 |
|---|---|---|---|
switch/case、状态枚举、状态字段转换 |
状态图 | stateDiagram-v2 |
订单状态流转、连接器生命周期 |
class、implements、extends、字段和方法定义 |
类图 | classDiagram |
领域模型、SDK 结构 |
| 函数调用链、if/else 分支、循环 | 流程图 | flowchart TD/LR/TB |
算法逻辑、业务流程、请求处理 |
| HTTP/RPC 调用、跨模块交互、消息传递 | 时序图 | sequenceDiagram |
API 调用链、微服务交互 |
CREATE TABLE、ORM 模型、外键关系 |
ER 图 | erDiagram |
数据库 schema、数据模型 |
| 多模块/多服务、目录结构分层 | 架构图 | flowchart TB + subgraph |
系统架构、微服务拓扑 |
| 甘特排期、任务依赖 | 甘特图 | gantt |
项目计划、迭代排期 |
| 占比数据、模块权重 | 饼图 | pie |
资源分布、模块占比 |
| Git 分支/合并 | Git 图 | gitGraph |
版本历史、分支策略 |
| 用户操作路径 | 用户旅程图 | journey |
用户体验、操作步骤 |
| 功能分解、层级结构 | 思维导图 | mindmap |
模块拆解、知识结构 |
| 版本演进、里程碑 | 时间线图 | timeline |
版本历史、演进路线 |
| 对比矩阵、优先级 | 象限图 | quadrantChart |
技术选型、优先级排序 |
| 数据趋势、性能指标 | XY 图 | xychart-beta |
性能监控、趋势分析 |
| 多容器部署、服务拓扑 | C4 架构 | C4Container |
微服务部署、云架构 |
图表类型快速判断:
- 代码中有状态转换 / 枚举 →
stateDiagram-v2 - 代码中有类定义 / 继承 →
classDiagram - 需要展示执行流程 / 分支逻辑 →
flowchart - 需要展示跨模块调用 →
sequenceDiagram - 需要展示数据模型 →
erDiagram - 需要展示系统整体结构 →
flowchart + subgraph或 SVG 架构图
语言规则:节点标签、连线说明必须使用中文。代码标识符保留原文。
曲线样式:如需使用弧线(曲线)连接,在图表开头添加配置:
%%{ init: { 'flowchart': { 'curve': 'basis' } } }%%
可选曲线类型:basis(平滑)、monotoneX、monotoneY、stepBefore、stepAfter。
曲线配置由官方 mermaid 渲染器原生支持,直接写在
.mmd中即可生效。
换行规则:节点文本中使用 <br/> 而非 \n。
特殊字符规则:
- 节点标签中禁止使用双引号
",直接移除或用单引号替代 - 节点标签中的括号
()、方括号[]、花括号{}会被解析为节点形状,如需显示用全角符号替代 - 代码标识符中的点号
.保留(如featurePut.SubTitle) - 冒号
:在中文环境下通常安全,但避免在连线标签中使用 &符号使用&转义
节点文本简化原则:
- 移除所有引号(单引号、双引号)
- 简化过长的变量名,保留关键信息
- 使用中文描述替代代码表达式(如 "创建者名 创建" 而非
creator.Name + " 创建")
选择主题
16 个内置主题,根据场景推荐:
| 主题 | 类型 | 推荐场景 |
|---|---|---|
markdown-preview(默认) |
亮色 | Markdown预览、浅色背景、系统字体 |
github-dark |
暗色 | GitHub README、技术文档 |
github-light |
亮色 | 明亮文档、演示文稿 |
tokyo-night |
暗色 | 优雅暗色、博客 |
tokyo-night-storm |
暗色 | Tokyo Night 变体 |
tokyo-night-light |
亮色 | Tokyo Night 亮色 |
catppuccin-mocha |
暗色 | 温暖暗色 |
catppuccin-latte |
亮色 | 温暖亮色 |
nord |
暗色 | 冷色调、北欧风 |
nord-light |
亮色 | 冷色调亮色 |
dracula |
暗色 | 经典暗色 |
one-dark |
暗色 | VS Code 风格 |
solarized-dark |
暗色 | 经典 Solarized |
solarized-light |
亮色 | 经典 Solarized 亮色 |
zinc-dark |
暗色 | 极简暗色 |
zinc-light |
亮色 | 极简亮色 |
主题快速判断:
- 用户说"暗色/dark/GitHub" →
github-dark - 用户说"亮色/light/明亮" →
github-light - 用户说"优雅/elegant" →
tokyo-night - 用户说"温暖/warm/mocha" →
catppuccin-mocha - 用户说"冷色/cold/北欧" →
nord - 用户说"紫色/dracula" →
dracula - 用户说"简约/minimal" →
zinc-dark或zinc-light - 用户说"Markdown预览/浅色/系统字体" →
markdown-preview - 默认 →
markdown-preview
生成逻辑解释文字
在生成图表源码的同时,编写代码逻辑解释文字。输出结构:
## 概述
<用 1-2 句话说明这段代码的整体功能和职责>
## 核心流程
<按执行顺序描述主要逻辑路径,每个步骤对应图表中的一个节点或子图>
## 关键组件
<列出图表中每个关键节点/子图对应的代码实体(类名、函数名、模块名),以及它们的职责>
## 设计要点
<描述代码中值得注意的设计模式、架构决策或边界处理>
渲染前校验
Mermaid 对语法要求严格。写入 .mmd 文件前,逐项检查:
- 第一行是合法的图表关键字(
flowchart、sequenceDiagram、classDiagram、stateDiagram-v2、erDiagram、gantt、pie、gitGraph、journey、mindmap、timeline、quadrantChart、xychart-beta) - 流程图关键字后需跟方向(
TD、LR、BT、RL) - 节点 ID 不含空格(用方括号/引号包裹的 label 来显示文本)
- 标签中的特殊字符用引号包裹:
A["节点 (含括号)"] - 箭头类型合法(
-->、->>、-->>、-.-、==>) - 节点标签中禁止使用双引号
",用单引号或移除 &符号使用&转义
如果渲染报解析错误,根据错误信息定位行号、修正 .mmd 文件后重试。
前置检测
渲染前必须检测依赖工具是否已安装(使用 --no-png 不生成图片时可跳过本步骤)。
Mermaid 引擎(必需 mmdc,即 Mermaid CLI):
which mmdc
已安装:返回路径,可继续
未安装:停止流程,提示用户安装:
未检测到
mmdc。请先安装:npm install -g @mermaid-js/mermaid-cli安装完成后告知我,我会继续。(也可使用 npx 免安装运行,脚本会自动尝试)
SVG 引擎(必需 rsvg-convert):
which rsvg-convert
未安装:提示用户安装:
# macOS brew install librsvg # Debian / Ubuntu apt-get install librsvg2-bin
写入 .mmd 文件并渲染
先用 Write 工具将 Mermaid 源码写入 .mmd 文件,然后调用:
node ~/.claude/skills/code-to-diagram/scripts/code_to_diagram.js render \
--file <路径/diagram.mmd> \
--theme <主题名> \
--name <输出文件基础名> \
--output-dir <保存目录>
脚本会生成两个文件:
<name>.png— 渲染后的高清 PNG 图片<name>.md— 基础 Markdown 文档(仅包含图表源码)
渲染成功后脚本会自动删除输入的 .mmd 中间文件,最终只保留 .md 和 .png。
如果用户明确表示不需要图片,在命令中加 --no-png 参数,此时只生成 .md 文件,不调用渲染器(也无需检测 rsvg-convert):
node ~/.claude/skills/code-to-diagram/scripts/code_to_diagram.js render \
--file <路径/diagram.mmd> \
--name <输出文件基础名> \
--output-dir <保存目录> \
--no-png
然后用 Read 工具读取生成的 .md 文件,用 Write 工具补充代码逻辑解释文字,最终内容结构:
- 代码逻辑解释文字(中文)
- Mermaid 图表源码(代码块)
第三步 B —— SVG 路径
1. 选择风格
读取 references/style-diagram-matrix.md 快速选择风格,然后读取对应的风格参考文件:
| 风格 | 参考文件 | 适用场景 |
|---|---|---|
| Flat Icon(默认) | references/style-1-flat-icon.md |
文档、博客、演示 |
| Dark Terminal | references/style-2-dark-terminal.md |
GitHub README、技术博客 |
| Blueprint | references/style-3-blueprint.md |
架构文档、RFC |
| Notion Clean | references/style-4-notion-clean.md |
Notion 嵌入、Wiki |
| Glassmorphism | references/style-5-glassmorphism.md |
营销页、发布会 |
风格快速判断:
- 用户说"暗色/dark/terminal/GitHub" → Dark Terminal
- 用户说"蓝图/blueprint/工程" → Blueprint
- 用户说"简洁/clean/Notion" → Notion Clean
- 用户说"毛玻璃/glass/现代" → Glassmorphism
- 默认或"文档/博客" → Flat Icon
2. 生成 SVG
读取选定的风格参考文件和 references/icons.md,按照以下步骤生成 SVG:
- 以风格参考文件中的 SVG 模板 为起点
- 根据内容规划布局(节点位置、连线路径)
- 使用
references/icons.md中的 语义图形 表示不同类型的组件 - 使用 产品图标 标识具体产品/服务
- 使用 箭头语义 区分不同类型的数据流
- 当使用 2+ 种箭头类型时,添加 图例
- 所有文字标签使用中文
关键约束:
- 禁止
@import url()——rsvg-convert 无法获取外部资源 - 字体通过内联
<style>声明 - viewBox 根据实际内容调整
3. 生成逻辑解释文字
在生成 SVG 的同时,编写代码逻辑解释文字。输出结构:
## 概述
<用 1-2 句话说明这段代码/系统的整体功能和职责>
## 核心流程
<按执行顺序描述主要逻辑路径,每个步骤对应图表中的一个节点或子图>
## 关键组件
<列出图表中每个关键节点/子图对应的代码实体(类名、函数名、模块名),以及它们的职责>
## 设计要点
<描述代码中值得注意的设计模式、架构决策或边界处理>
4. 写入 SVG 文件并渲染
先用 Write 工具将 SVG 写入 .svg 文件,然后调用:
node ~/.claude/skills/code-to-diagram/scripts/code_to_diagram.js render \
--engine svg \
--file <路径/diagram.svg> \
--name <输出文件基础名> \
--style <风格名> \
--output-dir <保存目录>
脚本会生成两个文件:
<name>.png— 渲染后的高清 PNG 图片<name>.md— 基础 Markdown 文档(仅包含图表源码)
然后用 Read 工具读取生成的 .md 文件,用 Write 工具补充代码逻辑解释文字,最终内容结构:
- 代码逻辑解释文字(中文)
- SVG 图表源码(代码块)
第四步 —— 补充逻辑解释
脚本已生成基础 .md 文件(仅包含图表源码),现在用 Read 工具读取该文件,然后用 Write 工具补充代码逻辑解释文字,最终内容结构:
- 代码逻辑解释:用中文描述代码的整体结构、关键逻辑流程、设计模式和架构决策
- 图表源码:Mermaid 或 SVG 源码(放在代码块中)
告知用户文件路径
脚本最后一行输出 JSON,包含输出文件路径:
{"md":"/path/to/diagram.md","png":"/path/to/diagram.png","engine":"mermaid","theme":"markdown-preview","renderer":"mmdc"}
使用 --no-png 时 png 字段为 null:
重要:不要使用 Read 工具读取 PNG 文件来内联展示图片。 PNG 图片体积大,直接读取会消耗大量上下文窗口,极易导致超限。只需将生成的文件路径告知用户即可,例如:
已生成图表:
- Markdown:
/path/to/diagram.md- PNG:
/path/to/diagram.png
用户可以自行打开文件查看图片。
语义图形速查表
使用 SVG 引擎时,根据组件类型选择对应图形(详见 references/icons.md):
| 图形 | 含义 | 使用时机 |
|---|---|---|
| 双边框圆角矩形 + ⚡ | LLM / 模型 | 大语言模型调用 |
| 六边形 | Agent / 编排器 | 自主代理、编排 |
| 圆柱体 + 内环 | 向量数据库 | Pinecone、Weaviate 等 |
| 圆柱体 | 传统数据库 | PostgreSQL、Redis 等 |
| 矩形 + ⚙ | 工具 / 函数 | API 调用、工具执行 |
| 菱形 | 决策点 | 条件判断 |
| 圆形 + 身体 | 用户 | 人类交互入口 |
| 虚线矩形 | 记忆节点 | 短期/长期记忆 |
| 水平管道 | 队列 / 消息流 | Kafka、RabbitMQ 等 |
| 小六边形 | API 网关 | 请求路由 |
| 红绿灯矩形 | 浏览器 | Web 客户端 |
| 折角矩形 | 文档 | 配置文件、日志 |
箭头语义速查表
| 类型 | 线型 | 含义 |
|---|---|---|
| 实线 2px | stroke-width="2" |
主数据流 |
| 虚线 1.5px | stroke-dasharray="5,3" |
记忆/缓存写入 |
| 点线 1.5px | stroke-dasharray="4,2" |
异步事件 |
| 曲线 1.5px | 贝塞尔曲线 | 反馈/循环 |
颜色编码:蓝=主数据流,红=错误/备选,绿=数据写入,紫=异步事件。
规则:使用 2+ 种箭头类型时,必须在左下角添加图例。
产品图标速查表
常用产品(完整列表见 references/icons.md):
| 类别 | 产品 | 品牌色 |
|---|---|---|
| AI/ML | OpenAI #10A37F · Anthropic #D97757 · Gemini #4285F4 · LLaMA #0467DF · Mistral #FF7000 |
|
| RAG | LangChain #1C3C3C · LlamaIndex #8B5CF6 · CrewAI #EF4444 · Mem0 #6366F1 |
|
| 向量库 | Pinecone #1C1C2E · Weaviate #FA0050 · Qdrant #DC244C · Chroma #FF6B35 |
|
| 数据库 | PostgreSQL #336791 · MySQL #4479A1 · MongoDB #47A248 · Redis #DC382D |
|
| 消息队列 | Kafka #231F20 · RabbitMQ #FF6600 |
|
| 云平台 | AWS #FF9900 · GCP #4285F4 · Azure #0089D6 · Docker #2496ED · K8s #326CE5 |
命令行参数
node code_to_diagram.js render [选项]
通用选项:
--file, -f <路径> 输入文件(.mmd 或 .svg)
--content, -c <字符串> Mermaid 源码(仅 mermaid 引擎)
--name, -n <字符串> 输出文件基础名(默认:diagram)
--output-dir, -o <路径> 输出目录(默认:当前工作目录)
--engine, -e <引擎> mermaid | svg(默认:mermaid)
--help, -h 帮助信息
Mermaid 引擎(官方 mmdc 渲染):
--theme, -t <主题> 16 个内置主题(默认:markdown-preview,映射为 themeVariables)
--transparent 透明背景
--bg, -b <颜色> 自定义背景色(覆盖主题)
--font <字体> 自定义字体(默认:系统中文字体)
--width, -W <像素> 画布宽度(默认:2400)
--height, -H <像素> 画布高度(默认:4000)
--scale, -s <倍数> 缩放系数(默认:3)
SVG 引擎:
--style <风格> flat-icon | dark-terminal | blueprint | notion-clean | glassmorphism
**图片尺寸说明**:Mermaid 引擎使用固定画布尺寸(可用 `--width` / `--height` / `--scale` 调整);SVG 引擎会根据图表内容自动计算最佳输出尺寸(viewBox 宽度 × 8/12/16,范围 1200-4800px),确保文字清晰可读。
依赖说明
Mermaid 引擎:
- mmdc(必需):Mermaid CLI,官方 mermaid 渲染器
SVG 引擎:依赖 rsvg-convert(来自 librsvg)。
安装 mmdc:
npm install -g @mermaid-js/mermaid-cli
安装 rsvg-convert:
# macOS
brew install librsvg
# Debian / Ubuntu
apt-get install librsvg2-bin
排错
| 问题 | 原因 | 解决方案 |
|---|---|---|
| 解析错误 / unknown error | Mermaid 语法不合法 | 阅读错误信息,定位行号,修正 .mmd 文件后重试 |
| 输出空白 | 图表关键字缺失或拼写错误 | 第一行必须以合法关键字开头,且流程图需跟方向(TD/LR) |
| 含特殊字符的标签出错 | 字符未转义 | 用引号包裹标签:A["节点 (文本)"] |
| 节点 ID 含空格失败 | ID 必须是单个标识符 | 用 camelCase 或下划线做 ID,文本放在 label 中 |
| rsvg-convert 报错(SVG 引擎) | 未安装 librsvg | 执行 brew install librsvg(macOS)或 apt-get install librsvg2-bin(Linux) |
| mmdc 未找到 | 未安装 Mermaid CLI | 执行 npm install -g @mermaid-js/mermaid-cli 或使用 npx |
| Puppeteer/Chrome 启动报错 | 无头浏览器不可用 | 创建 puppeteer-config.json 加 {"args": ["--no-sandbox"]},通过 -p 传入 |
| 大图被截断 | 默认页面过小 | 增大 --width / --height,或用 --scale 缩放 |
| SVG 引擎字体缺失 | 内联字体声明不完整 | 检查 <style> 中 @font-face 的 font-family 是否与文本一致 |