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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
var __create = Object.create;
var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __getProtoOf = Object.getPrototypeOf;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
for (var name in all)
__defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
if (from && typeof from === "object" || typeof from === "function") {
for (let key of __getOwnPropNames(from))
if (!__hasOwnProp.call(to, key) && key !== except)
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
}
return to;
};
var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
// If the importer is in node compatibility mode or this is not an ESM
// file that has been converted to a CommonJS file using a Babel-
// compatible transform (i.e. "__esModule" has not been set), then set
// "default" to the CommonJS "module.exports" for node compatibility.
isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
mod
));
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
// src/index.ts
var src_exports = {};
__export(src_exports, {
default: () => vitePluginRequireTransform
});
module.exports = __toCommonJS(src_exports);
var parser = __toESM(require("@babel/parser"));
var import_traverse = __toESM(require("@babel/traverse"));
var import_generator = __toESM(require("@babel/generator"));
var t = __toESM(require("@babel/types"));
var traverse = import_traverse.default.default || import_traverse.default;
var generate = import_generator.default.default || import_generator.default;
function vitePluginRequireTransform(params = {}) {
const {
fileRegex = /.ts$|.tsx$/,
importPrefix: prefix = "_vite_plugin_require_transform_",
importPathHandler = (path) => path.replace(/(.*\/)*([^.]+).*/ig, "$2").replace(/-/g, "_")
} = params;
return {
name: prefix,
async transform(code, id) {
if (!fileRegex.test(id)) {
return { code, map: null };
}
const importMap = /* @__PURE__ */ new Map();
const plugins = [];
const ast = parser.parse(code, {
sourceType: "module",
plugins,
sourceFilename: id
});
const declaredVariables = {};
traverse(ast, {
enter(path) {
var _a, _b, _c, _d;
const reportError = (message) => {
var _a2, _b2;
const loc = (_b2 = (_a2 = path.parentPath) == null ? void 0 : _a2.node.loc) == null ? void 0 : _b2.start;
console.error(message + " in " + id + (loc ? ":" + loc.line + ":" + loc.column : ""));
};
if (((_a = path.parentPath) == null ? void 0 : _a.node) && t.isVariableDeclarator(path.parentPath.node)) {
const name = path.parentPath.node.id.name;
if (!declaredVariables[name]) {
declaredVariables[name] = path.parentPath.node;
}
}
if (path.isIdentifier({ name: "require" }) && t.isCallExpression((_b = path == null ? void 0 : path.parentPath) == null ? void 0 : _b.node)) {
const argument = path.parentPath.node.arguments[0];
let requirePath = void 0;
if (t.isTemplateLiteral(argument)) {
const tl = argument;
let templateElementValue = "";
for (let i = 0; i < tl.quasis.length; i++) {
const element = tl.quasis[i];
const expression = tl.expressions[i];
if (expression === void 0) {
continue;
}
if (t.isIdentifier(expression)) {
const identifier2 = expression;
const variableValue = (_c = declaredVariables[identifier2.name]) == null ? void 0 : _c.init;
if (variableValue === void 0 || variableValue === null) {
reportError(`Unknown variable for template value: "${identifier2.name}"`);
continue;
}
if (t.isStringLiteral(variableValue)) {
const sl = variableValue;
templateElementValue += element.value.raw;
templateElementValue += variableValue.value;
} else {
reportError(`Unknown type of template value: "${variableValue.type}" for "${identifier2.name}"`);
}
} else {
reportError(`Unknown type of template expression: "${expression.type}"`);
}
}
requirePath = templateElementValue;
} else if (t.isStringLiteral(argument)) {
const sl = argument;
requirePath = sl.value;
} else {
reportError(`Unknown type of require argument: "${argument.type}"`);
return;
}
const nodes = (_d = importMap.get(requirePath)) != null ? _d : [];
nodes.push(path.parentPath);
importMap.set(requirePath, nodes);
}
}
});
let importIndex = 0;
for (const [requirePath, nodes] of importMap) {
const importVariableName = prefix + importPathHandler(requirePath) + "_" + importIndex++;
const identifier2 = t.identifier(importVariableName);
const firstNode = nodes[0];
const importNamespaceSpecifier2 = t.importNamespaceSpecifier(identifier2);
const importDeclaration2 = t.importDeclaration([importNamespaceSpecifier2], t.stringLiteral(requirePath));
importDeclaration2.loc = firstNode.node.loc;
const rootNode = firstNode.findParent((path) => {
var _a;
return ((_a = path.parentPath) == null ? void 0 : _a.isProgram()) || false;
});
rootNode == null ? void 0 : rootNode.insertBefore(importDeclaration2);
const identifierDefault = t.memberExpression(identifier2, t.identifier("default"));
const identifierDefaultOrIdentifier = t.logicalExpression("||", identifierDefault, identifier2);
nodes.forEach((node) => {
node.replaceWith(identifierDefaultOrIdentifier);
});
}
const output = generate(ast, { sourceMaps: true });
return { code: output.code, map: output.map };
}
};
}