第一版本

internal/handlers/handlers.go

@@ -0,0 +1,129 @@
+package handlers
+
+import (
+	"encoding/json"
+	"html/template"
+	"net/http"
+	"path/filepath"
+
+	"chuan/internal/models"
+	"chuan/internal/services"
+)
+
+type Handler struct {
+	p2pService *services.P2PService
+	templates  map[string]*template.Template
+}
+
+func NewHandler(p2pService *services.P2PService) *Handler {
+	h := &Handler{
+		p2pService: p2pService,
+		templates:  make(map[string]*template.Template),
+	}
+
+	// 加载模板
+	h.loadTemplates()
+	return h
+}
+
+// 加载模板
+func (h *Handler) loadTemplates() {
+	templateDir := "web/templates"
+
+	// 加载基础模板
+	baseTemplate := filepath.Join(templateDir, "base.html")
+
+	// 加载各个页面模板
+	templates := []string{"index.html"}
+
+	for _, tmplName := range templates {
+		tmplPath := filepath.Join(templateDir, tmplName)
+		tmpl, err := template.ParseFiles(baseTemplate, tmplPath)
+		if err != nil {
+			panic("加载模板失败: " + err.Error())
+		}
+		h.templates[tmplName] = tmpl
+		println("模板加载成功:", tmplName)
+	}
+}
+
+// IndexHandler 首页处理器
+func (h *Handler) IndexHandler(w http.ResponseWriter, r *http.Request) {
+	tmpl, exists := h.templates["index.html"]
+	if !exists {
+		http.Error(w, "模板不存在", http.StatusInternalServerError)
+		return
+	}
+
+	data := map[string]interface{}{
+		"Title": "P2P文件传输",
+	}
+
+	if err := tmpl.Execute(w, data); err != nil {
+		http.Error(w, "渲染模板失败: "+err.Error(), http.StatusInternalServerError)
+		return
+	}
+}
+
+// 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)
+}
+
+// HandleP2PWebSocket 处理P2P WebSocket连接
+func (h *Handler) HandleP2PWebSocket(w http.ResponseWriter, r *http.Request) {
+	h.p2pService.HandleWebSocket(w, r)
+}