AST 代码树结构生成技能

概述

此技能将输入的代码转化为 AST（抽象语法树）结构，支持多种编程语言，包括 Python、JavaScript、Java、C++ 等。它能够分析代码的语法结构，并以 JSON 格式输出详细的 AST 结构，便于代码分析和处理。

核心功能

• 多语言支持：支持 Python、JavaScript、Java、C++ 等多种编程语言
• AST 生成：将代码转化为抽象语法树结构
• JSON 输出：以 JSON 格式输出 AST 结构
• 详细分析：生成详细的 AST 节点和属性
• 文件支持：支持直接代码输入和文件输入

工作流程

步骤 1：接收代码输入

用户可以提供：

• 直接的代码片段
• 代码文件路径

步骤 2：解析代码

使用提供的脚本解析代码并生成 AST：

python /mnt/skills/custom/program_snippet/code-to-ast-new/scripts/convert.py \
  --language python \
  --code "def hello():\n    print('Hello, world!')" \
  --output /mnt/user-data/outputs/ast.json

步骤 3：生成 AST 结构

脚本会分析代码的语法结构，生成详细的 AST 节点，包括：

• 函数定义
• 类定义
• 变量声明
• 表达式
• 语句
• 控制流结构

步骤 4：输出结果

AST 结构将以 JSON 格式输出，可以直接在对话中呈现或保存到文件。

参数

参数	是否必需	描述
--language	是	编程语言（python, javascript, java, cpp）
--code	要么 --code 要么 --file	直接代码输入
--file	要么 --code 要么 --file	代码文件路径
--output	否	保存 AST 输出的路径
--indent	否	JSON 输出的缩进空格数（默认：2）

支持的语言

• Python：使用内置的 ast 模块
• JavaScript：使用 esprima 库
• Java：使用占位实现（需要 javaparser）
• C++：使用占位实现（需要 clang）

示例

Python 代码 AST 示例

输入：

def hello():
    print('Hello, world!')

命令：

python /mnt/skills/custom/program_snippet/code-to-ast-new/scripts/convert.py \
  --language python \
  --code "def hello():\n    print('Hello, world!')"

输出：

{
  "type": "Module",
  "body": [
    {
      "type": "FunctionDef",
      "name": "hello",
      "args": {
        "type": "arguments",
        "posonlyargs": [],
        "args": [],
        "kwonlyargs": [],
        "kw_defaults": [],
        "defaults": []
      },
      "body": [
        {
          "type": "Expr",
          "value": {
            "type": "Call",
            "func": {
              "type": "Name",
              "id": "print",
              "ctx": {
                "type": "Load"
              }
            },
            "args": [
              {
                "type": "Constant",
                "value": "Hello, world!",
                "kind": null
              }
            ],
            "keywords": []
          }
        }
      ],
      "decorator_list": [],
      "returns": null
    }
  ],
  "type_ignores": []
}

JavaScript 代码 AST 示例

输入：

function hello() {
  console.log('Hello, world!');
}

命令：

python /mnt/skills/custom/program_snippet/code-to-ast-new/scripts/convert.py \
  --language javascript \
  --code "function hello() {\n  console.log('Hello, world!');\n}"

输出：

{
  "type": "Program",
  "body": [
    {
      "type": "FunctionDeclaration",
      "id": {
        "type": "Identifier",
        "name": "hello"
      },
      "params": [],
      "body": {
        "type": "BlockStatement",
        "body": [
          {
            "type": "ExpressionStatement",
            "expression": {
              "type": "CallExpression",
              "callee": {
                "type": "MemberExpression",
                "object": {
                  "type": "Identifier",
                  "name": "console"
                },
                "property": {
                  "type": "Identifier",
                  "name": "log"
                },
                "computed": false
              },
              "arguments": [
                {
                  "type": "Literal",
                  "value": "Hello, world!",
                  "raw": "'Hello, world!'"
                }
              ]
            }
          }
        ]
      },
      "generator": false,
      "async": false
    }
  ],
  "sourceType": "script"
}

文件输入示例

命令：

python /mnt/skills/custom/program_snippet/code-to-ast-new/scripts/convert.py \
  --language python \
  --file /mnt/user-data/uploads/example.py \
  --output /mnt/user-data/outputs/ast.json

输出处理

• 对于简短的代码，直接在对话中呈现 AST 结构
• 对于较长的代码，导出到 JSON 文件并通过 present_files 工具分享
• 始终以 JSON 格式输出 AST 结构
• 提供详细的 AST 节点信息

注意事项

• AST 生成的质量取决于代码的语法正确性
• 非常大的代码文件可能需要更长的处理时间
• 不同语言的 AST 结构可能有所不同
• 对于最佳结果，提供语法正确的代码
• JavaScript AST 生成需要安装 esprima 库

依赖

• Python 3.7+
• 对于 JavaScript：esprima 库
• 对于 Java：javaparser（可选）
• 对于 C++：clang（可选）