file-transfer-go/cmd/router.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)
}