From 54603463c5545ec9baab1b12717d50b919a80420 Mon Sep 17 00:00:00 2001
From: Yuzhong Zhang <141388234+BetterAndBetterII@users.noreply.github.com>
Date: Sat, 5 Jul 2025 23:59:31 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=20handleUI=20=E5=87=BD?=
 =?UTF-8?q?=E6=95=B0=EF=BC=8C=E6=94=B9=E8=BF=9B=E6=96=87=E4=BB=B6=E8=B7=AF?=
 =?UTF-8?q?=E5=BE=84=E5=A4=84=E7=90=86=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

本次提交主要更改了 `handleUI` 函数的实现，具体包括：
1. 将返回类型从 `http.Handler` 修改为 `http.HandlerFunc`。
2. 改进了对请求路径的处理，确保根路径请求返回 `index.html`。
3. 增强了对嵌入文件系统中不存在文件的处理逻辑，支持客户端路由。
4. 更新了内容类型检测逻辑，确保根据请求路径正确设置响应的 Content-Type。

这些更改旨在提升文件处理的灵活性和准确性，改善用户体验。
---
 main.go | 56 +++++++++++++++++++++++++++++++++++---------------------
 1 file changed, 35 insertions(+), 21 deletions(-)

diff --git a/main.go b/main.go
index 7630f8d..298b489 100644
--- a/main.go
+++ b/main.go
@@ -45,33 +45,48 @@ type (
 //go:embed all:frontend
 var assets embed.FS
 
-func handleUI() http.Handler {
+func handleUI() http.HandlerFunc {
 	sub, err := fs.Sub(assets, "frontend")
 	if err != nil {
 		panic(err)
 	}
-	// Let's hot-patch all calls to firebase DB
-	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		originalPath := r.URL.Path
-		originalPath = strings.TrimPrefix(originalPath, "/")
 
-		// Redirect "/" to "index.html"
-		if originalPath == "" {
-			originalPath = "index.html"
+	return func(w http.ResponseWriter, r *http.Request) {
+		path := r.URL.Path
+		// If the path is empty, it means it's the root, so serve index.html
+		if path == "/" || path == "" {
+			path = "/index.html"
 		}
 
-		file, err := sub.Open(originalPath)
+		// Check if the file exists in the embedded filesystem.
+		f, err := sub.Open(strings.TrimPrefix(path, "/"))
 		if err != nil {
+			// If the file does not exist, and it's not a request for a static asset (like .js, .css),
+			// then it's likely a client-side route. In that case, we should serve the index.html
+			// and let the client-side router handle it.
+			if os.IsNotExist(err) && !strings.Contains(path, ".") {
+				path = "/index.html"
+				f, err = sub.Open("index.html")
+			} else {
+				// It's a genuine 404 for a missing asset.
+				http.NotFound(w, r)
+				return
+			}
+		}
+
+		if err != nil {
+			// If we still have an error, something is wrong.
 			http.Error(w, "File not found", http.StatusNotFound)
 			return
 		}
-		defer file.Close()
+		defer f.Close()
 
-		fileContent, err := io.ReadAll(file)
+		fileContent, err := io.ReadAll(f)
 		if err != nil {
 			http.Error(w, "Error reading file", http.StatusInternalServerError)
 			return
 		}
+
 		modifiedContent := strings.ReplaceAll(string(fileContent), "firestore.googleapis.com", "localhost:3002")
 		modifiedContent = strings.ReplaceAll(modifiedContent, "ssl=!0", "ssl=0")
 		modifiedContent = strings.ReplaceAll(modifiedContent, "ssl:!0", "ssl:0")
@@ -79,19 +94,19 @@ func handleUI() http.Handler {
 		// Set the correct Content-Type based on the file extension
 		contentType := http.DetectContentType([]byte(modifiedContent))
 		switch {
-		case strings.HasSuffix(originalPath, ".js"):
+		case strings.HasSuffix(path, ".js"):
 			contentType = "application/javascript"
-		case strings.HasSuffix(originalPath, ".html"):
+		case strings.HasSuffix(path, ".html"):
 			contentType = "text/html"
-		case strings.HasSuffix(originalPath, ".css"):
+		case strings.HasSuffix(path, ".css"):
 			contentType = "text/css"
-		case strings.HasSuffix(originalPath, ".wasm"):
+		case strings.HasSuffix(path, ".wasm"):
 			contentType = "application/wasm"
-		case strings.HasSuffix(originalPath, ".tsx"):
+		case strings.HasSuffix(path, ".tsx"):
 			contentType = "text/typescript"
-		case strings.HasSuffix(originalPath, ".png"):
+		case strings.HasSuffix(path, ".png"):
 			contentType = "image/png"
-		case strings.HasSuffix(originalPath, ".woff2"):
+		case strings.HasSuffix(path, ".woff2"):
 			contentType = "font/woff2"
 		}
 
@@ -102,8 +117,7 @@ func handleUI() http.Handler {
 			http.Error(w, "Error serving file", http.StatusInternalServerError)
 			return
 		}
-		return
-	})
+	}
 }
 
 func setupRouter(store stores.Store) *chi.Mux {
@@ -295,7 +309,7 @@ func main() {
 
 	ioo := setupSocketIO()
 	r.Mount("/socket.io/", ioo.ServeHandler(nil))
-	r.Mount("/", handleUI())
+	r.NotFound(handleUI())
 
 	logrus.WithField("addr", *listenAddress).Info("starting server")
 	go func() {