forked from awilliams/aws-inventory
/
main.go
77 lines (68 loc) · 1.54 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
package main
import (
"flag"
"fmt"
"os"
"github.com/mitchellh/goamz/aws"
"github.com/mitchellh/goamz/ec2"
)
var args = struct {
list bool
host bool
version bool
config string
}{
false,
false,
false,
"aws-inventory.ini",
}
func init() {
flag.BoolVar(&args.list, "list", false, "Print Ansible formatted inventory")
flag.BoolVar(&args.host, "host", false, "no-op since all information is given via --list")
flag.BoolVar(&args.version, "v", false, "Print version")
flag.StringVar(&args.config, "c", args.config, "Configuration filename")
}
func main() {
flag.Parse()
switch {
case args.version:
fmt.Printf("%s v%s\n", appName, appVersion)
return
case args.host:
fmt.Fprint(os.Stdout, "{}")
return
case args.list:
cfg, err := getConfig(args.config)
if err != nil {
die("Error reading configuration file:\n%s\n", err)
}
printList(cfg)
return
}
flag.PrintDefaults()
}
func printList(cfg *configuration) {
auth, err := aws.GetAuth(cfg.AccessKey, cfg.SecretKey)
if err != nil {
die("Error creating AWS auth:\n%s\n", err)
}
e := ec2.New(auth, aws.EUWest)
instances, err := e.Instances([]string{}, nil)
if err != nil {
die("Error fetching EC2 instances:\n%s\n", err)
}
inv, err := newInventory(instances)
if err != nil {
die("Error creating inventory from EC2 instances:\n%s\n", err)
}
invJSON, err := inv.toJSON()
if err != nil {
die("Error generatin inventory JSON:\n%s\n", err)
}
os.Stdout.Write(invJSON)
}
func die(msg string, args ...interface{}) {
fmt.Fprintf(os.Stderr, msg, args...)
os.Exit(1)
}