generated from songquanpeng/go-template
-
Notifications
You must be signed in to change notification settings - Fork 6
/
main.go
42 lines (40 loc) · 924 Bytes
/
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
package main
import (
"go-public/common"
"go-public/handler"
"os"
"strconv"
)
func main() {
if len(os.Args) == 3 && os.Args[1] == "init" {
if os.Args[2] == "server" {
common.InitConfigFile(true)
} else if os.Args[2] == "client" {
common.InitConfigFile(false)
} else {
println("Usage: go-public init <server|client>")
os.Exit(1)
}
os.Exit(0)
}
if len(os.Args) == 1 {
// Server mode
common.LoadConfigFile(true)
handler.ServeForever()
os.Exit(0)
}
if len(os.Args) == 3 {
// Client mode
common.LoadConfigFile(false)
localPort, err1 := strconv.Atoi(os.Args[1])
remotePort, err2 := strconv.Atoi(os.Args[2])
if err1 != nil || err2 != nil {
println("Usage: go-public <local_port> <remote_port>")
os.Exit(1)
}
handler.PublicPort(localPort, remotePort)
os.Exit(0)
}
println("Usage: go-public <local_port> <remote_port> or go-public init <server|client>")
os.Exit(1)
}