125 lines
2.5 KiB
Go
125 lines
2.5 KiB
Go
package handlers
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"platform/web/auth"
|
|
q "platform/web/queries"
|
|
"platform/web/services"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
// region CreateChannel
|
|
|
|
type CreateChannelReq struct {
|
|
ResourceId int32 `json:"resource_id" validate:"required"`
|
|
AuthType services.ChannelAuthType `json:"auth_type" validate:"required"`
|
|
Protocol services.ChannelProtocol `json:"protocol" validate:"required"`
|
|
Count int `json:"count" validate:"required"`
|
|
Prov string `json:"prov"`
|
|
City string `json:"city"`
|
|
Isp string `json:"isp"`
|
|
}
|
|
|
|
func CreateChannel(c *fiber.Ctx) error {
|
|
// 检查权限
|
|
authContext, err := auth.Protect(c, []services.PayloadType{services.PayloadUser}, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// 获取用户信息
|
|
user, err := q.User.
|
|
Where(q.User.ID.Eq(authContext.Payload.Id)).
|
|
Take()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if user.IDToken == "" {
|
|
return fiber.NewError(fiber.StatusForbidden, "账号未实名")
|
|
}
|
|
|
|
count, err := q.Whitelist.Where(
|
|
q.Whitelist.UserID.Eq(authContext.Payload.Id),
|
|
q.Whitelist.Host.Eq(c.IP()),
|
|
).Count()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if count == 0 {
|
|
return fiber.NewError(fiber.StatusForbidden, fmt.Sprintf("非白名单IP %s", c.IP()))
|
|
}
|
|
|
|
req := new(CreateChannelReq)
|
|
if err := c.BodyParser(req); err != nil {
|
|
return err
|
|
}
|
|
|
|
var isp string
|
|
switch req.Isp {
|
|
case "1":
|
|
isp = "电信"
|
|
case "2":
|
|
isp = "联通"
|
|
case "3":
|
|
isp = "移动"
|
|
}
|
|
|
|
result, err := services.Channel.CreateChannel(
|
|
c.Context(),
|
|
authContext,
|
|
req.ResourceId,
|
|
req.Protocol,
|
|
req.AuthType,
|
|
req.Count,
|
|
services.NodeFilterConfig{
|
|
Isp: isp,
|
|
Prov: req.Prov,
|
|
City: req.City,
|
|
},
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return c.JSON(result)
|
|
}
|
|
|
|
type CreateChannelResultType string
|
|
|
|
const (
|
|
CreateChannelResultTypeJson CreateChannelResultType = "json"
|
|
CreateChannelResultTypeText CreateChannelResultType = "text"
|
|
)
|
|
|
|
// endregion
|
|
|
|
// region RemoveChannels
|
|
|
|
type RemoveChannelsReq struct {
|
|
ByIds []int32 `json:"by_ids" validate:"required"`
|
|
}
|
|
|
|
func RemoveChannels(c *fiber.Ctx) error {
|
|
req := new(RemoveChannelsReq)
|
|
if err := c.BodyParser(req); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 获取用户信息
|
|
authCtx, ok := c.Locals("auth").(*services.AuthContext)
|
|
if !ok {
|
|
return errors.New("user not found")
|
|
}
|
|
|
|
// 删除通道
|
|
err := services.Channel.RemoveChannels(c.Context(), authCtx, req.ByIds...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return c.SendStatus(fiber.StatusOK)
|
|
}
|
|
|
|
// endregion
|