export const REACT_RE = /\/react\/|\/react-dom\/|\/@lazarv\/rsc\//;

export const createTreeshake = (config) => ({
  moduleSideEffects: (id, external) => {
    if (
      id.startsWith("virtual:rsc:react-client-reference:") ||
      id.startsWith("virtual:rsc:react-server-reference:") ||
      id.startsWith("virtual:ssr:react-client-reference:") ||
      id.startsWith("virtual:ssr:react-server-reference:") ||
      id.includes("@lazarv/react-server/dist/manifest-registry") ||
      id.includes("@lazarv/react-server/dist/client/manifest-registry") ||
      id.includes("server/manifest-registry") ||
      id.includes("server/client/manifest-registry")
    ) {
      return true;
    }
    if (REACT_RE.test(id)) {
      return true;
    }
    if (id.includes("/web-streams-polyfill/")) {
      return true;
    } else if (
      typeof config.build?.rollupOptions?.treeshake?.moduleSideEffects ===
      "function"
    ) {
      return config.build.rollupOptions.treeshake.moduleSideEffects(
        id,
        external
      );
    } else if (
      Array.isArray(config.build?.rollupOptions?.treeshake?.moduleSideEffects)
    ) {
      return config.build.rollupOptions.treeshake.moduleSideEffects.some(
        (pattern) =>
          (typeof pattern === "string" && id.includes(pattern)) ||
          (pattern instanceof RegExp && pattern.test(id))
      );
    } else if (
      config.build?.rollupOptions?.treeshake?.moduleSideEffects === true
    ) {
      return true;
    }
    return false;
  },
  ...config.build?.rollupOptions?.treeshake,
});