package util

import (
	"github.com/rancher/wrangler/pkg/condition"
	v1 "k8s.io/api/core/v1"

	"github.com/harvester/vm-import-controller/pkg/apis/common"
)

func ConditionExists(conditions []common.Condition, c condition.Cond, condType v1.ConditionStatus) bool {
	for _, v := range conditions {
		if v.Type == c && v.Status == condType {
			return true
		}
	}

	return false
}

func AddOrUpdateCondition(conditions []common.Condition, newCond common.Condition) []common.Condition {
	var found bool
	for _, v := range conditions {
		if v.Type == newCond.Type {
			found = true
			v.Status = newCond.Status
			v.LastTransitionTime = newCond.LastTransitionTime
			v.LastUpdateTime = newCond.LastUpdateTime
			v.Message = newCond.Message
			v.Reason = newCond.Reason
		}
	}

	if !found {
		conditions = append(conditions, newCond)
	}

	return conditions
}

func MergeConditions(srcConditions []common.Condition, newCond []common.Condition) []common.Condition {
	for _, v := range newCond {
		srcConditions = AddOrUpdateCondition(srcConditions, v)
	}

	return srcConditions
}

func RemoveCondition(conditions []common.Condition, c condition.Cond, condType v1.ConditionStatus) []common.Condition {
	var retConditions []common.Condition
	for _, v := range conditions {
		if v.Type != c || v.Status != condType {
			retConditions = append(retConditions, v)
		}
	}
	return retConditions
}