You've already forked agentic-coding-workflow
37 lines
1.3 KiB
Go
37 lines
1.3 KiB
Go
// Package httpx 内的 spa.go 提供 SPA 静态资源 + 404 fallback。
|
|
// 命中文件直接 ServeFile；不存在且最后一段无扩展名视为前端路由，回退 index.html；
|
|
// /api/ 前缀始终返回 404（避免把 API 拼写错误掩盖成 HTML）。
|
|
package httpx
|
|
|
|
import (
|
|
"io/fs"
|
|
"net/http"
|
|
stdpath "path"
|
|
"strings"
|
|
)
|
|
|
|
// NewSPAHandler 把 distFS 暴露成 SPA handler。
|
|
func NewSPAHandler(distFS fs.FS) http.Handler {
|
|
fileServer := http.FileServer(http.FS(distFS))
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if strings.HasPrefix(r.URL.Path, "/api/") {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
// Trim 两侧斜杠：带尾斜杠的路径（如 /p/x/w/）对 fs.Stat 是非法 fs 路径，
|
|
// 返回 ErrInvalid 而非 ErrNotExist，必须先归一化才能正确触发回退。
|
|
name := strings.Trim(r.URL.Path, "/")
|
|
if name == "" {
|
|
name = "index.html"
|
|
}
|
|
if _, err := fs.Stat(distFS, name); err != nil && stdpath.Ext(name) == "" {
|
|
// 前端路由：直接发 index.html 内容，不能改写 r.URL.Path 再走
|
|
// fileServer——FileServer 会把 /index.html 301 重定向到 "./"，
|
|
// 深层路由刷新会被改写成带尾斜杠的错误 URL。
|
|
http.ServeFileFS(w, r, distFS, "index.html")
|
|
return
|
|
}
|
|
fileServer.ServeHTTP(w, r)
|
|
})
|
|
}
|