package disk

import (
	"bsm/full/module/base/cloud/internal/impl"
	"bsm/full/module/base/cloud/internal/models"
	"git.apinb.com/bsm-sdk/core/errcode"
	"git.apinb.com/bsm-sdk/core/printer"
)

// defaultMaxStorage 默认容量配额100GB，与 Space.Get 创建默认空间时的取值保持一致
const defaultMaxStorage int64 = 100 * 1024 * 1024 * 1024

// checkQuota 校验「已用空间 + 本次新增」是否超出配额，超出返回明确错误
func checkQuota(passportID uint, addSize int64) error {
	if addSize <= 0 {
		return nil
	}

	// 配额取用户空间记录的 max_storage，无空间记录时使用默认配额
	maxStorage := defaultMaxStorage
	var space models.CloudSpace
	if err := impl.DBService.Where("passport_id = ?", passportID).First(&space).Error; err == nil && space.MaxStorage > 0 {
		maxStorage = space.MaxStorage
	}

	// 已用空间按云盘文件实际占用统计
	var usedStorage int64
	row := impl.DBService.Model(&models.CloudDiskFile{}).
		Joins("JOIN cloud_disk_dirs ON cloud_disk_files.directory_id = cloud_disk_dirs.id").
		Where("cloud_disk_dirs.passport_id = ?", passportID).
		Select("COALESCE(SUM(cloud_disk_files.size), 0)").
		Row()
	if err := row.Scan(&usedStorage); err != nil {
		printer.Error("Query used storage error: %v", err)
		return errcode.ErrDB
	}

	if usedStorage+addSize > maxStorage {
		return errcode.ErrResourceExhausted
	}

	return nil
}