from framework.ssh.ssh_connection import SSHConnection
from framework.validation.validation import validate_equals
from keywords.base_keyword import BaseKeyword
from keywords.cloud_platform.fault_management.alarms.alarm_list_keywords import AlarmListKeywords
from keywords.cloud_platform.system.application.system_application_list_keywords import SystemApplicationListKeywords
from keywords.cloud_platform.system.host.system_host_list_keywords import SystemHostListKeywords
from keywords.k8s.pods.kubectl_get_pods_keywords import KubectlGetPodsKeywords


class HealthKeywords(BaseKeyword):
    """Class for health Keywords for Cloud Platform"""

    def __init__(self, ssh_connection: SSHConnection):
        """Constructor

        Args:
            ssh_connection (SSHConnection): ssh object

        """
        self.ssh_connection = ssh_connection

    def validate_healty_cluster(self):
        """Function to validate the health of the cluster

        This function checks the health of the cluster which consists of
            1. validate all pods are healty
            2. validate no alarms are present
            3. validate all apps are healty and applied
        """
        # Validate all pods are healthy
        self.validate_pods_health()
        # Validate no alarms are present
        self.validate_no_alarms()
        # Validate all apps are healthy and applied
        self.validate_apps_health_and_applied()

    def validate_pods_health(self, namespace: str | None = None) -> bool:
        """Function to validate the health of all pods in the cluster

        Args:
           namespace (str | None): Namespace to check the pods in. If None, the default namespace will be used.

        Returns:
            bool: True if pod is in expected status else False.
        """
        healthy_status = ["Running", "Succeeded", "Completed"]
        return KubectlGetPodsKeywords(self.ssh_connection).wait_for_pods_to_reach_status(expected_status=healthy_status, namespace=namespace, timeout=300)

    def validate_no_alarms(self):
        """Function to validate no alarms are present in the cluster"""
        AlarmListKeywords(self.ssh_connection).wait_for_all_alarms_cleared()

    def validate_apps_health_and_applied(self):
        """Function to validate all apps are healthy and applied"""
        healthy_status = ["applied", "uploaded"]
        SystemApplicationListKeywords(self.ssh_connection).validate_all_apps_status(healthy_status)

    def validate_hosts_health(self):
        """Function to validate all hosts are healthy"""
        host_values = SystemHostListKeywords(self.ssh_connection).get_system_host_list().get_hosts()

        for host_value in host_values:
            validate_equals(host_value.get_availability(), "available", f"The host {host_value.get_host_name()} availability is {host_value.get_availability()}")
            validate_equals(host_value.get_administrative(), "unlocked", f"The host {host_value.get_host_name()} administrative is {host_value.get_administrative()}")
            validate_equals(host_value.get_operational(), "enabled", f"The host {host_value.get_host_name()} operational is {host_value.get_operational()}")