From 7e924511b340a37e30c23ccb93b55535d933dae0 Mon Sep 17 00:00:00 2001
From: Jack Westbrook <jack.westbrook@gmail.com>
Date: Wed, 6 Mar 2024 09:24:45 +0100
Subject: [PATCH] feat(vite): make production builds work

---
 .gitignore                     |  1 +
 pkg/api/webassets/webassets.go | 29 ++++++++++++++++-
 vite.config.ts                 | 58 +++++++++++++++++++++++++---------
 3 files changed, 72 insertions(+), 16 deletions(-)

diff --git a/.gitignore b/.gitignore
index c043094a016..611ba66e439 100644
--- a/.gitignore
+++ b/.gitignore
@@ -7,6 +7,7 @@ awsconfig
 /.awcache
 /dist
 /public/build
+/public/build_tmp
 /emails/dist
 /reports
 /e2e/tmp
diff --git a/pkg/api/webassets/webassets.go b/pkg/api/webassets/webassets.go
index c5931c6084e..880c7e9f03d 100644
--- a/pkg/api/webassets/webassets.go
+++ b/pkg/api/webassets/webassets.go
@@ -106,7 +106,7 @@ func readWebAssetsFromFile(manifestpath string) (*dtos.EntryPointAssets, error)
 }
 
 func readWebAssetsFromCDN(ctx context.Context, baseURL string) (*dtos.EntryPointAssets, error) {
-	req, err := http.NewRequestWithContext(ctx, http.MethodGet, baseURL+"public/build/vite/manifest.json", nil)
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, baseURL+"public/build/.vite/manifest.json", nil)
 	if err != nil {
 		return nil, err
 	}
@@ -222,6 +222,33 @@ func readWebAssets(r io.Reader) (*dtos.EntryPointAssets, error) {
 	return rsp, nil
 	// do some error handling shizzle here.
 
+	// return &dtos.EntryPointAssets{
+	// 	JSFiles: entryPointJSAssets,
+	// 	Dark:    darkCSS,
+	// 	Light:   lightCSS,
+	// }, nil
+	// var entryPointJSAssets []dtos.EntryPointAsset
+	// var darkCSS, lightCSS string
+
+	// for _, entry := range manifest {
+	// 	if entry.IsEntry {
+	// 		if filepath.Ext(entry.File) != ".css" {
+	// 			asset := dtos.EntryPointAsset{
+	// 				FilePath: entry.File,
+	// 				// Not sure what should happen with integrity here
+	// 				Integrity: "",
+	// 			}
+	// 			entryPointJSAssets = append(entryPointJSAssets, asset)
+	// 		}
+	// 		if entry.Src == "sass/grafana.dark.scss" && entry.IsEntry {
+	// 			darkCSS = entry.File
+	// 		}
+	// 		if entry.Src == "sass/grafana.light.scss" && entry.IsEntry {
+	// 			lightCSS = entry.File
+	// 		}
+	// 	}
+	// }
+
 	// return &dtos.EntryPointAssets{
 	// 	JSFiles: entryPointJSAssets,
 	// 	Dark:    darkCSS,
diff --git a/vite.config.ts b/vite.config.ts
index 8541d8157e1..cd9ef62084e 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -1,5 +1,7 @@
 import react from '@vitejs/plugin-react-swc';
+import { move, remove } from 'fs-extra';
 import { createRequire } from 'node:module';
+import { resolve } from 'node:path';
 import { fileURLToPath } from 'node:url';
 import { defineConfig } from 'vite';
 
@@ -7,22 +9,23 @@ const require = createRequire(import.meta.url);
 
 // https://vitejs.dev/config/
 export default defineConfig({
-  root: './public',
+  root: 'public',
   build: {
     // use manifest for backend integration in production
-    manifest: true,
+    manifest: 'public/build/.vite/manifest.json',
     rollupOptions: {
       input: ['./public/app/index.ts', './public/sass/grafana.dark.scss', './public/sass/grafana.light.scss'],
     },
-    outDir: './build',
-    assetsDir: './',
+    outDir: 'build_tmp',
+    assetsDir: 'public/build',
+    minify: 'esbuild',
   },
   server: {
     // vite binds to ipv6 by default... and that doesn't work for me locally on mac...
     host: '0.0.0.0',
     port: 5173,
   },
-  plugins: [react(), angularHtmlImport()],
+  plugins: [react(), angularHtmlImport(), { ...moveAssets(), apply: 'build' }],
   optimizeDeps: {
     // fixes for dependencies that don't support esm
     include: ['jquery'],
@@ -65,16 +68,16 @@ export default defineConfig({
 });
 
 /**
-This is a Vite plugin for handling angular templates.
-https://vitejs.dev/guide/api-plugin.html#simple-examples
- 
-The webpack output from https://github.com/WearyMonkey/ngtemplate-loader looks like this:
-var code = "\n<div class=\"graph-annotation\">\n\t<div class=\"graph-annotation__header\">\n\t\t</div></div>\n";
-Exports
-var _module_exports =code;;
-var path = 'public/app/features/annotations/partials/event_editor.html';
-window.angular.module('ng').run(['$templateCache', function(c) { c.put(path, _module_exports) }]);
-module.exports = path;
+ * This is a Vite plugin for handling angular templates.
+ * https://vitejs.dev/guide/api-plugin.html#simple-examples
+ *
+ * The webpack output from https://github.com/WearyMonkey/ngtemplate-loader looks like this:
+ * var code = "\n<div class=\"graph-annotation\">\n\t<div class=\"graph-annotation__header\">\n\t\t</div></div>\n";
+ * Exports
+ * var _module_exports =code;;
+ * var path = 'public/app/features/annotations/partials/event_editor.html';
+ * window.angular.module('ng').run(['$templateCache', function(c) { c.put(path, _module_exports) }]);
+ * module.exports = path;
  */
 
 function angularHtmlImport() {
@@ -97,3 +100,28 @@ angular.module('ng').run(['$templateCache', c => { c.put(path, htmlTemplate) }])
     },
   };
 }
+
+/**
+ * This is a Vite plugin for moving assets to the build folder.
+ * Due to Vite expecting html files to be in the root of the project
+ * and the manifest generating paths using the assetDir, we build to a temp folder first
+ * then move it to the build folder.
+ */
+
+function moveAssets() {
+  return {
+    name: 'move-assets',
+    async closeBundle() {
+      try {
+        console.log('Moving assets to build folder...');
+        const __dirname = fileURLToPath(new URL('.', import.meta.url));
+        const buildPath = resolve(__dirname, 'public/build');
+        const buildTmpPath = resolve(__dirname, 'public/build_tmp/public/build');
+        await move(buildTmpPath, buildPath, { overwrite: true });
+        await remove(resolve(__dirname, 'public/build_tmp'));
+      } catch (error) {
+        console.error('Failed to move assets', error);
+      }
+    },
+  };
+}