package exporter

import (
	"gorm.io/gorm"

	"github.com/break/junhong_cmp_fiber/internal/model"
	"github.com/break/junhong_cmp_fiber/pkg/constants"
)

func applyShopScopeForTask(query *gorm.DB, task *model.ExportTask) *gorm.DB {
	if task == nil {
		return query.Where("1 = 0")
	}

	switch task.CreatorUserType {
	case constants.UserTypeSuperAdmin, constants.UserTypePlatform:
		return query
	case constants.UserTypeAgent:
		if len(task.ScopeShopIDs) == 0 {
			return query.Where("1 = 0")
		}
		return query.Where("shop_id IN ?", []uint(task.ScopeShopIDs))
	default:
		return query.Where("1 = 0")
	}
}