mirror of
https://github.com/VaalaCat/frp-panel.git
synced 2025-09-26 19:31:18 +08:00
62 lines
1.7 KiB
Go
62 lines
1.7 KiB
Go
package server
|
|
|
|
import (
|
|
"github.com/VaalaCat/frp-panel/common"
|
|
"github.com/VaalaCat/frp-panel/models"
|
|
"github.com/VaalaCat/frp-panel/pb"
|
|
"github.com/VaalaCat/frp-panel/services/app"
|
|
"github.com/VaalaCat/frp-panel/services/dao"
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
func ListServersHandler(c *app.Context, req *pb.ListServersRequest) (*pb.ListServersResponse, error) {
|
|
var (
|
|
userInfo = common.GetUserInfo(c)
|
|
page = int(req.GetPage())
|
|
pageSize = int(req.GetPageSize())
|
|
keyword = req.GetKeyword()
|
|
servers []*models.ServerEntity
|
|
serverCounts int64
|
|
hasKeyword = len(keyword) > 0
|
|
err error
|
|
)
|
|
|
|
if !userInfo.Valid() {
|
|
return &pb.ListServersResponse{
|
|
Status: &pb.Status{Code: pb.RespCode_RESP_CODE_INVALID, Message: "invalid user"},
|
|
}, nil
|
|
}
|
|
|
|
if hasKeyword {
|
|
servers, err = dao.NewQuery(c).ListServersWithKeyword(userInfo, page, pageSize, keyword)
|
|
} else {
|
|
servers, err = dao.NewQuery(c).ListServers(userInfo, page, pageSize)
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if hasKeyword {
|
|
serverCounts, err = dao.NewQuery(c).CountServersWithKeyword(userInfo, keyword)
|
|
} else {
|
|
serverCounts, err = dao.NewQuery(c).CountServers(userInfo)
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &pb.ListServersResponse{
|
|
Status: &pb.Status{Code: pb.RespCode_RESP_CODE_SUCCESS, Message: "ok"},
|
|
Servers: lo.Map(servers, func(c *models.ServerEntity, _ int) *pb.Server {
|
|
return &pb.Server{
|
|
Id: lo.ToPtr(c.ServerID),
|
|
Config: lo.ToPtr(string(c.ConfigContent)),
|
|
Secret: lo.ToPtr(c.ConnectSecret),
|
|
Ip: lo.ToPtr(c.ServerIP),
|
|
Comment: lo.ToPtr(c.Comment),
|
|
}
|
|
}),
|
|
Total: lo.ToPtr(int32(serverCounts)),
|
|
}, nil
|
|
}
|