sun
dbaee35400
Bug 1: 安装模式下浏览器访问 / 看到 503 JSON
- 根因: middleware/installguard.go 默认白名单没包含 /;
浏览器访问 / 时 InstallGuard 先于 router 的 redirect 拦掉,返 503 JSON。
- 修复: InstallGuard 非白名单分支加 Accept 内容协商,
text/html → 302 Location: /setup(浏览器友好),
其它(API 客户端 / curl / 空 Accept)→ 维持 503 JSON(API 行为不变)。
- 额外: mux 路由未匹配时不走 r.Use() 中间件链,
手动设 r.NotFoundHandler 复用同一份 Accept 协商逻辑,
覆盖 /admin /api/voices 等任意未注册路径。
Bug 2: 装完访问 /setup 仍能进入(看到安装表单)
- 根因: router.go 的 /setup handler 没做 mode 校验。
- 修复: handler 顶部加 if ModeNormal → 302 Location: /admin
(M2 之后才有 /admin;目前会 404 也比继续显示表单好)。
测试(都被 .gitignore 排除,本地保留):
- middleware/installguard_test.go: 加 2 个 case(浏览器 redirect / API JSON)+ acceptsHTML 单元
- router/router_test.go(新): /setup 装后 redirect + NotFoundHandler 4 个 case
- 全套 go test ./... 全绿
验证(本机 e2e 跑通):
- setup 模式: 浏览器 /, /admin, /api/voices → 302 /setup
- setup 模式: API/curl / → 503 JSON(行为不变)
- normal 模式: /setup → 302 /admin(Bug 2)
- normal 模式: /admin → 404(M2 才有,行为不变)
- normal 模式: /api/setup/status → 200 {installed:true,mode:normal}
二进制大小: 15.87 → ~16MB(几乎不变)
82 lines
2.6 KiB
Go
82 lines
2.6 KiB
Go
package middleware
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/volcano-tts/tts-api/installer"
|
|
)
|
|
|
|
// InstallGuard 拦截所有非白名单路由,在安装模式下按 Accept 头做内容协商:
|
|
//
|
|
// - text/html 类(浏览器) → 302 重定向到 /setup
|
|
// - 其它(API 客户端、curl 等) → 503 + JSON
|
|
//
|
|
// 设计:放行白名单路径前缀,其余一律拦截。
|
|
// 中间件顺序:必须装在 RateLimit / ConcurrencyLimit / Logger 之前,
|
|
// 避免安装模式下被限流计数污染(参考 M1 风险点 #2)。
|
|
func InstallGuard(currentMode func() installer.Mode, allowPrefixes ...string) func(http.Handler) http.Handler {
|
|
defaults := []string{
|
|
"/setup", // 安装引导页
|
|
"/api/setup", // 安装相关 API
|
|
"/health", // 部署探针要能识别未安装状态
|
|
"/metrics", // Prometheus 拉取
|
|
"/static/", // 引导页静态资源(留口子)
|
|
}
|
|
allow := append(defaults, allowPrefixes...)
|
|
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if currentMode() != installer.ModeSetup {
|
|
next.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
// 安装模式:仅放行白名单
|
|
path := r.URL.Path
|
|
for _, p := range allow {
|
|
if strings.HasPrefix(path, p) {
|
|
next.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
}
|
|
log.Printf("[installguard] 安装模式下拒绝非白名单请求 - 路径=%s 客户端=%s accept=%q",
|
|
path, GetClientIP(r), r.Header.Get("Accept"))
|
|
// 内容协商:浏览器自动跳 /setup,API 客户端拿 JSON。
|
|
// / 不在白名单里,所以这里同时覆盖"敲域名根路径"和"敲其他路径"两种场景。
|
|
if acceptsHTML(r.Header.Get("Accept")) {
|
|
w.Header().Set("Location", "/setup")
|
|
w.WriteHeader(http.StatusFound) // 302
|
|
return
|
|
}
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
w.WriteHeader(http.StatusServiceUnavailable)
|
|
_, _ = w.Write([]byte(`{"error":"not installed","code":"install_required","redirect":"/setup"}`))
|
|
})
|
|
}
|
|
}
|
|
|
|
// acceptsHTML 判断客户端是否接受 HTML 响应。
|
|
// 严格匹配:Accept 必须显式包含 text/html 或 text/*,避免通配 */*(curl/API 默认)
|
|
// 走 302 路径影响 API 行为。
|
|
func acceptsHTML(accept string) bool {
|
|
if accept == "" {
|
|
return false
|
|
}
|
|
for _, part := range strings.Split(accept, ",") {
|
|
mt := strings.TrimSpace(part)
|
|
if mt == "" {
|
|
continue
|
|
}
|
|
// 去掉 q= 等参数
|
|
if idx := strings.Index(mt, ";"); idx >= 0 {
|
|
mt = strings.TrimSpace(mt[:idx])
|
|
}
|
|
mt = strings.ToLower(mt)
|
|
if mt == "text/html" || mt == "text/*" {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|