/
main.go
110 lines (98 loc) · 2.59 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
100
101
102
103
104
105
106
107
108
109
110
package main
import (
"log"
"io/ioutil"
"github.com/codegangsta/cli"
"gopkg.in/yaml.v2"
"fmt"
"github.com/jmcvetta/napping"
)
type ServiceManifest struct {
Image string `json:"image"`
Expose int `json:"expose,omitempty"`
Environment map[string]string `json:"environment,omitempty"`
Require struct {
Cpu int `json:"cpu,omitempty"`
Mem int `json:"mem,omitempty"`
} `json:"require,omitempty"`
}
func registerCommit(c *cli.Context, deployKey, commitSha, image string) {
url := fmt.Sprintf("http://%s/deploy/register", c.GlobalString("api-endpoint"))
var payload struct {
DeployKey string `json:"deploy_key"`
CommitSha string `json:"commit_sha"`
Manifest ServiceManifest `json:"manifest"`
}
payload.DeployKey = deployKey
payload.CommitSha = commitSha
data, err := ioutil.ReadFile(c.String("manifest-file"))
if err != nil {
log.Panic("mikro-cli: register-commit: cannot read manifest: ", err)
}
err = yaml.Unmarshal(data, &payload.Manifest)
if err != nil {
log.Panic("mikro-cli: register-commit: cannot decode manifest: ", err)
}
payload.Manifest.Image = image
_, err = napping.Post(url, &payload, nil, nil)
if err != nil {
log.Panic("mikro-cli: register-commit: register failed: ", err)
}
}
func cmdRegisterCommit(c *cli.Context) {
registerCommit(c, c.String("deploy-key"), c.String("commit-sha"),
c.Args().First())
}
func main() {
app := cli.NewApp()
app.Name = "mikro-cli"
app.Usage = "command-line interface to mikro.io"
app.Action = func(c *cli.Context) {
println("boom! I say!")
}
app.Flags = []cli.Flag{
cli.StringFlag{
Name: "api-endpoint",
Value: "api.mikro.io",
},
}
app.Commands = []cli.Command{
{
Name: "register-commit",
Action: cmdRegisterCommit,
Usage: "register an image and manifest with a commit",
Flags: []cli.Flag {
cli.StringFlag{
Name: "deploy-key, k",
EnvVar: "MIKRO_DEPLOY_KEY",
},
cli.StringFlag{
Name: "commit-sha, s",
},
cli.StringFlag{
Name: "manifest-file",
Value: "mikro.yml",
},
},
},
{
Name: "push",
Action: cmdPush,
Usage: "push image and register commit",
Flags: []cli.Flag {
cli.StringFlag{
Name: "deploy-key, k",
EnvVar: "MIKRO_DEPLOY_KEY",
},
cli.StringFlag{
Name: "commit-sha, s",
},
cli.StringFlag{
Name: "manifest-file",
Value: "mikro.yml",
},
},
},
}
app.RunAndExitOnError()
}