使用Esprima解析JavaScript代码

Esprima 是一个用于解析 JavaScript 代码的库，它可以将源代码转换为抽象语法树（AST）。Esprima 的主要目的是为 JavaScript 代码分析工具提供基础，例如代码检查、代码重构、代码高亮等。

安装

npm install esprima

使用

以下是一个简单的示例，演示如何使用 Esprima 解析 JavaScript 代码并获取抽象语法树(AST)：

const esprima = require('esprima');

const code = `
  function add(a, b) {
    return a + b;
  } 
`;

const ast = esprima.parseScript(code);

console.log(ast);

在上面的示例中，首先定义了一个简单的 JavaScript 代码字符串，然后使用 parseScript 方法将代码字符串解析为抽象语法树，并将结果存储在 ast 变量中。运行上面的例子，得到下面的输出：

node src/add.js
Script {
  type: 'Program',
  body: [
    FunctionDeclaration {
      type: 'FunctionDeclaration',
      id: [Identifier],
      params: [Array],
      body: [BlockStatement],
      generator: false,
      expression: false,
      async: false
    }
  ],
  sourceType: 'script'
}

抽象语法树

Esprima 解析后的抽象语法树是一个嵌套的对象结构，它表示了源代码的结构和语义。每个节点都有一个 type 属性，表示节点的类型，例如 FunctionDeclaration 表示函数声明，VariableDeclaration 表示变量声明等。

以下是一个简单的示例，演示如何遍历抽象语法树并获取函数声明的名称：

const esprima = require('esprima');

const code = `
  function add(a, b) {
    return a + b;
  }
  function subtract(a, b) {
    return a - b;
  }
`;

const ast = esprima.parseScript(code);

ast.body.forEach((node) => {
    if (node.type === 'FunctionDeclaration') {
        console.log(`Function: ${node.id.name}`);
        node.params.forEach((param) => {
            console.log(`Argument: ${param.name}`);
        });
    }
});

在上面代码中，我们遍历了抽象语法树的 body 属性，并检查每个节点的类型。如果节点类型是 FunctionDeclaration，则输出函数的名称和参数的名称。执行代码，得到如下输出

Function: add
Argument: a
Argument: b
Function: subtract
Argument: a
Argument: b