[优化] Phase2

openflare_server/controller/tunnel.go

@@ -0,0 +1,84 @@
+package controller
+
+import (
+	"openflare/service"
+
+	"github.com/gin-gonic/gin"
+)
+
+func GetTunnels(c *gin.Context) {
+	tunnels, err := service.ListTunnels()
+	if err != nil {
+		respondFailure(c, err.Error())
+		return
+	}
+	respondSuccess(c, tunnels)
+}
+
+func GetTunnel(c *gin.Context) {
+	id, ok := parseIDParam(c)
+	if !ok {
+		return
+	}
+	tunnel, err := service.GetTunnel(id)
+	if err != nil {
+		respondFailure(c, err.Error())
+		return
+	}
+	respondSuccess(c, tunnel)
+}
+
+func CreateTunnel(c *gin.Context) {
+	var input service.TunnelInput
+	if !bindJSON(c, &input) {
+		return
+	}
+	tunnel, err := service.CreateTunnel(input)
+	if err != nil {
+		respondFailure(c, err.Error())
+		return
+	}
+	respondSuccess(c, tunnel)
+}
+
+func UpdateTunnel(c *gin.Context) {
+	id, ok := parseIDParam(c)
+	if !ok {
+		return
+	}
+	var input service.TunnelInput
+	if !bindJSON(c, &input) {
+		return
+	}
+	tunnel, err := service.UpdateTunnel(id, input)
+	if err != nil {
+		respondFailure(c, err.Error())
+		return
+	}
+	respondSuccess(c, tunnel)
+}
+
+func DeleteTunnel(c *gin.Context) {
+	id, ok := parseIDParam(c)
+	if !ok {
+		return
+	}
+	if err := service.DeleteTunnel(id); err != nil {
+		respondFailure(c, err.Error())
+		return
+	}
+	respondSuccess(c, nil)
+}
+
+func RotateTunnelToken(c *gin.Context) {
+	id, ok := parseIDParam(c)
+	if !ok {
+		return
+	}
+	tunnel, err := service.RotateTunnelToken(id)
+	if err != nil {
+		respondFailure(c, err.Error())
+		return
+	}
+	respondSuccess(c, tunnel)
+}