Feat: Support Web frontend

.gitignore

@@ -53,4 +53,6 @@ CLAUDE.md
 docs/
 
 # Generated files
-s3-balance
+s3-balance
+
+dist/

cmd/s3-balance/main.go

@@ -20,6 +20,8 @@ import (
 	"github.com/DullJZ/s3-balance/internal/middleware"
 	"github.com/DullJZ/s3-balance/internal/scheduler"
 	"github.com/DullJZ/s3-balance/internal/storage"
+	"github.com/DullJZ/s3-balance/internal/web"
+	"github.com/DullJZ/s3-balance/internal/webui"
 	"github.com/DullJZ/s3-balance/pkg/presigner"
 	"github.com/gorilla/mux"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
@@ -28,9 +30,17 @@ import (
 func main() {
 	// 解析命令行参数
 	var configFile string
+	var onlyWeb bool
 	flag.StringVar(&configFile, "config", "config/config.yaml", "Path to configuration file")
+	flag.BoolVar(&onlyWeb, "only-web", false, "Only serve web UI, no backend services")
 	flag.Parse()
 
+	// 如果是只提供Web前端模式
+	if onlyWeb {
+		startWebOnlyMode(configFile)
+		return
+	}
+
 	// 创建配置管理器
 	configManager, err := config.NewManager(configFile)
 	if err != nil {
@@ -148,6 +158,15 @@ func main() {
 		log.Printf("Management API endpoints available at /api/*")
 	}
 
+	// 注册Web管理界面
+	distSubFS, err := webui.GetDistFS()
+	if err != nil {
+		log.Fatalf("Failed to load embedded web UI: %v", err)
+	}
+	webHandler := web.NewHandler(distSubFS)
+	router.PathPrefix("/web").Handler(http.StripPrefix("/web", webHandler))
+	log.Println("Web UI available at /web")
+
 	// 运行在S3兼容模式
 	log.Println("Running in S3-compatible mode")
 	s3Handler.RegisterS3Routes(router)
@@ -298,3 +317,77 @@ func cleanupS3MultipartUploads(_ context.Context, storageService *storage.Servic
 		}
 	}
 }
+
+// startWebOnlyMode 只启动Web前端服务，不启动后端服务
+func startWebOnlyMode(configFile string) {
+	log.Println("Starting in web-only mode (no backend services)")
+
+	// 加载配置文件以获取端口等信息
+	configManager, err := config.NewManager(configFile)
+	if err != nil {
+		log.Fatalf("Failed to load config: %v", err)
+	}
+	defer configManager.Close()
+
+	cfg := configManager.GetConfig()
+
+	// 创建路由器
+	router := mux.NewRouter()
+
+	// 加载嵌入的前端资源
+	distSubFS, err := webui.GetDistFS()
+	if err != nil {
+		log.Fatalf("Failed to load embedded web UI: %v", err)
+	}
+
+	// 注册Web前端路由
+	webHandler := web.NewHandler(distSubFS)
+
+	// 根路径重定向到 /web
+	router.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		http.Redirect(w, r, "/web/", http.StatusMovedPermanently)
+	})
+
+	// Web UI 路由
+	router.PathPrefix("/web").Handler(http.StripPrefix("/web", webHandler))
+
+	// 添加 CORS 和日志中间件
+	router.Use(corsMiddleware)
+	router.Use(loggingMiddleware)
+
+	// 使用配置文件中的端口
+	addr := fmt.Sprintf("%s:%d", cfg.Server.Host, cfg.Server.Port)
+	srv := &http.Server{
+		Addr:         addr,
+		Handler:      router,
+		ReadTimeout:  cfg.Server.ReadTimeout,
+		WriteTimeout: cfg.Server.WriteTimeout,
+		IdleTimeout:  cfg.Server.IdleTimeout,
+	}
+
+	log.Println("Web UI available at /web")
+	log.Printf("Starting web server on %s", srv.Addr)
+
+	// 启动服务器
+	go func() {
+		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			log.Fatalf("Failed to start server: %v", err)
+		}
+	}()
+
+	// 等待中断信号
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
+	<-sigChan
+
+	// 优雅关闭
+	log.Println("Shutting down web server...")
+	shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer shutdownCancel()
+
+	if err := srv.Shutdown(shutdownCtx); err != nil {
+		log.Printf("Server shutdown error: %v", err)
+	}
+
+	log.Println("Web server stopped")
+}

internal/web/handler.go

@@ -0,0 +1,87 @@
+package web
+
+import (
+	"io"
+	"io/fs"
+	"net/http"
+	"path"
+	"strings"
+)
+
+// Handler Web管理界面处理器
+type Handler struct {
+	fileSystem http.FileSystem
+}
+
+// NewHandler 创建Web处理器
+// distFS 应该是通过 embed.FS 嵌入的 dist 目录
+func NewHandler(distFS fs.FS) *Handler {
+	return &Handler{
+		fileSystem: http.FS(distFS),
+	}
+}
+
+// ServeHTTP 实现 http.Handler 接口
+// 处理单页应用的路由，将所有未找到的路径重定向到 index.html
+func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	// 清理路径
+	p := r.URL.Path
+	if !strings.HasPrefix(p, "/") {
+		p = "/" + p
+	}
+
+	// 尝试打开文件
+	f, err := h.fileSystem.Open(path.Clean(p))
+	if err != nil {
+		// 文件不存在，返回 index.html (用于支持前端路由)
+		indexFile, err := h.fileSystem.Open("index.html")
+		if err != nil {
+			http.Error(w, "File not found", http.StatusNotFound)
+			return
+		}
+		defer indexFile.Close()
+
+		// 读取 index.html 内容
+		stat, err := indexFile.Stat()
+		if err != nil {
+			http.Error(w, "Internal server error", http.StatusInternalServerError)
+			return
+		}
+
+		w.Header().Set("Content-Type", "text/html; charset=utf-8")
+		http.ServeContent(w, r, "index.html", stat.ModTime(), indexFile.(io.ReadSeeker))
+		return
+	}
+	defer f.Close()
+
+	// 文件存在，检查是否为目录
+	stat, err := f.Stat()
+	if err != nil {
+		http.Error(w, "Internal server error", http.StatusInternalServerError)
+		return
+	}
+
+	if stat.IsDir() {
+		// 如果是目录，尝试返回 index.html
+		indexPath := path.Join(p, "index.html")
+		indexFile, err := h.fileSystem.Open(indexPath)
+		if err != nil {
+			http.Error(w, "Forbidden", http.StatusForbidden)
+			return
+		}
+		defer indexFile.Close()
+
+		indexStat, err := indexFile.Stat()
+		if err != nil {
+			http.Error(w, "Internal server error", http.StatusInternalServerError)
+			return
+		}
+
+		w.Header().Set("Content-Type", "text/html; charset=utf-8")
+		http.ServeContent(w, r, "index.html", indexStat.ModTime(), indexFile.(io.ReadSeeker))
+		return
+	}
+
+	// 返回文件内容
+	http.ServeContent(w, r, stat.Name(), stat.ModTime(), f.(io.ReadSeeker))
+}

internal/webui/embed.go

@@ -0,0 +1,14 @@
+package webui
+
+import (
+	"embed"
+	"io/fs"
+)
+
+//go:embed dist
+var distFS embed.FS
+
+// GetDistFS 获取嵌入的前端静态文件系统
+func GetDistFS() (fs.FS, error) {
+	return fs.Sub(distFS, "dist")
+}