package model

import (
	"newaterobot-process/database"
	"newaterobot-process/entity"
)

// UserModel 用户模型
type UserModel struct{}

// Create 创建用户
func (m *UserModel) Create(user *entity.User) error {
	return database.DB.Create(user).Error
}

// GetByUsername 根据用户名获取用户
func (m *UserModel) GetByUsername(username string) (*entity.User, error) {
	var user entity.User
	err := database.DB.Where("username = ?", username).First(&user).Error
	return &user, err
}

// GetByID 根据ID获取用户
func (m *UserModel) GetByID(id uint) (*entity.User, error) {
	var user entity.User
	err := database.DB.First(&user, id).Error
	return &user, err
}

// Update 更新用户信息
func (m *UserModel) Update(user *entity.User) error {
	return database.DB.Save(user).Error
}

// Delete 删除用户
func (m *UserModel) Delete(id uint) error {
	return database.DB.Delete(&entity.User{}, id).Error
}