fix: Resolve SSR build failure related to #93 (#94)

* fix: Resolve SSR build failure related to [#93](#93).

* test: increase test timeout for integration tests to 180 seconds

Author: fossamagna

.changeset/nine-doors-float.md

@@ -0,0 +1,8 @@
+---
+"vite-plugin-react-router-amplify-hosting": patch
+---
+
+fix: Resolve SSR build failure related to [fossamagna/react-router-amplify#93](https://github.com/fossamagna/react-router-amplify/issues/93).
+
+- Revised the asset copy behavior during SSR build.
+- Fixed handling of the `unstable_viteEnvironmentApi` flag in `@react-router/dev`.

packages/vite-plugin-react-router-amplify-hosting/src/index.ts

@@ -1,4 +1,4 @@
-import { cp, mkdir, readFile, writeFile } from "node:fs/promises";
+import { copyFile, cp, mkdir, readFile, writeFile } from "node:fs/promises";
 import path from "node:path";
 import type {
   BuildEnvironmentOptions,
@@ -112,26 +112,33 @@ export function amplifyHosting(): Plugin {
       if (!pluginConfig) {
         return;
       }
-      if (!pluginConfig.isSsrBuild) {
+      const isClientBuild = pluginConfig.future.unstable_viteEnvironmentApi
+        ? this.environment.name === "client"
+        : !pluginConfig.isSsrBuild;
+      const isServerBuild = pluginConfig.future.unstable_viteEnvironmentApi
+        ? this.environment.name === "ssr"
+        : pluginConfig.isSsrBuild;
+
+      if (isClientBuild) {
         const staticDir = path.join(
           resolvedConfig.root,
           AMPLITY_HOSTING_STATIC_DIR,
         );
         await mkdir(staticDir, { recursive: true });
         const dir = options.dir ?? "";
         await cp(dir, staticDir, { recursive: true });
-      }
-      if (
-        pluginConfig.isSsrBuild ||
-        pluginConfig.future.unstable_viteEnvironmentApi
-      ) {
+      } else if (isServerBuild) {
+        // copy server.mjs to the compute default directory
         const computeDefaultDir = path.join(
           resolvedConfig.root,
           AMPLITY_HOSTING_COMPUTE_DEFAULT_DIR,
         );
         await mkdir(computeDefaultDir, { recursive: true });
         const dir = options.dir ?? "";
-        await cp(dir, computeDefaultDir, { recursive: true });
+        await copyFile(
+          path.join(dir, "server.mjs"),
+          path.join(computeDefaultDir, "server.mjs"),
+        );
         // write deploy-manifest.json
         const reactRouterVersion = await getPackageVersion(
           "react-router",

packages/vite-plugin-react-router-amplify-hosting/vitest.config.integration.ts

@@ -5,7 +5,7 @@ export default defineProject({
   test: {
     name: `${pkg.name}-integration`,
     include: ["integration/**/*.test.ts"],
-    testTimeout: 120_000,
+    testTimeout: 180_000,
     exclude: [
       "**/node_modules/**",
       "**/dist/**",

packages/vite-plugin-react-router-amplify-hosting/vitest.config.unit.ts

@@ -5,6 +5,6 @@ export default defineProject({
   test: {
     name: `${pkg.name}-unit`,
     include: ["**/*.test.ts"],
-    exclude: ["integration/**"],
+    exclude: ["integration/**", "node_modules/**", ".tmp/**"],
   },
 });