package model

import (
	"next-terminal/pkg/global"
	"next-terminal/pkg/utils"
	"reflect"
)

const (
	TypeUser  = "user"
	TypeAdmin = "admin"
)

type User struct {
	ID         string         `gorm:"primary_key" json:"id"`
	Username   string         `json:"username"`
	Password   string         `json:"password"`
	Nickname   string         `json:"nickname"`
	TOTPSecret string         `json:"-"`
	Online     bool           `json:"online"`
	Enabled    bool           `json:"enabled"`
	Created    utils.JsonTime `json:"created"`
	Type       string         `json:"type"`
}

func (r *User) TableName() string {
	return "users"
}

func (r *User) IsEmpty() bool {
	return reflect.DeepEqual(r, User{})
}

func FindAllUser() (o []User) {
	if global.DB.Find(&o).Error != nil {
		return nil
	}
	return
}

func FindPageUser(pageIndex, pageSize int, username, nickname string) (o []User, total int64, err error) {
	db := global.DB
	if len(username) > 0 {
		db = db.Where("username like ?", "%"+username+"%")
	}

	if len(nickname) > 0 {
		db = db.Where("nickname like ?", "%"+nickname+"%")
	}

	err = db.Order("created desc").Find(&o).Offset((pageIndex - 1) * pageSize).Limit(pageSize).Count(&total).Error
	if o == nil {
		o = make([]User, 0)
	}
	return
}

func CreateNewUser(o *User) (err error) {
	err = global.DB.Create(o).Error
	return
}

func FindUserById(id string) (o User, err error) {
	err = global.DB.Where("id = ?", id).First(&o).Error
	return
}

func FindUserByIdIn(ids []string) (o []User, err error) {
	err = global.DB.Where("id in ?", ids).First(&o).Error
	return
}

func FindUserByUsername(username string) (o User, err error) {
	err = global.DB.Where("username = ?", username).First(&o).Error
	return
}

func UpdateUserById(o *User, id string) {
	o.ID = id
	global.DB.Updates(o)
}

func DeleteUserById(id string) {
	global.DB.Where("id = ?", id).Delete(&User{})
}

func CountUser() (total int64, err error) {
	err = global.DB.Find(&User{}).Count(&total).Error
	return
}