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

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