-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
132 lines (119 loc) · 2.86 KB
/
main.go
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
128
129
130
131
132
package main
import (
"bufio"
"fmt"
"io/ioutil"
"os"
"github.com/rifflock/lfshook"
"github.com/sirupsen/logrus"
"github.com/urfave/cli"
"github.com/Fantom-foundation/go-lachesis/src/dummy"
)
var (
// NameFlag is the name of the client
NameFlag = cli.StringFlag{
Name: "name",
Usage: "Client Name",
}
// ProxyAddressFlag is the settings for the proxy address
ProxyAddressFlag = cli.StringFlag{
Name: "proxy_addr",
Usage: "IP:Port to bind Proxy Server",
Value: "127.0.0.1:1338",
}
// ClientAddressFlag is the settings for the address flag (can lookup)
ClientAddressFlag = cli.StringFlag{
Name: "client_addr",
Usage: "IP:Port of Client App",
Value: "127.0.0.1:1339",
}
// LogLevelFlag setting that the user wants
LogLevelFlag = cli.StringFlag{
Name: "log_level",
Usage: "debug, info, warn, error, fatal, panic",
Value: "debug",
}
)
func main() {
app := cli.NewApp()
app.Name = "dummy"
app.Usage = "Dummy Socket Client for Lachesis"
app.Flags = []cli.Flag{
NameFlag,
ProxyAddressFlag,
ClientAddressFlag,
LogLevelFlag,
}
app.Action = run
if err := app.Run(os.Args); err != nil {
fmt.Printf("Error in run: %v\n", err)
}
}
func run(c *cli.Context) error {
logger := newLogger()
logger.Level = logLevel(c.String(LogLevelFlag.Name))
name := c.String(NameFlag.Name)
address := c.String(ProxyAddressFlag.Name)
logger.WithFields(logrus.Fields{
"name": name,
"proxy_addr": address,
}).Debug("RUN")
//Create and run Dummy Socket Client
client, err := dummy.NewDummySocketClient(address, logger)
if err != nil {
return err
}
//Listen for input messages from tty
scanner := bufio.NewScanner(os.Stdin)
for scanner.Scan() {
fmt.Print("Enter your text: ")
text := scanner.Text()
message := fmt.Sprintf("%s: %s", name, text)
if err := client.SubmitTx([]byte(message)); err != nil {
fmt.Printf("Error in SubmitTx: %v\n", err)
}
}
return nil
}
func newLogger() *logrus.Logger {
logger := logrus.New()
pathMap := lfshook.PathMap{}
_, err := os.OpenFile("info.log", os.O_CREATE|os.O_WRONLY, 0666)
if err != nil {
logger.Info("Failed to open info.log file, using default stderr")
} else {
pathMap[logrus.InfoLevel] = "info.log"
}
_, err = os.OpenFile("debug.log", os.O_CREATE|os.O_WRONLY, 0666)
if err != nil {
logger.Info("Failed to open debug.log file, using default stderr")
} else {
pathMap[logrus.DebugLevel] = "debug.log"
}
if err == nil {
logger.Out = ioutil.Discard
}
logger.Hooks.Add(lfshook.NewHook(
pathMap,
&logrus.TextFormatter{},
))
return logger
}
func logLevel(l string) logrus.Level {
switch l {
case "debug":
return logrus.DebugLevel
case "info":
return logrus.InfoLevel
case "warn":
return logrus.WarnLevel
case "error":
return logrus.ErrorLevel
case "fatal":
return logrus.FatalLevel
case "panic":
return logrus.PanicLevel
default:
return logrus.DebugLevel
}
}