插入任意文本和打字稿编译器API

Question

// Input 
class Foo { 
    templateString = 'some value'; 
} 

// Output 
import __LIB__ from '@scope/lib/path/to/lib'; 

class Foo { 
    templateString = (function compiledTemplate(deps) { 
     // ... 
     return result; 
    })({lib: __LIB__}); 
} 

实际上在修改NodeArrays，我有两个问题：

1. 如何创建字符串AST片段？
2. 如何添加导入？

---

P.S. I tried各种方法createSourceFile和ts.createImportDeclaration，但他们都带来这样或那样的错误：[

Answer 1

要在任意文本中添加，我知道最好的办法是使用createIdentifier与要插入的字符串。

---

要在import语句添加，请记住，您要更新树不突变原。

ts.visitEachChild API需要nodesVisitor参数，该参数专门用于NodeArray s。通常，如果您未通过nodesVisitor，它将在NodeArray中的每个Node上使用您为visitor参数传递的第一个回调进行操作，但此处您特别要在整个节点阵列上进行操作。

在你的情况，你有兴趣在SourceFile的语句（这是一个NodeArray<Statement>您可以createImportDeclaration创建导入，并通过在nodesVisitor通过类似下面的更新SourceFile：

function addImport(statements: ts.NodeArray<ts.Statement>) { 
    const importStatement = ts.createImportStatement(/*...*/); 
    return ts.createNodeArray([importStatement, ...statements]); 
} 

visitEachChild(
    sourceFile, 
    /*replace this with something that controls traversal*/ x => x, 
    context, 
    addImport);