using k8s;

namespace KubernetesWorkflow
{
    public class K8sCluster
    {
        public K8sCluster(Configuration configuration)
        {
            Configuration = configuration;
        }
       
        public Configuration Configuration { get; }
        public string IP { get; private set; } = string.Empty;

        public KubernetesClientConfiguration GetK8sClientConfig()
        {
            var config = GetConfig();
            UpdateIp(config);
            return config;
        }

        public string GetNodeLabelForLocation(Location location)
        {
            if (location == Location.Unspecified) return string.Empty;
            return Configuration.LocationMap.Single(l => l.Location == location).WorkerName;
        }

        public TimeSpan K8sOperationTimeout()
        {
            return Configuration.OperationTimeout;
        }

        public TimeSpan WaitForK8sServiceDelay()
        {
            return Configuration.RetryDelay;
        }

        private KubernetesClientConfiguration GetConfig()
        {
            if (Configuration.KubeConfigFile != null)
            {
                return KubernetesClientConfiguration.BuildConfigFromConfigFile(Configuration.KubeConfigFile);
            }
            else
            {
                return KubernetesClientConfiguration.BuildDefaultConfig();
            }
        }

        private void UpdateIp(KubernetesClientConfiguration config)
        {
            var host = config.Host.Replace("https://", "");
            IP = host.Substring(0, host.IndexOf(':'));
        }
    }
}