wip: fix framework props for reexported components

Author: pcattori

packages/react-router-dev/vite/with-props.ts

@@ -22,6 +22,72 @@ export const decorateComponentExportsWithProps = (
     return uid;
   }
 
+  /**
+   * Rewrite any re-exports for named components (`default Component`, `HydrateFallback`, `ErrorBoundary`)
+   * into `export const <name> = <expr>` form in preparation for adding props HOCs in the next traversal.
+   *
+   * Case 1: `export { name, ... }` or `export { value as name, ... }`
+   * -> Rename `name` to `uid` where `uid` is a new unique identifier
+   * -> Insert `export const name = uid`
+   *
+   * Case 2: `export { name1, value as name 2, ... } from "source"`
+   * -> Insert `import { name as uid }` where `uid` is a new unique identifier
+   * -> Insert `export const name = uid`
+   */
+  traverse(ast, {
+    ExportNamedDeclaration(path) {
+      if (path.node.declaration) return;
+      const { source } = path.node;
+
+      const exports: Array<{
+        uid: Babel.Identifier;
+        local: Babel.Identifier;
+        exported: Babel.Identifier;
+      }> = [];
+      for (const specifier of path.get("specifiers")) {
+        if (specifier.isExportSpecifier()) {
+          const { local, exported } = specifier.node;
+          const { name } = local;
+          if (!t.isIdentifier(exported)) continue;
+          const uid = path.scope.generateUidIdentifier(`_${name}`);
+          if (exported.name === "default" || isNamedComponentExport(name)) {
+            exports.push({ uid, local, exported });
+            specifier.remove();
+          }
+        }
+      }
+      if (exports.length === 0) return;
+
+      if (source != null) {
+        // `import { local as uid } from "source"`
+        path.insertAfter([
+          t.importDeclaration(
+            exports.map(({ local, uid }) => t.importSpecifier(uid, local)),
+            source,
+          ),
+        ]);
+      } else {
+        for (const { local, uid } of exports) {
+          path.scope.getBinding(local.name)?.scope.rename(uid.name);
+        }
+      }
+
+      // `export const exported = uid`
+      path.insertAfter(
+        exports.map(({ uid, exported }) => {
+          if (exported.name === "default") {
+            return t.exportDefaultDeclaration(uid);
+          }
+          return t.exportNamedDeclaration(
+            t.variableDeclaration("const", [
+              t.variableDeclarator(exported, uid),
+            ]),
+          );
+        }),
+      );
+    },
+  });
+
   traverse(ast, {
     ExportDeclaration(path) {
       if (path.isExportDefaultDeclaration()) {