การท่องผ่าน AST ด้วยตัวเยี่ยมชม

AST คืออะไร

import ts from "typescript";
const sf = ts.createSourceFile(
  "example.ts",
  "const x = 1;",
  ts.ScriptTarget.Latest
);

ชนิดของโหนด

import ts from "typescript";
// Common kinds:
// ts.SyntaxKind.VariableDeclaration
// ts.SyntaxKind.FunctionDeclaration
// ts.SyntaxKind.Identifier

การเยี่ยมชมด้วย forEachChild

function visit(node: ts.Node) {
  if (ts.isFunctionDeclaration(node)) {
    console.log("Function:", node.name?.text);
  }
  ts.forEachChild(node, visit);
}
visit(sourceFile);

ฟังก์ชันตรวจสอบชนิดข้อมูล

ts.forEachChild(sf, node => {
  if (ts.isClassDeclaration(node)) {
    console.log("Class:", node.name?.getText());
  }
  if (ts.isInterfaceDeclaration(node)) {
    console.log("Interface:", node.name.getText());
  }
});

การรับข้อความของโหนด

ts.forEachChild(sf, node => {
  if (ts.isIdentifier(node)) {
    console.log("Id:", node.text);
  }
});

ตำแหน่งของโหนด

ts.forEachChild(sf, node => {
  console.log(node.kind, "at", node.pos, "-", node.end);
});

การเดินสำรวจด้วยตัวช่วยแบบเวียนเกิด

function walk(node: ts.Node, visitor: (n: ts.Node) => void) {
  visitor(node);
  ts.forEachChild(node, child => walk(child, visitor));
}

ค้นหาการเรียกฟังก์ชันทั้งหมด

walk(sf, node => {
  if (ts.isCallExpression(node)) {
    const expr = node.expression.getText(sf);
    console.log("Call:", expr);
  }
});

การเข้าถึงโหนดแม่

// For manually created source files:
// (ts as any).setParents(sf);

การใช้ TypeChecker กับโหนด

const checker = program.getTypeChecker();
walk(sf, node => {
  if (ts.isExpression(node)) {
    const type = checker.getTypeAtLocation(node);
    console.log(checker.typeToString(type));
  }
});

ทบทวน: การสำรวจ AST

ตรวจสอบความเข้าใจ

สิ่งที่ได้เรียนรู้