main.go

//go:generate go run pkg/codegen/cleanup/main.go
//go:generate go run pkg/codegen/main.go

package main

import (
	"context"
	"fmt"
	"net/http"
	_ "net/http/pprof"
	"os"

	"github.com/rancher/gitjob/pkg/controller"
	"github.com/rancher/gitjob/pkg/hooks"
	"github.com/rancher/gitjob/pkg/types"
	"github.com/rancher/wrangler/pkg/leader"
	"github.com/rancher/wrangler/pkg/ratelimit"
	"github.com/rancher/wrangler/pkg/resolvehome"
	"github.com/rancher/wrangler/pkg/signals"
	"github.com/sirupsen/logrus"
	"github.com/urfave/cli"
	"k8s.io/apimachinery/pkg/util/runtime"
	"k8s.io/client-go/tools/clientcmd"
)

var (
	Version   = "v0.0.0-dev"
	GitCommit = "HEAD"
)

func main() {
	app := cli.NewApp()
	app.Name = "gitjob"
	app.Version = fmt.Sprintf("%s (%s)", Version, GitCommit)
	app.Flags = []cli.Flag{
		cli.StringFlag{
			Name:   "kubeconfig",
			EnvVar: "KUBECONFIG",
		},
		cli.StringFlag{
			Name:   "namespace",
			EnvVar: "NAMESPACE",
			Value:  "kube-system",
		},
		cli.StringFlag{
			Name:  "listen",
			Value: ":8080",
		},
		cli.StringFlag{
			Name:  "tekton-image",
			Value: "rancher/tekton-utils:dev",
		},
	}
	app.Action = run

	if err := app.Run(os.Args); err != nil {
		logrus.Fatal(err)
	}
}

func run(c *cli.Context) {
	logrus.Info("Starting controller")
	ctx := signals.SetupSignalHandler(context.Background())

	kubeconfig, err := resolvehome.Resolve(c.String("kubeconfig"))
	if err != nil {
		logrus.Fatalf("Error resolving home dir: %s", err.Error())
	}

	cfg, err := clientcmd.BuildConfigFromFlags("", kubeconfig)
	if err != nil {
		logrus.Fatalf("Error building kubeconfig: %s", err.Error())
	}
	cfg.RateLimiter = ratelimit.None

	ctx, cont := types.BuildContext(ctx, c.String("namespace"), cfg)
	if err := cont.Start(ctx); err != nil {
		logrus.Fatal(err)
	}

	cont.Image = c.String("tekton-image")

	go func() {
		leader.RunOrDie(ctx, c.String("namespace"), "gitjob", cont.K8s, func(ctx context.Context) {
			controller.Register(ctx, cont)
			runtime.Must(cont.Start(ctx))
			<-ctx.Done()
		})
	}()

	logrus.Info("Setting up webhook listener")
	handler := hooks.HandleHooks(cont)
	addr := c.String("listen")
	if err := http.ListenAndServe(addr, handler); err != nil {
		logrus.Fatalf("Failed to listen on %s: %v", addr, err)
	}

	<-ctx.Done()
}