🚀 文件传输服务
⚠️ 前端界面未构建，当前显示的是后端 API 服务。
� 环境变量配置
FRONTEND_DIR - 指定外部前端文件目录
PORT - 自定义服务端口 (默认: 8080)
示例:
export FRONTEND_DIR=/path/to/frontend
export PORT=3000
./file-transfer-server
�📋 可用的 API 接口
POST /api/create-room - 创建WebRTC房间
GET /api/room-info - 获取房间信息
WebSocket /api/ws/webrtc - WebRTC 信令连接
🛠️ 构建前端
# 进入前端目录
cd chuan-next
# 安装依赖
npm install
# 构建静态文件
npm run build
# 方法1: 重新构建 Go 项目以嵌入前端文件
cd ..
go build -o file-transfer-server ./cmd
# 方法2: 使用外部前端目录
export FRONTEND_DIR=./chuan-next/out
./file-transfer-server
提示: 构建完成后刷新页面即可看到完整的前端界面。
`))
}
// externalSpaHandler 外部文件目录处理器
type externalSpaHandler struct {
baseDir string
}
func (h *externalSpaHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
// 清理路径
upath := strings.TrimPrefix(r.URL.Path, "/")
if upath == "" {
upath = "index.html"
}
// 构建完整文件路径
fullPath := filepath.Join(h.baseDir, upath)
// 安全检查：确保文件在基础目录内
absBasePath, err := filepath.Abs(h.baseDir)
if err != nil {
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
return
}
absFullPath, err := filepath.Abs(fullPath)
if err != nil {
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
return
}
if !strings.HasPrefix(absFullPath, absBasePath) {
http.Error(w, "Forbidden", http.StatusForbidden)
return
}
// 检查文件是否存在
if _, err := os.Stat(fullPath); os.IsNotExist(err) {
// 文件不存在，对于 SPA 应用返回 index.html
h.serveIndexHTML(w, r)
return
}
// 服务文件
http.ServeFile(w, r, fullPath)
}
// serveIndexHTML 服务外部目录的 index.html 文件
func (h *externalSpaHandler) serveIndexHTML(w http.ResponseWriter, r *http.Request) {
indexPath := filepath.Join(h.baseDir, "index.html")
if _, err := os.Stat(indexPath); os.IsNotExist(err) {
http.NotFound(w, r)
return
}
http.ServeFile(w, r, indexPath)
}
// 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
}
}