| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059 |
- package service
- import (
- "context"
- "encoding/json"
- "fmt"
- "strings"
- "time"
- "github.com/mhsanaei/3x-ui/v3/internal/database"
- "github.com/mhsanaei/3x-ui/v3/internal/database/model"
- "github.com/mhsanaei/3x-ui/v3/internal/logger"
- "github.com/mhsanaei/3x-ui/v3/internal/util/common"
- "github.com/mhsanaei/3x-ui/v3/internal/util/random"
- "github.com/mhsanaei/3x-ui/v3/internal/xray"
- )
- // delInboundClients removes several clients from a single inbound in one pass:
- // one settings rewrite, one runtime sweep, one Save and one SyncInbound for the
- // whole batch, instead of repeating the full per-client cycle. It mirrors the
- // semantics of DelInboundClientByEmail for each removed client. needRestart is
- // the OR across all removals.
- func (s *ClientService) delInboundClients(inboundSvc *InboundService, inboundId int, recs []*model.ClientRecord, keepTraffic bool) (bool, error) {
- if len(recs) == 0 {
- return false, nil
- }
- defer lockInbound(inboundId).Unlock()
- oldInbound, err := inboundSvc.GetInbound(inboundId)
- if err != nil {
- logger.Error("Load Old Data Error")
- return false, err
- }
- var settings map[string]any
- if err := json.Unmarshal([]byte(oldInbound.Settings), &settings); err != nil {
- return false, err
- }
- // Match by email — the client's stable identity (see Delete). Removes every
- // entry carrying a wanted email, independent of credential drift.
- wanted := make(map[string]struct{}, len(recs))
- for _, rec := range recs {
- if rec.Email != "" {
- wanted[rec.Email] = struct{}{}
- }
- }
- interfaceClients, ok := settings["clients"].([]any)
- if !ok {
- return false, common.NewError("invalid clients format in inbound settings")
- }
- type removedClient struct {
- email string
- needApiDel bool
- }
- removed := make([]removedClient, 0, len(wanted))
- newClients := make([]any, 0, len(interfaceClients))
- for _, client := range interfaceClients {
- c, ok := client.(map[string]any)
- if !ok {
- newClients = append(newClients, client)
- continue
- }
- email, _ := c["email"].(string)
- if _, hit := wanted[email]; hit && email != "" {
- enable, _ := c["enable"].(bool)
- removed = append(removed, removedClient{email: email, needApiDel: enable})
- continue
- }
- newClients = append(newClients, client)
- }
- if len(removed) == 0 {
- return false, nil
- }
- db := database.GetDB()
- newClients = compactOrphans(db, newClients)
- if newClients == nil {
- newClients = []any{}
- }
- settings["clients"] = newClients
- newSettings, err := json.MarshalIndent(settings, "", " ")
- if err != nil {
- return false, err
- }
- oldInbound.Settings = string(newSettings)
- var sharedSet map[string]bool
- if !keepTraffic {
- removedEmails := make([]string, 0, len(removed))
- for _, r := range removed {
- if r.email != "" {
- removedEmails = append(removedEmails, r.email)
- }
- }
- var sharedErr error
- sharedSet, sharedErr = inboundSvc.emailsUsedByOtherInbounds(removedEmails, inboundId)
- if sharedErr != nil {
- return false, sharedErr
- }
- }
- needRestart := false
- markDirty := false
- for _, r := range removed {
- email := r.email
- emailShared := sharedSet[strings.ToLower(strings.TrimSpace(email))]
- if !emailShared && !keepTraffic {
- if err := inboundSvc.DelClientIPs(db, email); err != nil {
- logger.Error("Error in delete client IPs")
- return needRestart, err
- }
- }
- if len(email) > 0 {
- var enables []bool
- if err := db.Model(xray.ClientTraffic{}).Where("email = ?", email).Limit(1).Pluck("enable", &enables).Error; err != nil {
- logger.Error("Get stats error")
- return needRestart, err
- }
- notDepleted := len(enables) > 0 && enables[0]
- if !emailShared && !keepTraffic {
- if err := inboundSvc.DelClientStat(db, email); err != nil {
- logger.Error("Delete stats Data Error")
- return needRestart, err
- }
- }
- if r.needApiDel && notDepleted && oldInbound.NodeID == nil {
- rt, rterr := inboundSvc.runtimeFor(oldInbound)
- if rterr != nil {
- needRestart = true
- } else if err1 := rt.RemoveUser(context.Background(), oldInbound, email); err1 != nil {
- if !strings.Contains(err1.Error(), fmt.Sprintf("User %s not found.", email)) {
- needRestart = true
- }
- }
- }
- }
- if oldInbound.NodeID != nil && len(email) > 0 {
- rt, push, dirty, perr := inboundSvc.nodePushPlan(oldInbound)
- if perr != nil {
- return needRestart, perr
- }
- if dirty {
- markDirty = true
- }
- if push {
- if err1 := rt.DeleteUser(context.Background(), oldInbound, email); err1 != nil {
- logger.Warning("Error in deleting client on", rt.Name(), ":", err1)
- markDirty = true
- }
- }
- }
- }
- if err := db.Save(oldInbound).Error; err != nil {
- return needRestart, err
- }
- finalClients, gcErr := inboundSvc.GetClients(oldInbound)
- if gcErr != nil {
- return needRestart, gcErr
- }
- if err := s.SyncInbound(db, inboundId, finalClients); err != nil {
- return needRestart, err
- }
- if markDirty && oldInbound.NodeID != nil {
- if dErr := (&NodeService{}).MarkNodeDirty(*oldInbound.NodeID); dErr != nil {
- logger.Warning("mark node dirty failed:", dErr)
- }
- }
- return needRestart, nil
- }
- func (s *ClientService) checkEmailsExistForClients(inboundSvc *InboundService, clients []model.Client, emailSubIDs map[string]string) (string, error) {
- if emailSubIDs == nil {
- var err error
- emailSubIDs, err = inboundSvc.getAllEmailSubIDs()
- if err != nil {
- return "", err
- }
- }
- seen := make(map[string]string, len(clients))
- for _, client := range clients {
- if client.Email == "" {
- continue
- }
- key := strings.ToLower(client.Email)
- if prev, ok := seen[key]; ok {
- if prev != client.SubID || client.SubID == "" {
- return client.Email, nil
- }
- continue
- }
- seen[key] = client.SubID
- if existingSub, ok := emailSubIDs[key]; ok {
- if client.SubID == "" || existingSub == "" || existingSub != client.SubID {
- return client.Email, nil
- }
- }
- }
- return "", nil
- }
- func (s *ClientService) AddInboundClient(inboundSvc *InboundService, data *model.Inbound) (bool, error) {
- return s.addInboundClient(inboundSvc, data, nil)
- }
- // addInboundClient is AddInboundClient with an optional precomputed email→subId
- // map. Bulk callers pass a single snapshot so the global getAllEmailSubIDs scan
- // runs once for the whole batch instead of once per target inbound; a nil map
- // makes it compute its own (the single-add path).
- func (s *ClientService) addInboundClient(inboundSvc *InboundService, data *model.Inbound, emailSubIDs map[string]string) (bool, error) {
- defer lockInbound(data.Id).Unlock()
- clients, err := inboundSvc.GetClients(data)
- if err != nil {
- return false, err
- }
- var settings map[string]any
- err = json.Unmarshal([]byte(data.Settings), &settings)
- if err != nil {
- return false, err
- }
- interfaceClients := settings["clients"].([]any)
- nowTs := time.Now().Unix() * 1000
- for i := range interfaceClients {
- if cm, ok := interfaceClients[i].(map[string]any); ok {
- if _, ok2 := cm["created_at"]; !ok2 {
- cm["created_at"] = nowTs
- }
- cm["updated_at"] = nowTs
- existingSub, _ := cm["subId"].(string)
- if strings.TrimSpace(existingSub) == "" {
- cm["subId"] = random.NumLower(16)
- }
- interfaceClients[i] = cm
- }
- }
- existEmail, err := s.checkEmailsExistForClients(inboundSvc, clients, emailSubIDs)
- if err != nil {
- return false, err
- }
- if existEmail != "" {
- return false, common.NewError("Duplicate email:", existEmail)
- }
- oldInbound, err := inboundSvc.GetInbound(data.Id)
- if err != nil {
- return false, err
- }
- for _, client := range clients {
- if strings.TrimSpace(client.Email) == "" {
- return false, common.NewError("client email is required")
- }
- switch oldInbound.Protocol {
- case "trojan":
- if client.Password == "" {
- return false, common.NewError("empty client ID")
- }
- case "shadowsocks":
- if client.Email == "" {
- return false, common.NewError("empty client ID")
- }
- case "hysteria":
- if client.Auth == "" {
- return false, common.NewError("empty client ID")
- }
- default:
- if client.ID == "" {
- return false, common.NewError("empty client ID")
- }
- }
- }
- var oldSettings map[string]any
- err = json.Unmarshal([]byte(oldInbound.Settings), &oldSettings)
- if err != nil {
- return false, err
- }
- if oldInbound.Protocol == model.Shadowsocks {
- applyShadowsocksClientMethod(interfaceClients, oldSettings)
- }
- oldClients := oldSettings["clients"].([]any)
- oldClients = compactOrphans(database.GetDB(), oldClients)
- oldClients = append(oldClients, interfaceClients...)
- oldSettings["clients"] = oldClients
- newSettings, err := json.MarshalIndent(oldSettings, "", " ")
- if err != nil {
- return false, err
- }
- oldInbound.Settings = string(newSettings)
- db := database.GetDB()
- tx := db.Begin()
- markDirty := false
- defer func() {
- if err != nil {
- tx.Rollback()
- return
- }
- tx.Commit()
- if markDirty && oldInbound.NodeID != nil {
- if dErr := (&NodeService{}).MarkNodeDirty(*oldInbound.NodeID); dErr != nil {
- logger.Warning("mark node dirty failed:", dErr)
- }
- }
- }()
- needRestart := false
- rt, push, dirty, perr := inboundSvc.nodePushPlan(oldInbound)
- if perr != nil {
- err = perr
- return false, err
- }
- if dirty {
- markDirty = true
- }
- if oldInbound.NodeID == nil {
- if !push {
- needRestart = true
- } else {
- for _, client := range clients {
- if len(client.Email) == 0 {
- needRestart = true
- continue
- }
- inboundSvc.AddClientStat(tx, data.Id, &client)
- if !client.Enable {
- continue
- }
- cipher := ""
- if oldInbound.Protocol == "shadowsocks" {
- cipher = oldSettings["method"].(string)
- }
- err1 := rt.AddUser(context.Background(), oldInbound, map[string]any{
- "email": client.Email,
- "id": client.ID,
- "auth": client.Auth,
- "security": client.Security,
- "flow": client.Flow,
- "password": client.Password,
- "cipher": cipher,
- })
- if err1 == nil {
- logger.Debug("Client added on", rt.Name(), ":", client.Email)
- } else {
- logger.Debug("Error in adding client on", rt.Name(), ":", err1)
- needRestart = true
- }
- }
- }
- } else {
- for _, client := range clients {
- if len(client.Email) > 0 {
- inboundSvc.AddClientStat(tx, data.Id, &client)
- }
- if push {
- if err1 := rt.AddClient(context.Background(), oldInbound, client); err1 != nil {
- logger.Warning("Error in adding client on", rt.Name(), ":", err1)
- markDirty = true
- push = false
- }
- }
- }
- }
- if err = tx.Save(oldInbound).Error; err != nil {
- return false, err
- }
- finalClients, gcErr := inboundSvc.GetClients(oldInbound)
- if gcErr != nil {
- err = gcErr
- return false, err
- }
- if err = s.SyncInbound(tx, oldInbound.Id, finalClients); err != nil {
- return false, err
- }
- return needRestart, nil
- }
- func (s *ClientService) UpdateInboundClient(inboundSvc *InboundService, data *model.Inbound, oldEmail string) (bool, error) {
- defer lockInbound(data.Id).Unlock()
- clients, err := inboundSvc.GetClients(data)
- if err != nil {
- return false, err
- }
- var settings map[string]any
- err = json.Unmarshal([]byte(data.Settings), &settings)
- if err != nil {
- return false, err
- }
- interfaceClients := settings["clients"].([]any)
- oldInbound, err := inboundSvc.GetInbound(data.Id)
- if err != nil {
- return false, err
- }
- oldClients, err := inboundSvc.GetClients(oldInbound)
- if err != nil {
- return false, err
- }
- newClientId := ""
- switch oldInbound.Protocol {
- case "trojan":
- newClientId = clients[0].Password
- case "shadowsocks":
- newClientId = clients[0].Email
- case "hysteria":
- newClientId = clients[0].Auth
- default:
- newClientId = clients[0].ID
- }
- // Locate the client to replace by email — the client's stable identity.
- // Credentials (uuid/password/auth) can drift from the inbound JSON, so they
- // are never used for matching.
- clientIndex := -1
- for index, oldClient := range oldClients {
- if strings.EqualFold(oldClient.Email, oldEmail) {
- oldEmail = oldClient.Email
- clientIndex = index
- break
- }
- }
- if newClientId == "" || clientIndex == -1 {
- return false, common.NewError("empty client ID")
- }
- if strings.TrimSpace(clients[0].Email) == "" {
- return false, common.NewError("client email is required")
- }
- if clients[0].Email != oldEmail {
- existEmail, err := s.checkEmailsExistForClients(inboundSvc, clients, nil)
- if err != nil {
- return false, err
- }
- if existEmail != "" {
- return false, common.NewError("Duplicate email:", existEmail)
- }
- }
- var oldSettings map[string]any
- err = json.Unmarshal([]byte(oldInbound.Settings), &oldSettings)
- if err != nil {
- return false, err
- }
- settingsClients := oldSettings["clients"].([]any)
- var preservedCreated any
- var preservedSubID string
- if clientIndex >= 0 && clientIndex < len(settingsClients) {
- if oldMap, ok := settingsClients[clientIndex].(map[string]any); ok {
- if v, ok2 := oldMap["created_at"]; ok2 {
- preservedCreated = v
- }
- preservedSubID, _ = oldMap["subId"].(string)
- }
- }
- if len(interfaceClients) > 0 {
- if newMap, ok := interfaceClients[0].(map[string]any); ok {
- if preservedCreated == nil {
- preservedCreated = time.Now().Unix() * 1000
- }
- newMap["created_at"] = preservedCreated
- newMap["updated_at"] = time.Now().Unix() * 1000
- newSub, _ := newMap["subId"].(string)
- if strings.TrimSpace(newSub) == "" {
- if strings.TrimSpace(preservedSubID) != "" {
- newMap["subId"] = preservedSubID
- } else {
- newMap["subId"] = random.NumLower(16)
- }
- }
- interfaceClients[0] = newMap
- }
- }
- if oldInbound.Protocol == model.Shadowsocks {
- applyShadowsocksClientMethod(interfaceClients, oldSettings)
- }
- settingsClients[clientIndex] = interfaceClients[0]
- oldSettings["clients"] = settingsClients
- if oldInbound.Protocol == model.VLESS {
- hasVisionFlow := false
- for _, c := range settingsClients {
- cm, ok := c.(map[string]any)
- if !ok {
- continue
- }
- if flow, _ := cm["flow"].(string); flow == "xtls-rprx-vision" {
- hasVisionFlow = true
- break
- }
- }
- if !hasVisionFlow {
- delete(oldSettings, "testseed")
- }
- }
- newSettings, err := json.MarshalIndent(oldSettings, "", " ")
- if err != nil {
- return false, err
- }
- oldInbound.Settings = string(newSettings)
- db := database.GetDB()
- tx := db.Begin()
- markDirty := false
- defer func() {
- if err != nil {
- tx.Rollback()
- return
- }
- tx.Commit()
- if markDirty && oldInbound.NodeID != nil {
- if dErr := (&NodeService{}).MarkNodeDirty(*oldInbound.NodeID); dErr != nil {
- logger.Warning("mark node dirty failed:", dErr)
- }
- }
- }()
- if len(clients[0].Email) > 0 {
- if len(oldEmail) > 0 {
- emailUnchanged := strings.EqualFold(oldEmail, clients[0].Email)
- targetExists := int64(0)
- if !emailUnchanged {
- if err = tx.Model(xray.ClientTraffic{}).Where("email = ?", clients[0].Email).Count(&targetExists).Error; err != nil {
- return false, err
- }
- }
- if emailUnchanged || targetExists == 0 {
- err = inboundSvc.UpdateClientStat(tx, oldEmail, &clients[0])
- if err != nil {
- return false, err
- }
- err = inboundSvc.UpdateClientIPs(tx, oldEmail, clients[0].Email)
- if err != nil {
- return false, err
- }
- } else {
- stillUsed, sErr := inboundSvc.emailUsedByOtherInbounds(oldEmail, data.Id)
- if sErr != nil {
- return false, sErr
- }
- if !stillUsed {
- if err = inboundSvc.DelClientStat(tx, oldEmail); err != nil {
- return false, err
- }
- if err = inboundSvc.DelClientIPs(tx, oldEmail); err != nil {
- return false, err
- }
- }
- if err = inboundSvc.UpdateClientStat(tx, clients[0].Email, &clients[0]); err != nil {
- return false, err
- }
- }
- } else {
- inboundSvc.AddClientStat(tx, data.Id, &clients[0])
- }
- } else {
- stillUsed, err := inboundSvc.emailUsedByOtherInbounds(oldEmail, data.Id)
- if err != nil {
- return false, err
- }
- if !stillUsed {
- err = inboundSvc.DelClientStat(tx, oldEmail)
- if err != nil {
- return false, err
- }
- err = inboundSvc.DelClientIPs(tx, oldEmail)
- if err != nil {
- return false, err
- }
- }
- }
- needRestart := false
- if len(oldEmail) > 0 {
- rt, push, dirty, perr := inboundSvc.nodePushPlan(oldInbound)
- if perr != nil {
- err = perr
- return false, err
- }
- if dirty {
- markDirty = true
- }
- if oldInbound.NodeID == nil {
- if !push {
- needRestart = true
- } else {
- if oldClients[clientIndex].Enable {
- err1 := rt.RemoveUser(context.Background(), oldInbound, oldEmail)
- if err1 == nil {
- logger.Debug("Old client deleted on", rt.Name(), ":", oldEmail)
- } else if strings.Contains(err1.Error(), fmt.Sprintf("User %s not found.", oldEmail)) {
- logger.Debug("User is already deleted. Nothing to do more...")
- } else {
- logger.Debug("Error in deleting client on", rt.Name(), ":", err1)
- needRestart = true
- }
- }
- if clients[0].Enable {
- cipher := ""
- if oldInbound.Protocol == "shadowsocks" {
- cipher = oldSettings["method"].(string)
- }
- err1 := rt.AddUser(context.Background(), oldInbound, map[string]any{
- "email": clients[0].Email,
- "id": clients[0].ID,
- "security": clients[0].Security,
- "flow": clients[0].Flow,
- "auth": clients[0].Auth,
- "password": clients[0].Password,
- "cipher": cipher,
- })
- if err1 == nil {
- logger.Debug("Client edited on", rt.Name(), ":", clients[0].Email)
- } else {
- logger.Debug("Error in adding client on", rt.Name(), ":", err1)
- needRestart = true
- }
- }
- }
- } else if push {
- if err1 := rt.UpdateUser(context.Background(), oldInbound, oldEmail, clients[0]); err1 != nil {
- logger.Warning("Error in updating client on", rt.Name(), ":", err1)
- markDirty = true
- }
- }
- } else {
- logger.Debug("Client old email not found")
- needRestart = true
- }
- if err = tx.Save(oldInbound).Error; err != nil {
- return false, err
- }
- finalClients, gcErr := inboundSvc.GetClients(oldInbound)
- if gcErr != nil {
- err = gcErr
- return false, err
- }
- if err = s.SyncInbound(tx, oldInbound.Id, finalClients); err != nil {
- return false, err
- }
- return needRestart, nil
- }
- func (s *ClientService) DelInboundClientByEmail(inboundSvc *InboundService, inboundId int, email string, keepTraffic bool) (bool, error) {
- defer lockInbound(inboundId).Unlock()
- oldInbound, err := inboundSvc.GetInbound(inboundId)
- if err != nil {
- logger.Error("Load Old Data Error")
- return false, err
- }
- var settings map[string]any
- if err := json.Unmarshal([]byte(oldInbound.Settings), &settings); err != nil {
- return false, err
- }
- interfaceClients, ok := settings["clients"].([]any)
- if !ok {
- return false, common.NewError("invalid clients format in inbound settings")
- }
- var newClients []any
- needApiDel := false
- found := false
- for _, client := range interfaceClients {
- c, ok := client.(map[string]any)
- if !ok {
- continue
- }
- if cEmail, ok := c["email"].(string); ok && cEmail == email {
- found = true
- needApiDel, _ = c["enable"].(bool)
- } else {
- newClients = append(newClients, client)
- }
- }
- if !found {
- return false, fmt.Errorf("%w for email: %s", ErrClientNotInInbound, email)
- }
- db := database.GetDB()
- newClients = compactOrphans(db, newClients)
- if newClients == nil {
- newClients = []any{}
- }
- settings["clients"] = newClients
- newSettings, err := json.MarshalIndent(settings, "", " ")
- if err != nil {
- return false, err
- }
- oldInbound.Settings = string(newSettings)
- emailShared, err := inboundSvc.emailUsedByOtherInbounds(email, inboundId)
- if err != nil {
- return false, err
- }
- if !emailShared && !keepTraffic {
- if err := inboundSvc.DelClientIPs(db, email); err != nil {
- logger.Error("Error in delete client IPs")
- return false, err
- }
- }
- needRestart := false
- markDirty := false
- if len(email) > 0 && !emailShared {
- if !keepTraffic {
- traffic, err := inboundSvc.GetClientTrafficByEmail(email)
- if err != nil {
- return false, err
- }
- if traffic != nil {
- if err := inboundSvc.DelClientStat(db, email); err != nil {
- logger.Error("Delete stats Data Error")
- return false, err
- }
- }
- }
- if oldInbound.NodeID == nil {
- // Local inbound: a disabled client isn't in the running Xray, so only
- // a live one (needApiDel) needs an API removal.
- if needApiDel {
- rt, push, dirty, perr := inboundSvc.nodePushPlan(oldInbound)
- if perr != nil {
- return false, perr
- }
- if dirty {
- markDirty = true
- }
- if !push {
- needRestart = true
- } else if err1 := rt.RemoveUser(context.Background(), oldInbound, email); err1 == nil {
- logger.Debug("Client deleted on", rt.Name(), ":", email)
- needRestart = false
- } else if strings.Contains(err1.Error(), fmt.Sprintf("User %s not found.", email)) {
- logger.Debug("User is already deleted. Nothing to do more...")
- } else {
- logger.Debug("Error in deleting client on", rt.Name(), ":", email)
- needRestart = true
- }
- }
- } else {
- // Node inbound: propagate the delete regardless of the enable flag —
- // the node's own DB still carries a disabled client and would
- // resurrect it on the next snapshot otherwise.
- rt, push, dirty, perr := inboundSvc.nodePushPlan(oldInbound)
- if perr != nil {
- return false, perr
- }
- if dirty {
- markDirty = true
- }
- if push {
- if err1 := rt.DeleteUser(context.Background(), oldInbound, email); err1 != nil {
- logger.Warning("Error in deleting client on", rt.Name(), ":", err1)
- markDirty = true
- }
- }
- }
- }
- if err := db.Save(oldInbound).Error; err != nil {
- return false, err
- }
- finalClients, gcErr := inboundSvc.GetClients(oldInbound)
- if gcErr != nil {
- return false, gcErr
- }
- if err := s.SyncInbound(db, inboundId, finalClients); err != nil {
- return false, err
- }
- if markDirty && oldInbound.NodeID != nil {
- if dErr := (&NodeService{}).MarkNodeDirty(*oldInbound.NodeID); dErr != nil {
- logger.Warning("mark node dirty failed:", dErr)
- }
- }
- return needRestart, nil
- }
- func (s *ClientService) SetClientTelegramUserID(inboundSvc *InboundService, trafficId int, tgId int64) (bool, error) {
- traffic, inbound, err := inboundSvc.GetClientInboundByTrafficID(trafficId)
- if err != nil {
- return false, err
- }
- if inbound == nil {
- return false, common.NewError("Inbound Not Found For Traffic ID:", trafficId)
- }
- clientEmail := traffic.Email
- oldClients, err := inboundSvc.GetClients(inbound)
- if err != nil {
- return false, err
- }
- found := false
- for _, oldClient := range oldClients {
- if oldClient.Email == clientEmail {
- found = true
- break
- }
- }
- if !found {
- return false, common.NewError("Client Not Found For Email:", clientEmail)
- }
- var settings map[string]any
- err = json.Unmarshal([]byte(inbound.Settings), &settings)
- if err != nil {
- return false, err
- }
- clients := settings["clients"].([]any)
- var newClients []any
- for client_index := range clients {
- c := clients[client_index].(map[string]any)
- if c["email"] == clientEmail {
- c["tgId"] = tgId
- c["updated_at"] = time.Now().Unix() * 1000
- newClients = append(newClients, any(c))
- }
- }
- settings["clients"] = newClients
- modifiedSettings, err := json.MarshalIndent(settings, "", " ")
- if err != nil {
- return false, err
- }
- inbound.Settings = string(modifiedSettings)
- needRestart, err := s.UpdateInboundClient(inboundSvc, inbound, clientEmail)
- return needRestart, err
- }
- func (s *ClientService) CheckIsEnabledByEmail(inboundSvc *InboundService, clientEmail string) (bool, error) {
- _, inbound, err := inboundSvc.GetClientInboundByEmail(clientEmail)
- if err != nil {
- return false, err
- }
- if inbound == nil {
- return false, common.NewError("Inbound Not Found For Email:", clientEmail)
- }
- clients, err := inboundSvc.GetClients(inbound)
- if err != nil {
- return false, err
- }
- isEnable := false
- for _, client := range clients {
- if client.Email == clientEmail {
- isEnable = client.Enable
- break
- }
- }
- return isEnable, err
- }
- func (s *ClientService) ToggleClientEnableByEmail(inboundSvc *InboundService, clientEmail string) (bool, bool, error) {
- _, inbound, err := inboundSvc.GetClientInboundByEmail(clientEmail)
- if err != nil {
- return false, false, err
- }
- if inbound == nil {
- return false, false, common.NewError("Inbound Not Found For Email:", clientEmail)
- }
- oldClients, err := inboundSvc.GetClients(inbound)
- if err != nil {
- return false, false, err
- }
- found := false
- clientOldEnabled := false
- for _, oldClient := range oldClients {
- if oldClient.Email == clientEmail {
- found = true
- clientOldEnabled = oldClient.Enable
- break
- }
- }
- if !found {
- return false, false, common.NewError("Client Not Found For Email:", clientEmail)
- }
- var settings map[string]any
- err = json.Unmarshal([]byte(inbound.Settings), &settings)
- if err != nil {
- return false, false, err
- }
- clients := settings["clients"].([]any)
- var newClients []any
- for client_index := range clients {
- c := clients[client_index].(map[string]any)
- if c["email"] == clientEmail {
- c["enable"] = !clientOldEnabled
- c["updated_at"] = time.Now().Unix() * 1000
- newClients = append(newClients, any(c))
- }
- }
- settings["clients"] = newClients
- modifiedSettings, err := json.MarshalIndent(settings, "", " ")
- if err != nil {
- return false, false, err
- }
- inbound.Settings = string(modifiedSettings)
- needRestart, err := s.UpdateInboundClient(inboundSvc, inbound, clientEmail)
- if err != nil {
- return false, needRestart, err
- }
- return !clientOldEnabled, needRestart, nil
- }
- func (s *ClientService) SetClientEnableByEmail(inboundSvc *InboundService, clientEmail string, enable bool) (bool, bool, error) {
- current, err := s.CheckIsEnabledByEmail(inboundSvc, clientEmail)
- if err != nil {
- return false, false, err
- }
- if current == enable {
- return false, false, nil
- }
- newEnabled, needRestart, err := s.ToggleClientEnableByEmail(inboundSvc, clientEmail)
- if err != nil {
- return false, needRestart, err
- }
- return newEnabled == enable, needRestart, nil
- }
- // applyClientFieldByEmail loads the inbound currently hosting clientEmail,
- // confirms the client exists, applies mutate to the matching client (plus a
- // refreshed updated_at), and hands a single-client update payload to
- // UpdateInboundClient. The rebuilt clients array intentionally contains only
- // the matched client — that is the input contract UpdateInboundClient expects
- // (clients[0] is the new data; clientEmail locates the row to replace). It
- // backs the single-field by-email setters below.
- // applyClientFieldByEmail mutates a client field on every inbound the email is
- // attached to. A multi-inbound client is one logical identity: patching only
- // the first inbound's JSON would leave the siblings stale, and the next
- // SyncInbound over a stale sibling would revert the edit in the normalized
- // records (#5039).
- func (s *ClientService) applyClientFieldByEmail(inboundSvc *InboundService, clientEmail string, mutate func(c map[string]any)) (bool, error) {
- inboundIds, err := s.GetInboundIdsForEmail(database.GetDB(), clientEmail)
- if err != nil {
- return false, err
- }
- if len(inboundIds) == 0 {
- // Legacy fallback for clients that only live in the inbound JSON and
- // were never normalized into client_inbounds.
- _, inbound, gErr := inboundSvc.GetClientInboundByEmail(clientEmail)
- if gErr != nil {
- return false, gErr
- }
- if inbound == nil {
- return false, common.NewError("Inbound Not Found For Email:", clientEmail)
- }
- inboundIds = []int{inbound.Id}
- }
- needRestart := false
- found := false
- for _, ibId := range inboundIds {
- inbound, gErr := inboundSvc.GetInbound(ibId)
- if gErr != nil {
- return needRestart, gErr
- }
- var settings map[string]any
- if uErr := json.Unmarshal([]byte(inbound.Settings), &settings); uErr != nil {
- return needRestart, uErr
- }
- clients, _ := settings["clients"].([]any)
- // UpdateInboundClient expects a single-client payload, so keep only the
- // matching entry in the scratch copy; it splices the result back into
- // the inbound's full client list itself.
- var newClients []any
- for client_index := range clients {
- c, ok := clients[client_index].(map[string]any)
- if !ok {
- continue
- }
- if c["email"] == clientEmail {
- mutate(c)
- c["updated_at"] = time.Now().Unix() * 1000
- newClients = append(newClients, any(c))
- }
- }
- if len(newClients) == 0 {
- continue
- }
- found = true
- settings["clients"] = newClients
- modifiedSettings, mErr := json.MarshalIndent(settings, "", " ")
- if mErr != nil {
- return needRestart, mErr
- }
- inbound.Settings = string(modifiedSettings)
- nr, uErr := s.UpdateInboundClient(inboundSvc, inbound, clientEmail)
- if uErr != nil {
- return needRestart, uErr
- }
- needRestart = needRestart || nr
- }
- if !found {
- return needRestart, common.NewError("Client Not Found For Email:", clientEmail)
- }
- return needRestart, nil
- }
- func (s *ClientService) ResetClientIpLimitByEmail(inboundSvc *InboundService, clientEmail string, count int) (bool, error) {
- return s.applyClientFieldByEmail(inboundSvc, clientEmail, func(c map[string]any) {
- c["limitIp"] = count
- })
- }
- func (s *ClientService) ResetClientExpiryTimeByEmail(inboundSvc *InboundService, clientEmail string, expiry_time int64) (bool, error) {
- return s.applyClientFieldByEmail(inboundSvc, clientEmail, func(c map[string]any) {
- c["expiryTime"] = expiry_time
- })
- }
- func (s *ClientService) ResetClientTrafficLimitByEmail(inboundSvc *InboundService, clientEmail string, totalGB int) (bool, error) {
- if totalGB < 0 {
- return false, common.NewError("totalGB must be >= 0")
- }
- return s.applyClientFieldByEmail(inboundSvc, clientEmail, func(c map[string]any) {
- c["totalGB"] = totalGB * 1024 * 1024 * 1024
- })
- }
|