Add build flag for production UI dist output embed.

cmd/server/fs_dev.go

@@ -1,13 +1,12 @@
-// +build !prod
+//go:build !prod
 
-package server
+package main
 
 import (
-  "io/fs"
+	"io/fs"
-  "os"
+	"os"
 )
 
 func getUIAssets() fs.FS {
-  return os.DirFS("ui/dist")
+	return os.DirFS("ui/dist")
 }
-

cmd/server/fs_prod.go

@@ -1,20 +1,12 @@
-// +build prod
+//go:build prod
 
-package server
+package main
 
 import (
-    "embed"
+	"github.com/smjklake/glancr/ui"
-    "io/fs"
+	"io/fs"
 )
 
-//go:embed frontend/dist
+func getUIAssets() fs.FS {
-var embedFrontend embed.FS
+	return ui.GetUIFS()
-
-func getFrontendAssets() fs.FS {
-  f, err := fs.Sub(embedFrontend, "ui/dist")
-  if err != nil {
-    panic(err)
-  }
-
-  return f
 }

cmd/server/main.go

@@ -1,5 +1,8 @@
-package server 
+package main
+
+import "fmt"
 
 func main() {
     frontend := getUIAssets()
+    fmt.Println(frontend)
 }

ui/embed.go

@@ -0,0 +1,19 @@
+package ui
+
+import (
+	"embed"
+	"io/fs"
+)
+
+//go:embed dist
+var embedUI embed.FS
+
+/// GetUIFS returns an embed FS for the UI components in the dist dir.
+func GetUIFS() fs.FS {
+	f, err := fs.Sub(embedUI, "build")
+	if err != nil {
+		panic(err)
+	}
+
+	return f
+}