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
|
package main
import (
"context"
"flag"
"fmt"
"github.com/jhalter/mobius/hotline"
"github.com/rivo/tview"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"log"
"os"
"os/signal"
"runtime"
"syscall"
)
func main() {
_, cancelRoot := context.WithCancel(context.Background())
sigChan := make(chan os.Signal, 1)
signal.Notify(sigChan, syscall.SIGTERM, syscall.SIGINT, os.Interrupt)
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")
logFile := flag.String("log-file", "", "output logs to file")
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)
}
// init DebugBuffer
db := &hotline.DebugBuffer{
TextView: tview.NewTextView(),
}
cores := []zapcore.Core{newZapCore(zapLvl, db)}
// Add file logger if optional log-file flag was passed
if *logFile != "" {
f, err := os.OpenFile(*logFile,
os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil {
log.Println(err)
}
defer f.Close()
if err != nil {
panic(err)
}
cores = append(cores, newZapCore(zapLvl, f))
}
l := zap.New(zapcore.NewTee(cores...))
defer func() { _ = l.Sync() }()
logger := l.Sugar()
logger.Infow("Started Mobius client", "Version", hotline.VERSION)
go func() {
sig := <-sigChan
logger.Infow("Stopping client", "signal", sig.String())
cancelRoot()
}()
client := hotline.NewUIClient(*configDir, logger)
client.DebugBuf = db
client.UI.Start()
}
func newZapCore(level zapcore.Level, syncer zapcore.WriteSyncer) zapcore.Core {
encoderCfg := zap.NewProductionEncoderConfig()
encoderCfg.TimeKey = "timestamp"
encoderCfg.EncodeTime = zapcore.ISO8601TimeEncoder
return zapcore.NewCore(
zapcore.NewConsoleEncoder(encoderCfg),
zapcore.Lock(syncer),
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 = "mobius-client-config.yaml"
case "darwin":
if _, err := os.Stat("/usr/local/etc/mobius-client-config.yaml"); err == nil {
cfgPath = "/usr/local/etc/mobius-client-config.yaml"
} else if _, err := os.Stat("/opt/homebrew/etc/mobius-client-config.yaml"); err == nil {
cfgPath = "/opt/homebrew/etc/mobius-client-config.yaml"
}
case "linux":
cfgPath = "/usr/local/etc/mobius-client-config.yaml"
default:
fmt.Printf("unsupported OS")
}
return cfgPath
}
|
