mirror of
https://github.com/MatrixSeven/file-transfer-go.git
synced 2026-02-15 01:34:46 +08:00
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"chuan/internal/handlers"
|
|
"chuan/internal/web"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
"github.com/go-chi/cors"
|
|
)
|
|
|
|
// setupRouter 设置路由和中间件
|
|
func setupRouter() http.Handler {
|
|
// 初始化处理器
|
|
h := handlers.NewHandler()
|
|
|
|
router := chi.NewRouter()
|
|
|
|
// 设置中间件
|
|
setupMiddleware(router)
|
|
|
|
// 设置API路由
|
|
setupAPIRoutes(router, h)
|
|
|
|
// 设置前端路由
|
|
router.Handle("/*", web.CreateFrontendHandler())
|
|
|
|
return router
|
|
}
|
|
|
|
// setupMiddleware 设置中间件
|
|
func setupMiddleware(r *chi.Mux) {
|
|
r.Use(middleware.Logger)
|
|
r.Use(middleware.Recoverer)
|
|
r.Use(middleware.Compress(5))
|
|
|
|
// CORS 配置
|
|
r.Use(cors.Handler(cors.Options{
|
|
AllowedOrigins: []string{"*"},
|
|
AllowedMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
|
|
AllowedHeaders: []string{"Accept", "Authorization", "Content-Type", "X-CSRF-Token"},
|
|
ExposedHeaders: []string{"Link"},
|
|
AllowCredentials: true,
|
|
MaxAge: 300,
|
|
}))
|
|
}
|
|
|
|
// setupAPIRoutes 设置API路由
|
|
func setupAPIRoutes(r *chi.Mux, h *handlers.Handler) {
|
|
// WebRTC信令WebSocket路由
|
|
r.Get("/api/ws/webrtc", h.HandleWebRTCWebSocket)
|
|
|
|
// WebRTC房间API
|
|
r.Post("/api/create-room", h.CreateRoomHandler)
|
|
r.Get("/api/room-info", h.WebRTCRoomStatusHandler)
|
|
}
|