package models

import (
	"MossChat/db"
	"MossChat/enums"
	"errors"
	"fmt"
	"golang.org/x/crypto/bcrypt"
	"gorm.io/gorm"
)

type User struct {
	gorm.Model
	Username  string           `json:"username"`
	Password  string           `json:"-"`
	Nickname  string           `json:"nickname"`
	Gender    enums.GenderType `json:"gender"`
	Introduce string           `json:"introduce"`
	Email     string           `json:"email"`
	Avatar    string           `json:"avatar"`
	IsBanned  bool             `json:"is_banned"`
	Phone     string           `json:"phone"`
}

func (user *User) UpdatePassword(password string) error {
	bcryptHash, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
	if err != nil {
		return errors.New("无法加密密码")
	}
	user.Password = string(bcryptHash)
	return nil
}
func (user *User) CheckPassword(password string) error {
	return bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(password))
}
func (user *User) Delete() error {
	if err := db.DB.Delete(user); err != nil {
		return errors.New("无法删除该用户")
	}
	return nil
}

func CreateUser(username string, password string, nickname string, gender enums.GenderType) (User, error) {
	user := User{Username: username,
		Nickname:  nickname,
		Gender:    gender,
		Introduce: "这个人很懒，什么都没留下。",
	}
	var count int64
	if err := db.DB.Model(&User{}).Where("username = ?", username).Count(&count).Error; err != nil {
		return User{}, errors.New("数据库查询错误")
	}
	if count != 0 {
		return User{}, errors.New("用户已存在")
	}
	//加密密码
	if err := user.UpdatePassword(password); err != nil {
		return User{}, err
	}
	if err := db.DB.Create(user).Error; err != nil {
		return User{}, errors.New("无法创建用户")
	}
	return user, nil
}

func DeleteUser(username string) error {
	if err := db.DB.Model(&User{}).Where("username = ?", username).Delete(&User{}).Error; err != nil {
		return errors.New(fmt.Sprintf("无法创建用户:%s", err.Error()))
	}
	return nil
}
func GetUserByUsername(username string) (User, error) {
	var user User
	if err := db.DB.Where("username = ?", username).First(&user).Error; err != nil {
		return User{}, errors.New("无法找到用户")
	}
	return user, nil
}
func GetUserByUid(uid uint) (User, error) {
	var user User
	if err := db.DB.Where("uid = ?", uid).First(&user).Error; err != nil {
		return User{}, errors.New("无法找到用户")
	}
	return user, nil
}