package utils

import (
	"context"

	"github.com/aws/amazon-vpc-cni-k8s/test/framework/resources/aws/services"
	"github.com/aws/amazon-vpc-cni-k8s/test/framework/utils"
	"github.com/pkg/errors"
	"k8s.io/apimachinery/pkg/util/wait"
)

func WaitTillAddonIsDeleted(eks services.EKS, addonName string, clusterName string) error {
	ctx := context.Background()
	return wait.PollImmediateUntil(utils.PollIntervalShort, func() (bool, error) {
		_, err := eks.DescribeAddon(&services.AddonInput{
			AddonName:   addonName,
			ClusterName: clusterName,
		})
		if err != nil {
			return false, err
		}
		return false, nil
	}, ctx.Done())
}

func WaitTillAddonIsActive(eks services.EKS, addonName string, clusterName string) error {
	ctx := context.Background()
	return wait.PollImmediateUntil(utils.PollIntervalShort, func() (bool, error) {
		describeAddonOutput, err := eks.DescribeAddon(&services.AddonInput{
			AddonName:   addonName,
			ClusterName: clusterName,
		})
		if err != nil {
			return false, err
		}

		status := *describeAddonOutput.Addon.Status
		if status == "CREATE_FAILED" || status == "DEGRADED" {
			return false, errors.Errorf("Create Addon Failed, addon status: %s", status)
		}
		if status == "ACTIVE" {
			return true, nil
		}
		return false, nil
	}, ctx.Done())
}