package dao

import (
	"errors"
	"github.com/sirupsen/logrus"
	"gorm.io/gorm"
	"hy2xs-admin/model/constant"
	"hy2xs-admin/model/entity"
	"time"
)

func GetAdminUser(query interface{}, args ...interface{}) (entity.AdminUser, error) {
	var admin entity.AdminUser
	if tx := sqliteDB.Model(&entity.AdminUser{}).Where(query, args...).First(&admin); tx.Error != nil {
		if tx.Error == gorm.ErrRecordNotFound {
			return admin, errors.New(constant.WrongPassword)
		}
		logrus.Errorf("%v", tx.Error)
		return admin, errors.New(constant.SysError)
	}
	return admin, nil
}

func SaveAdminUser(admin entity.AdminUser) (int64, error) {
	if tx := sqliteDB.Save(&admin); tx.Error != nil {
		logrus.Errorf("%v", tx.Error)
		return 0, errors.New(constant.SysError)
	}
	return *admin.Id, nil
}

func UpdateAdminUser(ids []int64, updates map[string]interface{}) error {
	if len(updates) == 0 {
		return nil
	}
	updates["update_time"] = time.Now().Format("2006-01-02 15:04:05")
	if tx := sqliteDB.Model(&entity.AdminUser{}).Where("id in ?", ids).Updates(updates); tx.Error != nil {
		logrus.Errorf("%v", tx.Error)
		return errors.New(constant.SysError)
	}
	return nil
}