@tanstack/router-plugin 1.124.2 → 1.125.1

package/dist/esm/core/router-hmr-plugin.js.map

@@ -1 +1 @@
-{"version":3,"file":"router-hmr-plugin.js","sources":["../../../src/core/router-hmr-plugin.ts"],"sourcesContent":["import { generateFromAst, logDiff, parseAst } from '@tanstack/router-utils'\nimport { routeHmrStatement } from './route-hmr-statement'\nimport { debug } from './utils'\nimport type { Config } from './config'\nimport type { UnpluginFactory } from 'unplugin'\n\n/**\n * This plugin adds HMR support for file routes.\n * It is only added to the composed plugin in dev when autoCodeSplitting is disabled, since the code splitting plugin\n * handles HMR for code-split routes itself.\n */\nexport const unpluginRouterHmrFactory: UnpluginFactory<\n  Partial<Config> | undefined\n> = () => {\n  return {\n    name: 'tanstack-router:hmr',\n    enforce: 'pre',\n    transform: {\n      filter: {\n        // this is necessary for webpack / rspack to avoid matching .html files\n        id: /\\.(m|c)?(j|t)sx?$/,\n        code: 'createFileRoute(',\n      },\n      handler(code, id) {\n        if (!globalThis.TSR_ROUTES_BY_ID_MAP?.has(id)) {\n          return null\n        }\n\n        if (debug) console.info('Adding HMR handling to route ', id)\n\n        const ast = parseAst({ code })\n        ast.program.body.push(routeHmrStatement)\n        const result = generateFromAst(ast, {\n          sourceMaps: true,\n          filename: id,\n          sourceFileName: id,\n        })\n        if (debug) {\n          logDiff(code, result.code)\n          console.log('Output:\\n', result.code + '\\n\\n')\n        }\n        return result\n      },\n    },\n  }\n}\n"],"names":[],"mappings":";;;AAWO,MAAM,2BAET,MAAM;AACD,SAAA;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,IACT,WAAW;AAAA,MACT,QAAQ;AAAA;AAAA,QAEN,IAAI;AAAA,QACJ,MAAM;AAAA,MACR;AAAA,MACA,QAAQ,MAAM,IAAI;;AAChB,YAAI,GAAC,gBAAW,yBAAX,mBAAiC,IAAI,MAAK;AACtC,iBAAA;AAAA,QAAA;AAGT,YAAI,MAAO,SAAQ,KAAK,iCAAiC,EAAE;AAE3D,cAAM,MAAM,SAAS,EAAE,MAAM;AACzB,YAAA,QAAQ,KAAK,KAAK,iBAAiB;AACjC,cAAA,SAAS,gBAAgB,KAAK;AAAA,UAClC,YAAY;AAAA,UACZ,UAAU;AAAA,UACV,gBAAgB;AAAA,QAAA,CACjB;AACD,YAAI,OAAO;AACD,kBAAA,MAAM,OAAO,IAAI;AACzB,kBAAQ,IAAI,aAAa,OAAO,OAAO,MAAM;AAAA,QAAA;AAExC,eAAA;AAAA,MAAA;AAAA,IACT;AAAA,EAEJ;AACF;"}
+{"version":3,"file":"router-hmr-plugin.js","sources":["../../../src/core/router-hmr-plugin.ts"],"sourcesContent":["import { generateFromAst, logDiff, parseAst } from '@tanstack/router-utils'\nimport { routeHmrStatement } from './route-hmr-statement'\nimport { debug } from './utils'\nimport { getConfig } from './config'\nimport type { UnpluginFactory } from 'unplugin'\nimport type { Config } from './config'\n\n/**\n * This plugin adds HMR support for file routes.\n * It is only added to the composed plugin in dev when autoCodeSplitting is disabled, since the code splitting plugin\n * handles HMR for code-split routes itself.\n */\n\nconst includeCode = [\n  'createFileRoute(',\n  'createRootRoute(',\n  'createRootRouteWithContext(',\n]\nexport const unpluginRouterHmrFactory: UnpluginFactory<\n  Partial<Config> | undefined\n> = (options = {}) => {\n  let ROOT: string = process.cwd()\n  let userConfig = options as Config\n\n  return {\n    name: 'tanstack-router:hmr',\n    enforce: 'pre',\n    transform: {\n      filter: {\n        // this is necessary for webpack / rspack to avoid matching .html files\n        id: /\\.(m|c)?(j|t)sx?$/,\n        code: {\n          include: includeCode,\n        },\n      },\n      handler(code, id) {\n        if (!globalThis.TSR_ROUTES_BY_ID_MAP?.has(id)) {\n          return null\n        }\n\n        if (debug) console.info('Adding HMR handling to route ', id)\n\n        const ast = parseAst({ code })\n        ast.program.body.push(routeHmrStatement)\n        const result = generateFromAst(ast, {\n          sourceMaps: true,\n          filename: id,\n          sourceFileName: id,\n        })\n        if (debug) {\n          logDiff(code, result.code)\n          console.log('Output:\\n', result.code + '\\n\\n')\n        }\n        return result\n      },\n    },\n    vite: {\n      configResolved(config) {\n        ROOT = config.root\n        userConfig = getConfig(options, ROOT)\n      },\n      applyToEnvironment(environment) {\n        if (userConfig.plugin?.vite?.environmentName) {\n          return userConfig.plugin.vite.environmentName === environment.name\n        }\n        return true\n      },\n    },\n  }\n}\n"],"names":[],"mappings":";;;;AAaA,MAAM,cAAc;AAAA,EAClB;AAAA,EACA;AAAA,EACA;AACF;AACO,MAAM,2BAET,CAAC,UAAU,OAAO;AAChB,MAAA,OAAe,QAAQ,IAAI;AAC/B,MAAI,aAAa;AAEV,SAAA;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,IACT,WAAW;AAAA,MACT,QAAQ;AAAA;AAAA,QAEN,IAAI;AAAA,QACJ,MAAM;AAAA,UACJ,SAAS;AAAA,QAAA;AAAA,MAEb;AAAA,MACA,QAAQ,MAAM,IAAI;;AAChB,YAAI,GAAC,gBAAW,yBAAX,mBAAiC,IAAI,MAAK;AACtC,iBAAA;AAAA,QAAA;AAGT,YAAI,MAAO,SAAQ,KAAK,iCAAiC,EAAE;AAE3D,cAAM,MAAM,SAAS,EAAE,MAAM;AACzB,YAAA,QAAQ,KAAK,KAAK,iBAAiB;AACjC,cAAA,SAAS,gBAAgB,KAAK;AAAA,UAClC,YAAY;AAAA,UACZ,UAAU;AAAA,UACV,gBAAgB;AAAA,QAAA,CACjB;AACD,YAAI,OAAO;AACD,kBAAA,MAAM,OAAO,IAAI;AACzB,kBAAQ,IAAI,aAAa,OAAO,OAAO,MAAM;AAAA,QAAA;AAExC,eAAA;AAAA,MAAA;AAAA,IAEX;AAAA,IACA,MAAM;AAAA,MACJ,eAAe,QAAQ;AACrB,eAAO,OAAO;AACD,qBAAA,UAAU,SAAS,IAAI;AAAA,MACtC;AAAA,MACA,mBAAmB,aAAa;;AAC1B,aAAA,sBAAW,WAAX,mBAAmB,SAAnB,mBAAyB,iBAAiB;AAC5C,iBAAO,WAAW,OAAO,KAAK,oBAAoB,YAAY;AAAA,QAAA;AAEzD,eAAA;AAAA,MAAA;AAAA,IACT;AAAA,EAEJ;AACF;"}

package/dist/esm/esbuild.d.ts

@@ -26,6 +26,11 @@ declare const TanStackRouterGeneratorEsbuild: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -68,6 +73,11 @@ declare const TanStackRouterCodeSplitterEsbuild: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -110,6 +120,11 @@ declare const TanStackRouterEsbuild: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -143,6 +158,11 @@ declare const tanstackRouter: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;

package/dist/esm/rspack.d.ts

@@ -30,6 +30,11 @@ declare const TanStackRouterGeneratorRspack: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -76,6 +81,11 @@ declare const TanStackRouterCodeSplitterRspack: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -122,6 +132,11 @@ declare const TanStackRouterRspack: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -155,6 +170,11 @@ declare const tanstackRouter: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;

package/dist/esm/vite.d.ts

@@ -17,6 +17,11 @@ declare const tanstackRouterAutoImport: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -59,6 +64,11 @@ declare const tanstackRouterGenerator: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -101,6 +111,11 @@ declare const tanStackRouterCodeSplitter: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -143,6 +158,11 @@ declare const tanstackRouter: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -179,6 +199,11 @@ declare const TanStackRouterVite: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;

package/dist/esm/webpack.d.ts

@@ -26,6 +26,11 @@ declare const TanStackRouterGeneratorWebpack: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -68,6 +73,11 @@ declare const TanStackRouterCodeSplitterWebpack: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -110,6 +120,11 @@ declare const TanStackRouterWebpack: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;
@@ -143,6 +158,11 @@ declare const tanstackRouter: (options?: Partial<{
     tmpDir: string;
     enableRouteGeneration?: boolean | undefined;
     codeSplittingOptions?: import('./core/config.js').CodeSplittingOptions | undefined;
+    plugin?: {
+        vite?: {
+            environmentName?: string | undefined;
+        } | undefined;
+    } | undefined;
     virtualRouteConfig?: string | import('@tanstack/virtual-file-routes').VirtualRootRoute | undefined;
     routeFilePrefix?: string | undefined;
     routeFileIgnorePattern?: string | undefined;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/router-plugin",
-  "version": "1.124.2",
+  "version": "1.125.1",
   "description": "Modern and scalable routing for React applications",
   "author": "Tanner Linsley",
   "license": "MIT",
@@ -97,10 +97,10 @@
     "chokidar": "^3.6.0",
     "unplugin": "^2.1.2",
     "zod": "^3.24.2",
-    "@tanstack/router-generator": "1.124.2",
+    "@tanstack/router-core": "^1.125.1",
     "@tanstack/router-utils": "1.121.21",
-    "@tanstack/router-core": "^1.124.2",
-    "@tanstack/virtual-file-routes": "^1.121.21"
+    "@tanstack/virtual-file-routes": "^1.121.21",
+    "@tanstack/router-generator": "1.125.1"
   },
   "devDependencies": {
     "@types/babel__core": "^7.20.5",
@@ -112,7 +112,7 @@
     "vite": ">=5.0.0 || >=6.0.0",
     "vite-plugin-solid": "^2.11.2",
     "webpack": ">=5.92.0",
-    "@tanstack/react-router": "^1.124.2"
+    "@tanstack/react-router": "^1.125.1"
   },
   "peerDependenciesMeta": {
     "@rsbuild/core": {

package/src/core/code-splitter/compilers.ts

@@ -12,7 +12,7 @@ import { createIdentifier } from './path-ids'
 import { getFrameworkOptions } from './framework-options'
 import type { GeneratorResult, ParseAstOptions } from '@tanstack/router-utils'
 import type { CodeSplitGroupings, SplitRouteIdentNodes } from '../constants'
-import type { Config } from '../config'
+import type { Config, DeletableNodes } from '../config'
 
 type SplitNodeMeta = {
   routeIdent: SplitRouteIdentNodes
@@ -92,13 +92,24 @@ function removeSplitSearchParamFromFilename(filename: string) {
   return bareFilename!
 }
 
+const splittableCreateRouteFns = ['createFileRoute']
+const unsplittableCreateRouteFns = [
+  'createRootRoute',
+  'createRootRouteWithContext',
+]
+const allCreateRouteFns = [
+  ...splittableCreateRouteFns,
+  ...unsplittableCreateRouteFns,
+]
+
 export function compileCodeSplitReferenceRoute(
   opts: ParseAstOptions & {
-    runtimeEnv: 'dev' | 'prod'
     codeSplitGroupings: CodeSplitGroupings
+    deleteNodes?: Set<DeletableNodes>
     targetFramework: Config['target']
     filename: string
     id: string
+    addHmr?: boolean
   },
 ): GeneratorResult {
   const ast = parseAst(opts)
@@ -116,6 +127,8 @@ export function compileCodeSplitReferenceRoute(
   const LAZY_ROUTE_COMPONENT_IDENT = frameworkOptions.idents.lazyRouteComponent
   const LAZY_FN_IDENT = frameworkOptions.idents.lazyFn
 
+  let createRouteFn: string
+
   babel.traverse(ast, {
     Program: {
       enter(programPath) {
@@ -136,29 +149,48 @@ export function compileCodeSplitReferenceRoute(
               return
             }
 
-            if (
-              !(
-                path.node.callee.name === 'createRoute' ||
-                path.node.callee.name === 'createFileRoute'
-              )
-            ) {
+            if (!allCreateRouteFns.includes(path.node.callee.name)) {
               return
             }
 
+            createRouteFn = path.node.callee.name
+
             function babelHandleReference(routeOptions: t.Node | undefined) {
               const hasImportedOrDefinedIdentifier = (name: string) => {
                 return programPath.scope.hasBinding(name)
               }
 
               if (t.isObjectExpression(routeOptions)) {
+                if (opts.deleteNodes && opts.deleteNodes.size > 0) {
+                  routeOptions.properties = routeOptions.properties.filter(
+                    (prop) => {
+                      if (t.isObjectProperty(prop)) {
+                        if (t.isIdentifier(prop.key)) {
+                          if (opts.deleteNodes?.has(prop.key.name as any)) {
+                            return false
+                          }
+                        }
+                      }
+                      return true
+                    },
+                  )
+                }
+                if (!splittableCreateRouteFns.includes(createRouteFn)) {
+                  // we can't split this route but we still add HMR handling if enabled
+                  if (opts.addHmr) {
+                    programPath.pushContainer('body', routeHmrStatement)
+                  }
+                  // exit traversal so this route is not split
+                  return programPath.stop()
+                }
                 routeOptions.properties.forEach((prop) => {
                   if (t.isObjectProperty(prop)) {
                     if (t.isIdentifier(prop.key)) {
+                      const key = prop.key.name
+
                       // If the user has not specified a split grouping for this key
                       // then we should not split it
-                      const codeSplitGroupingByKey = findIndexForSplitNode(
-                        prop.key.name,
-                      )
+                      const codeSplitGroupingByKey = findIndexForSplitNode(key)
                       if (codeSplitGroupingByKey === -1) {
                         return
                       }
@@ -168,7 +200,6 @@ export function compileCodeSplitReferenceRoute(
                         ),
                       ]
 
-                      const key = prop.key.name
                       // find key in nodeSplitConfig
                       const isNodeConfigAvailable = SPLIT_NODES_CONFIG.has(
                         key as any,
@@ -249,19 +280,12 @@ export function compileCodeSplitReferenceRoute(
                           ])
                         }
 
-                        // If it's a component, we need to pass the function to check the Route.ssr value
-                        if (key === 'component') {
-                          prop.value = template.expression(
-                            `${LAZY_ROUTE_COMPONENT_IDENT}(${splitNodeMeta.localImporterIdent}, '${splitNodeMeta.exporterIdent}', () => Route.ssr)`,
-                          )()
-                        } else {
-                          prop.value = template.expression(
-                            `${LAZY_ROUTE_COMPONENT_IDENT}(${splitNodeMeta.localImporterIdent}, '${splitNodeMeta.exporterIdent}')`,
-                          )()
-                        }
+                        prop.value = template.expression(
+                          `${LAZY_ROUTE_COMPONENT_IDENT}(${splitNodeMeta.localImporterIdent}, '${splitNodeMeta.exporterIdent}')`,
+                        )()
 
                         // add HMR handling
-                        if (opts.runtimeEnv !== 'prod') {
+                        if (opts.addHmr) {
                           programPath.pushContainer('body', routeHmrStatement)
                         }
                       }
@@ -416,12 +440,7 @@ export function compileCodeSplitVirtualRoute(
               return
             }
 
-            if (
-              !(
-                path.node.callee.name === 'createRoute' ||
-                path.node.callee.name === 'createFileRoute'
-              )
-            ) {
+            if (!splittableCreateRouteFns.includes(path.node.callee.name)) {
               return
             }
 
@@ -850,12 +869,10 @@ function resolveIdentifier(path: any, node: any): t.Node | undefined {
   return node
 }
 
-function removeIdentifierLiteral(path: any, node: any) {
-  if (t.isIdentifier(node)) {
-    const binding = path.scope.getBinding(node.name)
-    if (binding) {
-      binding.path.remove()
-    }
+function removeIdentifierLiteral(path: babel.NodePath, node: t.Identifier) {
+  const binding = path.scope.getBinding(node.name)
+  if (binding) {
+    binding.path.remove()
   }
 }
 

package/src/core/config.ts

@@ -55,12 +55,28 @@ export type CodeSplittingOptions = {
    * @default [['component'],['pendingComponent'],['errorComponent'],['notFoundComponent']]
    */
   defaultBehavior?: CodeSplitGroupings
+
+  /**
+   * The nodes that shall be deleted from the route.
+   * @default undefined
+   */
+  deleteNodes?: Array<DeletableNodes>
+
+  /**
+   * @default true
+   */
+  addHmr?: boolean
 }
 
+const DELETABLE_NODES = ['ssr'] as const
+export const deletableNodesSchema = z.enum(DELETABLE_NODES)
 const codeSplittingOptionsSchema = z.object({
   splitBehavior: z.function().optional(),
   defaultBehavior: splitGroupingsSchema.optional(),
+  deleteNodes: z.array(deletableNodesSchema).optional(),
+  addHmr: z.boolean().optional().default(true),
 })
+export type DeletableNodes = (typeof DELETABLE_NODES)[number]
 
 export const configSchema = generatorConfigSchema.extend({
   enableRouteGeneration: z.boolean().optional(),
@@ -69,6 +85,15 @@ export const configSchema = generatorConfigSchema.extend({
       return codeSplittingOptionsSchema.parse(v)
     })
     .optional(),
+  plugin: z
+    .object({
+      vite: z
+        .object({
+          environmentName: z.string().optional(),
+        })
+        .optional(),
+    })
+    .optional(),
 })
 
 export const getConfig = (inlineConfig: Partial<Config>, root: string) => {

package/src/core/router-code-splitter-plugin.ts

@@ -120,11 +120,12 @@ export const unpluginRouterCodeSplitterFactory: UnpluginFactory<
 
     const compiledReferenceRoute = compileCodeSplitReferenceRoute({
       code,
-      runtimeEnv: isProduction ? 'prod' : 'dev',
       codeSplitGroupings: splitGroupings,
       targetFramework: userConfig.target,
       filename: id,
       id,
+      deleteNodes: new Set(userConfig.codeSplittingOptions?.deleteNodes),
+      addHmr: options.codeSplittingOptions?.addHmr && !isProduction,
     })
 
     if (debug) {
@@ -171,6 +172,11 @@ export const unpluginRouterCodeSplitterFactory: UnpluginFactory<
     return result
   }
 
+  const includedCode = [
+    'createFileRoute(',
+    'createRootRoute(',
+    'createRootRouteWithContext(',
+  ]
   return [
     {
       name: 'tanstack-router:code-splitter:compile-reference-file',
@@ -183,11 +189,16 @@ export const unpluginRouterCodeSplitterFactory: UnpluginFactory<
             // this is necessary for webpack / rspack to avoid matching .html files
             include: /\.(m|c)?(j|t)sx?$/,
           },
-          code: 'createFileRoute(',
+          code: {
+            include: includedCode,
+          },
         },
         handler(code, id) {
           const generatorFileInfo = globalThis.TSR_ROUTES_BY_ID_MAP?.get(id)
-          if (generatorFileInfo && code.includes('createFileRoute(')) {
+          if (
+            generatorFileInfo &&
+            includedCode.some((included) => code.includes(included))
+          ) {
             for (const externalPlugin of bannedBeforeExternalPlugins) {
               if (!externalPlugin.frameworks.includes(framework)) {
                 continue
@@ -210,6 +221,12 @@ export const unpluginRouterCodeSplitterFactory: UnpluginFactory<
           ROOT = config.root
           userConfig = getConfig(options, ROOT)
         },
+        applyToEnvironment(environment) {
+          if (userConfig.plugin?.vite?.environmentName) {
+            return userConfig.plugin.vite.environmentName === environment.name
+          }
+          return true
+        },
       },
 
       rspack() {

package/src/core/router-generator-plugin.ts

@@ -73,14 +73,13 @@ export const unpluginRouterGeneratorFactory: UnpluginFactory<
       configResolved() {
         initConfigAndGenerator()
       },
-      async buildStart() {
-        // to support vite 5, we need to optionally chain the access to the environment
-        // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
-        if (this.environment?.config?.consumer === 'server') {
-          // When building in environment mode, we only need to generate routes
-          // for the client environment
-          return
+      applyToEnvironment(environment) {
+        if (userConfig.plugin?.vite?.environmentName) {
+          return userConfig.plugin.vite.environmentName === environment.name
         }
+        return true
+      },
+      async buildStart() {
         await generate()
       },
       sharedDuringBuild: true,

package/src/core/router-hmr-plugin.ts

@@ -1,17 +1,27 @@
 import { generateFromAst, logDiff, parseAst } from '@tanstack/router-utils'
 import { routeHmrStatement } from './route-hmr-statement'
 import { debug } from './utils'
-import type { Config } from './config'
+import { getConfig } from './config'
 import type { UnpluginFactory } from 'unplugin'
+import type { Config } from './config'
 
 /**
  * This plugin adds HMR support for file routes.
  * It is only added to the composed plugin in dev when autoCodeSplitting is disabled, since the code splitting plugin
  * handles HMR for code-split routes itself.
  */
+
+const includeCode = [
+  'createFileRoute(',
+  'createRootRoute(',
+  'createRootRouteWithContext(',
+]
 export const unpluginRouterHmrFactory: UnpluginFactory<
   Partial<Config> | undefined
-> = () => {
+> = (options = {}) => {
+  let ROOT: string = process.cwd()
+  let userConfig = options as Config
+
   return {
     name: 'tanstack-router:hmr',
     enforce: 'pre',
@@ -19,7 +29,9 @@ export const unpluginRouterHmrFactory: UnpluginFactory<
       filter: {
         // this is necessary for webpack / rspack to avoid matching .html files
         id: /\.(m|c)?(j|t)sx?$/,
-        code: 'createFileRoute(',
+        code: {
+          include: includeCode,
+        },
       },
       handler(code, id) {
         if (!globalThis.TSR_ROUTES_BY_ID_MAP?.has(id)) {
@@ -42,5 +54,17 @@ export const unpluginRouterHmrFactory: UnpluginFactory<
         return result
       },
     },
+    vite: {
+      configResolved(config) {
+        ROOT = config.root
+        userConfig = getConfig(options, ROOT)
+      },
+      applyToEnvironment(environment) {
+        if (userConfig.plugin?.vite?.environmentName) {
+          return userConfig.plugin.vite.environmentName === environment.name
+        }
+        return true
+      },
+    },
   }
 }