package basic

import (
	"fmt"

	"github.com/digitalocean/clusterlint/checks"
	"github.com/digitalocean/clusterlint/kube"
	corev1 "k8s.io/api/core/v1"
)

func init() {
	checks.Register(&podStatusCheck{})
}

type podStatusCheck struct{}

// Name returns a unique name for this check.
func (p *podStatusCheck) Name() string {
	return "pod-state"
}

// Groups returns a list of group names this check should be part of.
func (p *podStatusCheck) Groups() []string {
	return []string{"workload-health"}
}

// Description returns a detailed human-readable description of what this check
// does.
func (p *podStatusCheck) Description() string {
	return "Check if there are unhealthy pods in the cluster"
}

// Run runs this check on a set of Kubernetes objects. It can return warnings
// (low-priority problems) and errors (high-priority problems) as well as an
// error value indicating that the check failed to run.
func (p *podStatusCheck) Run(objects *kube.Objects) ([]checks.Diagnostic, error) {
	var diagnostics []checks.Diagnostic

	for _, pod := range objects.Pods.Items {
		if corev1.PodFailed == pod.Status.Phase || corev1.PodUnknown == pod.Status.Phase {
			d := checks.Diagnostic{
				Severity: checks.Warning,
				Message:  fmt.Sprintf("Pod '%s' in namespace '%s' has state: %s. Pod state should be `Running`, `Pending` or `Succeeded`.", pod.GetName(), pod.GetNamespace(), pod.Status.Phase),
				Kind:     checks.Pod,
				Object:   &pod.ObjectMeta,
				Owners:   pod.ObjectMeta.GetOwnerReferences(),
			}
			diagnostics = append(diagnostics, d)
		}
	}

	return diagnostics, nil
}