// Copyright (c) 2018 SAP SE or an SAP affiliate company. All rights reserved. This file is licensed under the Apache Software License, v. 2 except as noted otherwise in the LICENSE file
//
// 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 certificatesigningrequest

import (
	"context"
	"sync"
	"time"

	"github.com/gardener/gardener/pkg/client/kubernetes"
	"github.com/gardener/gardener/pkg/controllermanager"
	"github.com/gardener/gardener/pkg/controllerutils"
	"github.com/gardener/gardener/pkg/logger"

	"github.com/prometheus/client_golang/prometheus"
	kubeinformers "k8s.io/client-go/informers"
	certificatesv1beta1lister "k8s.io/client-go/listers/certificates/v1beta1"
	"k8s.io/client-go/tools/cache"
	"k8s.io/client-go/tools/record"
	"k8s.io/client-go/util/workqueue"
)

// Controller controls CertificateSigningRequests.
type Controller struct {
	k8sGardenClient kubernetes.Interface

	control  ControlInterface
	recorder record.EventRecorder

	csrLister certificatesv1beta1lister.CertificateSigningRequestLister
	csrQueue  workqueue.RateLimitingInterface
	csrSynced cache.InformerSynced

	workerCh               chan int
	numberOfRunningWorkers int
}

// NewCSRController takes a Kubernetes client for the Garden clusters <k8sGardenClient>, a struct
// holding information about the acting Gardener, a <kubeInformerFactory>, and a <recorder> for
// event recording. It creates a new CSR controller.
func NewCSRController(k8sGardenClient kubernetes.Interface, kubeInformerFactory kubeinformers.SharedInformerFactory, recorder record.EventRecorder) *Controller {
	var (
		certificatesV1beta1Informer = kubeInformerFactory.Certificates().V1beta1()
		csrInformer                 = certificatesV1beta1Informer.CertificateSigningRequests()
		csrLister                   = csrInformer.Lister()
	)

	csrController := &Controller{
		k8sGardenClient: k8sGardenClient,
		control:         NewDefaultControl(k8sGardenClient),
		recorder:        recorder,
		csrLister:       csrLister,
		csrQueue:        workqueue.NewNamedRateLimitingQueue(workqueue.DefaultControllerRateLimiter(), "CertificateSigningRequest"),
		workerCh:        make(chan int),
	}

	csrInformer.Informer().AddEventHandler(cache.ResourceEventHandlerFuncs{
		AddFunc:    csrController.csrAdd,
		UpdateFunc: csrController.csrUpdate,
	})
	csrController.csrSynced = csrInformer.Informer().HasSynced

	return csrController
}

// Run runs the Controller until the given context <ctx> is alive.
func (c *Controller) Run(ctx context.Context, workers int) {
	var waitGroup sync.WaitGroup

	if !cache.WaitForCacheSync(ctx.Done(), c.csrSynced) {
		logger.Logger.Error("Timed out waiting for caches to sync")
		return
	}

	// Count number of running workers.
	go func() {
		for res := range c.workerCh {
			c.numberOfRunningWorkers += res
			logger.Logger.Debugf("Current number of running CertificateSigningRequest workers is %d", c.numberOfRunningWorkers)
		}
	}()

	logger.Logger.Info("CertificateSigningRequest controller initialized.")

	for i := 0; i < workers; i++ {
		controllerutils.DeprecatedCreateWorker(ctx, c.csrQueue, "CertificateSigningRequest", c.reconcileCertificateSigningRequestKey, &waitGroup, c.workerCh)
	}

	// Shutdown handling
	<-ctx.Done()
	c.csrQueue.ShutDown()

	for {
		if c.csrQueue.Len() == 0 && c.numberOfRunningWorkers == 0 {
			logger.Logger.Debug("No running CertificateSigningRequest worker and no items left in the queues. Terminated CertificateSigningRequest controller...")
			break
		}
		logger.Logger.Debugf("Waiting for %d CertificateSigningRequest worker(s) to finish (%d item(s) left in the queues)...", c.numberOfRunningWorkers, c.csrQueue.Len())
		time.Sleep(5 * time.Second)
	}

	waitGroup.Wait()
}

// RunningWorkers returns the number of running workers.
func (c *Controller) RunningWorkers() int {
	return c.numberOfRunningWorkers
}

// CollectMetrics implements gardenmetrics.ControllerMetricsCollector interface
func (c *Controller) CollectMetrics(ch chan<- prometheus.Metric) {
	metric, err := prometheus.NewConstMetric(controllermanager.ControllerWorkerSum, prometheus.GaugeValue, float64(c.RunningWorkers()), "csr")
	if err != nil {
		controllermanager.ScrapeFailures.With(prometheus.Labels{"kind": "csr-controller"}).Inc()
		return
	}
	ch <- metric
}