package task

import (
	entity "github.com/tidbcloud/devops/entity/k8supgrade"
	"github.com/tidbcloud/devops/pkg/infra/config"
)

const (
	MaintenancePolicyModeLegacy             = entity.MaintenancePolicyModeLegacy
	MaintenancePolicyModeClusterMaintenance = entity.MaintenancePolicyModeClusterMaintenance
)

func ResolveMaintenancePolicy(gate config.ScopedFeatureGate, tenantID, projectID string) string {
	if gate.Enabled(tenantID, projectID) {
		return MaintenancePolicyModeClusterMaintenance
	}
	return MaintenancePolicyModeLegacy
}

func IsClusterMaintenanceMode(mode string) bool {
	return mode == MaintenancePolicyModeClusterMaintenance
}