Honor --mode flag in routes.ts context

Author: markdalgleish

.changeset/dirty-balloons-stare.md

@@ -0,0 +1,5 @@
+---
+"@react-router/dev": patch
+---
+
+Ensure `--mode` flag is honored within `routes.ts` when accessing via `import.meta.env.MODE`

packages/react-router-dev/cli/commands.ts

@@ -21,10 +21,14 @@ export async function routes(
   flags: {
     config?: string;
     json?: boolean;
+    mode?: string;
   } = {}
 ): Promise<void> {
   let rootDirectory = reactRouterRoot ?? process.cwd();
-  let configResult = await loadConfig({ rootDirectory });
+  let configResult = await loadConfig({
+    rootDirectory,
+    mode: flags.mode ?? "production",
+  });
 
   if (!configResult.ok) {
     console.error(colors.red(configResult.error));
@@ -81,6 +85,7 @@ export async function generateEntry(
   flags: {
     typescript?: boolean;
     config?: string;
+    mode?: string;
   } = {}
 ) {
   // if no entry passed, attempt to create both
@@ -91,7 +96,10 @@ export async function generateEntry(
   }
 
   let rootDirectory = reactRouterRoot ?? process.cwd();
-  let configResult = await loadConfig({ rootDirectory });
+  let configResult = await loadConfig({
+    rootDirectory,
+    mode: flags.mode ?? "production",
+  });
 
   if (!configResult.ok) {
     console.error(colors.red(configResult.error));
@@ -198,17 +206,25 @@ async function createClientEntry(
   return contents;
 }
 
-export async function typegen(root: string, flags: { watch: boolean }) {
+export async function typegen(
+  root: string,
+  flags: {
+    watch: boolean;
+    mode?: string;
+  }
+) {
   root ??= process.cwd();
 
+  let mode = flags.mode ?? "production";
+
   if (flags.watch) {
     await preloadVite();
     const vite = getVite();
     const logger = vite.createLogger("info", { prefix: "[react-router]" });
 
-    await Typegen.watch(root, { logger });
+    await Typegen.watch(root, { mode, logger });
     await new Promise(() => {}); // keep alive
     return;
   }
-  await Typegen.run(root);
+  await Typegen.run(root, { mode });
 }

packages/react-router-dev/config/config.ts

@@ -545,15 +545,17 @@ export type ConfigLoader = {
 export async function createConfigLoader({
   rootDirectory: root,
   watch,
+  mode,
 }: {
   watch: boolean;
   rootDirectory?: string;
+  mode: string;
 }): Promise<ConfigLoader> {
   root = root ?? process.env.REACT_ROUTER_ROOT ?? process.cwd();
 
   let viteNodeContext = await ViteNode.createContext({
     root,
-    mode: watch ? "development" : "production",
+    mode,
   });
 
   let reactRouterConfigFile = findEntry(root, "react-router.config", {
@@ -656,9 +658,16 @@ export async function createConfigLoader({
   };
 }
 
-export async function loadConfig({ rootDirectory }: { rootDirectory: string }) {
+export async function loadConfig({
+  rootDirectory,
+  mode,
+}: {
+  rootDirectory: string;
+  mode: string;
+}) {
   let configLoader = await createConfigLoader({
     rootDirectory,
+    mode,
     watch: false,
   });
   let config = await configLoader.getConfig();

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

@@ -14,8 +14,8 @@ import { getTypesDir, getTypesPath } from "./paths";
 import * as Params from "./params";
 import * as Route from "./route";
 
-export async function run(rootDirectory: string) {
-  const ctx = await createContext({ rootDirectory, watch: false });
+export async function run(rootDirectory: string, { mode }: { mode: string }) {
+  const ctx = await createContext({ rootDirectory, mode, watch: false });
   await writeAll(ctx);
 }
 
@@ -25,9 +25,9 @@ export type Watcher = {
 
 export async function watch(
   rootDirectory: string,
-  { logger }: { logger?: vite.Logger } = {}
+  { mode, logger }: { mode: string; logger?: vite.Logger }
 ): Promise<Watcher> {
-  const ctx = await createContext({ rootDirectory, watch: true });
+  const ctx = await createContext({ rootDirectory, mode, watch: true });
   await writeAll(ctx);
   logger?.info(pc.green("generated types"), { timestamp: true, clear: true });
 
@@ -55,11 +55,13 @@ export async function watch(
 async function createContext({
   rootDirectory,
   watch,
+  mode,
 }: {
   rootDirectory: string;
   watch: boolean;
+  mode: string;
 }): Promise<Context> {
-  const configLoader = await createConfigLoader({ rootDirectory, watch });
+  const configLoader = await createConfigLoader({ rootDirectory, mode, watch });
   const configResult = await configLoader.getConfig();
 
   if (!configResult.ok) {

packages/react-router-dev/vite/build.ts

@@ -35,7 +35,10 @@ export async function build(root: string, viteBuildOptions: ViteBuildOptions) {
   await preloadVite();
   let vite = getVite();
 
-  let configResult = await loadConfig({ rootDirectory: root });
+  let configResult = await loadConfig({
+    rootDirectory: root,
+    mode: viteBuildOptions.mode ?? "production",
+  });
 
   if (!configResult.ok) {
     throw new Error(configResult.error);

packages/react-router-dev/vite/cloudflare-dev-proxy.ts

@@ -57,7 +57,7 @@ export const cloudflareDevProxyVitePlugin = <Env, Cf extends CfProperties>(
 
   return {
     name: PLUGIN_NAME,
-    config: async (config) => {
+    config: async (config, configEnv) => {
       await preloadVite();
       const vite = getVite();
       // This is a compatibility layer for Vite 5. Default conditions were
@@ -74,6 +74,7 @@ export const cloudflareDevProxyVitePlugin = <Env, Cf extends CfProperties>(
 
       let configResult = await loadConfig({
         rootDirectory: config.root ?? process.cwd(),
+        mode: configEnv.mode,
       });
 
       if (!configResult.ok) {

packages/react-router-dev/vite/plugin.ts

@@ -1187,15 +1187,19 @@ export const reactRouterVitePlugin: ReactRouterVitePlugin = () => {
         rootDirectory =
           viteUserConfig.root ?? process.env.REACT_ROUTER_ROOT ?? process.cwd();
 
+        let mode = viteConfigEnv.mode;
+
         if (viteCommand === "serve") {
           typegenWatcherPromise = Typegen.watch(rootDirectory, {
+            mode,
             // ignore `info` logs from typegen since they are redundant when Vite plugin logs are active
             logger: vite.createLogger("warn", { prefix: "[react-router]" }),
           });
         }
 
         reactRouterConfigLoader = await createConfigLoader({
           rootDirectory,
+          mode,
           watch: viteCommand === "serve",
         });