mirror of
https://github.com/sipeed/picoclaw.git
synced 2026-06-12 18:08:54 +00:00
Cytown
80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"embed"
|
|
"fmt"
|
|
"io/fs"
|
|
"mime"
|
|
"net/http"
|
|
"path"
|
|
"strings"
|
|
|
|
"github.com/sipeed/picoclaw/pkg/logger"
|
|
)
|
|
|
|
//go:embed all:dist
|
|
var frontendFS embed.FS
|
|
|
|
// registerEmbedRoutes sets up the HTTP handler to serve the embedded frontend files
|
|
func registerEmbedRoutes(mux *http.ServeMux) {
|
|
// Register correct MIME type for SVG files
|
|
// Go's built-in mime.TypeByExtension returns "image/svg" which is incorrect
|
|
// The correct MIME type per RFC 6838 is "image/svg+xml"
|
|
if err := mime.AddExtensionType(".svg", "image/svg+xml"); err != nil {
|
|
logger.ErrorC("web", fmt.Sprintf("Warning: failed to register SVG MIME type: %v", err))
|
|
}
|
|
|
|
// Attempt to get the subdirectory 'dist' where Vite usually builds
|
|
subFS, err := fs.Sub(frontendFS, "dist")
|
|
if err != nil {
|
|
// Log a warning if dist doesn't exist yet (e.g., during development before a frontend build)
|
|
logger.WarnC("web",
|
|
"Warning: no 'dist' folder found in embedded frontend. "+
|
|
"Ensure you run `pnpm build:backend` in the frontend directory "+
|
|
"before building the Go backend.",
|
|
)
|
|
return
|
|
}
|
|
|
|
fileServer := http.FileServer(http.FS(subFS))
|
|
|
|
// Serve static assets and fallback to index.html for SPA routes.
|
|
mux.Handle(
|
|
"/",
|
|
http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodGet && r.Method != http.MethodHead {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
// Keep unknown API paths as 404 instead of falling back to SPA entry.
|
|
if r.URL.Path == "/api" || strings.HasPrefix(r.URL.Path, "/api/") {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
cleanPath := path.Clean(strings.TrimPrefix(r.URL.Path, "/"))
|
|
if cleanPath == "." {
|
|
cleanPath = ""
|
|
}
|
|
|
|
// Existing static files/directories should be served directly.
|
|
if cleanPath != "" {
|
|
if _, statErr := fs.Stat(subFS, cleanPath); statErr == nil {
|
|
fileServer.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
// Missing asset-like paths should remain 404.
|
|
if strings.Contains(path.Base(cleanPath), ".") {
|
|
fileServer.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
}
|
|
|
|
indexReq := r.Clone(r.Context())
|
|
indexReq.URL.Path = "/"
|
|
fileServer.ServeHTTP(w, indexReq)
|
|
}),
|
|
)
|
|
}
|