diff --git a/database/db.go b/database/db.go index 251ca3e7..19cf0478 100644 --- a/database/db.go +++ b/database/db.go @@ -7,6 +7,7 @@ import ( "log" "os" "path" + "slices" "x-ui/config" "x-ui/database/model" @@ -34,6 +35,7 @@ func initModels() error { &model.Setting{}, &model.InboundClientIps{}, &xray.ClientTraffic{}, + &model.HistoryOfSeeders{}, } for _, model := range models { if err := db.AutoMigrate(model); err != nil { @@ -61,6 +63,40 @@ func initUser() error { return nil } +func runSeeders(isUsersEmpty bool) error { + empty, err := isTableEmpty("history_of_seeders") + if err != nil { + log.Printf("Error checking if users table is empty: %v", err) + return err + } + + if empty && isUsersEmpty { + hashSeeder := &model.HistoryOfSeeders{ + SeederName: "UserPasswordHash", + } + return db.Create(hashSeeder).Error + } else { + var seedersHistory []string + db.Model(&model.HistoryOfSeeders{}).Pluck("seeder_name", &seedersHistory) + + if !slices.Contains(seedersHistory, "UserPasswordHash") && !isUsersEmpty { + var users []model.User + db.Find(&users) + + for _, user := range users { + db.Model(&user).Update("password", crypto.HashSHA256(user.Password)) + } + + hashSeeder := &model.HistoryOfSeeders{ + SeederName: "UserPasswordHash", + } + return db.Create(hashSeeder).Error + } + } + + return nil +} + func isTableEmpty(tableName string) (bool, error) { var count int64 err := db.Table(tableName).Count(&count).Error @@ -93,11 +129,16 @@ func InitDB(dbPath string) error { if err := initModels(); err != nil { return err } - if err := initUser(); err != nil { - return err - } + empty, err := isTableEmpty("users") + if empty { + if err := initUser(); err != nil { + return err + } - return nil + } + return runSeeders(empty) + + // return nil } func CloseDB() error { diff --git a/database/model/model.go b/database/model/model.go index e9d1836f..7a20de16 100644 --- a/database/model/model.go +++ b/database/model/model.go @@ -63,6 +63,11 @@ type InboundClientIps struct { Ips string `json:"ips" form:"ips"` } +type HistoryOfSeeders struct { + Id int `json:"id" gorm:"primaryKey;autoIncrement"` + SeederName string `json:"seederName"` +} + func (i *Inbound) GenXrayInboundConfig() *xray.InboundConfig { listen := i.Listen if listen != "" {