package db

import "gorm.io/gorm"

type User struct {
	gorm.Model
	Name  string `gorm:"not null"`
	Token string `gorm:"unique;not null"`
}

// 添加用户
func AddUser(name, token string) error {
	user := &User{Name: name, Token: token}
	result := db.Create(&user)
	if result.Error != nil {
		return result.Error
	}
	return nil
}

// 删除用户
func DeleteUser(id uint) error {
	result := db.Delete(&User{}, id)
	if result.Error != nil {
		return result.Error
	}
	return nil
}

// 修改用户
func UpdateUser(id uint, name, token string) error {
	user := &User{Name: name, Token: token}
	result := db.Model(&User{}).Where("id = ?", id).Updates(user)
	if result.Error != nil {
		return result.Error
	}
	return nil
}