git.r.bdr.sh - rbdr/mobius/blob - server/main.go

   1 package main
   2
   3 import (
   4         "context"
   5         "flag"
   6         "fmt"
   7         "github.com/jhalter/mobius"
   8         "go.uber.org/zap"
   9         "go.uber.org/zap/zapcore"
  10         "os"
  11 )
  12
  13 const (
  14         defaultConfigPath = "/usr/local/var/mobius/config/" // matches Homebrew default config location
  15         defaultPort       = 5500
  16 )
  17
  18 func main() {
  19         ctx, cancelRoot := context.WithCancel(context.Background())
  20
  21         basePort := flag.Int("bind", defaultPort, "Bind address and port")
  22         configDir := flag.String("config", defaultConfigPath, "Path to config root")
  23         version := flag.Bool("version", false, "print version and exit")
  24         logLevel := flag.String("log-level", "info", "Log level")
  25         flag.Parse()
  26
  27         if *version {
  28                 fmt.Printf("v%s\n", hotline.VERSION)
  29                 os.Exit(0)
  30         }
  31
  32         zapLvl, ok := zapLogLevel[*logLevel]
  33         if !ok {
  34                 fmt.Printf("Invalid log level %s.  Must be debug, info, warn, or error.\n", *logLevel)
  35                 os.Exit(0)
  36         }
  37
  38         cores := []zapcore.Core{newStdoutCore(zapLvl)}
  39         l := zap.New(zapcore.NewTee(cores...))
  40         defer func() { _ = l.Sync() }()
  41         logger := l.Sugar()
  42
  43         if _, err := os.Stat(*configDir); os.IsNotExist(err) {
  44                 logger.Fatalw("Configuration directory not found", "path", configDir)
  45         }
  46
  47         srv, err := hotline.NewServer(*configDir, "", *basePort, logger)
  48         if err != nil {
  49                 logger.Fatal(err)
  50         }
  51
  52         // Serve Hotline requests until program exit
  53         logger.Fatal(srv.ListenAndServe(ctx, cancelRoot))
  54 }
  55
  56 func newStdoutCore(level zapcore.Level) zapcore.Core {
  57         encoderCfg := zap.NewProductionEncoderConfig()
  58         encoderCfg.TimeKey = "timestamp"
  59         encoderCfg.EncodeTime = zapcore.ISO8601TimeEncoder
  60
  61         return zapcore.NewCore(
  62                 zapcore.NewConsoleEncoder(encoderCfg),
  63                 zapcore.Lock(os.Stdout),
  64                 level,
  65         )
  66 }
  67
  68 var zapLogLevel = map[string]zapcore.Level{
  69         "debug": zap.DebugLevel,
  70         "info":  zap.InfoLevel,
  71         "warn":  zap.WarnLevel,
  72         "error": zap.ErrorLevel,
  73 }