package dao

import (
	"errors"
	"github.com/sirupsen/logrus"
	"gorm.io/gorm"
	"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 errors.Is(tx.Error, gorm.ErrRecordNotFound) {
			return admin, ErrAdminUserNotFound
		}
		logrus.Errorf("%v", tx.Error)
		return admin, ErrStorage
	}
	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, ErrStorage
	}
	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 ErrStorage
	}
	return nil
}