common

type KaiwoLabelContext struct {
	User    string
	Name    string
	Type    string
	RunId   string
	Managed string
}

type KaiwoWorkload interface {
	GetUser() string
	GetObjectMeta() *metav1.ObjectMeta
	GetStatus() string
	GetType() string
	GetPods(ctx context.Context, k8sClient client.Client) ([]corev1.Pod, error)
	GetServices(ctx context.Context, k8sClient client.Client) ([]corev1.Service, error)
	GetDuration() *metav1.Duration
	GetStartTime() *metav1.Time
	GetClusterQueue() string
	GetGPUVendor() string
}

type LocalQueueReconciler struct {
	ResourceReconcilerBase[*kueuev1beta1.LocalQueue]
}

type Reconciler[T client.Object] interface {
	// Reconcile runs through the reconciliation loop
	Reconcile(ctx context.Context, k8sClient client.Client, scheme *runtime.Scheme) (ctrl.Result, error)
}

type ReconcilerBase[T client.Object] struct {
	ObjectKey client.ObjectKey
	Object    T
	Self      Reconciler[T]
}

type ResourceReconciler[T client.Object] interface {
	// Build will build the client object without creating it
	Build(ctx context.Context, k8sClient client.Client) (desired T, err error)

	// Get will fetch the client object
	Get(ctx context.Context, k8sClient client.Client) (actual T, err error)

	// Create will create the client object
	Create(ctx context.Context, k8sClient client.Client, scheme *runtime.Scheme, desired T, owner client.Object, recorder record.EventRecorder) error

	// Update will update the client object
	Update(ctx context.Context, k8sClient client.Client, desired *T, actual T, recorder record.EventRecorder) error

	// Reconcile will build and then create
	Reconcile(ctx context.Context, k8sClient client.Client, scheme *runtime.Scheme, owner client.Object, recorder record.EventRecorder) (actual T, result *ctrl.Result, err error)

	// ShouldContinue returns a reconciliation result if there is an intermediate result to return
	ShouldContinue(ctx context.Context, actual T) *ctrl.Result

	GetEmptyObject() T

	// ValidateBeforeCreateOrUpdate provides a way to abort the reconciliation based on the currently existing object's state
	ValidateBeforeCreateOrUpdate(ctx context.Context, actual T) (*ctrl.Result, error)
}

type ResourceReconcilerBase[T client.Object] struct {
	ObjectKey client.ObjectKey
	Self      ResourceReconciler[T]
	Desired   T
}

type StorageReconciler struct {
	ResourceReconcilerBase[*corev1.PersistentVolumeClaim]
	Amount           string
	StorageClassName string
	AccessMode       corev1.PersistentVolumeAccessMode
}