package main

import (
	"github.com/caddyserver/ingress/internal/caddy"
	"github.com/caddyserver/ingress/internal/controller"
	"go.uber.org/zap"
	v1 "k8s.io/api/core/v1"
	"k8s.io/apimachinery/pkg/util/wait"
	"k8s.io/apimachinery/pkg/version"
	"k8s.io/client-go/kubernetes"
	"k8s.io/client-go/tools/clientcmd"
	"os"
	"os/signal"
	"syscall"
	"time"
)

const (
	// high enough QPS to fit all expected use cases.
	defaultQPS = 1e6

	// high enough Burst to fit all expected use cases.
	defaultBurst = 1e6
)

func createLogger(verbose bool) *zap.SugaredLogger {
	prodCfg := zap.NewProductionConfig()

	if verbose {
		prodCfg.Level = zap.NewAtomicLevelAt(zap.DebugLevel)
	}
	logger, _ := prodCfg.Build()

	return logger.Sugar()
}

func main() {
	// parse any flags required to configure the caddy ingress controller
	cfg := parseFlags()

	logger := createLogger(cfg.Verbose)

	if cfg.WatchNamespace == "" {
		cfg.WatchNamespace = v1.NamespaceAll
		logger.Warn("-namespace flag is unset, caddy ingress controller will monitor ingress resources in all namespaces.")
	}

	// get client to access the kubernetes service api
	kubeClient, _, err := createApiserverClient(logger)
	if err != nil {
		logger.Fatalf("Could not establish a connection to the Kubernetes API Server. %v", err)
	}

	stopCh := make(chan struct{}, 1)

	c := controller.NewCaddyController(logger, kubeClient, cfg, caddy.Converter{}, stopCh)

	// start the ingress controller
	logger.Info("Starting the caddy ingress controller")
	go c.Run()

	// Listen for SIGINT and SIGTERM signals
	sigs := make(chan os.Signal, 1)
	signal.Notify(sigs, os.Interrupt, syscall.SIGTERM)
	<-sigs
	close(stopCh)

	// Let controller exit the process
	select {}
}

// createApiserverClient creates a new Kubernetes REST client. We assume the
// controller runs inside Kubernetes and use the in-cluster config.
func createApiserverClient(logger *zap.SugaredLogger) (*kubernetes.Clientset, *version.Info, error) {
	cfg, err := clientcmd.BuildConfigFromFlags("", "")
	if err != nil {
		return nil, nil, err
	}

	logger.Infof("Creating API client for %s", cfg.Host)

	cfg.QPS = defaultQPS
	cfg.Burst = defaultBurst
	cfg.ContentType = "application/vnd.kubernetes.protobuf"
	client, err := kubernetes.NewForConfig(cfg)
	if err != nil {
		return nil, nil, err
	}

	// The client may fail to connect to the API server on the first request
	defaultRetry := wait.Backoff{
		Steps:    10,
		Duration: 1 * time.Second,
		Factor:   1.5,
		Jitter:   0.1,
	}

	var v *version.Info
	var retries int
	var lastErr error

	err = wait.ExponentialBackoff(defaultRetry, func() (bool, error) {
		v, err = client.Discovery().ServerVersion()
		if err == nil {
			return true, nil
		}

		lastErr = err
		logger.Infof("Unexpected error discovering Kubernetes version (attempt %v): %v", retries, err)
		retries++
		return false, nil
	})

	// err is returned in case of timeout in the exponential backoff (ErrWaitTimeout)
	if err != nil {
		return nil, nil, lastErr
	}

	if retries > 0 {
		logger.Warnf("Initial connection to the Kubernetes API server was retried %d times.", retries)
	}

	return client, v, nil
}