Spaces:
Sleeping
Sleeping
| ; | |
| exports.__esModule = true; | |
| exports.PossibleGlobalObjects = void 0; | |
| exports.createUtilsGetter = createUtilsGetter; | |
| exports.getImportSource = getImportSource; | |
| exports.getRequireSource = getRequireSource; | |
| exports.has = has; | |
| exports.intersection = intersection; | |
| exports.resolveKey = resolveKey; | |
| exports.resolveSource = resolveSource; | |
| var _babel = _interopRequireWildcard(require("@babel/core")); | |
| function _interopRequireWildcard(e, t) { if ("function" == typeof WeakMap) var r = new WeakMap(), n = new WeakMap(); return (_interopRequireWildcard = function (e, t) { if (!t && e && e.__esModule) return e; var o, i, f = { __proto__: null, default: e }; if (null === e || "object" != typeof e && "function" != typeof e) return f; if (o = t ? n : r) { if (o.has(e)) return o.get(e); o.set(e, f); } for (const t in e) "default" !== t && {}.hasOwnProperty.call(e, t) && ((i = (o = Object.defineProperty) && Object.getOwnPropertyDescriptor(e, t)) && (i.get || i.set) ? o(f, t, i) : f[t] = e[t]); return f; })(e, t); } | |
| const { | |
| types: t, | |
| template: template | |
| } = _babel.default || _babel; | |
| const PossibleGlobalObjects = exports.PossibleGlobalObjects = new Set(["global", "globalThis", "self", "window"]); | |
| function intersection(a, b) { | |
| const result = new Set(); | |
| a.forEach(v => b.has(v) && result.add(v)); | |
| return result; | |
| } | |
| function has(object, key) { | |
| return Object.prototype.hasOwnProperty.call(object, key); | |
| } | |
| function resolve(path, seen = new Set()) { | |
| if (seen.has(path)) return; | |
| seen.add(path); | |
| if (path.isVariableDeclarator()) { | |
| if (path.get("id").isIdentifier()) { | |
| return resolve(path.get("init"), seen); | |
| } | |
| } else if (path.isReferencedIdentifier()) { | |
| const binding = path.scope.getBinding(path.node.name); | |
| if (!binding) return path; | |
| if (!binding.constant) return; | |
| return resolve(binding.path, seen); | |
| } | |
| return path; | |
| } | |
| function resolveId(path) { | |
| if (path.isIdentifier() && !path.scope.hasBinding(path.node.name, /* noGlobals */true)) { | |
| return path.node.name; | |
| } | |
| // globalThis.Object / window.Array / self.Map / global.Set -> resolve to | |
| // the property name, because accessing a built-in through a global object | |
| // reference is equivalent to accessing it directly. | |
| if (path.isMemberExpression() && !path.node.computed) { | |
| const object = path.get("object"); | |
| const property = path.get("property"); | |
| if (object.isIdentifier() && !object.scope.hasBinding(object.node.name, /* noGlobals */true) && PossibleGlobalObjects.has(object.node.name) && property.isIdentifier()) { | |
| return property.node.name; | |
| } | |
| } | |
| const resolved = resolve(path); | |
| if (resolved != null && resolved.isIdentifier()) { | |
| return resolved.node.name; | |
| } | |
| } | |
| function resolveKey(path, computed = false) { | |
| const { | |
| scope | |
| } = path; | |
| if (path.isStringLiteral()) return path.node.value; | |
| const isIdentifier = path.isIdentifier(); | |
| if (isIdentifier && !(computed || path.parent.computed)) { | |
| return path.node.name; | |
| } | |
| if (computed && path.isMemberExpression() && path.get("object").isIdentifier({ | |
| name: "Symbol" | |
| }) && !scope.hasBinding("Symbol", /* noGlobals */true)) { | |
| const sym = resolveKey(path.get("property"), path.node.computed); | |
| if (sym) return "Symbol." + sym; | |
| } | |
| if (isIdentifier ? scope.hasBinding(path.node.name, /* noGlobals */true) : path.isPure()) { | |
| const { | |
| value | |
| } = path.evaluate(); | |
| if (typeof value === "string") return value; | |
| } | |
| } | |
| function resolveInstance(obj, seen) { | |
| const source = resolveSource(obj, seen); | |
| return source.placement === "prototype" ? source.id : null; | |
| } | |
| function resolveSource(obj, seen) { | |
| if (seen.has(obj)) { | |
| return { | |
| id: null, | |
| placement: null | |
| }; | |
| } | |
| seen.add(obj); | |
| if (obj.isMemberExpression() && obj.get("property").isIdentifier({ | |
| name: "prototype" | |
| })) { | |
| const id = resolveId(obj.get("object")); | |
| if (id) { | |
| return { | |
| id, | |
| placement: "prototype" | |
| }; | |
| } | |
| return { | |
| id: null, | |
| placement: null | |
| }; | |
| } | |
| const id = resolveId(obj); | |
| if (id) { | |
| return { | |
| id, | |
| placement: "static" | |
| }; | |
| } | |
| const path = resolve(obj); | |
| switch (path == null ? void 0 : path.type) { | |
| case "NullLiteral": | |
| return { | |
| id: null, | |
| placement: null | |
| }; | |
| case "RegExpLiteral": | |
| return { | |
| id: "RegExp", | |
| placement: "prototype" | |
| }; | |
| case "StringLiteral": | |
| case "TemplateLiteral": | |
| return { | |
| id: "String", | |
| placement: "prototype" | |
| }; | |
| case "NumericLiteral": | |
| return { | |
| id: "Number", | |
| placement: "prototype" | |
| }; | |
| case "BooleanLiteral": | |
| return { | |
| id: "Boolean", | |
| placement: "prototype" | |
| }; | |
| case "BigIntLiteral": | |
| return { | |
| id: "BigInt", | |
| placement: "prototype" | |
| }; | |
| case "ObjectExpression": | |
| return { | |
| id: "Object", | |
| placement: "prototype" | |
| }; | |
| case "ArrayExpression": | |
| return { | |
| id: "Array", | |
| placement: "prototype" | |
| }; | |
| case "FunctionExpression": | |
| case "ArrowFunctionExpression": | |
| case "ClassExpression": | |
| return { | |
| id: "Function", | |
| placement: "prototype" | |
| }; | |
| // new Constructor() -> resolve the constructor name | |
| case "NewExpression": | |
| { | |
| const calleeId = resolveId(path.get("callee")); | |
| if (calleeId) return { | |
| id: calleeId, | |
| placement: "prototype" | |
| }; | |
| return { | |
| id: null, | |
| placement: null | |
| }; | |
| } | |
| // Unary expressions -> result type depends on operator | |
| case "UnaryExpression": | |
| { | |
| const { | |
| operator | |
| } = path.node; | |
| if (operator === "typeof") return { | |
| id: "String", | |
| placement: "prototype" | |
| }; | |
| if (operator === "!" || operator === "delete") return { | |
| id: "Boolean", | |
| placement: "prototype" | |
| }; | |
| // Unary + always produces Number (throws on BigInt) | |
| if (operator === "+") return { | |
| id: "Number", | |
| placement: "prototype" | |
| }; | |
| // Unary - and ~ can produce Number or BigInt depending on operand | |
| if (operator === "-" || operator === "~") { | |
| const arg = resolveInstance(path.get("argument"), seen); | |
| if (arg === "BigInt") return { | |
| id: "BigInt", | |
| placement: "prototype" | |
| }; | |
| if (arg !== null) return { | |
| id: "Number", | |
| placement: "prototype" | |
| }; | |
| return { | |
| id: null, | |
| placement: null | |
| }; | |
| } | |
| return { | |
| id: null, | |
| placement: null | |
| }; | |
| } | |
| // ++i, i++ produce Number or BigInt depending on the argument | |
| case "UpdateExpression": | |
| { | |
| const arg = resolveInstance(path.get("argument"), seen); | |
| if (arg === "BigInt") return { | |
| id: "BigInt", | |
| placement: "prototype" | |
| }; | |
| if (arg !== null) return { | |
| id: "Number", | |
| placement: "prototype" | |
| }; | |
| return { | |
| id: null, | |
| placement: null | |
| }; | |
| } | |
| // Binary expressions -> result type depends on operator | |
| case "BinaryExpression": | |
| { | |
| const { | |
| operator | |
| } = path.node; | |
| if (operator === "==" || operator === "!=" || operator === "===" || operator === "!==" || operator === "<" || operator === ">" || operator === "<=" || operator === ">=" || operator === "instanceof" || operator === "in") { | |
| return { | |
| id: "Boolean", | |
| placement: "prototype" | |
| }; | |
| } | |
| // >>> always produces Number | |
| if (operator === ">>>") { | |
| return { | |
| id: "Number", | |
| placement: "prototype" | |
| }; | |
| } | |
| // Arithmetic and bitwise operators can produce Number or BigInt | |
| if (operator === "-" || operator === "*" || operator === "/" || operator === "%" || operator === "**" || operator === "&" || operator === "|" || operator === "^" || operator === "<<" || operator === ">>") { | |
| const left = resolveInstance(path.get("left"), seen); | |
| const right = resolveInstance(path.get("right"), seen); | |
| if (left === "BigInt" && right === "BigInt") { | |
| return { | |
| id: "BigInt", | |
| placement: "prototype" | |
| }; | |
| } | |
| if (left !== null && right !== null) { | |
| return { | |
| id: "Number", | |
| placement: "prototype" | |
| }; | |
| } | |
| return { | |
| id: null, | |
| placement: null | |
| }; | |
| } | |
| // + depends on operand types: string wins, otherwise number or bigint | |
| if (operator === "+") { | |
| const left = resolveInstance(path.get("left"), seen); | |
| const right = resolveInstance(path.get("right"), seen); | |
| if (left === "String" || right === "String") { | |
| return { | |
| id: "String", | |
| placement: "prototype" | |
| }; | |
| } | |
| if (left === "Number" && right === "Number") { | |
| return { | |
| id: "Number", | |
| placement: "prototype" | |
| }; | |
| } | |
| if (left === "BigInt" && right === "BigInt") { | |
| return { | |
| id: "BigInt", | |
| placement: "prototype" | |
| }; | |
| } | |
| } | |
| return { | |
| id: null, | |
| placement: null | |
| }; | |
| } | |
| // (a, b, c) -> the result is the last expression | |
| case "SequenceExpression": | |
| { | |
| const expressions = path.get("expressions"); | |
| return resolveSource(expressions[expressions.length - 1], seen); | |
| } | |
| // a = b -> the result is the right side | |
| case "AssignmentExpression": | |
| { | |
| if (path.node.operator === "=") { | |
| return resolveSource(path.get("right"), seen); | |
| } | |
| return { | |
| id: null, | |
| placement: null | |
| }; | |
| } | |
| // a ? b : c -> if both branches resolve to the same type, use it | |
| case "ConditionalExpression": | |
| { | |
| const consequent = resolveSource(path.get("consequent"), seen); | |
| const alternate = resolveSource(path.get("alternate"), seen); | |
| if (consequent.id && consequent.id === alternate.id) { | |
| return consequent; | |
| } | |
| return { | |
| id: null, | |
| placement: null | |
| }; | |
| } | |
| // (expr) -> unwrap parenthesized expressions | |
| case "ParenthesizedExpression": | |
| return resolveSource(path.get("expression"), seen); | |
| // TypeScript / Flow type wrappers -> unwrap to the inner expression | |
| case "TSAsExpression": | |
| case "TSSatisfiesExpression": | |
| case "TSNonNullExpression": | |
| case "TSInstantiationExpression": | |
| case "TSTypeAssertion": | |
| case "TypeCastExpression": | |
| return resolveSource(path.get("expression"), seen); | |
| } | |
| return { | |
| id: null, | |
| placement: null | |
| }; | |
| } | |
| function getImportSource({ | |
| node | |
| }) { | |
| if (node.specifiers.length === 0) return node.source.value; | |
| } | |
| function getRequireSource({ | |
| node | |
| }) { | |
| if (!t.isExpressionStatement(node)) return; | |
| const { | |
| expression | |
| } = node; | |
| if (t.isCallExpression(expression) && t.isIdentifier(expression.callee) && expression.callee.name === "require" && expression.arguments.length === 1 && t.isStringLiteral(expression.arguments[0])) { | |
| return expression.arguments[0].value; | |
| } | |
| } | |
| function hoist(node) { | |
| // @ts-expect-error | |
| node._blockHoist = 3; | |
| return node; | |
| } | |
| function createUtilsGetter(cache) { | |
| return path => { | |
| const prog = path.findParent(p => p.isProgram()); | |
| return { | |
| injectGlobalImport(url, moduleName) { | |
| cache.storeAnonymous(prog, url, moduleName, (isScript, source) => { | |
| return isScript ? template.statement.ast`require(${source})` : t.importDeclaration([], source); | |
| }); | |
| }, | |
| injectNamedImport(url, name, hint = name, moduleName) { | |
| return cache.storeNamed(prog, url, name, moduleName, (isScript, source, name) => { | |
| const id = prog.scope.generateUidIdentifier(hint); | |
| return { | |
| node: isScript ? hoist(template.statement.ast` | |
| var ${id} = require(${source}).${name} | |
| `) : t.importDeclaration([t.importSpecifier(id, name)], source), | |
| name: id.name | |
| }; | |
| }); | |
| }, | |
| injectDefaultImport(url, hint = url, moduleName) { | |
| return cache.storeNamed(prog, url, "default", moduleName, (isScript, source) => { | |
| const id = prog.scope.generateUidIdentifier(hint); | |
| return { | |
| node: isScript ? hoist(template.statement.ast`var ${id} = require(${source})`) : t.importDeclaration([t.importDefaultSpecifier(id)], source), | |
| name: id.name | |
| }; | |
| }); | |
| } | |
| }; | |
| }; | |
| } |