// Copyright (c) 2024 Tigera, Inc. All rights reserved.

// 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 clientv3

import (
	"context"
	"errors"
	"time"

	apiv3 "github.com/projectcalico/api/pkg/apis/projectcalico/v3"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"

	"github.com/projectcalico/calico/libcalico-go/lib/options"
	validator "github.com/projectcalico/calico/libcalico-go/lib/validator/v3"
	"github.com/projectcalico/calico/libcalico-go/lib/watch"
)

// KubeControllersConfigurationInterface has methods to work with KubeControllersConfiguration resources.
type KubeControllersConfigurationInterface interface {
	Create(ctx context.Context, res *apiv3.KubeControllersConfiguration, opts options.SetOptions) (*apiv3.KubeControllersConfiguration, error)
	Update(ctx context.Context, res *apiv3.KubeControllersConfiguration, opts options.SetOptions) (*apiv3.KubeControllersConfiguration, error)
	Delete(ctx context.Context, name string, opts options.DeleteOptions) (*apiv3.KubeControllersConfiguration, error)
	Get(ctx context.Context, name string, opts options.GetOptions) (*apiv3.KubeControllersConfiguration, error)
	List(ctx context.Context, opts options.ListOptions) (*apiv3.KubeControllersConfigurationList, error)
	Watch(ctx context.Context, opts options.ListOptions) (watch.Interface, error)
}

// KubeControllersConfiguration implements KubeControllersConfigurationInterface
type kubeControllersConfiguration struct {
	client client
}

func (r kubeControllersConfiguration) fillDefaults(res *apiv3.KubeControllersConfiguration) {
	if res.Spec.PrometheusMetricsPort == nil {
		var defaultPort = 9094
		res.Spec.PrometheusMetricsPort = &defaultPort
	}

	if res.Spec.Controllers.Node != nil {
		if res.Spec.Controllers.Node.LeakGracePeriod == nil {
			res.Spec.Controllers.Node.LeakGracePeriod = &metav1.Duration{Duration: 15 * time.Minute}
		}
	}

	if res.Spec.Controllers.LoadBalancer != nil {
		if res.Spec.Controllers.LoadBalancer.AssignIPs == "" {
			res.Spec.Controllers.LoadBalancer.AssignIPs = apiv3.AllServices
		}
	}
}

// Create takes the representation of a KubeControllersConfiguration and creates it.
// Returns the stored representation of the KubeControllersConfiguration, and an error
// if there is any.
func (r kubeControllersConfiguration) Create(ctx context.Context, res *apiv3.KubeControllersConfiguration, opts options.SetOptions) (*apiv3.KubeControllersConfiguration, error) {
	r.fillDefaults(res)
	if err := validator.Validate(res); err != nil {
		return nil, err
	}

	if res.ObjectMeta.GetName() != "default" {
		return nil, errors.New("Cannot create a Kube Controllers Configuration resource with a name other than \"default\"")
	}
	out, err := r.client.resources.Create(ctx, opts, apiv3.KindKubeControllersConfiguration, res)
	if out != nil {
		return out.(*apiv3.KubeControllersConfiguration), err
	}
	return nil, err
}

// Update takes the representation of a KubeControllersConfiguration and updates it.
// Returns the stored representation of the KubeControllersConfiguration, and an error
// if there is any.
func (r kubeControllersConfiguration) Update(ctx context.Context, res *apiv3.KubeControllersConfiguration, opts options.SetOptions) (*apiv3.KubeControllersConfiguration, error) {
	r.fillDefaults(res)
	if err := validator.Validate(res); err != nil {
		return nil, err
	}

	out, err := r.client.resources.Update(ctx, opts, apiv3.KindKubeControllersConfiguration, res)
	if out != nil {
		return out.(*apiv3.KubeControllersConfiguration), err
	}
	return nil, err
}

// Delete takes name of the KubeControllersConfiguration and deletes it. Returns an
// error if one occurs.
func (r kubeControllersConfiguration) Delete(ctx context.Context, name string, opts options.DeleteOptions) (*apiv3.KubeControllersConfiguration, error) {
	out, err := r.client.resources.Delete(ctx, opts, apiv3.KindKubeControllersConfiguration, noNamespace, name)
	if out != nil {
		return out.(*apiv3.KubeControllersConfiguration), err
	}
	return nil, err
}

// Get takes name of the KubeControllersConfiguration, and returns the corresponding
// KubeControllersConfiguration object, and an error if there is any.
func (r kubeControllersConfiguration) Get(ctx context.Context, name string, opts options.GetOptions) (*apiv3.KubeControllersConfiguration, error) {
	out, err := r.client.resources.Get(ctx, opts, apiv3.KindKubeControllersConfiguration, noNamespace, name)
	if out != nil {
		return out.(*apiv3.KubeControllersConfiguration), err
	}
	return nil, err
}

// List returns the list of KubeControllersConfiguration objects that match the supplied options.
func (r kubeControllersConfiguration) List(ctx context.Context, opts options.ListOptions) (*apiv3.KubeControllersConfigurationList, error) {
	res := &apiv3.KubeControllersConfigurationList{}
	if err := r.client.resources.List(ctx, opts, apiv3.KindKubeControllersConfiguration, apiv3.KindKubeControllersConfigurationList, res); err != nil {
		return nil, err
	}
	return res, nil
}

// Watch returns a watch.Interface that watches the KubeControllersConfiguration that
// match the supplied options.
func (r kubeControllersConfiguration) Watch(ctx context.Context, opts options.ListOptions) (watch.Interface, error) {
	return r.client.resources.Watch(ctx, opts, apiv3.KindKubeControllersConfiguration, nil)
}
