Alireza Ahmadi
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package service
|
|
|
|
import (
|
|
"s-ui/database"
|
|
"s-ui/database/model"
|
|
"s-ui/logger"
|
|
"s-ui/util/common"
|
|
"time"
|
|
)
|
|
|
|
type UserService struct {
|
|
}
|
|
|
|
func (s *UserService) GetFirstUser() (*model.User, error) {
|
|
db := database.GetDB()
|
|
|
|
user := &model.User{}
|
|
err := db.Model(model.User{}).
|
|
First(user).
|
|
Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return user, nil
|
|
}
|
|
|
|
func (s *UserService) UpdateFirstUser(username string, password string) error {
|
|
if username == "" {
|
|
return common.NewError("username can not be empty")
|
|
} else if password == "" {
|
|
return common.NewError("password can not be empty")
|
|
}
|
|
db := database.GetDB()
|
|
user := &model.User{}
|
|
err := db.Model(model.User{}).First(user).Error
|
|
if database.IsNotFound(err) {
|
|
user.Username = username
|
|
user.Password = password
|
|
return db.Model(model.User{}).Create(user).Error
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
user.Username = username
|
|
user.Password = password
|
|
return db.Save(user).Error
|
|
}
|
|
|
|
func (s *UserService) Login(username string, password string, remoteIP string) (string, error) {
|
|
user := s.CheckUser(username, password, remoteIP)
|
|
if user == nil {
|
|
return "", common.NewError("wrong user or password! IP: ", remoteIP)
|
|
}
|
|
return user.Username, nil
|
|
}
|
|
|
|
func (s *UserService) CheckUser(username string, password string, remoteIP string) *model.User {
|
|
db := database.GetDB()
|
|
|
|
user := &model.User{}
|
|
err := db.Model(model.User{}).
|
|
Where("username = ? and password = ?", username, password).
|
|
First(user).
|
|
Error
|
|
if database.IsNotFound(err) {
|
|
return nil
|
|
} else if err != nil {
|
|
logger.Warning("check user err:", err, " IP: ", remoteIP)
|
|
return nil
|
|
}
|
|
|
|
lastLoginTxt := time.Now().Format("2006-01-02 15:04:05") + " " + remoteIP
|
|
err = db.Model(model.User{}).
|
|
Where("username = ?", username).
|
|
Update("last_logins", &lastLoginTxt).Error
|
|
if err != nil {
|
|
logger.Warning("unable to log login data", err)
|
|
}
|
|
return user
|
|
}
|
|
|
|
func (s *UserService) GetUsers() (*[]model.User, error) {
|
|
var users []model.User
|
|
db := database.GetDB()
|
|
err := db.Model(model.User{}).Select("id,username,last_logins").Scan(&users).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &users, nil
|
|
}
|
|
|
|
func (s *UserService) ChangePass(id string, oldPass string, newUser string, newPass string) error {
|
|
db := database.GetDB()
|
|
user := &model.User{}
|
|
err := db.Model(model.User{}).Where("id = ? AND password = ?", id, oldPass).First(user).Error
|
|
if err != nil || database.IsNotFound(err) {
|
|
return err
|
|
}
|
|
user.Username = newUser
|
|
user.Password = newPass
|
|
return db.Save(user).Error
|
|
}
|