forked from gabrtv/deisctl
/
deisctl.go
130 lines (120 loc) · 3.6 KB
/
deisctl.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
package main
import (
"fmt"
"os"
"strconv"
"github.com/deis/deisctl/backend/fleet"
"github.com/deis/deisctl/client"
"github.com/deis/deisctl/utils"
docopt "github.com/docopt/docopt-go"
)
const (
// Version of deisctl client
Version string = "0.13.0-dev"
)
func exit(err error, code int) {
fmt.Printf("Error: %v\n", err)
os.Exit(code)
}
func setGlobalFlags(args map[string]interface{}) {
fleet.Flags.Endpoint = args["--endpoint"].(string)
fleet.Flags.EtcdKeyPrefix = args["--etcd-key-prefix"].(string)
fleet.Flags.EtcdKeyFile = args["--etcd-keyfile"].(string)
fleet.Flags.EtcdCertFile = args["--etcd-certfile"].(string)
fleet.Flags.EtcdCAFile = args["--etcd-cafile"].(string)
//fleet.Flags.UseAPI = args["--experimental-api"].(bool)
fleet.Flags.KnownHostsFile = args["--known-hosts-file"].(string)
fleet.Flags.StrictHostKeyChecking = args["--strict-host-key-checking"].(bool)
timeout, _ := strconv.ParseFloat(args["--request-timeout"].(string), 64)
fleet.Flags.RequestTimeout = timeout
tunnel := args["--tunnel"].(string)
if tunnel != "" {
fleet.Flags.Tunnel = tunnel
} else {
fleet.Flags.Tunnel = os.Getenv("DEISCTL_TUNNEL")
}
}
func main() {
deisctlMotd := utils.DeisIfy("Deis Control Utility")
usage := deisctlMotd + `
Usage:
deisctl <command> [<target>...] [options]
Commands:
deisctl install [<service> | platform]
deisctl uninstall [<service> | platform]
deisctl list
deisctl scale [<service>=<num>]
deisctl start [<service> | platform]
deisctl stop [<service> | platform]
deisctl restart [<service> | platform]
deisctl journal <service>
deisctl config <component> <get|set> <args>
deisctl update
deisctl refresh-units
Example Commands:
deisctl install platform
deisctl uninstall builder@1
deisctl scale router=2
deisctl start router@2
deisctl stop router builder
deisctl status controller
deisctl journal controller
Options:
--version print version and exit
--endpoint=<url> etcd endpoint for fleet [default: http://127.0.0.1:4001]
--etcd-key-prefix=<path> keyspace for fleet data in etcd [default: /_coreos.com/fleet/]
--etcd-keyfile=<path> etcd key file authentication [default: ]
--etcd-certfile=<path> etcd cert file authentication [default: ]
--etcd-cafile=<path> etcd CA file authentication [default: ]
--known-hosts-file=<path> file used to store remote machine fingerprints [default: ~/.ssh/known_hosts]
--strict-host-key-checking verify SSH host keys [default: true]
--tunnel=<host> establish an SSH tunnel for communication with fleet and etcd [default: ]
--request-timeout=<secs> amount of time to allow a single request before considering it failed. [default: 3.0]
`
// parse command-line arguments
args, err := docopt.Parse(usage, nil, true, Version, true)
if err != nil {
exit(err, 2)
}
command := args["<command>"]
targets := args["<target>"].([]string)
setGlobalFlags(args)
// construct a client
c, err := client.NewClient("fleet")
if err != nil {
exit(err, 1)
}
// dispatch the command
switch command {
case "list":
err = c.List()
case "scale":
err = c.Scale(targets)
case "start":
err = c.Start(targets)
case "restart":
err = c.Restart(targets)
case "stop":
err = c.Stop(targets)
case "status":
err = c.Status(targets)
case "journal":
err = c.Journal(targets)
case "install":
err = c.Install(targets)
case "uninstall":
err = c.Uninstall(targets)
case "config":
err = c.Config()
case "update":
err = c.Update()
case "refresh-units":
err = c.RefreshUnits()
default:
fmt.Printf(usage)
os.Exit(2)
}
if err != nil {
exit(err, 1)
}
}