Insérer un texte arbitraire et modifier NodeArrays dans le En fait, API compilateur tapuscrit

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__}); 
} 

J'ai deux questions:

1. Comment créer fragment d'AST de chaîne?
2. Comment ajouter une importation?

---

post-scriptum I tried diverses méthodes createSourceFile et ts.createImportDeclaration, mais ils conduisent tous à telle ou telle erreur: [

Answer 1

Pour ajouter dans le texte arbitraire, la meilleure façon que je connaisse est d'utiliser createIdentifier avec la chaîne que vous souhaitez insérer.

---

Pour ajouter dans une déclaration d'importation, rappelez-vous que vous mettez à jour l'arbre sans l'original muter.

L'API ts.visitEachChild prend un paramètre nodesVisitor qui fonctionne spécialement sur NodeArray s. Habituellement, si vous ne passez pas un nodesVisitor dedans, il fonctionnera sur chaque Node dans le NodeArray en utilisant le premier rappel que vous passez pour le paramètre visitor, mais ici vous voulez spécialement opérer sur le tableau de nœuds complet.

Dans votre cas, vous êtes intéressé par une des déclarations de SourceFile (qui est un NodeArray<Statement> Vous pouvez créer l'importation avec createImportDeclaration, et mettre à jour le SourceFile en passant dans un nodesVisitor comme ce qui suit:.

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);