feat: 文字图片传输支持,去掉无用交互和接口

This commit is contained in:
MatrixSeven
2025-08-07 14:03:53 +08:00
parent 7cb0d34fb1
commit baf67742f3
9 changed files with 666 additions and 244 deletions

View File

@@ -2,6 +2,7 @@ package handlers
import (
"encoding/json"
"log"
"net/http"
"chuan/internal/services"
@@ -37,6 +38,7 @@ func (h *Handler) CreateRoomHandler(w http.ResponseWriter, r *http.Request) {
// 创建新房间
code := h.webrtcService.CreateNewRoom()
log.Printf("创建房间成功: %s", code)
// 构建响应
response := map[string]interface{}{
@@ -48,8 +50,8 @@ func (h *Handler) CreateRoomHandler(w http.ResponseWriter, r *http.Request) {
json.NewEncoder(w).Encode(response)
}
// RoomStatusHandler 获取WebRTC房间状态API
func (h *Handler) RoomStatusHandler(w http.ResponseWriter, r *http.Request) {
// WebRTCRoomStatusHandler WebRTC房间状态API
func (h *Handler) WebRTCRoomStatusHandler(w http.ResponseWriter, r *http.Request) {
// 设置响应为JSON格式
w.Header().Set("Content-Type", "application/json")
@@ -61,35 +63,45 @@ func (h *Handler) RoomStatusHandler(w http.ResponseWriter, r *http.Request) {
return
}
// 从查询参数获取房间代码
code := r.URL.Query().Get("code")
if code == "" || len(code) != 6 {
if code == "" {
json.NewEncoder(w).Encode(map[string]interface{}{
"success": false,
"message": "请提供正确的6位房间码",
"message": "缺少房间码",
})
return
}
// 获取WebRTC房间状态
webrtcStatus := h.webrtcService.GetRoomStatus(code)
// 获取房间状态
status := h.webrtcService.GetRoomStatus(code)
// 如果房间不存在,返回不存在状态
if !webrtcStatus["exists"].(bool) {
json.NewEncoder(w).Encode(map[string]interface{}{
"success": false,
"message": "房间不存在",
})
return
}
// 构建响应
response := map[string]interface{}{
"success": true,
"message": "房间状态获取成功",
"sender_online": webrtcStatus["sender_online"],
"receiver_online": webrtcStatus["receiver_online"],
"created_at": webrtcStatus["created_at"],
}
json.NewEncoder(w).Encode(response)
json.NewEncoder(w).Encode(status)
}
// GetRoomStatusHandler 获取房间状态API
func (h *Handler) GetRoomStatusHandler(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
if r.Method != http.MethodGet {
json.NewEncoder(w).Encode(map[string]interface{}{
"success": false,
"message": "方法不允许",
})
return
}
// 获取房间码
code := r.URL.Query().Get("code")
if code == "" {
json.NewEncoder(w).Encode(map[string]interface{}{
"success": false,
"message": "房间码不能为空",
})
return
}
// 获取房间状态
status := h.webrtcService.GetRoomStatus(code)
json.NewEncoder(w).Encode(status)
}

View File

@@ -237,8 +237,9 @@ func (ws *WebRTCService) CreateNewRoom() string {
// generatePickupCode 生成6位取件码
func (ws *WebRTCService) generatePickupCode() string {
rand.Seed(time.Now().UnixNano())
code := rand.Intn(900000) + 100000
source := rand.NewSource(time.Now().UnixNano())
rng := rand.New(source)
code := rng.Intn(900000) + 100000
return fmt.Sprintf("%d", code)
}
@@ -312,11 +313,14 @@ func (ws *WebRTCService) GetRoomStatus(code string) map[string]interface{} {
room := ws.rooms[code]
if room == nil {
return map[string]interface{}{
"exists": false,
"success": false,
"exists": false,
"message": "房间不存在或已过期",
}
}
return map[string]interface{}{
"success": true,
"exists": true,
"sender_online": room.Sender != nil,
"receiver_online": room.Receiver != nil,