one/dist/esm/babel-plugins/remove-server-code.js

One is a new React Framework that makes Vite serve both native and web.

import * as t from "@babel/types";
import { deadCodeElimination, findReferencedIdentifiers } from "babel-dead-code-elimination";
const SERVER_EXPORTS = ["loader", "generateStaticParams"];
function removeServerCodePlugin(_, options) {
  const { routerRoot = "app" } = options;
  return {
    name: "one-remove-server-code",
    visitor: {
      Program: {
        enter(path, state) {
          const filename = state.filename;
          if (!filename || !new RegExp(`[/\\\\]${routerRoot}[/\\\\]`).test(filename) || filename.includes("node_modules"))
            return;
          const code = path.toString();
          if (/generateStaticParams|loader/.test(code))
            try {
              state.referenced = findReferencedIdentifiers(path.node);
            } catch (error) {
              console.warn(
                `[one/metro] Skipping tree shaking for ${filename} due to identifier analysis error:`,
                error instanceof Error ? error.message : String(error)
              );
            }
        },
        exit(path, state) {
          const filename = state.filename;
          if (!filename || !new RegExp(`[/\\\\]${routerRoot}[/\\\\]`).test(filename) || filename.includes("node_modules") || !state.referenced)
            return;
          const removedExports = /* @__PURE__ */ new Set();
          if (path.traverse({
            ExportNamedDeclaration(exportPath) {
              const declaration = exportPath.node.declaration;
              if (t.isFunctionDeclaration(declaration) && declaration.id) {
                const name = declaration.id.name;
                SERVER_EXPORTS.includes(name) && (removedExports.add(name), exportPath.remove());
              } else if (t.isVariableDeclaration(declaration))
                for (let i = declaration.declarations.length - 1; i >= 0; i--) {
                  const declarator = declaration.declarations[i];
                  if (t.isIdentifier(declarator.id)) {
                    const name = declarator.id.name;
                    SERVER_EXPORTS.includes(name) && (removedExports.add(name), declaration.declarations.length === 1 ? exportPath.remove() : declaration.declarations.splice(i, 1));
                  }
                }
            },
            // Also remove helper functions created by babel's async-to-generator transform
            // These are named _loader, _generateStaticParams, etc.
            FunctionDeclaration(funcPath) {
              if (!funcPath.node.id) return;
              const name = funcPath.node.id.name;
              for (const serverExport of SERVER_EXPORTS)
                if (name === `_${serverExport}`) {
                  let isAsyncHelper = !1;
                  funcPath.traverse({
                    Identifier(idPath) {
                      idPath.node.name === "_asyncToGenerator" && (isAsyncHelper = !0, idPath.stop());
                    }
                  }), isAsyncHelper && (removedExports.add(serverExport), funcPath.remove());
                }
            }
          }), removedExports.size === 0)
            return;
          try {
            deadCodeElimination(path.node, state.referenced);
          } catch (error) {
            console.warn(
              `[one/metro] Dead code elimination failed for ${filename}:`,
              error instanceof Error ? error.message : String(error)
            );
          }
          const stubs = [];
          removedExports.has("loader") && stubs.push(
            t.exportNamedDeclaration(
              t.functionDeclaration(
                t.identifier("loader"),
                [],
                t.blockStatement([t.returnStatement(t.stringLiteral("__vxrn__loader__"))])
              )
            )
          ), removedExports.has("generateStaticParams") && stubs.push(
            t.exportNamedDeclaration(
              t.functionDeclaration(
                t.identifier("generateStaticParams"),
                [],
                t.blockStatement([])
              )
            )
          );
          for (const stub of stubs)
            path.pushContainer("body", stub);
          console.info(
            ` \u{1F9F9} [one/metro] ${filename} removed ${removedExports.size} server-only exports`
          );
        }
      }
    }
  };
}
var remove_server_code_default = removeServerCodePlugin;
export {
  remove_server_code_default as default
};
//# sourceMappingURL=remove-server-code.js.map