vite-plugin-enter-dev 0.0.12 → 0.0.13

package/dist/configure.js

@@ -31,12 +31,15 @@ function configureVite(source) {
     if (!types.isIdentifier(config.callee) || !defineNames.includes(config.callee.name) || config.arguments.length !== 1) throw new Error("Unsupported Vite config wrapper");
     config = config.arguments[0];
   }
+  let setup = [];
   if (types.isArrowFunctionExpression(config) || types.isFunctionExpression(config)) {
     if (types.isBlockStatement(config.body)) {
-      if (config.body.body.length !== 1 || !types.isReturnStatement(config.body.body[0])) {
+      const last = config.body.body.at(-1);
+      if (!types.isReturnStatement(last)) {
         throw new Error("Unsupported Vite config: expected a single unconditional return");
       }
-      config = config.body.body[0].argument ?? void 0;
+      setup = config.body.body.slice(0, -1);
+      config = last.argument ?? void 0;
     } else {
       config = config.body;
     }
@@ -54,10 +57,12 @@ function configureVite(source) {
     config.properties.push(plugins);
     changed = true;
   }
-  if (!types.isObjectProperty(plugins) || !types.isArrayExpression(plugins.value)) {
+  if (!types.isObjectProperty(plugins)) throw new Error("Unsupported plugins property");
+  const pluginArrays = resolvePluginArrays(plugins.value, setup);
+  if (!pluginArrays.length) {
     throw new Error("Unsupported Vite config: expected a literal plugins array");
   }
-  const calls = plugins.value.elements.map((node) => types.isSpreadElement(node) ? node.argument : node);
+  const calls = pluginArrays.flatMap((array) => array.elements).map((node) => types.isSpreadElement(node) ? node.argument : node);
   if (calls.some((node) => !types.isCallExpression(node) && node !== null && !types.isBooleanLiteral(node))) {
     throw new Error("Unsupported Vite config: dynamic plugin list");
   }
@@ -77,7 +82,7 @@ function configureVite(source) {
     let call = matching[0];
     if (!call) {
       call = types.callExpression(types.identifier(local), []);
-      plugins.value.elements.push(types.spreadElement(call));
+      pluginArrays[0].elements.push(types.spreadElement(call));
       changed = true;
     }
     if (name !== "enterDevPlugin" || !hostReact || !types.isCallExpression(call)) continue;
@@ -119,5 +124,39 @@ function configureViteFile(path, check = false) {
   }
   return true;
 }
+function resolvePluginArrays(value, setup) {
+  if (setup.length === 0 && types.isArrayExpression(value)) return [value];
+  if (types.isTSAsExpression(value) || types.isTSSatisfiesExpression(value)) value = value.expression;
+  if (types.isCallExpression(value) && types.isMemberExpression(value.callee) && !value.callee.computed && types.isIdentifier(value.callee.property, { name: "filter" }) && value.arguments.length === 1 && types.isIdentifier(value.arguments[0], { name: "Boolean" })) value = value.callee.object;
+  if (!types.isIdentifier(value)) throw new Error("Unsupported dynamic plugins reference");
+  const name = value.name;
+  const first = setup[0];
+  if (!types.isVariableDeclaration(first, { kind: "const" }) || first.declarations.length !== 1) {
+    throw new Error("Unsupported plugins initializer");
+  }
+  const declaration = first.declarations[0];
+  if (!types.isIdentifier(declaration.id, { name }) || !types.isArrayExpression(declaration.init)) {
+    throw new Error("Unsupported plugins initializer");
+  }
+  const arrays = [declaration.init];
+  const visit = (statement) => {
+    if (types.isIfStatement(statement)) {
+      visit(statement.consequent);
+      if (statement.alternate) visit(statement.alternate);
+      return;
+    }
+    if (types.isBlockStatement(statement)) {
+      statement.body.forEach(visit);
+      return;
+    }
+    const call = types.isExpressionStatement(statement) ? statement.expression : void 0;
+    if (!types.isCallExpression(call) || !types.isMemberExpression(call.callee) || call.callee.computed || !types.isIdentifier(call.callee.object, { name }) || !types.isIdentifier(call.callee.property, { name: "push" }) || call.arguments.some((argument) => types.isJSXNamespacedName(argument) || types.isArgumentPlaceholder(argument))) {
+      throw new Error("Unsupported plugins mutation");
+    }
+    arrays.push(types.arrayExpression(call.arguments));
+  };
+  setup.slice(1).forEach(visit);
+  return arrays;
+}
 
 export { configureVite, configureViteFile };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vite-plugin-enter-dev",
-  "version": "0.0.12",
+  "version": "0.0.13",
   "type": "module",
   "private": false,
   "description": "A Vite plugin for development enhancements",