package services

import (
	"github.com/rancher/rke/hosts"
	"github.com/rancher/types/apis/cluster.cattle.io/v1"
	"github.com/sirupsen/logrus"
)

func RunControlPlane(controlHosts []hosts.Host, etcdHosts []hosts.Host, controlServices v1.RKEConfigServices) error {
	logrus.Infof("[%s] Building up Controller Plane..", ControlRole)
	for _, host := range controlHosts {
		// run kubeapi
		err := runKubeAPI(host, etcdHosts, controlServices.KubeAPI)
		if err != nil {
			return err
		}
		// run kubecontroller
		err = runKubeController(host, controlServices.KubeController)
		if err != nil {
			return err
		}
		// run scheduler
		err = runScheduler(host, controlServices.Scheduler)
		if err != nil {
			return err
		}
	}
	logrus.Infof("[%s] Successfully started Controller Plane..", ControlRole)
	return nil
}

func UpgradeControlPlane(controlHosts []hosts.Host, etcdHosts []hosts.Host, controlServices v1.RKEConfigServices) error {
	logrus.Infof("[%s] Upgrading the Controller Plane..", ControlRole)
	for _, host := range controlHosts {
		// upgrade KubeAPI
		if err := upgradeKubeAPI(host, etcdHosts, controlServices.KubeAPI); err != nil {
			return err
		}

		// upgrade KubeController
		if err := upgradeKubeController(host, controlServices.KubeController); err != nil {
			return nil
		}

		// upgrade scheduler
		err := upgradeScheduler(host, controlServices.Scheduler)
		if err != nil {
			return err
		}
	}
	logrus.Infof("[%s] Successfully upgraded Controller Plane..", ControlRole)
	return nil
}

func RemoveControlPlane(controlHosts []hosts.Host) error {
	logrus.Infof("[%s] Tearing down the Controller Plane..", ControlRole)
	for _, host := range controlHosts {
		// remove KubeAPI
		if err := removeKubeAPI(host); err != nil {
			return err
		}

		// remove KubeController
		if err := removeKubeController(host); err != nil {
			return nil
		}

		// remove scheduler
		err := removeScheduler(host)
		if err != nil {
			return err
		}
	}
	logrus.Infof("[%s] Successfully teared down Controller Plane..", ControlRole)
	return nil
}