nebula/cmd/nebula-service/main.go

package main

import (
	"flag"
	"fmt"
	"os"
	"runtime/debug"
	"strings"

	"github.com/slackhq/nebula"
	"github.com/slackhq/nebula/config"
	"github.com/slackhq/nebula/logging"
	"github.com/slackhq/nebula/util"
)

// A version string that can be set with
//
//	-ldflags "-X main.Build=SOMEVERSION"
//
// at compile-time.
var Build string

func init() {
	if Build == "" {
		info, ok := debug.ReadBuildInfo()
		if !ok {
			return
		}

		Build = strings.TrimPrefix(info.Main.Version, "v")
	}
}

func main() {
	serviceFlag := flag.String("service", "", "Control the system service.")
	configPath := flag.String("config", "", "Path to either a file or directory to load configuration from")
	configTest := flag.Bool("test", false, "Test the config and print the end result. Non zero exit indicates a faulty config")
	printVersion := flag.Bool("version", false, "Print version")
	printUsage := flag.Bool("help", false, "Print command line usage")

	flag.Parse()

	if *printVersion {
		fmt.Printf("Version: %s\n", Build)
		os.Exit(0)
	}

	if *printUsage {
		flag.Usage()
		os.Exit(0)
	}

	l := logging.NewLogger(os.Stdout)

	if *serviceFlag != "" {
		if *configTest {
			fmt.Println("-test is not supported with -service, run the config test without -service")
			os.Exit(1)
		}

		if err := doService(configPath, Build, serviceFlag); err != nil {
			l.Error("Service command failed", "error", err)
			os.Exit(1)
		}
		return
	}

	if *configPath == "" {
		p, err := config.DefaultPath()
		if err != nil {
			fmt.Println(err)
			os.Exit(1)
		}
		*configPath = p
	}

	c := config.NewC(l)
	err := c.Load(*configPath)
	if err != nil {
		fmt.Printf("failed to load config: %s", err)
		os.Exit(1)
	}

	if err := logging.ApplyConfig(l, c); err != nil {
		fmt.Printf("failed to apply logging config: %s", err)
		os.Exit(1)
	}
	c.RegisterReloadCallback(func(c *config.C, r *config.Reload) {
		if err := logging.ApplyConfig(l, c); err != nil {
			l.Error("Failed to reconfigure logger on reload", "error", err)
			r.Failed()
		}
	})

	ctrl, err := nebula.Main(c, *configTest, Build, l, nil)
	if err != nil {
		util.LogWithContextIfNeeded("Failed to start", err, l)
		os.Exit(1)
	}

	if !*configTest {
		if err := ctrl.Start(); err != nil {
			util.LogWithContextIfNeeded("Error while running", err, l)
			os.Exit(1)
		}

		go ctrl.ShutdownBlock()

		if err := ctrl.Wait(); err != nil {
			l.Error("Nebula stopped due to fatal error", "error", err)
			os.Exit(2)
		}

		l.Info("Goodbye")
	}

	os.Exit(0)
}