76 lines
1.5 KiB
Go
76 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/slackhq/nebula"
|
|
)
|
|
|
|
// A version string that can be set with
|
|
//
|
|
// -ldflags "-X main.Build=SOMEVERSION"
|
|
//
|
|
// at compile-time.
|
|
var Build string
|
|
|
|
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)
|
|
}
|
|
|
|
if *serviceFlag != "" {
|
|
doService(configPath, configTest, Build, serviceFlag)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if *configPath == "" {
|
|
fmt.Println("-config flag must be set")
|
|
flag.Usage()
|
|
os.Exit(1)
|
|
}
|
|
|
|
config := nebula.NewConfig()
|
|
err := config.Load(*configPath)
|
|
if err != nil {
|
|
fmt.Printf("failed to load config: %s", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
l := logrus.New()
|
|
l.Out = os.Stdout
|
|
c, err := nebula.Main(config, *configTest, Build, l, nil)
|
|
|
|
switch v := err.(type) {
|
|
case nebula.ContextualError:
|
|
v.Log(l)
|
|
os.Exit(1)
|
|
case error:
|
|
l.WithError(err).Error("Failed to start")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if !*configTest {
|
|
c.Start()
|
|
c.ShutdownBlock()
|
|
}
|
|
|
|
os.Exit(0)
|
|
}
|