fix ui build

Author: mattdholloway

script/build-ui

@@ -13,18 +13,5 @@ fi
 echo "Building UI..."
 npm run build
 
-# Move files to expected locations
-echo "Organizing output..."
-mkdir -p dist
-mv dist/src/apps/get-me/index.html dist/get-me.html 2>/dev/null || true
-mv dist/src/apps/issue-write/index.html dist/issue-write.html 2>/dev/null || true
-mv dist/src/apps/pr-write/index.html dist/pr-write.html 2>/dev/null || true
-rm -rf dist/src 2>/dev/null || true
-
-# Copy to Go embed location
-echo "Copying to pkg/github/ui_dist/..."
-mkdir -p ../pkg/github/ui_dist
-cp dist/*.html ../pkg/github/ui_dist/
-
 echo "UI build complete. Output:"
-ls -la dist/
+ls -la ../pkg/github/ui_dist/*.html

ui/vite.config.ts

@@ -1,4 +1,4 @@
-import { defineConfig } from "vite";
+import { defineConfig, Plugin } from "vite";
 import react from "@vitejs/plugin-react";
 import { viteSingleFile } from "vite-plugin-singlefile";
 import { resolve } from "path";
@@ -10,10 +10,30 @@ if (!app) {
   throw new Error("APP environment variable must be set");
 }
 
+// Plugin to rename the output file and remove the nested directory structure
+function renameOutput(): Plugin {
+  return {
+    name: "rename-output",
+    enforce: "post",
+    generateBundle(_, bundle) {
+      // Find the HTML file and rename it
+      for (const fileName of Object.keys(bundle)) {
+        if (fileName.endsWith("index.html")) {
+          const chunk = bundle[fileName];
+          chunk.fileName = `${app}.html`;
+          delete bundle[fileName];
+          bundle[`${app}.html`] = chunk;
+          break;
+        }
+      }
+    },
+  };
+}
+
 export default defineConfig({
-  plugins: [react(), viteSingleFile()],
+  plugins: [react(), viteSingleFile(), renameOutput()],
   build: {
-    outDir: "dist",
+    outDir: resolve(__dirname, "../pkg/github/ui_dist"),
     emptyOutDir: false,
     rollupOptions: {
       input: resolve(__dirname, `src/apps/${app}/index.html`),