package handler

import (
	"errors"
	"fmt"

	"gitee.com/qhitc_admin/gweb/internal/pkg/db/model"
)

func (h *DbHandler) InserUser(user model.User) (model.User, error) {

	if h.UserExist(user.UserName) {
		return user, errors.New("exist user " + user.UserName)
	}

	res := h.db.Create(&user)
	return user, res.Error
}

func (h *DbHandler) UserExist(username string) bool {
	var count int64
	res := h.db.Model(model.User{}).Where("user_name = ?", username).Count(&count)
	if res.Error != nil {
		fmt.Println(res.Error)
		return true
	}

	return count != 0
}