TypeScrip:如何根据数据结构生成和打印AST [英] Typescript: How to generate and print an AST based on a data structure
本文介绍了TypeScrip:如何根据数据结构生成和打印AST的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
我正在启动一个新项目,作为其接口的一部分,我们有一大堆"令牌",这是一个带有字符串值的递归对象,如下所示:
const colors = {
accent: '#f90',
primary: {
active: '#fff',
inactive: 'silver'
}
};
我们提供了一个通过基于字符串的路径使用这些内容的实用程序(例如,在本例中,primary.active
for#fff
)。将所有可能的路径提取到一个数组中是很容易的,但是我们希望为这个包的使用者提供更好的自动补全,所以不是‘字符串’,而是这些可能的路径的联合或枚举。有没有人有这方面的经验?我最初的方法是编写一个简单的脚本,获取一个数组并使用模板或类似的东西将其打印为联合,但考虑到我们希望更频繁地这样做,并且我们的用例将增加复杂性,我认为生成和打印AST可能是一种更好的方法。我写了巴别塔和重塑代码之前,我只是寻找一些关于现有的工具集,例子等方面的指导。我做了一个快速的谷歌,但找不到任何东西。理想情况下,这些代码将与我正常的"监视"过程一起重新编译,但这是一个扩展目标^_^。
推荐答案
您可以使用编译器API提取对象类型并创建联合类型
import * as ts from 'typescript'
import * as fs from 'fs'
var cmd = ts.parseCommandLine(['test.ts']); // replace with target file
// Create the program
let program = ts.createProgram(cmd.fileNames, cmd.options);
type ObjectDictionary = { [key: string]: string | ObjectDictionary}
function extractAllObjects(program: ts.Program, file: ts.SourceFile): ObjectDictionary {
let empty = ()=> {};
// Dummy transformation context
let context: ts.TransformationContext = {
startLexicalEnvironment: empty,
suspendLexicalEnvironment: empty,
resumeLexicalEnvironment: empty,
endLexicalEnvironment: ()=> [],
getCompilerOptions: ()=> program.getCompilerOptions(),
hoistFunctionDeclaration: empty,
hoistVariableDeclaration: empty,
readEmitHelpers: ()=>undefined,
requestEmitHelper: empty,
enableEmitNotification: empty,
enableSubstitution: empty,
isEmitNotificationEnabled: ()=> false,
isSubstitutionEnabled: ()=> false,
onEmitNode: empty,
onSubstituteNode: (hint, node)=>node,
};
let typeChecker = program.getTypeChecker();
function extractObject(node: ts.ObjectLiteralExpression): ObjectDictionary {
var result : ObjectDictionary = {};
for(let propDeclaration of node.properties){
if(!ts.isPropertyAssignment( propDeclaration )) continue;
const propName = propDeclaration.name.getText()
if(!propName) continue;
if(ts.isObjectLiteralExpression(propDeclaration.initializer)) {
result[propName] = extractObject(propDeclaration.initializer);
}else{
result[propName] = propDeclaration.initializer.getFullText()
}
}
return result;
}
let foundVariables: ObjectDictionary = {};
function visit(node: ts.Node, context: ts.TransformationContext): ts.Node {
if(ts.isVariableDeclarationList(node)) {
let triviaWidth = node.getLeadingTriviaWidth()
let sourceText = node.getSourceFile().text;
let trivia = sourceText.substr(node.getFullStart(), triviaWidth);
if(trivia.indexOf("Generate_Union") != -1) // Will generate fro variables with a comment Generate_Union above them
{
for(let declaration of node.declarations) {
if(declaration.initializer && ts.isObjectLiteralExpression(declaration.initializer)){
foundVariables[declaration.name.getText()] = extractObject(declaration.initializer)
}
}
}
}
return ts.visitEachChild(node, child => visit(child, context), context);
}
ts.visitEachChild(file, child => visit(child, context), context);
return foundVariables;
}
let result = extractAllObjects(program, program.getSourceFile("test.ts")!); // replace with file name
function generateUnions(dic: ObjectDictionary) {
function toPaths(dic: ObjectDictionary) : string[] {
let result: string[] = []
function extractPath(parent: string, object: ObjectDictionary) {
for (const key of Object.keys(object)) {
let value = object[key];
if(typeof value === "string") {
result.push(parent + key);
}else{
extractPath(key + ".", value);
}
}
}
extractPath("", dic);
return result;
}
return Object.entries(dic)
.map(([name, values])=>
{
let paths = toPaths(values as ObjectDictionary)
.map(ts.createStringLiteral)
.map(ts.createLiteralTypeNode);
let unionType = ts.createUnionTypeNode(paths);
return ts.createTypeAliasDeclaration(undefined, undefined, name + "Paths", undefined, unionType);
})
}
var source = ts.createSourceFile("d.ts", "", ts.ScriptTarget.ES2015);
source = ts.updateSourceFileNode(source, generateUnions(result));
var printer = ts.createPrinter({ });
let r = printer.printFile(source);
fs.writeFileSync("union.ts", r);
这篇关于TypeScrip:如何根据数据结构生成和打印AST的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文