-
-
Notifications
You must be signed in to change notification settings - Fork 561
/
customTransformers.ts
78 lines (63 loc) 路 2.32 KB
/
customTransformers.ts
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
import { BuilderProgram, CustomTransformers, Program, TypeChecker } from 'typescript';
import type { CustomTransformerFactories, TransformerStage, TransformerFactory } from '../types';
/**
* Merges all received custom transformer definitions into a single CustomTransformers object
*/
export function mergeTransformers(
builder: BuilderProgram,
...input: Array<CustomTransformerFactories | undefined>
): CustomTransformers {
// List of all transformer stages
const transformerTypes: TransformerStage[] = ['after', 'afterDeclarations', 'before'];
const accumulator: Required<CustomTransformers> = {
after: [],
afterDeclarations: [],
before: []
};
let program: Program;
let typeChecker: TypeChecker;
input.forEach((transformers) => {
if (!transformers) {
// Skip empty arguments lists
return;
}
transformerTypes.forEach((stage) => {
getTransformers<typeof stage>(transformers[stage]).forEach((transformer) => {
if (!transformer) {
// Skip empty
return;
}
if ('type' in transformer) {
if (typeof transformer.factory === 'function') {
// Allow custom factories to grab the extra information required
program = program || builder.getProgram();
typeChecker = typeChecker || program.getTypeChecker();
let factory: ReturnType<typeof transformer.factory>;
if (transformer.type === 'program') {
program = program || builder.getProgram();
factory = transformer.factory(program);
} else {
program = program || builder.getProgram();
typeChecker = typeChecker || program.getTypeChecker();
factory = transformer.factory(typeChecker);
}
// Forward the requested reference to the custom transformer factory
if (factory) {
accumulator[stage].push(factory as any);
}
}
} else {
// Add normal transformer factories as is
accumulator[stage].push(transformer as any);
}
});
});
});
return accumulator;
}
function getTransformers<T extends TransformerStage>(
transformers?: Array<TransformerFactory<T>>
): Array<TransformerFactory<T>> {
return transformers || [];
}
export default mergeTransformers;