FilesExpand file tree

master

/

index.ts

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

45 lines (37 loc) · 1.86 KB

master

/

index.ts

Copy path

Top

File metadata and controls

• Code
• Blame

45 lines (37 loc) · 1.86 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

}