66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
"github.com/gitcodestatic/gitcodestatic/internal/api/handlers"
|
|
"github.com/gitcodestatic/gitcodestatic/internal/service"
|
|
)
|
|
|
|
// Router 路由配置
|
|
type Router struct {
|
|
repoHandler *handlers.RepoHandler
|
|
statsHandler *handlers.StatsHandler
|
|
}
|
|
|
|
// NewRouter 创建路由
|
|
func NewRouter(repoService *service.RepoService, statsService *service.StatsService) *Router {
|
|
return &Router{
|
|
repoHandler: handlers.NewRepoHandler(repoService),
|
|
statsHandler: handlers.NewStatsHandler(statsService),
|
|
}
|
|
}
|
|
|
|
// Setup 设置路由
|
|
func (rt *Router) Setup() http.Handler {
|
|
r := chi.NewRouter()
|
|
|
|
// 中间件
|
|
r.Use(middleware.RequestID)
|
|
r.Use(middleware.RealIP)
|
|
r.Use(middleware.Logger)
|
|
r.Use(middleware.Recoverer)
|
|
|
|
// Health check
|
|
r.Get("/health", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(http.StatusOK)
|
|
w.Write([]byte(`{"status":"healthy"}`))
|
|
})
|
|
|
|
// API routes
|
|
r.Route("/api/v1", func(r chi.Router) {
|
|
// 仓库管理
|
|
r.Route("/repos", func(r chi.Router) {
|
|
r.Post("/batch", rt.repoHandler.AddBatch)
|
|
r.Get("/", rt.repoHandler.List)
|
|
r.Get("/{id}", rt.repoHandler.Get)
|
|
r.Post("/{id}/switch-branch", rt.repoHandler.SwitchBranch)
|
|
r.Post("/{id}/update", rt.repoHandler.Update)
|
|
r.Post("/{id}/reset", rt.repoHandler.Reset)
|
|
r.Delete("/{id}", rt.repoHandler.Delete)
|
|
})
|
|
|
|
// 统计
|
|
r.Route("/stats", func(r chi.Router) {
|
|
r.Post("/calculate", rt.statsHandler.Calculate)
|
|
r.Get("/result", rt.statsHandler.QueryResult)
|
|
r.Get("/commit-count", rt.statsHandler.CountCommits)
|
|
})
|
|
})
|
|
|
|
return r
|
|
}
|