mirror of
https://github.com/MatrixSeven/file-transfer-go.git
synced 2026-02-04 03:25:03 +08:00
316 lines
8.4 KiB
Go
316 lines
8.4 KiB
Go
package web
|
||
|
||
import (
|
||
"embed"
|
||
"io"
|
||
"io/fs"
|
||
"net/http"
|
||
"os"
|
||
"path"
|
||
"path/filepath"
|
||
"strings"
|
||
)
|
||
|
||
// 前端文件嵌入 - 这个路径会在构建脚本中被替换
|
||
//
|
||
//go:embed frontend/*
|
||
var FrontendFiles embed.FS
|
||
|
||
// hasFrontendFiles 检查是否有前端文件
|
||
func hasFrontendFiles() bool {
|
||
entries, err := FrontendFiles.ReadDir("frontend")
|
||
if err != nil {
|
||
return false
|
||
}
|
||
return len(entries) > 0
|
||
}
|
||
|
||
// CreateFrontendHandler 创建前端文件处理器
|
||
func CreateFrontendHandler() http.Handler {
|
||
// 检查是否配置了外部前端目录
|
||
if frontendDir := os.Getenv("FRONTEND_DIR"); frontendDir != "" {
|
||
if info, err := os.Stat(frontendDir); err == nil && info.IsDir() {
|
||
// 使用外部前端目录
|
||
return &externalSpaHandler{baseDir: frontendDir}
|
||
}
|
||
}
|
||
|
||
// 使用内嵌的前端文件
|
||
if !hasFrontendFiles() {
|
||
return &placeholderHandler{}
|
||
}
|
||
|
||
frontendFS, err := fs.Sub(FrontendFiles, "frontend")
|
||
if err != nil {
|
||
return &placeholderHandler{}
|
||
}
|
||
|
||
return &spaHandler{fs: frontendFS}
|
||
}
|
||
|
||
// placeholderHandler 占位处理器
|
||
type placeholderHandler struct{}
|
||
|
||
func (h *placeholderHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||
w.WriteHeader(http.StatusOK)
|
||
w.Write([]byte(`
|
||
<!DOCTYPE html>
|
||
<html lang="zh-CN">
|
||
<head>
|
||
<title>文件传输服务</title>
|
||
<meta charset="utf-8">
|
||
<meta name="viewport" content="width=device-width, initial-scale=1">
|
||
<style>
|
||
body { font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", sans-serif; margin: 0; padding: 20px; background: #f5f5f5; }
|
||
.container { max-width: 800px; margin: 0 auto; background: white; padding: 40px; border-radius: 8px; box-shadow: 0 2px 10px rgba(0,0,0,0.1); }
|
||
h1 { color: #333; margin-bottom: 20px; }
|
||
.status { padding: 15px; background: #fff3cd; border: 1px solid #ffeaa7; border-radius: 4px; margin: 20px 0; }
|
||
.commands { background: #f8f9fa; padding: 15px; border-radius: 4px; margin: 20px 0; }
|
||
pre { margin: 0; overflow-x: auto; }
|
||
.api-list { margin: 20px 0; }
|
||
.api-item { margin: 10px 0; padding: 10px; background: #e3f2fd; border-radius: 4px; }
|
||
.env-config { background: #e8f5e8; padding: 15px; border-radius: 4px; margin: 20px 0; }
|
||
</style>
|
||
</head>
|
||
<body>
|
||
<div class="container">
|
||
<h1>🚀 文件传输服务</h1>
|
||
|
||
<div class="status">
|
||
⚠️ 前端界面未构建，当前显示的是后端 API 服务。
|
||
</div>
|
||
|
||
<h2><3E> 环境变量配置</h2>
|
||
<div class="env-config">
|
||
<strong>FRONTEND_DIR</strong> - 指定外部前端文件目录<br>
|
||
<strong>PORT</strong> - 自定义服务端口 (默认: 8080)<br><br>
|
||
<strong>示例:</strong><br>
|
||
<pre>export FRONTEND_DIR=/path/to/frontend
|
||
export PORT=3000
|
||
./file-transfer-server</pre>
|
||
</div>
|
||
|
||
<h2><3E>📋 可用的 API 接口</h2>
|
||
<div class="api-list">
|
||
<div class="api-item"><strong>POST</strong> /api/create-room - 创建WebRTC房间</div>
|
||
<div class="api-item"><strong>GET</strong> /api/room-info - 获取房间信息</div>
|
||
<div class="api-item"><strong>WebSocket</strong> /api/ws/webrtc - WebRTC 信令连接</div>
|
||
</div>
|
||
|
||
<h2>🛠️ 构建前端</h2>
|
||
<div class="commands">
|
||
<pre># 进入前端目录
|
||
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</pre>
|
||
</div>
|
||
|
||
<p><strong>提示:</strong> 构建完成后刷新页面即可看到完整的前端界面。</p>
|
||
</div>
|
||
</body>
|
||
</html>
|
||
`))
|
||
}
|
||
|
||
// 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
|
||
}
|
||
}
|