forked from rancher/rancher-compose
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
98 lines (87 loc) · 1.98 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
package main
import (
"fmt"
"os"
log "github.com/Sirupsen/logrus"
"github.com/codegangsta/cli"
"github.com/rancherio/go-rancher/client"
"github.com/rancherio/rancher-compose/project"
)
func main() {
app := cli.NewApp()
app.Name = "rancher-compose"
app.Usage = "Docker-compose to Rancher"
app.Version = "0.1.0"
app.Author = "Rancher"
app.Email = ""
app.Flags = []cli.Flag{
cli.StringFlag{
Name: "api-url",
Usage: fmt.Sprintf(
"Specify the Rancher API Endpoint URL",
),
},
cli.StringFlag{
Name: "access-key",
Usage: fmt.Sprintf(
"Specify api access key",
),
},
cli.StringFlag{
Name: "secret-key",
Usage: fmt.Sprintf(
"Specify api secret key",
),
EnvVar: "RANCHER_SECRET_KEY",
},
cli.StringFlag{
Name: "f",
Usage: "docker-compose yml file to use",
Value: "docker-compose.yml",
},
}
app.Commands = []cli.Command{
{
Name: "up",
Usage: "Bring all services up",
Action: func(c *cli.Context) {
prj := getProject(c)
err := prj.StartAllServices()
if err != nil {
log.Fatalf("Died trying to create Servers")
}
},
},
{
Name: "rm",
Usage: "Remove all containers and services",
Action: func(c *cli.Context) {
prj := getProject(c)
err := prj.RmAllServices()
if err != nil {
log.Fatal("Could not remove all services. %s", err)
}
},
},
}
app.Run(os.Args)
}
func getRancherClient(c *cli.Context) *client.RancherClient {
url := c.GlobalString("api-url")
accessKey := c.GlobalString("access-key")
secretKey := c.GlobalString("secret-key")
rClient, err := GetRancherClient(url, accessKey, secretKey)
if err != nil {
log.Fatalf("Unable to get Rancher client: %s", err)
}
return rClient
}
func getProject(c *cli.Context) *project.Project {
filename := c.GlobalString("f")
rClient := getRancherClient(c)
prj, err := project.NewProject("rc", filename, rClient)
if err != nil {
log.Fatalf("Could not create project from file. %v", filename, err)
}
return prj
}