diff --git a/web/controller/inbound.go b/web/controller/inbound.go deleted file mode 100644 index 176d3bf8..00000000 --- a/web/controller/inbound.go +++ /dev/null @@ -1,413 +0,0 @@ -package controller - -import ( - "encoding/json" - "fmt" - "strconv" - - "x-ui/database/model" - "x-ui/web/service" - "x-ui/web/session" - - "github.com/gin-gonic/gin" -) - -type InboundController struct { - inboundService service.InboundService - xrayService service.XrayService -} - -type AddClientPayload struct { - Settings string `json:"settings"` - InboundIds []int `json:"inboundIds"` -} - -func NewInboundController(g *gin.RouterGroup) *InboundController { - a := &InboundController{} - a.initRouter(g) - return a -} - -func (a *InboundController) initRouter(g *gin.RouterGroup) { - g = g.Group("/inbound") - - g.POST("/list", a.getInbounds) - g.POST("/add", a.addInbound) - g.POST("/del/:id", a.delInbound) - g.POST("/update/:id", a.updateInbound) - g.POST("/clientIps/:email", a.getClientIps) - g.POST("/clearClientIps/:email", a.clearClientIps) - g.POST("/addClient", a.addInboundClient) - g.POST("/addClientInbounds", a.addClientToMultipleInbounds) - g.POST("/:id/delClient/:clientId", a.delInboundClient) - g.POST("/updateClient/:clientId", a.updateInboundClient) - g.POST("/updateClientInbounds/:subId", a.updateClientInMultipleInbounds) - g.POST("/:id/resetClientTraffic/:email", a.resetClientTraffic) - g.POST("/resetAllTraffics", a.resetAllTraffics) - g.POST("/resetAllClientTraffics/:id", a.resetAllClientTraffics) - g.POST("/delDepletedClients/:id", a.delDepletedClients) - g.POST("/import", a.importInbound) - g.POST("/onlines", a.onlines) -} - -func (a *InboundController) getInbounds(c *gin.Context) { - user := session.GetLoginUser(c) - inbounds, err := a.inboundService.GetInbounds(user.Id) - if err != nil { - jsonMsg(c, I18nWeb(c, "pages.inbounds.toasts.obtain"), err) - return - } - jsonObj(c, inbounds, nil) -} - -func (a *InboundController) getInbound(c *gin.Context) { - id, err := strconv.Atoi(c.Param("id")) - if err != nil { - jsonMsg(c, I18nWeb(c, "get"), err) - return - } - inbound, err := a.inboundService.GetInbound(id) - if err != nil { - jsonMsg(c, I18nWeb(c, "pages.inbounds.toasts.obtain"), err) - return - } - jsonObj(c, inbound, nil) -} - -func (a *InboundController) getClientTraffics(c *gin.Context) { - email := c.Param("email") - clientTraffics, err := a.inboundService.GetClientTrafficByEmail(email) - if err != nil { - jsonMsg(c, "Error getting traffics", err) - return - } - jsonObj(c, clientTraffics, nil) -} - -func (a *InboundController) getClientTrafficsById(c *gin.Context) { - id := c.Param("id") - clientTraffics, err := a.inboundService.GetClientTrafficByID(id) - if err != nil { - jsonMsg(c, "Error getting traffics", err) - return - } - jsonObj(c, clientTraffics, nil) -} - -func (a *InboundController) addInbound(c *gin.Context) { - inbound := &model.Inbound{} - err := c.ShouldBind(inbound) - if err != nil { - jsonMsg(c, I18nWeb(c, "pages.inbounds.create"), err) - return - } - user := session.GetLoginUser(c) - inbound.UserId = user.Id - if inbound.Listen == "" || inbound.Listen == "0.0.0.0" || inbound.Listen == "::" || inbound.Listen == "::0" { - inbound.Tag = fmt.Sprintf("inbound-%v", inbound.Port) - } else { - inbound.Tag = fmt.Sprintf("inbound-%v:%v", inbound.Listen, inbound.Port) - } - - needRestart := false - inbound, needRestart, err = a.inboundService.AddInbound(inbound) - jsonMsgObj(c, I18nWeb(c, "pages.inbounds.create"), inbound, err) - if err == nil && needRestart { - a.xrayService.SetToNeedRestart() - } -} - -func (a *InboundController) delInbound(c *gin.Context) { - id, err := strconv.Atoi(c.Param("id")) - if err != nil { - jsonMsg(c, I18nWeb(c, "delete"), err) - return - } - needRestart := true - needRestart, err = a.inboundService.DelInbound(id) - jsonMsgObj(c, I18nWeb(c, "delete"), id, err) - if err == nil && needRestart { - a.xrayService.SetToNeedRestart() - } -} - -func (a *InboundController) updateInbound(c *gin.Context) { - id, err := strconv.Atoi(c.Param("id")) - if err != nil { - jsonMsg(c, I18nWeb(c, "pages.inbounds.update"), err) - return - } - inbound := &model.Inbound{ - Id: id, - } - err = c.ShouldBind(inbound) - if err != nil { - jsonMsg(c, I18nWeb(c, "pages.inbounds.update"), err) - return - } - needRestart := true - inbound, needRestart, err = a.inboundService.UpdateInbound(inbound) - jsonMsgObj(c, I18nWeb(c, "pages.inbounds.update"), inbound, err) - if err == nil && needRestart { - a.xrayService.SetToNeedRestart() - } -} - -func (a *InboundController) getClientIps(c *gin.Context) { - email := c.Param("email") - - ips, err := a.inboundService.GetInboundClientIps(email) - if err != nil || ips == "" { - jsonObj(c, "No IP Record", nil) - return - } - - jsonObj(c, ips, nil) -} - -func (a *InboundController) clearClientIps(c *gin.Context) { - email := c.Param("email") - - err := a.inboundService.ClearClientIps(email) - if err != nil { - jsonMsg(c, "Update", err) - return - } - jsonMsg(c, "Log Cleared", nil) -} - -func (a *InboundController) addInboundClient(c *gin.Context) { - data := &model.Inbound{} - err := c.ShouldBind(data) - if err != nil { - jsonMsg(c, I18nWeb(c, "pages.inbounds.update"), err) - return - } - - needRestart := true - - needRestart, err = a.inboundService.AddInboundClient(data) - if err != nil { - jsonMsg(c, "Something went wrong!", err) - return - } - jsonMsg(c, "Client(s) added", nil) - if needRestart { - a.xrayService.SetToNeedRestart() - } -} - - -func (a *InboundController) addClientToMultipleInbounds(c *gin.Context) { - var payload AddClientPayload - if err := c.ShouldBindJSON(&payload); err != nil { - jsonMsg(c, "Invalid request payload", err) - return - } - - data := &model.Inbound{ - Settings: payload.Settings, - } - - // If no specific inbound IDs are provided, add to all inbounds - if len(payload.InboundIds) == 0 { - allInbounds, err := a.inboundService.GetAllInbounds() - if err != nil { - jsonMsg(c, "Could not retrieve inbounds", err) - return - } - - for _, inbound := range allInbounds { - payload.InboundIds = append(payload.InboundIds, inbound.Id) - } - } - - needRestart, err := a.inboundService.AddClientToMultipleInbounds(data, payload.InboundIds) - if err != nil { - jsonMsg(c, "Something went wrong!", err) - return - } - - jsonMsg(c, "Client(s) added to multiple inbounds", nil) - if needRestart { - a.xrayService.SetToNeedRestart() - } -} - - -func (a *InboundController) updateClientInMultipleInbounds(c *gin.Context) { - var payload AddClientPayload - subId := c.Param("subId") - - if err := c.ShouldBindJSON(&payload); err != nil { - jsonMsg(c, "Invalid request payload", err) - return - } - - data := &model.Inbound{ - Settings: payload.Settings, - } - - // If no specific inbound IDs are provided, add to all inbounds - if len(payload.InboundIds) == 0 { - allInbounds, err := a.inboundService.GetAllInbounds() - if err != nil { - jsonMsg(c, "Could not retrieve inbounds", err) - return - } - - for _, inbound := range allInbounds { - payload.InboundIds = append(payload.InboundIds, inbound.Id) - } - } - - needRestart, err := a.inboundService.UpdateClientInMultipleInbounds(data, subId, payload.InboundIds) - if err != nil { - jsonMsg(c, "Something went wrong!", err) - return - } - jsonMsg(c, "Client updated in multiple inbounds", nil) - if needRestart { - a.xrayService.SetToNeedRestart() - } -} - - - -func (a *InboundController) delInboundClient(c *gin.Context) { - id, err := strconv.Atoi(c.Param("id")) - if err != nil { - jsonMsg(c, I18nWeb(c, "pages.inbounds.update"), err) - return - } - clientId := c.Param("clientId") - - needRestart := true - - needRestart, err = a.inboundService.DelInboundClient(id, clientId) - if err != nil { - jsonMsg(c, "Something went wrong!", err) - return - } - jsonMsg(c, "Client deleted", nil) - if needRestart { - a.xrayService.SetToNeedRestart() - } -} - -func (a *InboundController) updateInboundClient(c *gin.Context) { - clientId := c.Param("clientId") - - inbound := &model.Inbound{} - err := c.ShouldBind(inbound) - if err != nil { - jsonMsg(c, I18nWeb(c, "pages.inbounds.update"), err) - return - } - - needRestart := true - - needRestart, err = a.inboundService.UpdateInboundClient(inbound, clientId) - if err != nil { - jsonMsg(c, "Something went wrong!", err) - return - } - jsonMsg(c, "Client updated", nil) - if needRestart { - a.xrayService.SetToNeedRestart() - } -} - -func (a *InboundController) resetClientTraffic(c *gin.Context) { - id, err := strconv.Atoi(c.Param("id")) - if err != nil { - jsonMsg(c, I18nWeb(c, "pages.inbounds.update"), err) - return - } - email := c.Param("email") - - needRestart, err := a.inboundService.ResetClientTraffic(id, email) - if err != nil { - jsonMsg(c, "Something went wrong!", err) - return - } - jsonMsg(c, "Traffic has been reset", nil) - if needRestart { - a.xrayService.SetToNeedRestart() - } -} - -func (a *InboundController) resetAllTraffics(c *gin.Context) { - err := a.inboundService.ResetAllTraffics() - if err != nil { - jsonMsg(c, "Something went wrong!", err) - return - } else { - a.xrayService.SetToNeedRestart() - } - jsonMsg(c, "all traffic has been reset", nil) -} - -func (a *InboundController) resetAllClientTraffics(c *gin.Context) { - id, err := strconv.Atoi(c.Param("id")) - if err != nil { - jsonMsg(c, I18nWeb(c, "pages.inbounds.update"), err) - return - } - - err = a.inboundService.ResetAllClientTraffics(id) - if err != nil { - jsonMsg(c, "Something went wrong!", err) - return - } else { - a.xrayService.SetToNeedRestart() - } - jsonMsg(c, "All traffic from the client has been reset.", nil) -} - -func (a *InboundController) importInbound(c *gin.Context) { - inbound := &model.Inbound{} - err := json.Unmarshal([]byte(c.PostForm("data")), inbound) - if err != nil { - jsonMsg(c, "Something went wrong!", err) - return - } - user := session.GetLoginUser(c) - inbound.Id = 0 - inbound.UserId = user.Id - if inbound.Listen == "" || inbound.Listen == "0.0.0.0" || inbound.Listen == "::" || inbound.Listen == "::0" { - inbound.Tag = fmt.Sprintf("inbound-%v", inbound.Port) - } else { - inbound.Tag = fmt.Sprintf("inbound-%v:%v", inbound.Listen, inbound.Port) - } - - for index := range inbound.ClientStats { - inbound.ClientStats[index].Id = 0 - inbound.ClientStats[index].Enable = true - } - - needRestart := false - inbound, needRestart, err = a.inboundService.AddInbound(inbound) - jsonMsgObj(c, I18nWeb(c, "pages.inbounds.create"), inbound, err) - if err == nil && needRestart { - a.xrayService.SetToNeedRestart() - } -} - -func (a *InboundController) delDepletedClients(c *gin.Context) { - id, err := strconv.Atoi(c.Param("id")) - if err != nil { - jsonMsg(c, I18nWeb(c, "pages.inbounds.update"), err) - return - } - err = a.inboundService.DelDepletedClients(id) - if err != nil { - jsonMsg(c, "Something went wrong!", err) - return - } - jsonMsg(c, "All depleted clients are deleted", nil) -} - -func (a *InboundController) onlines(c *gin.Context) { - jsonObj(c, a.inboundService.GetOnlineClients(), nil) -}