代码之家  ›  专栏  ›  技术社区  ›  A T

分析emit typescript?-` const a={foo:'bar'}`到'const a={foo:'bar',can:'haz'}`?

  •  0
  • A T  · 技术社区  · 5 年前

    跟着 the guide ,找到了一些方法:

    export function delint(sourceFile: ts.SourceFile) {
      delintNode(sourceFile);
    
      function delintNode(node: ts.Node) {
        if (node.kind === ts.SyntaxKind.VariableDeclaration) {
          // something
        }
    
        ts.forEachChild(node, delintNode);
    

    我如何获得其余的方法,即:向对象文字添加一个新的东西,然后以相同的属性顺序输出编辑代码,但在最后是一个新的属性?

    1 回复  |  直到 5 年前
        1
  •  0
  •   David Sherret    5 年前

    如果您只想修改源文件文本以添加属性,则可以这样做:

    1. 从ast中获取对象文本表达式(使用 https://ts-ast-viewer.com 有助于理解ast的外观)。
    2. 使用 ts.updateObjectLiteral 为具有新属性的旧节点获取新节点。
    3. 打印出新节点的文本,并用打印的节点文本修改原始文本。
    4. 必要时重新分析到新的ast,或者只使用修改后的文本。

    请注意,打印对象文字将删除以前使用的任何自定义格式,因为打印机以自己的方式打印节点。之后,您可能需要通过prettier或其他格式化程序运行代码。

    例子:

    import * as ts from "typescript";
    
    // parse the AST
    const sourceFile = ts.createSourceFile("file.ts", "const a = { foo: 'bar' };",
        ts.ScriptTarget.Latest, false);
    
    // get the object literal expression
    const objectLiteralExpression = sourceFile.statements.find(ts.isVariableStatement)!
        .declarationList.declarations[0].initializer as ts.ObjectLiteralExpression;
    
    // get a transformed node with the new property and print it
    const transformedOle = ts.updateObjectLiteral(objectLiteralExpression, [
        ...objectLiteralExpression.properties,
        ts.createPropertyAssignment("can", ts.createStringLiteral("haz"))
    ]);
    const newNodeText = ts.createPrinter()
        .printNode(ts.EmitHint.Unspecified, transformedOle, sourceFile);
    
    // get the new source file text and reparse it to a new AST
    const oldText = sourceFile.text;
    const newText = oldText.substring(0, objectLiteralExpression.getStart(sourceFile, true))
        + newNodeText + oldText.substring(objectLiteralExpression.end);
    const newSourceFile = ts.createSourceFile("file.ts", newText, ts.ScriptTarget.Latest, false);
    
    // outputs: `const a = { foo: "bar", can: "haz" };`
    console.log(newText);
    

    如果您希望在发射时执行此操作,请考虑为 Program#emit(...) 是的。