package evict

import (
	"context"
	"fmt"
	"pandax/base/global"

	policy "k8s.io/api/policy/v1beta1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/client-go/kubernetes"
)

var (
	systemNamespace = "kube-system"
)

func EvictsNodePods(client *kubernetes.Clientset, nodeName string) error {
	/*
		驱逐节点上不在 kube-system 命名空间中的所有 pod
	*/
	pods, err := client.CoreV1().Pods("").List(context.TODO(), metav1.ListOptions{
		FieldSelector: "spec.nodeName=" + nodeName,
	})

	if err != nil {
		return err
	}
	for _, i := range pods.Items {
		if i.Namespace == systemNamespace {
			continue
		} else {
			global.Log.Info(fmt.Sprintf("开始驱逐Node: %v, 节点Namespace: %v下的Pod: %v", nodeName, i.Namespace, i.Name))
			err := EvictsPod(client, i.Name, i.Namespace)
			if err != nil {
				global.Log.Error(fmt.Sprintf("驱逐Pod：%v失败", i.Name))
			}
		}
	}
	global.Log.Info(fmt.Sprintf("已成功从节点: %v 中驱逐所有pod", nodeName))
	return nil
}

func EvictsPod(client *kubernetes.Clientset, name, namespace string) error {
	// Pod优雅退出时间, 默认退出时间30s, 如果未指定, 则默认为每个对象的值。0表示立即删除。
	var gracePeriodSeconds int64 = 0
	propagationPolicy := metav1.DeletePropagationForeground
	deleteOptions := &metav1.DeleteOptions{
		GracePeriodSeconds: &gracePeriodSeconds,
		PropagationPolicy:  &propagationPolicy,
	}
	return client.PolicyV1beta1().Evictions(namespace).Evict(context.TODO(), &policy.Eviction{
		ObjectMeta: metav1.ObjectMeta{
			Name:      name,
			Namespace: namespace,
		},
		DeleteOptions: deleteOptions,
	})
}