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
|
package main
import (
"context"
"flag"
"fmt"
hotline "github.com/jhalter/mobius"
"github.com/rivo/tview"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"os"
"os/signal"
"syscall"
"time"
)
//var defaultTrackerList = []string{
// "hltracker.com:5498",
//}
const connectTimeout = 3 * time.Second
func main() {
_, cancelRoot := context.WithCancel(context.Background())
sigChan := make(chan os.Signal, 1)
signal.Notify(sigChan, syscall.SIGTERM, syscall.SIGINT, os.Interrupt)
version := flag.Bool("version", false, "print version and exit")
logLevel := flag.String("log-level", "info", "Log level")
userName := flag.String("name", "unnamed", "User name")
//srvAddr := flag.String("server", "localhost:5500", "Hostname/Port of server")
//login := flag.String("login", "guest", "Login Name")
//pass := flag.String("password", "", "Login Password")
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{
newDebugCore(zapLvl, db),
//newStderrCore(zapLvl),
}
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.NewClient(*userName, logger)
client.DebugBuf = db
client.UI.Start()
}
func newDebugCore(level zapcore.Level, db *hotline.DebugBuffer) zapcore.Core {
encoderCfg := zap.NewProductionEncoderConfig()
encoderCfg.TimeKey = "timestamp"
encoderCfg.EncodeTime = zapcore.ISO8601TimeEncoder
return zapcore.NewCore(
zapcore.NewConsoleEncoder(encoderCfg),
zapcore.Lock(db),
level,
)
}
func newStderrCore(level zapcore.Level) zapcore.Core {
encoderCfg := zap.NewProductionEncoderConfig()
encoderCfg.TimeKey = "timestamp"
encoderCfg.EncodeTime = zapcore.ISO8601TimeEncoder
return zapcore.NewCore(
zapcore.NewConsoleEncoder(encoderCfg),
zapcore.Lock(os.Stderr),
level,
)
}
var zapLogLevel = map[string]zapcore.Level{
"debug": zap.DebugLevel,
"info": zap.InfoLevel,
"warn": zap.WarnLevel,
"error": zap.ErrorLevel,
}
|
