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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
package main
import (
"context"
"encoding/json"
"flag"
"fmt"
"github.com/jhalter/mobius/hotline"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"io"
"log"
"math/rand"
"net/http"
"os"
"runtime"
"time"
)
const (
defaultPort = 5500
)
func main() {
rand.Seed(time.Now().UnixNano())
ctx, cancelRoot := context.WithCancel(context.Background())
basePort := flag.Int("bind", defaultPort, "Bind address and port")
statsPort := flag.String("stats-port", "", "Enable stats HTTP endpoint on 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)
}
srv, err := hotline.NewServer(*configDir, "", *basePort, logger, &hotline.OSFileStore{})
if err != nil {
logger.Fatal(err)
}
sh := statHandler{hlServer: srv}
if *statsPort != "" {
http.HandleFunc("/", sh.RenderStats)
go func(srv *hotline.Server) {
// Use the default DefaultServeMux.
err = http.ListenAndServe(":"+*statsPort, nil)
if err != nil {
log.Fatal(err)
}
}(srv)
}
// Serve Hotline requests until program exit
logger.Fatal(srv.ListenAndServe(ctx, cancelRoot))
}
type statHandler struct {
hlServer *hotline.Server
}
func (sh *statHandler) RenderStats(w http.ResponseWriter, _ *http.Request) {
u, err := json.Marshal(sh.hlServer.Stats)
if err != nil {
panic(err)
}
_, _ = io.WriteString(w, string(u))
}
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,
}
func defaultConfigPath() (cfgPath string) {
switch runtime.GOOS {
case "windows":
cfgPath = "config"
case "darwin":
if _, err := os.Stat("/usr/local/var/mobius/config/"); err == nil {
cfgPath = "/usr/local/var/mobius/config/"
} else if _, err := os.Stat("/opt/homebrew/var/mobius/config"); err == nil {
cfgPath = "/opt/homebrew/var/mobius/config/"
}
case "linux":
cfgPath = "/usr/local/var/mobius/config/"
default:
fmt.Printf("unsupported OS")
}
return cfgPath
}
|
