/
main.go
99 lines (83 loc) · 1.93 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
package main
import (
"fmt"
"github.com/e154/smart-home/api/mobile"
"github.com/e154/smart-home/api/server"
"github.com/e154/smart-home/api/websocket"
"github.com/e154/smart-home/system/backup"
"github.com/e154/smart-home/system/graceful_service"
"github.com/e154/smart-home/system/initial"
l "github.com/e154/smart-home/system/logging"
"github.com/op/go-logging"
"os"
)
var (
log = logging.MustGetLogger("main")
)
func main() {
args := os.Args[1:]
for _, arg := range args {
switch arg {
case "-v", "--version":
fmt.Printf(shortVersionBanner, GetHumanVersion())
return
case "-backup":
container := BuildContainer()
container.Invoke(func(
backup *backup.Backup,
graceful *graceful_service.GracefulService) {
if err := backup.New(); err != nil {
log.Error(err.Error())
}
graceful.Shutdown()
})
return
case "-restore":
if len(os.Args) < 3 {
log.Error("need backup name")
return
}
container := BuildContainer()
container.Invoke(func(
backup *backup.Backup,
graceful *graceful_service.GracefulService) {
if err := backup.Restore(os.Args[2]); err != nil {
log.Error(err.Error())
}
graceful.Shutdown()
})
return
case "-reset":
container := BuildContainer()
container.Invoke(func(
initialService *initial.InitialService) {
initialService.Reset()
})
return
default:
fmt.Printf(verboseVersionBanner, "v2", os.Args[0])
return
}
}
start()
}
func start() {
fmt.Printf(shortVersionBanner, "")
container := BuildContainer()
err := container.Invoke(func(server *server.Server,
graceful *graceful_service.GracefulService,
back *l.LogBackend,
initialService *initial.InitialService,
ws *websocket.WebSocket,
mobileServer *mobile.MobileServer) {
l.Initialize(back)
go server.Start()
go mobileServer.Start()
go ws.Start()
//go streamProxy.Start()
graceful.Wait()
})
if err != nil {
panic(err.Error())
}
}