commands.go

package framework

import (
	"os"
	"path/filepath"
	"strings"

	releasev1alpha1 "github.com/aws/eks-anywhere/release/api/v1alpha1"
)

type CommandOpt func(*string, *[]string) (err error)

func appendOpt(new ...string) CommandOpt {
	return func(binaryPath *string, args *[]string) (err error) {
		*args = append(*args, new...)
		return nil
	}
}

func withKubeconfig(kubeconfigFile string) CommandOpt {
	return appendOpt("--kubeconfig", kubeconfigFile)
}

func WithForce() CommandOpt {
	return appendOpt("--force-cleanup")
}

func WithControlPlaneWaitTimeout(timeout string) CommandOpt {
	return appendOpt("--control-plane-wait-timeout", timeout)
}

func WithExternalEtcdWaitTimeout(timeout string) CommandOpt {
	return appendOpt("--external-etcd-wait-timeout", timeout)
}

func WithPerMachineWaitTimeout(timeout string) CommandOpt {
	return appendOpt("--per-machine-wait-timeout", timeout)
}

func ExecuteWithEksaRelease(release *releasev1alpha1.EksARelease) CommandOpt {
	return executeWithBinaryCommandOpt(func() (string, error) {
		return getBinary(release)
	})
}

// PackagedBinary represents a binary that can be extracted
// executed from local disk.
type PackagedBinary interface {
	// BinaryPath returns the local disk path to the binary.
	BinaryPath() (string, error)
}

// ExecuteWithBinary executes the command with a binary from an specific path.
func ExecuteWithBinary(eksa PackagedBinary) CommandOpt {
	return executeWithBinaryCommandOpt(func() (string, error) {
		return eksa.BinaryPath()
	})
}

// WithSudo add prefix "sudo" to the command. And preserve PATH.
func WithSudo(user string) CommandOpt {
	return func(binaryPath *string, args *[]string) (err error) {
		*args = append([]string{*binaryPath}, *args...)
		*binaryPath = "sudo"
		if user != "" {
			*args = append([]string{"-E", "PATH=$PATH", "-u", user}, *args...)
		}
		return nil
	}
}

// WithBundlesOverride modify bundles-override.
func WithBundlesOverride(bundles string) CommandOpt {
	return appendOpt("--bundles-override", bundles)
}

type binaryFetcher func() (binaryPath string, err error)

func executeWithBinaryCommandOpt(fetcher binaryFetcher) CommandOpt {
	return func(binaryPath *string, args *[]string) (err error) {
		b, err := fetcher()
		if err != nil {
			return err
		}
		*binaryPath = b
		if err = setEksctlVersionEnvVar(); err != nil {
			return err
		}

		// When bundles override is present, the manifest belongs to the current
		// build of the CLI and it's intended to be used only with that version
		removeFlag("--bundles-override", args)

		return nil
	}
}

func removeFlag(flag string, args *[]string) {
	for i, a := range *args {
		if a == flag {
			elementsToDelete := 1
			// If it's not the last arg and next arg is not a flag,
			// that means it's the value for the current flag, remove it as well
			if i < len(*args)-1 && !strings.HasPrefix((*args)[i+1], "-") {
				elementsToDelete = 2
			}

			*args = append((*args)[:i], (*args)[i+elementsToDelete:]...)
			break
		}
	}
}

// DefaultLocalEKSABinaryPath returns the full path for the local eks-a binary being tested.
func DefaultLocalEKSABinaryPath() (string, error) {
	binDir, err := DefaultLocalEKSABinDir()
	if err != nil {
		return "", err
	}

	return filepath.Join(binDir, "eksctl-anywhere"), nil
}

// DefaultLocalEKSABinDir returns the full path for the local directory where
// the tested eks-a binary lives.
func DefaultLocalEKSABinDir() (string, error) {
	workDir, err := os.Getwd()
	if err != nil {
		return "", err
	}

	return filepath.Join(workDir, "bin"), nil
}