基础框架先行，编译器已无报错，项目正式创建

2026-07-23 12:41:14 +08:00
parent 9e561e5ae6
commit 9118972a7a
12 changed files with 1004 additions and 0 deletions
@@ -0,0 +1,247 @@
+package main
+
+import (
+	"net/http"
+	"strconv"
+
+	"github.com/gin-gonic/gin"
+	"golang.org/x/crypto/bcrypt"
+)
+
+func SetupRouter() *gin.Engine {
+	r := gin.Default()
+
+	// 静态文件（前端）
+	r.Static("/static", "./static")
+
+	// API 路由
+	api := r.Group("/api")
+	{
+		// 登录（无需鉴权）
+		api.POST("/login", loginHandler)
+
+		// 需要鉴权的路由
+		auth := api.Group("/")
+		auth.Use(AuthMiddleware())
+		{
+			// 全局配置
+			auth.GET("/config", getConfigHandler)
+			auth.PUT("/config", updateConfigHandler)
+
+			// 隧道管理
+			auth.GET("/proxies", getProxiesHandler)
+			auth.GET("/proxy/:id", getProxyHandler)
+			auth.POST("/proxy", createProxyHandler)
+			auth.PUT("/proxy/:id", updateProxyHandler)
+			auth.DELETE("/proxy/:id", deleteProxyHandler)
+
+			// frpc 控制
+			auth.POST("/frpc/reload", reloadFrpcHandler)
+			auth.POST("/frpc/start", startFrpcHandler)
+			auth.POST("/frpc/stop", stopFrpcHandler)
+			auth.GET("/frpc/status", getFrpcStatusHandler)
+		}
+	}
+
+	// 前端入口
+	r.GET("/", func(c *gin.Context) {
+		c.File("./static/index.html")
+	})
+
+	return r
+}
+
+// ========== 登录 ==========
+func loginHandler(c *gin.Context) {
+	var req struct {
+		Username string `json:"username"`
+		Password string `json:"password"`
+	}
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"code": 1, "msg": "请求参数错误"})
+		return
+	}
+
+	user, err := GetUserByUsername(req.Username)
+	if err != nil {
+		c.JSON(http.StatusUnauthorized, gin.H{"code": 1, "msg": "用户名或密码错误"})
+		return
+	}
+
+	err = bcrypt.CompareHashAndPassword([]byte(user.PasswordHash), []byte(req.Password))
+	if err != nil {
+		c.JSON(http.StatusUnauthorized, gin.H{"code": 1, "msg": "用户名或密码错误"})
+		return
+	}
+
+	token, err := GenerateJWT(user.Username)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "生成Token失败"})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "登录成功", "data": gin.H{"token": token}})
+}
+
+// ========== 全局配置 ==========
+func getConfigHandler(c *gin.Context) {
+	cfg, err := GetGlobalConfig()
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "读取配置失败"})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"code": 0, "data": cfg})
+}
+
+func updateConfigHandler(c *gin.Context) {
+	var cfg GlobalConfig
+	if err := c.ShouldBindJSON(&cfg); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"code": 1, "msg": "请求参数错误"})
+		return
+	}
+
+	if err := UpdateGlobalConfig(&cfg); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "更新配置失败"})
+		return
+	}
+
+	// 自动生成配置文件并热加载
+	if err := GenerateFrpcConfig(); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "生成配置文件失败: " + err.Error()})
+		return
+	}
+
+	if err := ReloadFrpc(); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "热加载失败: " + err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "配置更新成功"})
+}
+
+// ========== 隧道管理 ==========
+func getProxiesHandler(c *gin.Context) {
+	proxies, err := GetProxies()
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "读取隧道列表失败"})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"code": 0, "data": proxies})
+}
+
+func getProxyHandler(c *gin.Context) {
+	id, _ := strconv.Atoi(c.Param("id"))
+	p, err := GetProxy(id)
+	if err != nil {
+		c.JSON(http.StatusNotFound, gin.H{"code": 1, "msg": "隧道不存在"})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"code": 0, "data": p})
+}
+
+func createProxyHandler(c *gin.Context) {
+	var p Proxy
+	if err := c.ShouldBindJSON(&p); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"code": 1, "msg": "请求参数错误"})
+		return
+	}
+	p.Enabled = true // 默认启用
+
+	if err := CreateProxy(&p); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "创建隧道失败"})
+		return
+	}
+
+	// 重新生成配置并热加载
+	if err := generateAndReload(); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "配置生效失败: " + err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "隧道创建成功", "data": gin.H{"id": p.ID}})
+}
+
+func updateProxyHandler(c *gin.Context) {
+	id, _ := strconv.Atoi(c.Param("id"))
+	var p Proxy
+	if err := c.ShouldBindJSON(&p); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"code": 1, "msg": "请求参数错误"})
+		return
+	}
+	p.ID = id
+
+	if err := UpdateProxy(&p); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "更新隧道失败"})
+		return
+	}
+
+	// 重新生成配置并热加载
+	if err := generateAndReload(); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "配置生效失败: " + err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "隧道更新成功"})
+}
+
+func deleteProxyHandler(c *gin.Context) {
+	id, _ := strconv.Atoi(c.Param("id"))
+	if err := DeleteProxy(id); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "删除隧道失败"})
+		return
+	}
+
+	// 重新生成配置并热加载
+	if err := generateAndReload(); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "配置生效失败: " + err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "隧道删除成功"})
+}
+
+// ========== frpc 控制 ==========
+func reloadFrpcHandler(c *gin.Context) {
+	if err := GenerateFrpcConfig(); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "生成配置失败: " + err.Error()})
+		return
+	}
+	if err := ReloadFrpc(); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "热加载失败: " + err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "热加载成功"})
+}
+
+func startFrpcHandler(c *gin.Context) {
+	if err := StartFrpc(); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "启动失败: " + err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "frpc 启动成功"})
+}
+
+func stopFrpcHandler(c *gin.Context) {
+	if err := StopFrpc(); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "停止失败: " + err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "frpc 已停止"})
+}
+
+func getFrpcStatusHandler(c *gin.Context) {
+	running, err := GetFrpcStatus()
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "查询状态失败"})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"code": 0, "data": gin.H{"running": running}})
+}
+
+// ========== 辅助函数 ==========
+func generateAndReload() error {
+	if err := GenerateFrpcConfig(); err != nil {
+		return err
+	}
+	return ReloadFrpc()
+}