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