package models

import (
	"github.com/ouqiang/gocron/modules/utils"
	"time"
)

const PasswordSaltLength = 6

// 用户model
type User struct {
	Id        int       `xorm:"pk autoincr notnull "`
	Name      string    `xorm:"varchar(32) notnull unique"`             // 用户名
	Password  string    `xorm:"char(32) notnull "`                      // 密码
	Salt      string    `xorm:"char(6) notnull "`                       // 密码盐值
	Email     string    `xorm:"varchar(50) notnull unique default '' "` // 邮箱
	Created   time.Time `xorm:"datetime notnull created"`
	Updated   time.Time `xorm:"datetime updated"`
	IsAdmin   int8      `xorm:"tinyint notnull default 0"` // 是否是管理员 1:管理员 0:普通用户
	Status    Status    `xorm:"tinyint notnull default 1"` // 1: 正常 0:禁用
	BaseModel `xorm:"-"`
}

// 新增
func (user *User) Create() (insertId int, err error) {
	user.Status = Enabled
	user.Salt = user.generateSalt()
	user.Password = user.encryptPassword(user.Password, user.Salt)

	_, err = Db.Insert(user)
	if err == nil {
		insertId = user.Id
	}

	return
}

// 更新
func (user *User) Update(id int, data CommonMap) (int64, error) {
	return Db.Table(user).ID(id).Update(data)
}

func (user *User) UpdatePassword(id int, password string) (int64, error) {
	salt := user.generateSalt()
	safePassword := user.encryptPassword(password, salt)

	return user.Update(id, CommonMap{"password": safePassword, "salt": salt})
}

// 删除
func (user *User) Delete(id int) (int64, error) {
	return Db.Id(id).Delete(user)
}

// 禁用
func (user *User) Disable(id int) (int64, error) {
	return user.Update(id, CommonMap{"status": Disabled})
}

// 激活
func (user *User) Enable(id int) (int64, error) {
	return user.Update(id, CommonMap{"status": Enabled})
}

// 验证用户名和密码
func (user *User) Match(username, password string) bool {
	where := "(name = ? OR email = ?) AND status =? "
	_, err := Db.Where(where, username, username, Enabled).Get(user)
	if err != nil {
		return false
	}
	hashPassword := user.encryptPassword(password, user.Salt)
	if hashPassword != user.Password {
		return false
	}

	return true
}

// 获取用户详情
func (user *User) Find(id int) (error) {
	_, err := Db.Id(id).Get(user)

	return err
}

// 用户名是否存在
func (user *User) UsernameExists(username string, uid int) (int64, error) {
	if uid > 0 {
		return Db.Where("name = ? AND id != ?", username, uid).Count(user)
	}

	return Db.Where("name = ?", username).Count(user)
}

// 邮箱地址是否存在
func (user *User) EmailExists(email string, uid int) (int64, error) {
	if uid > 0 {
		return Db.Where("email = ? AND id != ?", email, uid).Count(user)
	}

	return Db.Where("email = ?", email).Count(user)
}

func (user *User) List(params CommonMap) ([]User, error) {
	user.parsePageAndPageSize(params)
	list := make([]User, 0)
	err := Db.Desc("id").Limit(user.PageSize, user.pageLimitOffset()).Find(&list)

	return list, err
}

func (user *User) Total() (int64, error) {
	return Db.Count(user)
}

// 密码加密
func (user *User) encryptPassword(password, salt string) string {
	return utils.Md5(password + salt)
}

// 生成密码盐值
func (user *User) generateSalt() string {
	return utils.RandString(PasswordSaltLength)
}