57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package static
|
||
|
||
import (
|
||
"embed"
|
||
"io/fs"
|
||
"net/http"
|
||
"strings"
|
||
)
|
||
|
||
//go:embed dist/*
|
||
var StaticFiles embed.FS
|
||
|
||
// Handler 返回静态文件处理器
|
||
// 用于服务嵌入的前端静态文件
|
||
func Handler() http.Handler {
|
||
// 提取 dist 子目录
|
||
distFS, err := fs.Sub(StaticFiles, "dist")
|
||
if err != nil {
|
||
// 如果 dist 目录不存在，返回空处理器
|
||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
http.Error(w, "Static files not available", http.StatusNotFound)
|
||
})
|
||
}
|
||
|
||
fileServer := http.FileServer(http.FS(distFS))
|
||
|
||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
// 对于 API 路径，不处理
|
||
if strings.HasPrefix(r.URL.Path, "/api/") {
|
||
http.NotFound(w, r)
|
||
return
|
||
}
|
||
|
||
// 尝试服务静态文件
|
||
// 对于 SPA，如果文件不存在，返回 index.html
|
||
path := r.URL.Path
|
||
if path == "/" {
|
||
path = "/index.html"
|
||
}
|
||
|
||
// 检查文件是否存在
|
||
_, err := fs.Stat(distFS, strings.TrimPrefix(path, "/"))
|
||
if err != nil {
|
||
// 文件不存在，返回 index.html（支持 SPA 路由）
|
||
r.URL.Path = "/"
|
||
}
|
||
|
||
fileServer.ServeHTTP(w, r)
|
||
})
|
||
}
|
||
|
||
// IsAvailable 检查静态文件是否可用
|
||
func IsAvailable() bool {
|
||
_, err := fs.Sub(StaticFiles, "dist")
|
||
return err == nil
|
||
}
|