typegen: register lookup from route ID to route module

Author: pcattori

packages/react-router-dev/typegen/generate.ts

@@ -105,13 +105,16 @@ export function generateRoutes(ctx: Context): Array<VirtualFile> {
           interface Register {
             pages: Pages
             routeFiles: RouteFiles
+            routeModules: RouteModules
           }
         }
       ` +
       "\n\n" +
       Babel.generate(pagesType(allPages)).code +
       "\n\n" +
-      Babel.generate(routeFilesType({ fileToRoutes, routeToPages })).code,
+      Babel.generate(routeFilesType({ fileToRoutes, routeToPages })).code +
+      "\n\n" +
+      Babel.generate(routeModulesType(ctx)).code,
   };
 
   // **/+types/*.ts
@@ -193,6 +196,29 @@ function routeFilesType({
   );
 }
 
+function routeModulesType(ctx: Context) {
+  return t.tsTypeAliasDeclaration(
+    t.identifier("RouteModules"),
+    null,
+    t.tsTypeLiteral(
+      Object.values(ctx.config.routes).map((route) =>
+        t.tsPropertySignature(
+          t.stringLiteral(route.id),
+          t.tsTypeAnnotation(
+            t.tsTypeQuery(
+              t.tsImportType(
+                t.stringLiteral(
+                  `./${Path.relative(ctx.rootDirectory, ctx.config.appDirectory)}/${route.file}`,
+                ),
+              ),
+            ),
+          ),
+        ),
+      ),
+    ),
+  );
+}
+
 function isInAppDirectory(ctx: Context, routeFile: string): boolean {
   const path = Path.resolve(ctx.config.appDirectory, routeFile);
   return path.startsWith(ctx.config.appDirectory);