forked from flynn/flynn
/
init.go
47 lines (40 loc) · 1.17 KB
/
init.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
package cli
import (
"fmt"
"github.com/flynn/flynn/bootstrap/discovery"
"github.com/flynn/flynn/host/config"
"github.com/flynn/go-docopt"
)
func init() {
Register("init", runInit, `
usage: flynn-host init [options]
options:
--init-discovery create and join a discovery token
--discovery=TOKEN join cluster with discovery token
--peer-ips=IPLIST join cluster using host IPs (must be already bootstrapped)
--external-ip=IP external IP address of host, defaults to the first IPv4 address of eth0
--file=NAME file to write to [default: /etc/flynn/host.json]
`)
}
func runInit(args *docopt.Args) error {
c := config.New()
discoveryToken := args.String["--discovery"]
if args.Bool["--init-discovery"] {
var err error
discoveryToken, err = discovery.NewToken()
if err != nil {
return err
}
fmt.Println(discoveryToken)
}
if discoveryToken != "" {
c.Args = append(c.Args, "--discovery", discoveryToken)
}
if ip := args.String["--external-ip"]; ip != "" {
c.Args = append(c.Args, "--external-ip", ip)
}
if ips := args.String["--peer-ips"]; ips != "" {
c.Args = append(c.Args, "--peer-ips", ips)
}
return c.WriteTo(args.String["--file"])
}