464 lines
11 KiB
Go
464 lines
11 KiB
Go
package handlers
|
||
|
||
import (
|
||
"platform/web/auth"
|
||
"platform/web/core"
|
||
g "platform/web/globals"
|
||
s "platform/web/services"
|
||
"time"
|
||
|
||
"github.com/gofiber/fiber/v2"
|
||
)
|
||
|
||
func PageProxyByAdmin(c *fiber.Ctx) error {
|
||
_, err := auth.GetAuthCtx(c).PermitAdmin(core.ScopeProxyRead)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
var req core.PageReq
|
||
if err := g.Validator.ParseBody(c, &req); err != nil {
|
||
return err
|
||
}
|
||
|
||
list, total, err := s.Proxy.Page(req)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return c.JSON(core.PageResp{
|
||
List: list,
|
||
Total: int(total),
|
||
Page: req.GetPage(),
|
||
Size: req.GetSize(),
|
||
})
|
||
}
|
||
|
||
func AllProxyByAdmin(c *fiber.Ctx) error {
|
||
_, err := auth.GetAuthCtx(c).PermitAdmin(core.ScopeProxyRead)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
list, err := s.Proxy.All()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return c.JSON(list)
|
||
}
|
||
|
||
func CreateProxy(c *fiber.Ctx) error {
|
||
_, err := auth.GetAuthCtx(c).PermitAdmin(core.ScopeProxyWrite)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
var req s.CreateProxy
|
||
if err := g.Validator.ParseBody(c, &req); err != nil {
|
||
return err
|
||
}
|
||
|
||
if err := s.Proxy.Create(&req); err != nil {
|
||
return err
|
||
}
|
||
|
||
return c.JSON(nil)
|
||
}
|
||
|
||
func UpdateProxy(c *fiber.Ctx) error {
|
||
_, err := auth.GetAuthCtx(c).PermitAdmin(core.ScopeProxyWrite)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
var req s.UpdateProxy
|
||
if err := g.Validator.ParseBody(c, &req); err != nil {
|
||
return err
|
||
}
|
||
|
||
if err := s.Proxy.Update(&req); err != nil {
|
||
return err
|
||
}
|
||
|
||
return c.JSON(nil)
|
||
}
|
||
|
||
func UpdateProxyStatus(c *fiber.Ctx) error {
|
||
_, err := auth.GetAuthCtx(c).PermitAdmin(core.ScopeProxyWriteStatus)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
var req s.UpdateProxyStatus
|
||
if err := g.Validator.ParseBody(c, &req); err != nil {
|
||
return err
|
||
}
|
||
|
||
if err := s.Proxy.UpdateStatus(&req); err != nil {
|
||
return err
|
||
}
|
||
|
||
return c.JSON(nil)
|
||
}
|
||
|
||
func RemoveProxy(c *fiber.Ctx) error {
|
||
_, err := auth.GetAuthCtx(c).PermitAdmin(core.ScopeProxyWrite)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
var req core.IdReq
|
||
if err := g.Validator.ParseBody(c, &req); err != nil {
|
||
return err
|
||
}
|
||
|
||
if err := s.Proxy.Remove(req.Id); err != nil {
|
||
return err
|
||
}
|
||
|
||
return c.JSON(nil)
|
||
}
|
||
|
||
// region 报告上线
|
||
func ProxyReportOnline(c *fiber.Ctx) (err error) {
|
||
return c.JSON(map[string]any{
|
||
"error": "接口暂不可用",
|
||
})
|
||
|
||
// // 检查接口权限
|
||
// _, err = auth2.GetAuthCtx(c).PermitSecretClient()
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
|
||
// // 验证请求参数
|
||
// var req = new(ProxyReportOnlineReq)
|
||
// err = g.Validator.Validate(c, req)
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
|
||
// // 创建代理
|
||
// var ip = c.Context().RemoteIP()
|
||
|
||
// var secretBytes = make([]byte, 16)
|
||
// if _, err := rand.Read(secretBytes); err != nil {
|
||
// return err
|
||
// }
|
||
// var secret = base32.StdEncoding.
|
||
// WithPadding(base32.NoPadding).
|
||
// EncodeToString(secretBytes)
|
||
// slog.Debug("生成随机密钥", "ip", ip, "secret", secret)
|
||
|
||
// var proxy = &m.Proxy{
|
||
// Mac: req.Name,
|
||
// Version: int32(req.Version),
|
||
// Type: m.ProxyTypeSelfHosted,
|
||
// IP: ip,
|
||
// Secret: &secret,
|
||
// Status: 1,
|
||
// }
|
||
// err = q.Proxy.
|
||
// Clauses(clause.OnConflict{
|
||
// UpdateAll: true,
|
||
// Columns: []clause.Column{
|
||
// {Name: q.Proxy.Mac.ColumnName().String()},
|
||
// },
|
||
// }).
|
||
// Create(proxy)
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
|
||
// // 获取边缘节点信息
|
||
// data, err := q.Edge.Where(
|
||
// q.Edge.ProxyID.Eq(proxy.ID),
|
||
// ).Find()
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
// edges := make([]*ProxyEdge, len(data))
|
||
// for i, edge := range data {
|
||
// edges[i] = &ProxyEdge{
|
||
// Id: edge.ID,
|
||
// Port: edge.ProxyPort,
|
||
// Prov: &edge.Prov,
|
||
// City: &edge.City,
|
||
// Isp: u.P(edge2.ISP(edge.Isp).String()),
|
||
// Status: &edge.Status,
|
||
// Loss: edge.Loss,
|
||
// Rtt: edge.Rtt,
|
||
// }
|
||
// }
|
||
|
||
// // 获取许可配置
|
||
// channels, err := q.Channel.Where(
|
||
// q.Channel.ProxyID.Eq(proxy.ID),
|
||
// q.Channel.Expiration.Gt(orm.LocalDateTime(time.Now())),
|
||
// ).Find()
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
|
||
// var permits = make([]*ProxyPermit, len(channels))
|
||
// for i, channel := range channels {
|
||
// if channel.EdgeID == nil {
|
||
// return core.NewBizErr(fmt.Sprintf("权限解析异常，通道缺少边缘节点ID %d", channel.ID))
|
||
// }
|
||
// permits[i] = &ProxyPermit{
|
||
// Id: *channel.EdgeID,
|
||
// Expire: time.Time(channel.Expiration),
|
||
// Whitelists: u.P(strings.Split(u.Z(channel.Whitelists), ",")),
|
||
// Username: channel.Username,
|
||
// Password: channel.Password,
|
||
// }
|
||
// }
|
||
|
||
// slog.Debug("注册转发服务", "ip", ip, "id", proxy.ID)
|
||
// return c.JSON(&ProxyReportOnlineResp{
|
||
// Id: proxy.ID,
|
||
// Secret: secret,
|
||
// Edges: edges,
|
||
// Permits: permits,
|
||
// })
|
||
}
|
||
|
||
type ProxyReportOnlineReq struct {
|
||
Name string `json:"name" validate:"required"`
|
||
Version int `json:"version" validate:"required"`
|
||
}
|
||
|
||
type ProxyReportOnlineResp struct {
|
||
Id int32 `json:"id"`
|
||
Secret string `json:"secret"`
|
||
Permits []*ProxyPermit `json:"permits"`
|
||
Edges []*ProxyEdge `json:"edges"`
|
||
}
|
||
|
||
// region 报告下线
|
||
func ProxyReportOffline(c *fiber.Ctx) (err error) {
|
||
return c.JSON(map[string]any{
|
||
"error": "接口暂不可用",
|
||
})
|
||
|
||
// // 检查接口权限
|
||
// _, err = auth2.GetAuthCtx(c).PermitSecretClient()
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
|
||
// // 验证请求参数
|
||
// var req = new(ProxyReportOfflineReq)
|
||
// err = g.Validator.Validate(c, req)
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
|
||
// // 下线转发服务
|
||
// _, err = q.Proxy.
|
||
// Where(q.Proxy.ID.Eq(req.Id)).
|
||
// UpdateSimple(q.Proxy.Status.Value(0))
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
|
||
// // 下线所有相关的边缘节点
|
||
// _, err = q.Edge.
|
||
// Where(q.Edge.ProxyID.Eq(req.Id)).
|
||
// UpdateSimple(q.Edge.Status.Value(0))
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
|
||
// return nil
|
||
}
|
||
|
||
type ProxyReportOfflineReq struct {
|
||
Id int32 `json:"id" validate:"required"`
|
||
}
|
||
|
||
// region 报告更新
|
||
func ProxyReportUpdate(c *fiber.Ctx) (err error) {
|
||
return c.JSON(map[string]any{
|
||
"error": "接口暂不可用",
|
||
})
|
||
|
||
// // 检查接口权限
|
||
// _, err = auth2.GetAuthCtx(c).PermitSecretClient()
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
|
||
// // 验证请求参数
|
||
// var req = new(ProxyReportUpdateReq)
|
||
// err = g.Validator.Validate(c, req)
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
|
||
// // 更新节点信息
|
||
// var idsActive = make([]int32, 0, len(req.Edges))
|
||
// var idsInactive = make([]int32, 0, len(req.Edges))
|
||
// var idsIspUnknown = make([]int32, 0, len(req.Edges))
|
||
// var idsIspTelecom = make([]int32, 0, len(req.Edges))
|
||
// var idsIspUnicom = make([]int32, 0, len(req.Edges))
|
||
// var idsIspMobile = make([]int32, 0, len(req.Edges))
|
||
// var otherEdges = make([]*ProxyEdge, 0, len(req.Edges))
|
||
// for _, edge := range req.Edges {
|
||
|
||
// // 检查更新ISP
|
||
// if edge.Isp != nil {
|
||
// switch edge2.ISPFromStr(*edge.Isp) {
|
||
// case edge2.IspUnknown:
|
||
// idsIspUnknown = append(idsIspUnknown, edge.Id)
|
||
// case edge2.IspChinaTelecom:
|
||
// idsIspTelecom = append(idsIspTelecom, edge.Id)
|
||
// case edge2.IspChinaUnicom:
|
||
// idsIspUnicom = append(idsIspUnicom, edge.Id)
|
||
// case edge2.IspChinaMobile:
|
||
// idsIspMobile = append(idsIspMobile, edge.Id)
|
||
// }
|
||
// }
|
||
|
||
// // 检查更新状态
|
||
// if edge.Status != nil {
|
||
// if *edge.Status == 1 {
|
||
// idsActive = append(idsActive, edge.Id)
|
||
// } else {
|
||
// idsInactive = append(idsInactive, edge.Id)
|
||
// }
|
||
// }
|
||
|
||
// // 无法分类更新
|
||
// if edge.Host != nil || edge.Port != nil || edge.Prov != nil || edge.City != nil {
|
||
// otherEdges = append(otherEdges, edge)
|
||
// continue
|
||
// }
|
||
// }
|
||
|
||
// slog.Debug("更新边缘节点信息",
|
||
// "active", len(idsActive),
|
||
// "inactive", len(idsInactive),
|
||
// "isp_unknown", len(idsIspUnknown),
|
||
// "isp_telecom", len(idsIspTelecom),
|
||
// "isp_unicom", len(idsIspUnicom),
|
||
// "isp_mobile", len(idsIspMobile),
|
||
// "other_edges", len(otherEdges),
|
||
// )
|
||
|
||
// err = q.Q.Transaction(func(q *q.Query) error {
|
||
// // 更新边缘节点状态
|
||
// if len(idsActive) > 0 {
|
||
// _, err = q.Edge.Debug().
|
||
// Where(q.Edge.ID.In(idsActive...)).
|
||
// UpdateSimple(q.Edge.Status.Value(1))
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
// }
|
||
|
||
// if len(idsInactive) > 0 {
|
||
// _, err = q.Edge.Debug().
|
||
// Where(q.Edge.ID.In(idsInactive...)).
|
||
// UpdateSimple(q.Edge.Status.Value(0))
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
// }
|
||
|
||
// // 更新边缘节点ISP
|
||
// if len(idsIspUnknown) > 0 {
|
||
// _, err = q.Edge.Debug().
|
||
// Where(q.Edge.ID.In(idsIspUnknown...)).
|
||
// UpdateSimple(q.Edge.Isp.Value(int32(edge2.IspUnknown)))
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
// }
|
||
|
||
// if len(idsIspTelecom) > 0 {
|
||
// _, err = q.Edge.Debug().
|
||
// Where(q.Edge.ID.In(idsIspTelecom...)).
|
||
// UpdateSimple(q.Edge.Isp.Value(int32(edge2.IspChinaTelecom)))
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
// }
|
||
|
||
// if len(idsIspUnicom) > 0 {
|
||
// _, err = q.Edge.Debug().
|
||
// Where(q.Edge.ID.In(idsIspUnicom...)).
|
||
// UpdateSimple(q.Edge.Isp.Value(int32(edge2.IspChinaUnicom)))
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
// }
|
||
|
||
// if len(idsIspMobile) > 0 {
|
||
// _, err = q.Edge.Debug().
|
||
// Where(q.Edge.ID.In(idsIspMobile...)).
|
||
// UpdateSimple(q.Edge.Isp.Value(int32(edge2.IspChinaMobile)))
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
// }
|
||
|
||
// // 更新其他边缘节点信息
|
||
// for _, edge := range otherEdges {
|
||
// do := q.Edge.Debug().Where(q.Edge.ID.Eq(edge.Id))
|
||
|
||
// var assigns = make([]field.AssignExpr, 0, 5)
|
||
// if edge.Host != nil {
|
||
// assigns = append(assigns, q.Edge.Host.Value(*edge.Host))
|
||
// }
|
||
// if edge.Port != nil {
|
||
// assigns = append(assigns, q.Edge.ProxyPort.Value(*edge.Port))
|
||
// }
|
||
// if edge.Prov != nil {
|
||
// assigns = append(assigns, q.Edge.Prov.Value(*edge.Prov))
|
||
// }
|
||
// if edge.City != nil {
|
||
// assigns = append(assigns, q.Edge.City.Value(*edge.City))
|
||
// }
|
||
|
||
// // 更新边缘节点
|
||
// _, err := do.UpdateSimple(assigns...)
|
||
// if err != nil {
|
||
// return fmt.Errorf("更新边缘节点 %d 失败: %w", edge.Id, err)
|
||
// }
|
||
// }
|
||
// return nil
|
||
// })
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
|
||
// return nil
|
||
}
|
||
|
||
type ProxyReportUpdateReq struct {
|
||
Id int32 `json:"id" validate:"required"`
|
||
Edges []*ProxyEdge `json:"edges" validate:"required"`
|
||
}
|
||
|
||
type ProxyPermit struct {
|
||
Id int32 `json:"id"`
|
||
Expire time.Time `json:"expire"`
|
||
Whitelists *[]string `json:"whitelists"`
|
||
Username *string `json:"username"`
|
||
Password *string `json:"password"`
|
||
}
|
||
|
||
type ProxyEdge struct {
|
||
Id int32 `json:"id"`
|
||
Host *string `json:"host,omitempty"` // 边缘节点地址
|
||
Port *int32 `json:"port,omitempty"` // 边缘节点代理端口
|
||
Prov *string `json:"prov,omitempty"`
|
||
City *string `json:"city,omitempty"`
|
||
Isp *string `json:"isp,omitempty"`
|
||
Status *int32 `json:"status,omitempty"`
|
||
Loss *int32 `json:"loss,omitempty"` // 丢包率
|
||
Rtt *int32 `json:"latency,omitempty"` // 延迟
|
||
}
|