//go:build e2e
// +build e2e

package e2e

import (
	"github.com/aws/eks-anywhere/pkg/kubeconfig"
	"github.com/aws/eks-anywhere/test/framework"
)

const (
	prometheusPackageName            = "prometheus"
	prometheusPackagePrefix          = "generated"
	prometheusPackageTargetNamespace = "observability"
)

func runCuratedPackagesPrometheusInstall(test *framework.ClusterE2ETest) {
	packageFullName := prometheusPackagePrefix + "-" + prometheusPackageName
	test.InstallLocalStorageProvisioner()
	test.CreateNamespace(prometheusPackageTargetNamespace)
	test.SetPackageBundleActive()
	test.InstallCuratedPackage(prometheusPackageName, packageFullName,
		kubeconfig.FromClusterName(test.ClusterName),
		"--set server.persistentVolume.storageClass=local-path")
	test.VerifyPrometheusPackageInstalled(packageFullName, prometheusPackageTargetNamespace)
	test.VerifyPrometheusNodeExporterStates(packageFullName, prometheusPackageTargetNamespace)
	test.VerifyPrometheusPrometheusServerStates(packageFullName, prometheusPackageTargetNamespace, "deployment")
}

func runCuratedPackagesPrometheusUpdate(test *framework.ClusterE2ETest) {
	packageFullName := prometheusPackagePrefix + "-" + prometheusPackageName

	test.InstallLocalStorageProvisioner()
	test.CreateNamespace(prometheusPackageTargetNamespace)
	test.SetPackageBundleActive()
	test.InstallCuratedPackage(prometheusPackageName, packageFullName,
		kubeconfig.FromClusterName(test.ClusterName),
		"--set server.persistentVolume.storageClass=local-path")

	test.ApplyPrometheusPackageServerStatefulSetFile(packageFullName, prometheusPackageTargetNamespace)
	test.VerifyPrometheusPackageInstalled(packageFullName, prometheusPackageTargetNamespace)
	test.VerifyPrometheusPrometheusServerStates(packageFullName, prometheusPackageTargetNamespace, "statefulset")
	test.VerifyPrometheusNodeExporterStates(packageFullName, prometheusPackageTargetNamespace)

	test.ApplyPrometheusPackageServerDeploymentFile(packageFullName, prometheusPackageTargetNamespace)
	test.VerifyPrometheusPackageInstalled(packageFullName, prometheusPackageTargetNamespace)
	test.VerifyPrometheusPrometheusServerStates(packageFullName, prometheusPackageTargetNamespace, "deployment")
	test.VerifyPrometheusNodeExporterStates(packageFullName, prometheusPackageTargetNamespace)
}

func runCuratedPackagesPrometheusInstallSimpleFlow(test *framework.ClusterE2ETest) {
	test.WithCluster(runCuratedPackagesPrometheusInstall)
}

func runCuratedPackagesPrometheusUpdateFlow(test *framework.ClusterE2ETest) {
	test.WithCluster(runCuratedPackagesPrometheusUpdate)
}

func runCuratedPackagesPrometheusInstallTinkerbellSimpleFlow(test *framework.ClusterE2ETest) {
	test.GenerateClusterConfig()
	test.GenerateHardwareConfig()
	test.PowerOnHardware()
	test.CreateCluster(framework.WithControlPlaneWaitTimeout("20m"))
	test.ValidateControlPlaneNodes(framework.ValidateControlPlaneNoTaints, framework.ValidateControlPlaneLabels)
	runCuratedPackagesPrometheusInstall(test)
	test.DeleteCluster()
	test.PowerOffHardware()
	test.ValidateHardwareDecommissioned()
}