Scoreboard/server/internal/httpapi/server.go

// Package httpapi 提供 REST 接口、静态资源托管与统一响应结构。
package httpapi

import (
	"context"
	"database/sql"
	"fmt"
	"io/fs"
	"log/slog"
	"net/http"
	"path/filepath"
	"strings"
	"time"

	"git.yeij.top/AskaEth/Scoreboard/server/internal/auth"
	"git.yeij.top/AskaEth/Scoreboard/server/internal/export"
	"git.yeij.top/AskaEth/Scoreboard/server/internal/middleware"
	"git.yeij.top/AskaEth/Scoreboard/server/internal/webui"
)

// DisplaySender 把大屏控制命令推给实时层；未接入时命令只记审计、不影响接口可用性。
type DisplaySender interface {
	// SendDisplayCommand 下发一条大屏指令。targets 为空表示广播给所有大屏（公告靠它），
	// 否则只发给这些实例 id（GET /api/display/instances 里的 id），并返回命中的实例数。
	SendDisplayCommand(action string, payload any, targets []string) (matched int, err error)
}

// Options 是构建路由所需的依赖。
type Options struct {
	Logger  *slog.Logger
	Version string
	// System 是系统库连接（users/sessions）。
	System *sql.DB
	// Event 是赛事库连接（teams/events/results/ledger/audit…）。
	Event *sql.DB
	// Auth 可选；为空时用 System 自动构建。
	Auth *auth.Service
	// DataDir 用于备份目录 data/backups。
	DataDir string
	// Backup 可选，手动/发布前备份共用的服务；为空时按 System/Event/DataDir 自动构建。
	Backup *export.Service
	// Shutdown 可选，由 main 注入：触发与 SIGINT/SIGTERM 相同的优雅关闭流程。
	// 为空时 POST /api/shutdown 返回 503，避免把「没接关闭」误报成已受理。
	Shutdown func()
	// Display 可选，实时层接入后传入。
	Display DisplaySender
	// Realtime 可选，实时层接入后传入：写事务提交后推送变更，并提供在线大屏实例。
	Realtime Realtime
	// WS 可选，实时层的 WebSocket 处理器，挂到 /ws。
	WS http.Handler
	// ProjectDir 可选，当前赛事项目目录（含 uploads/），用于赛事包导出。
	ProjectDir string
	// WebFS 可选，默认使用内嵌前端资源。
	WebFS fs.FS
}

type server struct {
	logger     *slog.Logger
	version    string
	sys        *sql.DB
	ev         *sql.DB
	auth       *auth.Service
	dataDir    string
	projectDir string
	backups    *export.Service
	shutdown   func()
	display    DisplaySender
	realtime   Realtime
	idem       *idempotencyStore
}

// New 组装并返回服务端的 HTTP 处理器。
func New(opts Options) http.Handler {
	logger := opts.Logger
	if logger == nil {
		logger = slog.Default()
	}
	svc := opts.Auth
	if svc == nil {
		svc = auth.New(opts.System, auth.Options{Logger: logger})
	}
	backups := opts.Backup
	if backups == nil {
		dir := ""
		if strings.TrimSpace(opts.DataDir) != "" {
			dir = filepath.Join(opts.DataDir, "backups")
		}
		backups = export.NewService(opts.System, opts.Event, dir, 0, logger)
	}

	s := &server{
		logger:     logger,
		version:    opts.Version,
		sys:        opts.System,
		ev:         opts.Event,
		auth:       svc,
		dataDir:    opts.DataDir,
		projectDir: opts.ProjectDir,
		backups:    backups,
		shutdown:   opts.Shutdown,
		display:    opts.Display,
		realtime:   opts.Realtime,
		idem:       newIdempotencyStore(10 * time.Minute),
	}

	mux := http.NewServeMux()
	s.routes(mux)
	s.exportRoutes(mux)
	s.projectRoutes(mux)
	s.rulesRoutes(mux)
	s.systemRoutes(mux)
	if opts.WS != nil {
		mux.Handle("/ws", opts.WS)
	}

	assets := opts.WebFS
	if assets == nil {
		var err error
		assets, err = webui.FS()
		if err != nil {
			logger.Error("前端资源不可用", "err", err)
		}
	}
	if assets != nil {
		mux.Handle("/", staticHandler(assets))
	}

	var h http.Handler = mux
	h = s.broadcastChanges(h)
	h = middleware.NewAuditor(opts.Event, logger).Middleware(h)
	h = middleware.RateLimitWrite(svc, logger)(h)
	h = middleware.CSRF(logger, h)
	h = middleware.MustChangePassword(logger, h)
	h = middleware.Authenticate(svc, logger)(h)
	h = middleware.Log(logger)(h)
	h = middleware.Recover(logger)(h)
	h = middleware.RequestID(h)
	return h
}

func middlewareUserID(r *http.Request) int64 {
	u := middleware.User(r.Context())
	if u.Role == auth.RoleGuest {
		return 0
	}
	return u.ID
}

// public 注册公开接口（guest 可读）。
func (s *server) public(fn handler) http.HandlerFunc { return s.handle(fn) }

// read 注册「guest 也不可读」的接口，权限不足一律 403。
func (s *server) read(role auth.Role, fn handler) http.HandlerFunc {
	return middleware.RequireRole(s.logger, role)(s.handle(fn)).ServeHTTP
}

// write 注册写接口：未登录 401，角色不足 403；成功后按 X-Request-Id 支持幂等回放。
func (s *server) write(role auth.Role, fn handler) http.HandlerFunc {
	return middleware.RequireWrite(s.logger, role)(s.cached(s.handle(fn))).ServeHTTP
}

// writeVoid 注册无返回体的写接口。
func (s *server) writeVoid(role auth.Role, fn func(w http.ResponseWriter, r *http.Request) error) http.HandlerFunc {
	return middleware.RequireWrite(s.logger, role)(s.cached(s.handleVoid(fn))).ServeHTTP
}

func (s *server) routes(mux *http.ServeMux) {
	// 健康检查不走 {ok,data} 包装（05 第 7 节）。
	//
	// 语义分两层，都写进响应体的不同字段，谁也不覆盖谁：
	//   ok      —— HTTP 服务活着（Console 拿状态码当存活信号，这里恒回 200）；
	//   db      —— 数据库能应答（新增，之前不探库，「进程活着但连不上库」会被当成健康）。
	// 状态码保持 200：Console 判定不健康会重启进程，而数据库故障重启治不了，
	// 只会变成重启循环。让守护脚本读 db 字段自行决策，比替它做决定安全。
	// 不返回最近备份时间：本接口会被 Console 高频轮询，不做目录扫描。
	mux.HandleFunc("GET /api/health", func(w http.ResponseWriter, r *http.Request) {
		ctx, cancel := context.WithTimeout(r.Context(), healthProbeTimeout)
		defer cancel()
		resp := map[string]any{"ok": true, "version": s.version, "db": "ok"}
		if err := s.probeDatabases(ctx); err != nil {
			resp["db"] = "error"
			resp["dbError"] = err.Error()
		}
		writeJSON(w, http.StatusOK, resp)
	})

	// 认证
	mux.HandleFunc("POST /api/auth/login", s.handle(s.login))
	mux.HandleFunc("POST /api/auth/logout", s.handle(s.logout))
	mux.HandleFunc("GET /api/auth/me", s.handle(s.me))
	mux.HandleFunc("POST /api/auth/password", s.handle(s.changePassword))

	// 系统配置（guest 一律 403）
	mux.HandleFunc("GET /api/settings", s.read(auth.RoleAdmin, s.getSettingsHandler))
	mux.HandleFunc("PUT /api/settings", s.write(auth.RoleAdmin, s.updateSettingsHandler))

	// 基础配置
	mux.HandleFunc("GET /api/teams", s.public(s.listTeamsHandler))
	mux.HandleFunc("POST /api/teams", s.write(auth.RoleRecorder, s.createTeamHandler))
	mux.HandleFunc("GET /api/teams/{id}", s.public(s.getTeamHandler))
	mux.HandleFunc("PUT /api/teams/{id}", s.write(auth.RoleRecorder, s.updateTeamHandler))
	mux.HandleFunc("DELETE /api/teams/{id}", s.writeVoid(auth.RoleRecorder, s.deleteTeamHandler))

	mux.HandleFunc("GET /api/sports", s.public(s.listSportsHandler))
	mux.HandleFunc("POST /api/sports", s.write(auth.RoleRecorder, s.createSportHandler))
	mux.HandleFunc("GET /api/sports/{id}", s.public(s.getSportHandler))
	mux.HandleFunc("PUT /api/sports/{id}", s.write(auth.RoleRecorder, s.updateSportHandler))
	mux.HandleFunc("DELETE /api/sports/{id}", s.writeVoid(auth.RoleRecorder, s.deleteSportHandler))

	mux.HandleFunc("GET /api/events", s.public(s.listEventsHandler))
	mux.HandleFunc("POST /api/events", s.write(auth.RoleRecorder, s.createEventHandler))
	mux.HandleFunc("GET /api/events/{id}", s.public(s.getEventHandler))
	mux.HandleFunc("PUT /api/events/{id}", s.write(auth.RoleRecorder, s.updateEventHandler))
	mux.HandleFunc("DELETE /api/events/{id}", s.writeVoid(auth.RoleRecorder, s.deleteEventHandler))
	mux.HandleFunc("GET /api/events/{id}/entries", s.public(s.listEntriesHandler))
	mux.HandleFunc("GET /api/events/{id}/categories", s.public(s.listCategoriesHandler))
	mux.HandleFunc("PUT /api/events/{id}/categories", s.write(auth.RoleRecorder, s.replaceCategoriesHandler))
	mux.HandleFunc("GET /api/events/{id}/channels", s.public(s.listChannelsHandler))
	mux.HandleFunc("PUT /api/events/{id}/channels", s.write(auth.RoleRecorder, s.replaceChannelsHandler))

	mux.HandleFunc("GET /api/scoring-profiles", s.public(s.listProfilesHandler))
	mux.HandleFunc("POST /api/scoring-profiles", s.write(auth.RoleRecorder, s.createProfileHandler))
	mux.HandleFunc("GET /api/scoring-profiles/{id}", s.public(s.getProfileHandler))
	mux.HandleFunc("PUT /api/scoring-profiles/{id}", s.write(auth.RoleRecorder, s.updateProfileHandler))

	mux.HandleFunc("GET /api/athletes", s.public(s.listAthletesHandler))
	mux.HandleFunc("POST /api/athletes", s.write(auth.RoleRecorder, s.createAthleteHandler))
	mux.HandleFunc("GET /api/athletes/{id}", s.public(s.getAthleteHandler))
	mux.HandleFunc("PUT /api/athletes/{id}", s.write(auth.RoleRecorder, s.updateAthleteHandler))
	mux.HandleFunc("DELETE /api/athletes/{id}", s.write(auth.RoleRecorder, s.deleteAthleteHandler))

	mux.HandleFunc("POST /api/entries/import", s.write(auth.RoleRecorder, s.importEntriesHandler))
	mux.HandleFunc("POST /api/entries/import/resolve", s.write(auth.RoleRecorder, s.resolveImportHandler))
	// 批次（heat）是录入 / 展示分组，不参与算分，所以没有独立的「发布」概念。
	mux.HandleFunc("PUT /api/entries/heat", s.write(auth.RoleRecorder, s.setEntriesHeatHandler))

	// 成绩
	mux.HandleFunc("GET /api/results", s.public(s.listResultsHandler))
	mux.HandleFunc("PUT /api/results/draft", s.write(auth.RoleRecorder, s.saveDraftHandler))
	mux.HandleFunc("POST /api/results/publish", s.write(auth.RoleRecorder, s.publishHandler))
	mux.HandleFunc("POST /api/results/unpublish", s.write(auth.RoleRecorder, s.unpublishHandler))
	mux.HandleFunc("POST /api/results/{id}/void", s.write(auth.RoleRecorder, s.voidResultHandler))

	// 积分与排名
	mux.HandleFunc("GET /api/standings", s.public(s.standingsHandler))
	mux.HandleFunc("GET /api/standings/athletes", s.public(s.athleteStandingsHandler))
	mux.HandleFunc("GET /api/standings/teams/{id}/breakdown", s.public(s.breakdownHandler))
	mux.HandleFunc("GET /api/adjustments", s.public(s.listAdjustmentsHandler))
	mux.HandleFunc("POST /api/adjustments", s.write(auth.RoleRecorder, s.createAdjustmentHandler))
	mux.HandleFunc("POST /api/adjustments/{id}/void", s.write(auth.RoleRecorder, s.voidAdjustmentHandler))

	// 纪录
	mux.HandleFunc("GET /api/records", s.public(s.listRecordsHandler))
	mux.HandleFunc("POST /api/record-breaks", s.write(auth.RoleRecorder, s.createRecordBreakHandler))
	mux.HandleFunc("POST /api/record-breaks/{id}/approve", s.write(auth.RoleRecorder, s.approveRecordBreakHandler))

	// 大屏控制与运维
	mux.HandleFunc("POST /api/display/command", s.write(auth.RoleRecorder, s.displayCommandHandler))
	mux.HandleFunc("GET /api/display/instances", s.read(auth.RoleRecorder, s.displayInstancesHandler))
	// 「实时赛况」页的当前项目与已发布成绩；大屏（guest）要能直接拉，故对外只读开放。
	mux.HandleFunc("GET /api/display/live", s.public(s.liveEventsHandler))
	mux.HandleFunc("GET /api/audit-logs", s.read(auth.RoleAdmin, s.listAuditLogsHandler))
	mux.HandleFunc("POST /api/backup", s.write(auth.RoleAdmin, s.backupHandler))

	// 用户管理（系统配置，仅 admin；guest 读 403、写 401）
	mux.HandleFunc("GET /api/users", s.read(auth.RoleAdmin, s.listUsersHandler))
	mux.HandleFunc("POST /api/users", s.write(auth.RoleAdmin, s.createUserHandler))
	mux.HandleFunc("PUT /api/users/{id}", s.write(auth.RoleAdmin, s.updateUserHandler))
	mux.HandleFunc("POST /api/users/{id}/password", s.write(auth.RoleAdmin, s.resetUserPasswordHandler))

	// 变更历史（recorder 及以上可读）
	mux.HandleFunc("GET /api/revisions", s.read(auth.RoleRecorder, s.listRevisionsHandler))
	mux.HandleFunc("GET /api/revisions/{id}", s.read(auth.RoleRecorder, s.getRevisionDetailHandler))

	// 图片上传与静态访问（会标/背景/单位图标）。上传落 <ProjectDir>/uploads/，随赛事包导出；
	// 静态访问要匿名放行，大屏的 <img> 才能不带会话直接取图。
	mux.HandleFunc("POST /api/uploads", s.write(auth.RoleRecorder, s.uploadHandler))
	mux.HandleFunc("GET /uploads/{name}", s.serveUpload)

	// 字体：4 款内置字体（前端 @font-face）由 GET /api/fonts 列出，上传的落
	// <项目目录>/uploads/fonts/，随赛事包导出；静态访问同样是匿名放行——大屏的
	// @font-face 不带会话。字体目录比图片多一层，静态路由必须单独注册。
	mux.HandleFunc("GET /api/fonts", s.read(auth.RoleRecorder, s.listFontsHandler))
	mux.HandleFunc("POST /api/fonts", s.write(auth.RoleAdmin, s.uploadFontHandler))
	mux.HandleFunc("DELETE /api/fonts/{name}", s.writeVoid(auth.RoleAdmin, s.deleteFontHandler))
	mux.HandleFunc("GET /uploads/fonts/{name}", s.serveFont)

	// 由其他模块负责的接口：先占位，避免落到静态资源处理器。
	mux.HandleFunc("/api/export/", s.notWired("导出"))
	mux.HandleFunc("/api/rules/", s.notWired("规则导入导出"))
	mux.HandleFunc("/api/", func(w http.ResponseWriter, r *http.Request) {
		s.writeAPIError(w, r, notFound("接口不存在"))
	})
}

// notWired 为待接线的接口返回 501，避免请求被静态资源兜底成 HTML。
func (s *server) notWired(name string) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		s.writeAPIError(w, r, notImplemented(name+"接口尚未接入"))
	}
}

// healthProbeTimeout 是健康检查探库的超时。
// 取 5s 与 busy_timeout 同量级：库被备份、重算这类长事务占着单连接时，
// 探针宁可等一会儿，也不要报出一个假的「数据库故障」。
const healthProbeTimeout = 5 * time.Second

// probeDatabases 探活系统库与赛事库，返回第一个出错的原因。
// 用 `SELECT 1` 而不是 Ping：sqlite 驱动的 Ping 可能只到连接池就返回，
// 探不到「文件还在、库已经读不动」这种情况。
func (s *server) probeDatabases(ctx context.Context) error {
	for _, target := range []struct {
		name string
		db   *sql.DB
	}{{"system", s.sys}, {"event", s.ev}} {
		if target.db == nil {
			continue
		}
		var one int
		if err := target.db.QueryRowContext(ctx, `SELECT 1`).Scan(&one); err != nil {
			return fmt.Errorf("%s 库无响应: %w", target.name, err)
		}
	}
	return nil
}

// staticHandler 把 URL 映射到多页面前端产物。
//
//	/            → index.html
//	/display     → display.html
//	/admin       → admin.html
//	/其他         → 按文件路径查找
func staticHandler(assets fs.FS) http.Handler {
	files := http.FileServerFS(assets)

	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		p := strings.TrimPrefix(r.URL.Path, "/")

		switch p {
		case "", "index.html":
			http.ServeFileFS(w, r, assets, "index.html")
		case "display", "display/":
			http.ServeFileFS(w, r, assets, "display.html")
		case "admin", "admin/":
			http.ServeFileFS(w, r, assets, "admin.html")
		default:
			if _, err := fs.Stat(assets, p); err != nil {
				http.NotFound(w, r)
				return
			}
			files.ServeHTTP(w, r)
		}
	})
}