// Copyright 2021 The Kubernetes Authors.
// SPDX-License-Identifier: Apache-2.0

package flagutils

import (
	"fmt"

	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"sigs.k8s.io/cli-utils/pkg/inventory"
)

const (
	InventoryPolicyFlag       = "inventory-policy"
	InventoryPolicyStrict     = "strict"
	InventoryPolicyAdopt      = "adopt"
	InventoryPolicyForceAdopt = "force-adopt"
)

// ConvertPropagationPolicy converts a propagationPolicy described as a
// string to a DeletionPropagation type that is passed into the Applier.
func ConvertPropagationPolicy(propagationPolicy string) (metav1.DeletionPropagation, error) {
	switch propagationPolicy {
	case string(metav1.DeletePropagationForeground):
		return metav1.DeletePropagationForeground, nil
	case string(metav1.DeletePropagationBackground):
		return metav1.DeletePropagationBackground, nil
	case string(metav1.DeletePropagationOrphan):
		return metav1.DeletePropagationOrphan, nil
	default:
		return metav1.DeletePropagationBackground, fmt.Errorf(
			"prune propagation policy must be one of Background, Foreground, Orphan")
	}
}

func ConvertInventoryPolicy(policy string) (inventory.Policy, error) {
	switch policy {
	case InventoryPolicyStrict:
		return inventory.PolicyMustMatch, nil
	case InventoryPolicyAdopt:
		return inventory.PolicyAdoptIfNoInventory, nil
	case InventoryPolicyForceAdopt:
		return inventory.PolicyAdoptAll, nil
	default:
		return inventory.PolicyMustMatch, fmt.Errorf(
			"inventory policy must be one of strict, adopt")
	}
}

// PathFromArgs returns the path which is a positional arg from args list
// returns "-" if there is length of args is 0, which implies no path is provided
func PathFromArgs(args []string) string {
	if len(args) == 0 {
		return "-"
	}
	return args[0]
}