feat:UI大调整,WEBRTC切换

internal/handlers/handlers.go

@@ -6,19 +6,20 @@ import (
 	"net/http"
 	"path/filepath"
 
-	"chuan/internal/models"
 	"chuan/internal/services"
 )
 
 type Handler struct {
-	p2pService *services.P2PService
-	templates  map[string]*template.Template
+	p2pService    *services.P2PService
+	webrtcService *services.WebRTCService
+	templates     map[string]*template.Template
 }
 
 func NewHandler(p2pService *services.P2PService) *Handler {
 	h := &Handler{
-		p2pService: p2pService,
-		templates:  make(map[string]*template.Template),
+		p2pService:    p2pService,
+		webrtcService: services.NewWebRTCService(),
+		templates:     make(map[string]*template.Template),
 	}
 
 	// 加载模板
@@ -65,127 +66,6 @@ func (h *Handler) IndexHandler(w http.ResponseWriter, r *http.Request) {
 	}
 }
 
-// CreateRoomHandler 创建房间API
-func (h *Handler) CreateRoomHandler(w http.ResponseWriter, r *http.Request) {
-	if r.Method != http.MethodPost {
-		http.Error(w, "方法不允许", http.StatusMethodNotAllowed)
-		return
-	}
-
-	var req struct {
-		Files []models.FileTransferInfo `json:"files"`
-	}
-
-	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
-		http.Error(w, "解析请求失败", http.StatusBadRequest)
-		return
-	}
-
-	// 创建房间
-	code := h.p2pService.CreateRoom(req.Files)
-
-	response := map[string]interface{}{
-		"success": true,
-		"code":    code,
-		"message": "房间创建成功",
-	}
-
-	w.Header().Set("Content-Type", "application/json")
-	json.NewEncoder(w).Encode(response)
-}
-
-// GetRoomInfoHandler 获取房间信息API
-func (h *Handler) GetRoomInfoHandler(w http.ResponseWriter, r *http.Request) {
-	code := r.URL.Query().Get("code")
-	if code == "" {
-		http.Error(w, "缺少取件码", http.StatusBadRequest)
-		return
-	}
-
-	room, exists := h.p2pService.GetRoomByCode(code)
-	if !exists {
-		response := map[string]interface{}{
-			"success": false,
-			"message": "取件码不存在或已过期",
-		}
-		w.Header().Set("Content-Type", "application/json")
-		json.NewEncoder(w).Encode(response)
-		return
-	}
-
-	response := map[string]interface{}{
-		"success": true,
-		"files":   room.Files,
-		"message": "房间信息获取成功",
-	}
-
-	w.Header().Set("Content-Type", "application/json")
-	json.NewEncoder(w).Encode(response)
-}
-
-// GetRoomStatusHandler 获取房间状态API
-func (h *Handler) GetRoomStatusHandler(w http.ResponseWriter, r *http.Request) {
-	code := r.URL.Query().Get("code")
-	if code == "" {
-		http.Error(w, "缺少取件码", http.StatusBadRequest)
-		return
-	}
-
-	status, exists := h.p2pService.GetRoomStatusByCode(code)
-	if !exists {
-		response := map[string]interface{}{
-			"success": false,
-			"message": "取件码不存在或已过期",
-		}
-		w.Header().Set("Content-Type", "application/json")
-		json.NewEncoder(w).Encode(response)
-		return
-	}
-
-	response := map[string]interface{}{
-		"success": true,
-		"status":  status,
-		"message": "房间状态获取成功",
-	}
-
-	w.Header().Set("Content-Type", "application/json")
-	json.NewEncoder(w).Encode(response)
-}
-
-// UpdateRoomFilesHandler 更新房间文件列表API
-func (h *Handler) UpdateRoomFilesHandler(w http.ResponseWriter, r *http.Request) {
-	if r.Method != http.MethodPost {
-		http.Error(w, "方法不允许", http.StatusMethodNotAllowed)
-		return
-	}
-
-	var req struct {
-		Code  string                    `json:"code"`
-		Files []models.FileTransferInfo `json:"files"`
-	}
-
-	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
-		http.Error(w, "解析请求失败", http.StatusBadRequest)
-		return
-	}
-
-	// 更新房间文件列表
-	success := h.p2pService.UpdateRoomFiles(req.Code, req.Files)
-
-	response := map[string]interface{}{
-		"success": success,
-	}
-
-	if success {
-		response["message"] = "文件列表更新成功"
-	} else {
-		response["message"] = "房间不存在或更新失败"
-	}
-
-	w.Header().Set("Content-Type", "application/json")
-	json.NewEncoder(w).Encode(response)
-}
-
 // CreateTextRoomHandler 创建文字传输房间API
 func (h *Handler) CreateTextRoomHandler(w http.ResponseWriter, r *http.Request) {
 	if r.Method != http.MethodPost {
@@ -255,7 +135,7 @@ func (h *Handler) GetTextContentHandler(w http.ResponseWriter, r *http.Request)
 	json.NewEncoder(w).Encode(response)
 }
 
-// HandleP2PWebSocket 处理P2P WebSocket连接
-func (h *Handler) HandleP2PWebSocket(w http.ResponseWriter, r *http.Request) {
-	h.p2pService.HandleWebSocket(w, r)
+// HandleWebRTCWebSocket 处理WebRTC信令WebSocket连接
+func (h *Handler) HandleWebRTCWebSocket(w http.ResponseWriter, r *http.Request) {
+	h.webrtcService.HandleWebSocket(w, r)
 }