mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2025-11-29 10:52:54 +00:00
113 lines
2.6 KiB
Go
113 lines
2.6 KiB
Go
package controller
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/mhsanaei/3x-ui/v2/database/model"
|
|
"github.com/mhsanaei/3x-ui/v2/web/service"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type MultiServerController struct {
|
|
multiServerService service.MultiServerService
|
|
}
|
|
|
|
func NewMultiServerController(g *gin.RouterGroup) *MultiServerController {
|
|
c := &MultiServerController{}
|
|
c.initRouter(g)
|
|
return c
|
|
}
|
|
|
|
func (c *MultiServerController) initRouter(g *gin.RouterGroup) {
|
|
|
|
g.GET("/list", c.getServers)
|
|
g.POST("/add", c.addServer)
|
|
g.POST("/del/:id", c.delServer)
|
|
g.POST("/update/:id", c.updateServer)
|
|
g.GET("/onlines", c.getOnlineClients)
|
|
g.POST("/sync/:id", c.syncServer)
|
|
}
|
|
|
|
func (c *MultiServerController) getServers(ctx *gin.Context) {
|
|
servers, err := c.multiServerService.GetServers()
|
|
if err != nil {
|
|
jsonMsg(ctx, "Failed to get servers", err)
|
|
return
|
|
}
|
|
jsonObj(ctx, servers, nil)
|
|
}
|
|
|
|
func (c *MultiServerController) getOnlineClients(ctx *gin.Context) {
|
|
clients, err := c.multiServerService.GetOnlineClients()
|
|
if err != nil {
|
|
jsonMsg(ctx, "Failed to get online clients", err)
|
|
return
|
|
}
|
|
jsonObj(ctx, clients, nil)
|
|
}
|
|
|
|
func (c *MultiServerController) addServer(ctx *gin.Context) {
|
|
server := &model.Server{}
|
|
err := ctx.ShouldBind(server)
|
|
if err != nil {
|
|
jsonMsg(ctx, "Invalid data", err)
|
|
return
|
|
}
|
|
err = c.multiServerService.AddServer(server)
|
|
if err != nil {
|
|
jsonMsg(ctx, "Failed to add server", err)
|
|
return
|
|
}
|
|
jsonMsg(ctx, "Server added successfully", nil)
|
|
}
|
|
|
|
func (c *MultiServerController) delServer(ctx *gin.Context) {
|
|
id, err := strconv.Atoi(ctx.Param("id"))
|
|
if err != nil {
|
|
jsonMsg(ctx, "Invalid ID", err)
|
|
return
|
|
}
|
|
err = c.multiServerService.DeleteServer(id)
|
|
if err != nil {
|
|
jsonMsg(ctx, "Failed to delete server", err)
|
|
return
|
|
}
|
|
jsonMsg(ctx, "Server deleted successfully", nil)
|
|
}
|
|
|
|
func (c *MultiServerController) updateServer(ctx *gin.Context) {
|
|
id, err := strconv.Atoi(ctx.Param("id"))
|
|
if err != nil {
|
|
jsonMsg(ctx, "Invalid ID", err)
|
|
return
|
|
}
|
|
server := &model.Server{
|
|
Id: id,
|
|
}
|
|
err = ctx.ShouldBind(server)
|
|
if err != nil {
|
|
jsonMsg(ctx, "Invalid data", err)
|
|
return
|
|
}
|
|
err = c.multiServerService.UpdateServer(server)
|
|
if err != nil {
|
|
jsonMsg(ctx, "Failed to update server", err)
|
|
return
|
|
}
|
|
jsonMsg(ctx, "Server updated successfully", nil)
|
|
}
|
|
|
|
func (c *MultiServerController) syncServer(ctx *gin.Context) {
|
|
id, err := strconv.Atoi(ctx.Param("id"))
|
|
if err != nil {
|
|
jsonMsg(ctx, "Invalid ID", err)
|
|
return
|
|
}
|
|
err = c.multiServerService.SyncServer(id)
|
|
if err != nil {
|
|
jsonMsg(ctx, "Failed to sync server", err)
|
|
return
|
|
}
|
|
jsonMsg(ctx, "Server synced successfully", nil)
|
|
}
|