-
-
Notifications
You must be signed in to change notification settings - Fork 185
Expand file tree
/
Copy pathindex.ts
More file actions
45 lines (37 loc) · 1.86 KB
/
index.ts
File metadata and controls
45 lines (37 loc) · 1.86 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
import * as ts from "typescript";
import * as lua from "../LuaAST";
import { getOrUpdate } from "../utils";
import { ObjectVisitor, TransformationContext, VisitorMap, Visitors } from "./context";
import { standardVisitors } from "./visitors";
import { usingTransformer } from "./pre-transformers/using-transformer";
export function createVisitorMap(customVisitors: Visitors[]): VisitorMap {
const objectVisitorMap: Map<ts.SyntaxKind, Array<ObjectVisitor<ts.Node>>> = new Map();
for (const visitors of [standardVisitors, ...customVisitors]) {
const priority = visitors === standardVisitors ? -Infinity : 0;
for (const [syntaxKindKey, visitor] of Object.entries(visitors)) {
if (!visitor) continue;
const syntaxKind = Number(syntaxKindKey) as ts.SyntaxKind;
const nodeVisitors = getOrUpdate(objectVisitorMap, syntaxKind, () => []);
const objectVisitor: ObjectVisitor<any> =
typeof visitor === "function" ? { transform: visitor, priority } : visitor;
nodeVisitors.push(objectVisitor);
}
}
const result: VisitorMap = new Map();
for (const [kind, nodeVisitors] of objectVisitorMap) {
result.set(
kind,
nodeVisitors.sort((a, b) => (a.priority ?? 0) - (b.priority ?? 0)).map(visitor => visitor.transform)
);
}
return result;
}
export function transformSourceFile(program: ts.Program, sourceFile: ts.SourceFile, visitorMap: VisitorMap) {
const context = new TransformationContext(program, sourceFile, visitorMap);
// TS -> TS pre-transformation
const preTransformers = [usingTransformer(context)];
const result = ts.transform(sourceFile, preTransformers);
// TS -> Lua transformation
const [file] = context.transformNode(result.transformed[0]) as [lua.File];
return { file, diagnostics: context.diagnostics };
}