/*
Copyright 2024.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package v1alpha1

import (
	corev1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

// A ConditionType represents a condition a resource could be in.
type ConditionType string

// Condition types.
const (
	// TypeReady resources are believed to be ready to handle work.
	TypeReady ConditionType = "Ready"

	// TypeSynced resources are believed to be in sync with the
	// Kubernetes resources that manage their lifecycle.
	TypeSynced ConditionType = "Synced"

	// TypePending resources are believed to be pending.
	TypePending ConditionType = "Pending"
)

// A ConditionReason represents the reason a resource is in a condition.
type ConditionReason string

// Reasons a resource is or is not synced.
const (
	ReasonReconcileSuccess ConditionReason = "ReconcileSuccess"
	ReasonReconcileError   ConditionReason = "ReconcileError"
	ReasonDeleting         ConditionReason = "Deleting"
)

// A Condition that may apply to a resource.
type Condition struct {
	// Type of this condition. At most one of each condition type may apply to
	// a resource at any point in time.
	Type ConditionType `json:"type"`

	// Status of this condition; is it currently True, False, or Unknown?
	Status corev1.ConditionStatus `json:"status"`

	// LastTransitionTime is the last time this condition transitioned from one
	// status to another.
	LastTransitionTime metav1.Time `json:"lastTransitionTime"`

	// A Reason for this condition's last transition from one status to another.
	Reason ConditionReason `json:"reason"`

	// A Message containing details about this condition's last transition from
	// one status to another, if any.
	// +optional
	Message string `json:"message,omitempty"`

	// ObservedGeneration represents the .metadata.generation that the condition was set based upon.
	// For instance, if .metadata.generation is currently 12, but the .status.conditions[x].observedGeneration is 9, the condition is out of date
	// with respect to the current state of the instance.
	// +optional
	ObservedGeneration int64 `json:"observedGeneration,omitempty"`
}

// Equal returns true if the condition is identical to the supplied condition,
// ignoring the LastTransitionTime and ObservedGeneration.
func (c Condition) Equal(other Condition) bool {
	return c.Type == other.Type &&
		c.Status == other.Status &&
		c.Reason == other.Reason &&
		c.Message == other.Message
}

// WithMessage returns a condition by adding the provided message to existing
// condition.
func (c Condition) WithMessage(msg string) Condition {
	c.Message = msg
	return c
}

// WithObservedGeneration returns a condition by adding the provided observed generation
// to existing condition.
func (c Condition) WithObservedGeneration(gen int64) Condition {
	c.ObservedGeneration = gen
	return c
}

// SetConditions sets the supplied conditions, replacing any existing conditions
// of the same type. This is a no-op if all supplied conditions are identical,
// ignoring the last transition time, to those already set.
// Observed generation is updated if higher than the existing one.
func (r *ArgoCDRole) SetConditions(c ...Condition) {
	for _, new := range c {
		exists := false
		for i, existing := range r.Status.Conditions {
			if existing.Type != new.Type {
				continue
			}

			if existing.Equal(new) {
				exists = true
				if r.Status.Conditions[i].ObservedGeneration < new.ObservedGeneration {
					r.Status.Conditions[i].ObservedGeneration = new.ObservedGeneration
				}
				continue
			}

			r.Status.Conditions[i] = new
			exists = true
		}
		if !exists {
			r.Status.Conditions = append(r.Status.Conditions, new)
		}
	}
}

func (r *ArgoCDRole) SetArgoCDRoleBindingRef(ref string) {
	r.Status.ArgoCDRoleBindingRef = ref
}

func (r *ArgoCDRole) HasArgoCDRoleBindingRef() bool {
	return r.Status.ArgoCDRoleBindingRef != ""
}

// SetConditions sets the supplied conditions, replacing any existing conditions
// of the same type. This is a no-op if all supplied conditions are identical,
// ignoring the last transition time, to those already set.
// Observed generation is updated if higher than the existing one.
func (r *ArgoCDProjectRole) SetConditions(c ...Condition) {
	for _, new := range c {
		exists := false
		for i, existing := range r.Status.Conditions {
			if existing.Type != new.Type {
				continue
			}

			if existing.Equal(new) {
				exists = true
				if r.Status.Conditions[i].ObservedGeneration < new.ObservedGeneration {
					r.Status.Conditions[i].ObservedGeneration = new.ObservedGeneration
				}
				continue
			}

			r.Status.Conditions[i] = new
			exists = true
		}
		if !exists {
			r.Status.Conditions = append(r.Status.Conditions, new)
		}
	}
}

func (r *ArgoCDProjectRole) SetArgoCDProjectRoleBindingRef(ref string) {
	r.Status.ArgoCDProjectRoleBindingRef = ref
}

func (r *ArgoCDProjectRole) HasArgoCDProjectRoleBindingRef() bool {
	return r.Status.ArgoCDProjectRoleBindingRef != ""
}

// SetConditions sets the supplied conditions, replacing any existing conditions
// of the same type. This is a no-op if all supplied conditions are identical,
// ignoring the last transition time, to those already set.
// Observed generation is updated if higher than the existing one.
func (rb *ArgoCDRoleBinding) SetConditions(c ...Condition) {
	for _, new := range c {
		exists := false
		for i, existing := range rb.Status.Conditions {
			if existing.Type != new.Type {
				continue
			}

			if existing.Equal(new) {
				exists = true
				if rb.Status.Conditions[i].ObservedGeneration < new.ObservedGeneration {
					rb.Status.Conditions[i].ObservedGeneration = new.ObservedGeneration
				}
				continue
			}

			rb.Status.Conditions[i] = new
			exists = true
		}
		if !exists {
			rb.Status.Conditions = append(rb.Status.Conditions, new)
		}
	}
}

// SetConditions sets the supplied conditions, replacing any existing conditions
// of the same type. This is a no-op if all supplied conditions are identical,
// ignoring the last transition time, to those already set.
// Observed generation is updated if higher than the existing one.
func (rb *ArgoCDProjectRoleBinding) SetConditions(c ...Condition) {
	for _, new := range c {
		exists := false
		for i, existing := range rb.Status.Conditions {
			if existing.Type != new.Type {
				continue
			}

			if existing.Equal(new) {
				exists = true
				if rb.Status.Conditions[i].ObservedGeneration < new.ObservedGeneration {
					rb.Status.Conditions[i].ObservedGeneration = new.ObservedGeneration
				}
				continue
			}

			rb.Status.Conditions[i] = new
			exists = true
		}
		if !exists {
			rb.Status.Conditions = append(rb.Status.Conditions, new)
		}
	}
}

// Deleting returns a condition that indicates the resource is currently
// being deleted.
func Deleting() Condition {
	return Condition{
		Type:               TypeReady,
		Status:             corev1.ConditionFalse,
		LastTransitionTime: metav1.Now(),
		Reason:             ReasonDeleting,
	}
}

// Pending returns a condition that indicates the resource is currently pending.
func Pending(err error) Condition {
	return Condition{
		Type:               TypePending,
		Status:             corev1.ConditionFalse,
		LastTransitionTime: metav1.Now(),
		Reason:             ReasonReconcileError,
		Message:            err.Error(),
	}
}

// ReconcileSuccess returns a condition indicating that the Controller successfully
// completed the most recent reconciliation of the resource.
func ReconcileSuccess() Condition {
	return Condition{
		Type:               TypeSynced,
		Status:             corev1.ConditionTrue,
		LastTransitionTime: metav1.Now(),
		Reason:             ReasonReconcileSuccess,
	}
}

// ReconcileError returns a condition indicating that the Controller encountered an
// error while reconciling the resource.
func ReconcileError(err error) Condition {
	return Condition{
		Type:               TypeSynced,
		Status:             corev1.ConditionFalse,
		LastTransitionTime: metav1.Now(),
		Reason:             ReasonReconcileError,
		Message:            err.Error(),
	}
}
