package main

import (
	"flag"
	"github.com/example/mysql-database-operator/api/v1alpha1"
	"github.com/example/mysql-database-operator/controllers"
	"k8s.io/apimachinery/pkg/runtime"
	"os"
	"sigs.k8s.io/controller-runtime/pkg/client/config"
	"sigs.k8s.io/controller-runtime/pkg/controller"
	"sigs.k8s.io/controller-runtime/pkg/healthz"
	"sigs.k8s.io/controller-runtime/pkg/log/zap"
	"sigs.k8s.io/controller-runtime/pkg/manager"
	"sigs.k8s.io/controller-runtime/pkg/metrics/server"
)

func main() {
	var metricsAddr, probeAddr string
	var leader bool
	flag.StringVar(&metricsAddr, "metrics-bind-address", ":8080", "metrics address")
	flag.StringVar(&probeAddr, "health-probe-bind-address", ":8081", "health probe address")
	flag.BoolVar(&leader, "leader-elect", true, "enable leader election")
	var zo zap.Options
	zo.BindFlags(flag.CommandLine)
	flag.Parse()
	controller.SetLogger(zap.New(zap.UseFlagOptions(&zo)))
	scheme := runtime.NewScheme()
	_ = v1alpha1.AddToScheme(scheme)
	mgr, err := manager.New(config.GetConfigOrDie(), manager.Options{Scheme: scheme, Metrics: server.Options{BindAddress: metricsAddr}, HealthProbeBindAddress: probeAddr, LeaderElection: leader, LeaderElectionID: "mysql-database-operator"})
	if err != nil {
		os.Exit(1)
	}
	if err := (&controllers.MySQLDatabaseReconciler{Client: mgr.GetClient(), Scheme: scheme}).SetupWithManager(mgr); err != nil {
		os.Exit(1)
	}
	if err := mgr.AddHealthzCheck("health", healthz.Ping); err != nil {
		os.Exit(1)
	}
	if err := mgr.AddReadyzCheck("ready", healthz.Ping); err != nil {
		os.Exit(1)
	}
	if err := mgr.Start(controller.SetupSignalHandler()); err != nil {
		os.Exit(1)
	}
}