1 | "use strict";
|
2 |
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 | var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
10 | if (k2 === undefined) k2 = k;
|
11 | Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
|
12 | }) : (function(o, m, k, k2) {
|
13 | if (k2 === undefined) k2 = k;
|
14 | o[k2] = m[k];
|
15 | }));
|
16 | var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
17 | Object.defineProperty(o, "default", { enumerable: true, value: v });
|
18 | }) : function(o, v) {
|
19 | o["default"] = v;
|
20 | });
|
21 | var __importStar = (this && this.__importStar) || function (mod) {
|
22 | if (mod && mod.__esModule) return mod;
|
23 | var result = {};
|
24 | if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
|
25 | __setModuleDefault(result, mod);
|
26 | return result;
|
27 | };
|
28 | Object.defineProperty(exports, "__esModule", { value: true });
|
29 | exports.findTopLevelFunctions = exports.getPrefixFunctionsTransformer = void 0;
|
30 | const ts = __importStar(require("typescript"));
|
31 | const ast_utils_1 = require("../helpers/ast-utils");
|
32 | function getPrefixFunctionsTransformer() {
|
33 | return (context) => {
|
34 | const transformer = (sf) => {
|
35 | const topLevelFunctions = findTopLevelFunctions(sf);
|
36 | const visitor = (node) => {
|
37 |
|
38 | if (topLevelFunctions.has(node)) {
|
39 | const newNode = (0, ast_utils_1.addPureComment)(node);
|
40 |
|
41 | return ts.visitEachChild(newNode, visitor, context);
|
42 | }
|
43 |
|
44 | return ts.visitEachChild(node, visitor, context);
|
45 | };
|
46 | return ts.visitNode(sf, visitor);
|
47 | };
|
48 | return transformer;
|
49 | };
|
50 | }
|
51 | exports.getPrefixFunctionsTransformer = getPrefixFunctionsTransformer;
|
52 | function findTopLevelFunctions(parentNode) {
|
53 | const topLevelFunctions = new Set();
|
54 | function cb(node) {
|
55 |
|
56 |
|
57 |
|
58 |
|
59 |
|
60 |
|
61 | if (ts.isFunctionLike(node) ||
|
62 | ts.isClassLike(node) ||
|
63 | ts.isArrowFunction(node) ||
|
64 | ts.isMethodDeclaration(node)) {
|
65 | return;
|
66 | }
|
67 | let noPureComment = !(0, ast_utils_1.hasPureComment)(node);
|
68 | let innerNode = node;
|
69 | while (innerNode && ts.isParenthesizedExpression(innerNode)) {
|
70 | innerNode = innerNode.expression;
|
71 | noPureComment = noPureComment && !(0, ast_utils_1.hasPureComment)(innerNode);
|
72 | }
|
73 | if (!innerNode) {
|
74 | return;
|
75 | }
|
76 | if ((ts.isFunctionExpression(innerNode) || ts.isArrowFunction(innerNode)) &&
|
77 | ts.isParenthesizedExpression(node)) {
|
78 |
|
79 |
|
80 |
|
81 | return;
|
82 | }
|
83 | if (noPureComment) {
|
84 | if (ts.isNewExpression(innerNode)) {
|
85 | topLevelFunctions.add(node);
|
86 | }
|
87 | else if (ts.isCallExpression(innerNode)) {
|
88 | let expression = innerNode.expression;
|
89 | if (ts.isIdentifier(expression) && (0, ast_utils_1.getCleanHelperName)(expression.text)) {
|
90 | return;
|
91 | }
|
92 | while (expression && ts.isParenthesizedExpression(expression)) {
|
93 | expression = expression.expression;
|
94 | }
|
95 | if (expression) {
|
96 | if (ts.isFunctionExpression(expression)) {
|
97 |
|
98 |
|
99 | if (innerNode.arguments.length === 0) {
|
100 | topLevelFunctions.add(node);
|
101 | }
|
102 | }
|
103 | else {
|
104 | topLevelFunctions.add(node);
|
105 | }
|
106 | }
|
107 | }
|
108 | }
|
109 | ts.forEachChild(innerNode, cb);
|
110 | }
|
111 | ts.forEachChild(parentNode, cb);
|
112 | return topLevelFunctions;
|
113 | }
|
114 | exports.findTopLevelFunctions = findTopLevelFunctions;
|