-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
124 lines (100 loc) · 1.84 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
/**
* @program: send
*
* @description:
*
* @author: lemo
*
* @create: 2022-09-09 15:40
**/
package main
import (
"os"
"path/filepath"
"time"
"github.com/lemonyxk/console"
"github.com/lemonyxk/utils/v3"
)
func init() {
console.SetFlags(0)
// console.Colorful(false)
console.SetInfoColor(console.FgGreen)
if HasArgs("--debug", os.Args) {
console.SetFlags(console.FILE)
}
}
var (
savePath = ""
serverAddr = ""
)
func main() {
if len(os.Args) < 2 {
console.Exit(help())
}
switch os.Args[1] {
case "server":
startServer()
case "client":
startClient()
default:
console.Exit(help())
}
}
func startServer() {
var pwd, err = os.Getwd()
if err != nil {
console.Exit(err)
}
absPwd, err := filepath.Abs(pwd)
if err != nil {
console.Exit(err)
}
savePath = absPwd
if HasArgs("--path", os.Args) {
var p = GetArgs([]string{"--path"}, os.Args)
if p == "" {
console.Exit("save path is empty")
}
var absP, err = filepath.Abs(p)
if err != nil {
console.Exit(err)
}
savePath = absP
}
if HasArgs("--addr", os.Args) {
var p = GetArgs([]string{"--addr"}, os.Args)
if p == "" {
console.Exit("addr is empty")
}
serverAddr = p
}
go runTcpServer()
utils.Signal.ListenKill().Done(func(sig os.Signal) {
console.Exit(sig)
})
}
func startClient() {
var addr = GetArgs([]string{"--addr"}, os.Args)
if addr == "" {
console.Exit("addr is empty")
}
<-runTcpClient(addr)
if HasArgs("--file", os.Args) || HasArgs("-f", os.Args) {
var p = GetArgs([]string{"--file", "-f"}, os.Args)
var absP, err = filepath.Abs(p)
if err != nil {
console.Exit(err)
}
sendFile(absP)
return
}
if HasArgs("--string", os.Args) || HasArgs("-s", os.Args) {
var p = GetArgs([]string{"--string", "-s"}, os.Args)
if p == "" {
console.Exit("string is empty")
}
sendStr(p)
return
}
time.Sleep(time.Second)
}