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
79
80
81
82
83
84
85
86
87
88
89
|
import { applyReplacements } from "./replacements";
/**
* Slices a string until it hits a }, but keeping in mind JS comments,
* regex, template literals, comments, and matching {
*
* Used to extract function bodies without parsing the code.
*
* If you pass replace=true, it will run replacements on the code
*/
export function sliceSourceCode(
contents: string,
replace: boolean,
): { result: string; rest: string; usesThis: boolean } {
let bracketCount = 0;
let i = 0;
let result = "";
let usesThis = false;
while (contents.length) {
// TODO: template literal, regexp
// these are important because our replacement logic would replace intrinsics
// within these, when it should remain as the literal dollar.
// but this isn't used in the codebase
i = contents.match(/\/\*|\/\/|'|"|{|}|`/)?.index ?? contents.length;
const chunk = replace ? applyReplacements(contents.slice(0, i)) : contents.slice(0, i);
if (chunk.includes("this")) usesThis = true;
result += chunk;
contents = contents.slice(i);
if (!contents.length) break;
if (contents.startsWith("/*")) {
i = contents.slice(2).indexOf("*/") + 2;
} else if (contents.startsWith("//")) {
i = contents.slice(2).indexOf("\n") + 2;
} else if (contents.startsWith("'")) {
i = contents.slice(1).match(/(?<!\\)'/)!.index! + 2;
} else if (contents.startsWith('"')) {
i = contents.slice(1).match(/(?<!\\)"/)!.index! + 2;
} else if (contents.startsWith("`")) {
const { result: result2, rest } = sliceTemplateLiteralSourceCode(contents.slice(1), replace);
result += "`" + result2;
contents = rest;
continue;
} else if (contents.startsWith("{")) {
bracketCount++;
i = 1;
} else if (contents.startsWith("}")) {
bracketCount--;
if (bracketCount <= 0) {
result += "}";
contents = contents.slice(1);
break;
}
i = 1;
} else {
throw new Error("TODO");
}
result += contents.slice(0, i);
contents = contents.slice(i);
}
return { result, rest: contents, usesThis };
}
function sliceTemplateLiteralSourceCode(contents: string, replace: boolean) {
let i = 0;
let result = "";
let usesThis = false;
while (contents.length) {
i = contents.match(/`|\${/)!.index!;
result += contents.slice(0, i);
contents = contents.slice(i);
if (!contents.length) break;
if (contents.startsWith("`")) {
result += "`";
contents = contents.slice(1);
break;
} else if (contents.startsWith("$")) {
const { result: result2, rest, usesThis: usesThisVal } = sliceSourceCode(contents.slice(1), replace);
result += "$" + result2;
contents = rest;
usesThis ||= usesThisVal;
continue;
} else {
throw new Error("TODO");
}
}
return { result, rest: contents, usesThis };
}
|
