- {/* 模式切换 */}
-
- );
-}
diff --git a/chuan-next/src/hooks/useWebSocket-new.ts b/chuan-next/src/hooks/useWebSocket-new.ts
deleted file mode 100644
index 637c614..0000000
--- a/chuan-next/src/hooks/useWebSocket-new.ts
+++ /dev/null
@@ -1,119 +0,0 @@
-"use client";
-
-import { useState, useEffect, useCallback, useRef } from 'react';
-import { UseWebSocketReturn, WebSocketMessage } from '@/types';
-
-export function useWebSocket(): UseWebSocketReturn {
- const [websocket, setWebsocket] = useState
-
-
- {mode === 'send' ? (
-
-
-
-
-
-
- ) : (
-
-
-
-
-
- 传送文字
-- {isRoomCreated ? '实时编辑，对方可以同步看到' : '输入要传输的文本内容'} -
- {connectedUsers > 1 && ( -
-
- )}
-
-
-
-
- {isRoomCreated && (
-
-
-
-
- )}
-
-
- 实时同步
-
-
- {textContent.length} 字符
- 最大 50,000 字符
-
-
- {!isRoomCreated ? (
-
- ) : (
-
-
- )}
-
- {/* 图片展示区域 */}
- {images.length > 0 && (
-
-
-
-
- 房间码
-{roomCode}
-
-
-
-
-
-
- )}
-
-
-
- {images.map((img, index) => (
-
-
- 
window.open(img, '_blank')}
- />
-
-
- ))}
-
-
- )}
-
-
-
-
-
- 接收文字
-输入6位房间码来获取文字内容
-
- setRoomCode(e.target.value.toUpperCase().slice(0, 6))}
- placeholder="请输入房间码"
- className="text-center text-2xl sm:text-3xl tracking-[0.3em] sm:tracking-[0.5em] font-mono h-12 sm:h-16 border-2 border-slate-200 rounded-xl focus:border-emerald-500 focus:ring-emerald-500 bg-white/80 backdrop-blur-sm"
- maxLength={6}
- disabled={isLoading}
- />
-
-
-
- {(receivedText || textContent) && (
-
-
-
- )}
-
- {/* 接收到的图片展示 */}
- {images.length > 0 && (
-
-
- {currentRole === 'receiver' && isConnected && (
-
-
-
-
- )}
-
-
- 实时同步
-
-
-
-
-
- )}
-
-
-
- {images.map((img, index) => (
-
-
- window.open(img, '_blank')}
- />
-
-
- ))}
-
+
+
+
+ `))
+}
+
+// spaHandler SPA 应用处理器
+type spaHandler struct {
+ fs fs.FS
+}
+
+func (h *spaHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+ // 清理路径
+ upath := strings.TrimPrefix(r.URL.Path, "/")
+ if upath == "" {
+ upath = "index.html"
+ }
+
+ // 尝试打开请求的文件
+ file, err := h.fs.Open(upath)
+ if err != nil {
+ // 文件不存在，对于 SPA 应用返回 index.html
+ h.serveIndexHTML(w, r)
+ return
+ }
+ defer file.Close()
+
+ // 获取文件信息
+ stat, err := file.Stat()
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusInternalServerError)
+ return
+ }
+
+ // 如果是目录，查找 index.html
+ if stat.IsDir() {
+ indexPath := path.Join(upath, "index.html")
+ indexFile, err := h.fs.Open(indexPath)
+ if err != nil {
+ h.serveIndexHTML(w, r)
+ return
+ }
+ defer indexFile.Close()
+
+ h.serveFile(w, r, "index.html", indexFile)
+ return
+ }
+
+ // 服务静态文件
+ h.serveFile(w, r, stat.Name(), file)
+}
+
+// serveIndexHTML 服务 index.html 文件
+func (h *spaHandler) serveIndexHTML(w http.ResponseWriter, r *http.Request) {
+ file, err := h.fs.Open("index.html")
+ if err != nil {
+ http.NotFound(w, r)
+ return
+ }
+ defer file.Close()
+
+ h.serveFile(w, r, "index.html", file)
+}
+
+// serveFile 服务文件
+func (h *spaHandler) serveFile(w http.ResponseWriter, r *http.Request, filename string, file fs.File) {
+ // 设置 Content-Type
+ setContentType(w, filename)
+
+ // 获取文件信息
+ stat, err := file.Stat()
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusInternalServerError)
+ return
+ }
+
+ // 读取文件内容
+ content, err := io.ReadAll(file)
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusInternalServerError)
+ return
+ }
+
+ // 设置缓存头
+ if shouldCache(filename) {
+ w.Header().Set("Cache-Control", "public, max-age=31536000") // 1年
+ } else {
+ w.Header().Set("Cache-Control", "no-cache")
+ }
+
+ // 服务内容
+ http.ServeContent(w, r, filename, stat.ModTime(), strings.NewReader(string(content)))
+}
+
+// setContentType 设置 Content-Type
+func setContentType(w http.ResponseWriter, filename string) {
+ ext := path.Ext(filename)
+ switch ext {
+ case ".html":
+ w.Header().Set("Content-Type", "text/html; charset=utf-8")
+ case ".css":
+ w.Header().Set("Content-Type", "text/css; charset=utf-8")
+ case ".js":
+ w.Header().Set("Content-Type", "application/javascript; charset=utf-8")
+ case ".json":
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ case ".png":
+ w.Header().Set("Content-Type", "image/png")
+ case ".jpg", ".jpeg":
+ w.Header().Set("Content-Type", "image/jpeg")
+ case ".gif":
+ w.Header().Set("Content-Type", "image/gif")
+ case ".svg":
+ w.Header().Set("Content-Type", "image/svg+xml")
+ case ".ico":
+ w.Header().Set("Content-Type", "image/x-icon")
+ case ".woff":
+ w.Header().Set("Content-Type", "font/woff")
+ case ".woff2":
+ w.Header().Set("Content-Type", "font/woff2")
+ case ".ttf":
+ w.Header().Set("Content-Type", "font/ttf")
+ case ".txt":
+ w.Header().Set("Content-Type", "text/plain; charset=utf-8")
+ default:
+ w.Header().Set("Content-Type", "application/octet-stream")
+ }
+}
+
+// shouldCache 判断文件是否应该被缓存
+func shouldCache(filename string) bool {
+ ext := path.Ext(filename)
+ switch ext {
+ case ".css", ".js", ".png", ".jpg", ".jpeg", ".gif", ".svg", ".ico", ".woff", ".woff2", ".ttf":
+ return true
+ default:
+ return false
+ }
+}
diff --git a/internal/web/static.go b/internal/web/static.go
new file mode 100644
index 0000000..55a7a8a
--- /dev/null
+++ b/internal/web/static.go
@@ -0,0 +1,52 @@
+package web
+
+import (
+ "embed"
+ "io/fs"
+ "net/http"
+)
+
+// 嵌入静态文件
+//
+//go:embed all:static
+var StaticFiles embed.FS
+
+// StaticFileServer 创建静态文件服务器
+func StaticFileServer() http.Handler {
+ // 获取嵌入的文件系统
+ staticFS, err := fs.Sub(StaticFiles, "static")
+ if err != nil {
+ panic(err)
+ }
+
+ return http.FileServer(http.FS(staticFS))
+}
+
+// FrontendFileServer 创建前端文件服务器
+func FrontendFileServer() http.Handler {
+ return &frontendHandler{}
+}
+
+// frontendHandler 处理前端文件请求，支持 SPA 路由
+type frontendHandler struct{}
+
+func (h *frontendHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+ // 返回一个简单的占位页面
+ w.Header().Set("Content-Type", "text/html; charset=utf-8")
+ w.WriteHeader(http.StatusOK)
+ w.Write([]byte(`
+
+
+
+ 🚀 文件传输服务
+ +
+ ⚠️ 前端界面未构建，当前显示的是后端 API 服务。
+
+
+ 📋 可用的 API 接口
+
+
+
+ POST /api/create-room - 创建文件传输房间
+ GET /api/room-info - 获取房间信息
+ GET /api/room-status - 获取房间状态
+ POST /api/update-room-files - 更新房间文件
+ POST /api/create-text-room - 创建文本传输房间
+ GET /api/get-text-content/* - 获取文本内容
+ WebSocket /ws/p2p - P2P 连接
+ 🛠️ 构建前端
+
+
+
+ # 进入前端目录 +cd chuan-next + +# 安装依赖 +yarn install + +# 构建静态文件 +yarn build:ssg + +# 重新构建 Go 项目以嵌入前端文件 +cd .. +go build -o file-transfer-server ./cmd+
提示: 构建完成后刷新页面即可看到完整的前端界面。
+文件传输服务
+前端文件未嵌入，请先构建前端项目。
+运行以下命令构建前端：
+cd chuan-next && yarn build:ssg+ + + `)) +} diff --git a/internal/web/static/.gitkeep b/internal/web/static/.gitkeep new file mode 100644 index 0000000..ee68a30 --- /dev/null +++ b/internal/web/static/.gitkeep @@ -0,0 +1,3 @@ +# Static Files Directory + +This directory contains static files that will be embedded into the Go binary.