main.go

// Copyright 2019 HAProxy Technologies LLC
//
// 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.

//nolint:forbidigo
package main

import (
	_ "embed"
	"fmt"
	"os"
	"os/signal"
	"strconv"
	"strings"
	"syscall"

	//nolint:gosec
	_ "net/http/pprof"

	"k8s.io/apimachinery/pkg/watch"

	"github.com/go-test/deep"
	"github.com/jessevdk/go-flags"

	"github.com/haproxytech/kubernetes-ingress/pkg/annotations"
	"github.com/haproxytech/kubernetes-ingress/pkg/controller"
	"github.com/haproxytech/kubernetes-ingress/pkg/job"
	"github.com/haproxytech/kubernetes-ingress/pkg/k8s"
	"github.com/haproxytech/kubernetes-ingress/pkg/k8s/meta"
	k8ssync "github.com/haproxytech/kubernetes-ingress/pkg/k8s/sync"
	"github.com/haproxytech/kubernetes-ingress/pkg/store"
	"github.com/haproxytech/kubernetes-ingress/pkg/utils"
	"github.com/haproxytech/kubernetes-ingress/pkg/version"

	_ "go.uber.org/automaxprocs"
)

//go:embed fs/usr/local/etc/haproxy/haproxy.cfg
var haproxyConf []byte

func main() {
	exitCode := 0
	defer func() {
		if r := recover(); r != nil {
			fmt.Println("Error : ", r)
		}
		os.Exit(exitCode)
	}()

	// Parse Controller Args
	var osArgs utils.OSArgs
	var err error
	parser := flags.NewParser(&osArgs, flags.IgnoreUnknown)
	if _, err = parser.Parse(); err != nil {
		fmt.Println(err)
		exitCode = 1
		return
	}

	// Set Logger
	logger := utils.GetLogger()
	err = version.Set()
	if err != nil {
		logger.Error(err)
		os.Exit(1) //nolint:gocritic
	}

	logger.SetLevel(osArgs.LogLevel.LogLevel)
	if len(osArgs.Help) > 0 && osArgs.Help[0] {
		parser.WriteHelp(os.Stdout)
		return
	}

	if osArgs.JobCheckCRD {
		logger.Print(version.IngressControllerInfo)
		logger.Print(job.IngressControllerCRDUpdater)
		logger.Infof("HAProxy Ingress Controller CRD Updater %s %s", version.GitTag, version.GitCommit)
		logger.Infof("Build from: %s", version.GitRepo)
		logger.Infof("Build date: %s\n", version.GitCommitDate)

		err := job.CRDRefresh(logger, osArgs)
		if err != nil {
			logger.Error(err)
			os.Exit(1)
		}
		// exit, this is just a job
		os.Exit(0)
	}
	if osArgs.InitialSyncPeriod == 0 {
		osArgs.InitialSyncPeriod = osArgs.SyncPeriod
	}

	logger.ShowFilename(false)
	exit := logInfo(logger, osArgs)
	if exit {
		return
	}
	logger.ShowFilename(true)

	annotations.DisableConfigSnippets(osArgs.DisableConfigSnippets)

	// backwards compatibility with 1.7
	if osArgs.PprofEnabled && osArgs.ControllerPort == 0 {
		osArgs.ControllerPort = 6060
	}
	if osArgs.PrometheusEnabled && osArgs.ControllerPort == 0 {
		osArgs.ControllerPort = 6060
	}

	deep.NilMapsAreEmpty = true
	deep.NilSlicesAreEmpty = true

	// Default annotations
	defaultBackendSvc := fmt.Sprint(osArgs.DefaultBackendService)
	defaultCertificate := fmt.Sprint(osArgs.DefaultCertificate)
	annotations.SetDefaultValue("default-backend-service", defaultBackendSvc)
	annotations.SetDefaultValue("default-backend-port", strconv.Itoa(osArgs.DefaultBackendPort))
	annotations.SetDefaultValue("ssl-certificate", defaultCertificate)

	// Start Controller
	var chanSize int64 = int64(watch.DefaultChanSize * 6)
	if osArgs.ChannelSize > 0 {
		chanSize = osArgs.ChannelSize
	}
	eventChan := make(chan k8ssync.SyncDataEvent, chanSize)
	stop := make(chan struct{})

	publishService := getNamespaceValue(osArgs.PublishService)

	s := store.NewK8sStore(osArgs)
	k := k8s.New(
		osArgs,
		s.NamespacesAccess.Whitelist,
		publishService,
	)

	if osArgs.Test {
		meta.GetMetaStore().ProcessedResourceVersion.SetTestMode()
	}

	c := controller.NewBuilder().
		WithHaproxyCfgFile(haproxyConf).
		WithEventChan(eventChan).
		WithStore(s).
		WithClientSet(k.GetClientset()).
		WithRestClientSet(k.GetRestClientset()).
		WithArgs(osArgs).Build()

	isGatewayAPIInstalled := k.IsGatewayAPIInstalled(osArgs.GatewayControllerName)

	c.SetGatewayAPIInstalled(isGatewayAPIInstalled)

	go k.MonitorChanges(eventChan, stop, osArgs, isGatewayAPIInstalled)
	go c.Start()
	// Catch QUIT signals
	signalC := make(chan os.Signal, 1)
	signal.Notify(signalC, os.Interrupt, syscall.SIGTERM, syscall.SIGUSR1)
	<-signalC
	c.Stop()
	close(stop)
}

func logInfo(logger utils.Logger, osArgs utils.OSArgs) bool {
	if len(osArgs.Version) > 0 {
		fmt.Printf("HAProxy Ingress Controller %s %s\n", version.GitTag, version.GitCommit)
		fmt.Printf("Build from: %s\n", version.GitRepo)
		fmt.Printf("Git commit date: %s\n", version.GitCommitDate)
		if len(osArgs.Version) > 1 {
			fmt.Printf("ConfigMap: %s\n", osArgs.ConfigMap)
			fmt.Printf("Ingress class: %s\n", osArgs.IngressClass)
			fmt.Printf("Empty Ingress class: %t\n", osArgs.EmptyIngressClass)
		}
		return true
	}

	logger.Print(version.IngressControllerInfo)
	logger.Printf("HAProxy Ingress Controller %s %s", version.GitTag, version.GitCommit)
	logger.Printf("Build from: %s", version.GitRepo)
	logger.Printf("Git commit date: %s", version.GitCommitDate)
	logger.Printf("ConfigMap: %s", osArgs.ConfigMap)
	logger.Printf("Ingress class: %s", osArgs.IngressClass)
	logger.Printf("Empty Ingress class: %t", osArgs.EmptyIngressClass)
	if osArgs.GatewayControllerName != "" {
		// display log message only if Gateway API is used
		logger.Printf("Gateway API controller name: %s", osArgs.GatewayControllerName)
	}
	logger.Printf("Publish service: %s", osArgs.PublishService)
	if osArgs.DefaultBackendService.String() != "" {
		logger.Printf("Default backend service: %s", osArgs.DefaultBackendService)
	} else {
		logger.Printf("Using local backend service on port: %d", osArgs.DefaultBackendPort)
	}
	logger.Printf("Default ssl certificate: %s", osArgs.DefaultCertificate)
	if !osArgs.DisableHTTP {
		logger.Printf("Frontend HTTP listening on: %s:%d", osArgs.IPV4BindAddr, osArgs.HTTPBindPort)
	}
	if !osArgs.DisableHTTPS {
		logger.Printf("Frontend HTTPS listening on: %s:%d", osArgs.IPV4BindAddr, osArgs.HTTPSBindPort)
	}
	if osArgs.DisableHTTP {
		logger.Printf("Disabling HTTP frontend")
	}
	if osArgs.DisableHTTPS {
		logger.Printf("Disabling HTTPS frontend")
	}
	if osArgs.DisableIPV4 {
		logger.Printf("Disabling IPv4 support")
	}
	if osArgs.DisableIPV6 {
		logger.Printf("Disabling IPv6 support")
	}
	if osArgs.ConfigMapTCPServices.Name != "" {
		logger.Printf("TCP Services provided in '%s'", osArgs.ConfigMapTCPServices)
	}
	if osArgs.ConfigMapErrorFiles.Name != "" {
		logger.Printf("Errorfiles provided in '%s'", osArgs.ConfigMapErrorFiles)
	}
	if osArgs.ConfigMapPatternFiles.Name != "" {
		logger.Printf("Pattern files provided in '%s'", osArgs.ConfigMapPatternFiles)
	}
	if osArgs.DisableConfigSnippets != "" {
		logger.Printf("Disabling config snippets for [%s]", osArgs.DisableConfigSnippets)
	}
	if osArgs.DisableDelayedWritingOnlyIfReload {
		logger.Printf("Disabling the delayed writing of files to disk only in case of haproxy reload (write to disk even if no reload)")
	}
	logger.Debugf("Kubernetes Informers resync period: %s", osArgs.CacheResyncPeriod.String())
	logger.Printf("Controller initial sync period: %s", osArgs.InitialSyncPeriod.String())
	logger.Printf("Controller sync period: %s\n", osArgs.SyncPeriod.String())

	hostname, err := os.Hostname()
	logger.Error(err)
	logger.Printf("Running on %s", hostname)
	return false
}

func getNamespaceValue(name string) *utils.NamespaceValue {
	parts := strings.Split(name, "/")
	var result *utils.NamespaceValue
	if len(parts) == 2 {
		result = &utils.NamespaceValue{
			Namespace: parts[0],
			Name:      parts[1],
		}
	}
	return result
}