Alireza Ahmadi
92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
package service
|
|
|
|
import (
|
|
"encoding/json"
|
|
"s-ui/database"
|
|
"s-ui/database/model"
|
|
"s-ui/logger"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type ClientService struct {
|
|
}
|
|
|
|
func (s *ClientService) GetAll() ([]model.Client, error) {
|
|
db := database.GetDB()
|
|
clients := []model.Client{}
|
|
err := db.Model(model.Client{}).Scan(&clients).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return clients, nil
|
|
}
|
|
|
|
func (s *ClientService) Save(tx *gorm.DB, changes []model.Changes) error {
|
|
var err error
|
|
for _, change := range changes {
|
|
client := model.Client{}
|
|
err = json.Unmarshal(change.Obj, &client)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
switch change.Action {
|
|
case "new":
|
|
err = tx.Create(&client).Error
|
|
case "del":
|
|
err = tx.Where("id = ?", change.Index).Delete(model.Client{}).Error
|
|
default:
|
|
err = tx.Save(client).Error
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (s *ClientService) DepleteClients() ([]string, []string, error) {
|
|
var err error
|
|
var clients []model.Client
|
|
var changes []model.Changes
|
|
now := time.Now().Unix()
|
|
db := database.GetDB()
|
|
err = db.Model(model.Client{}).Where("enable = true AND ((volume >0 AND up+down > volume) OR (expiry > 0 AND expiry < ?))", now).Scan(&clients).Error
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
dt := time.Now().Unix()
|
|
var users, inbounds []string
|
|
for _, client := range clients {
|
|
logger.Debug("Client ", client.Name, " is going to be disabled")
|
|
users = append(users, client.Name)
|
|
var userInbounds []string
|
|
json.Unmarshal(client.Inbounds, &userInbounds)
|
|
inbounds = append(inbounds, userInbounds...)
|
|
changes = append(changes, model.Changes{
|
|
DateTime: dt,
|
|
Actor: "DepleteJob",
|
|
Key: "clients",
|
|
Action: "disable",
|
|
Obj: json.RawMessage("\"" + client.Name + "\""),
|
|
})
|
|
}
|
|
|
|
// Save changes
|
|
if len(changes) > 0 {
|
|
err = db.Model(model.Client{}).Where("enable = true AND ((volume >0 AND up+down > volume) OR (expiry > 0 AND expiry < ?))", now).Update("enable", false).Error
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
err = db.Model(model.Changes{}).Create(&changes).Error
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
LastUpdate = dt
|
|
}
|
|
|
|
return users, inbounds, nil
|
|
}
|