Files

35 lines
1020 B
Go

package common
import (
"net/http"
"os"
"path/filepath"
"strings"
"github.com/gogf/gf/v2/net/ghttp"
)
// SpaFallback 单页应用 history 路由回退中间件:prefix 路径下请求对应的静态文件不存在
// (即前端路由,如 /admin/orders)时回退 index.html;真实静态文件交给后续静态服务处理。
// root 为前端构建产物目录,prefix 为托管路径前缀(如 "/admin")。
func SpaFallback(root, prefix string) ghttp.HandlerFunc {
root = filepath.Clean(root)
index := filepath.Join(root, "index.html")
return func(r *ghttp.Request) {
path := r.URL.Path
if r.Method == http.MethodGet && (path == prefix || strings.HasPrefix(path, prefix+"/")) {
rel := strings.TrimPrefix(path, prefix)
candidate := filepath.Clean(filepath.Join(root, rel))
if !strings.HasPrefix(candidate, root) {
r.Middleware.Next()
return
}
if _, err := os.Stat(candidate); err != nil {
r.Response.ServeFile(index)
return
}
}
r.Middleware.Next()
}
}