dinou/dinou/rollup/rollup-plugins/rollup-plugin-server-functions.js

Dinou is a modern React 19 framework with React Server Components, Server Functions, and streaming SSR.

// rollup-plugin-server-functions.js
const path = require("path");
const parser = require("@babel/parser");
const traverse = require("@babel/traverse").default;
const manifestGeneratorPlugin = require("./manifest-generator-plugin");

function parseExports(code) {
  const ast = parser.parse(code, {
    sourceType: "module",
    plugins: ["jsx", "typescript"],
  });
  const exports = new Set();

  traverse(ast, {
    ExportDefaultDeclaration() {
      exports.add("default");
    },
    ExportNamedDeclaration(p) {
      if (p.node.declaration) {
        if (p.node.declaration.type === "FunctionDeclaration") {
          exports.add(p.node.declaration.id.name);
        } else if (p.node.declaration.type === "VariableDeclaration") {
          p.node.declaration.declarations.forEach((d) => {
            if (d.id.type === "Identifier") {
              exports.add(d.id.name);
            }
          });
        }
      } else if (p.node.specifiers) {
        p.node.specifiers.forEach((s) => {
          if (s.type === "ExportSpecifier") {
            exports.add(s.exported.name);
          }
        });
      }
    },
  });

  return Array.from(exports);
}

function serverFunctionsPlugin() {
  return {
    name: "server-functions-proxy",
    transform(code, id) {
      if (!code.trim().startsWith('"use server"')) return null;

      const exports = parseExports(code);
      if (exports.length === 0) return null;

      const fileUrl = `file:///${path.relative(process.cwd(), id)}`;

      // Generamos un módulo que exporta proxies en lugar del código real
      let proxyCode = `
        import { createServerFunctionProxy } from "/__SERVER_FUNCTION_PROXY__";
      `;

      for (const exp of exports) {
        const key =
          exp === "default" ? `${fileUrl}#default` : `${fileUrl}#${exp}`;
        if (exp === "default") {
          proxyCode += `export default createServerFunctionProxy(${JSON.stringify(
            key
          )});\n`;
        } else {
          proxyCode += `export const ${exp} = createServerFunctionProxy(${JSON.stringify(
            key
          )});\n`;
        }
      }

      return {
        code: proxyCode,
        map: null,
      };
    },
    // 🪄 After manifest exists, replace the placeholder with the final URL
    generateBundle(options, bundle) {
      const manifest = manifestGeneratorPlugin.manifestData;
      const hashedPath =
        "/" + (manifest["serverFunctionProxy.js"] || "serverFunctionProxy.js");

      for (const file of Object.keys(bundle)) {
        const chunk = bundle[file];
        if (chunk.type === "asset" || !chunk.code) continue;
        if (chunk.code.includes("/__SERVER_FUNCTION_PROXY__")) {
          chunk.code = chunk.code.replace(
            /\/__SERVER_FUNCTION_PROXY__/g,
            hashedPath
          );
        }
      }
    },
  };
}

module.exports = serverFunctionsPlugin;