package app

import (
	"OneAuth/libs/auth"
	"OneAuth/libs/oerr"
	"OneAuth/models"
	"errors"
	"gorm.io/gorm"
)

func AddUser(tx *gorm.DB, appID uint, userID uint, roleID uint) error {
	au := &models.AppUser{}
	au.AppID = appID
	au.UserID = userID
	err := tx.Where(au).First(au).Error
	if err == nil {
		return oerr.ResourceDuplicated
	}
	if errors.Is(err, gorm.ErrRecordNotFound) {
		err = tx.Create(au).Error
		if err != nil {
			return err
		}
		err = auth.BindUserRole(tx, userID, roleID)
		if err != nil {
			return err
		}
		return tx.Model(&models.App{}).Where("id = ?", appID).Update("user_count", gorm.Expr("user_count + ?", 1)).Error
	}
	return err
}
func EnableUser(tx *gorm.DB, appID uint, userID uint) error {
	au := &models.AppUser{}
	au.AppID = appID
	au.UserID = userID
	err := tx.Where(au).First(au).Error
	if err != nil {
		return err
	}
	return tx.Where(au).Update("disabled", false).Error
}

func DisableUser(tx *gorm.DB, appID uint, userID uint) error {
	au := &models.AppUser{}
	au.AppID = appID
	au.UserID = userID
	return tx.Where(au).Update("disabled", true).Error
}