通道接口预实现

web/handlers/channel.go

@@ -0,0 +1,92 @@
+package handlers
+
+import (
+	"errors"
+	"fmt"
+	"platform/web/services"
+	"strings"
+
+	"github.com/gofiber/fiber/v2"
+)
+
+// region CreateChannel
+
+type CreateChannelReq struct {
+	Region          string                       `json:"region" validate:"required"`
+	Provider        string                       `json:"provider" validate:"required"`
+	Protocol        services.Protocol            `json:"protocol" validate:"required,oneof=socks5 http https"`
+	ResourceId      int                          `json:"resource_id" validate:"required"`
+	Count           int                          `json:"count" validate:"required"`
+	ResultType      CreateChannelResultType      `json:"return_type" validate:"required,oneof=json text"`
+	ResultSeparator CreateChannelResultSeparator `json:"return_separator" validate:"required,oneof=enter line both tab"`
+}
+
+func CreateChannel(c *fiber.Ctx) error {
+	req := new(CreateChannelReq)
+	if err := c.BodyParser(req); err != nil {
+		return err
+	}
+
+	// 建立连接通道
+	auth, ok := c.Locals("user").(*services.AuthContext)
+	if !ok {
+		return errors.New("user not found")
+	}
+
+	channels, err := services.Channel.CreateChannel(
+		auth.Payload.Id,
+		req.Region,
+		req.Provider,
+		req.Protocol,
+		req.ResourceId,
+		req.Count,
+	)
+	if err != nil {
+		return err
+	}
+
+	// 返回连接通道列表
+	var result []string
+	for _, channel := range channels {
+		url := fmt.Sprintf("%s://%s:%d", channel.Protocol, channel.UserHost, channel.NodePort)
+		result = append(result, url)
+	}
+
+	switch req.ResultType {
+	case CreateChannelResultTypeJson:
+		return c.JSON(fiber.Map{
+			"result": result,
+		})
+	case CreateChannelResultTypeText:
+		switch req.ResultSeparator {
+		case CreateChannelResultSeparatorEnter:
+			return c.SendString(strings.Join(result, "\r"))
+		case CreateChannelResultSeparatorLine:
+			return c.SendString(strings.Join(result, "\n"))
+		case CreateChannelResultSeparatorBoth:
+			return c.SendString(strings.Join(result, "\r\n"))
+		case CreateChannelResultSeparatorTab:
+			return c.SendString(strings.Join(result, "\t"))
+		}
+	}
+
+	return errors.New("无效的返回类型")
+}
+
+type CreateChannelResultType string
+
+const (
+	CreateChannelResultTypeJson CreateChannelResultType = "json"
+	CreateChannelResultTypeText CreateChannelResultType = "text"
+)
+
+type CreateChannelResultSeparator string
+
+const (
+	CreateChannelResultSeparatorEnter CreateChannelResultSeparator = "enter"
+	CreateChannelResultSeparatorLine  CreateChannelResultSeparator = "line"
+	CreateChannelResultSeparatorBoth  CreateChannelResultSeparator = "both"
+	CreateChannelResultSeparatorTab   CreateChannelResultSeparator = "tab"
+)
+
+// endregion