Add network command

Author: atsushi-ishibashi

.gitignore

@@ -12,3 +12,7 @@
 
 # Project-local glide cache, RE: https://github.com/Masterminds/glide/issues/736
 .glide/
+vendor
+aws-state-report
+*.pdf
+bin

Makefile

@@ -0,0 +1,13 @@
+build: mkdir_bin build_mac build_linux build_win
+
+build_mac:
+	GOOS=darwin GOARCH=amd64 go build -o bin/aws-state-report-for-mac
+
+build_linux:
+	GOOS=linux GOARCH=amd64 go build -o bin/aws-state-report-for-linux
+
+build_win:
+	GOOS=windows GOARCH=386 go build -o bin/aws-state-report-for-win.exe
+
+mkdir_bin:
+	mkdir -p ./bin

README.md

@@ -1 +1,16 @@
 # aws-state-report
+aws-state-report export infrastructure in aws to pdf file.
+
+## Usage
+### aws-state-report deploy
+```
+$ aws-state-report network --help
+NAME:
+   aws-state-report network - export vpcs, route tables and subnets information
+
+USAGE:
+   aws-state-report network [arguments...]
+
+Examples:
+  $ aws-state-report --awsconf default network
+```

cmd/network.go

@@ -0,0 +1,237 @@
+package cmd
+
+import (
+	"fmt"
+	"math"
+
+	"github.com/atsushi-ishibashi/aws-state-report/svc"
+	"github.com/atsushi-ishibashi/aws-state-report/util"
+	"github.com/aws/aws-sdk-go/service/ec2"
+	"github.com/jung-kurt/gofpdf"
+	"github.com/urfave/cli"
+)
+
+func NewNetworkCommand() cli.Command {
+	return cli.Command{
+		Name:  "network",
+		Usage: "export vpcs, route tables and subnets information",
+		Flags: []cli.Flag{},
+		Action: func(c *cli.Context) error {
+			if err := util.ConfigAWS(c); err != nil {
+				return util.ErrorRed(err.Error())
+			}
+			mng, err := svc.NewManager()
+			if err != nil {
+				return util.ErrorRed(err.Error())
+			}
+			ntw := &Network{
+				manager: mng,
+				Errs:    make([]error, 0),
+			}
+			if err := ntw.recursiveConstruct(); err != nil {
+				return util.ErrorRed(err.Error())
+			}
+			return nil
+		},
+	}
+}
+
+type Network struct {
+	Vpcs    []*Vpc
+	manager *svc.Manager
+	Errs    []error
+}
+
+func (nt *Network) recursiveConstruct() error {
+	nt.constructVpcs().constructRouteTables().constructSubnets().associateRouteTableSubnet()
+	fmt.Printf("%#v\n", nt.Vpcs)
+	nt.convertPdf()
+	return nt.flattenErrs()
+}
+
+func (nt *Network) constructVpcs() *Network {
+	result, err := nt.manager.FetchVpcs()
+	if err != nil {
+		return nt.stackError(err)
+	}
+	nt.Vpcs = parseDescribeVpcsOutputToVpcs(result)
+	return nt
+}
+
+func (nt *Network) constructRouteTables() *Network {
+	for _, vpc := range nt.Vpcs {
+		if result, err := nt.manager.FetchRouteTablesWithVpc(vpc.ID); err != nil {
+			nt.stackError(err)
+		} else {
+			vpc.RouteTables = parseDescribeRouteTablesOutputToRouteTables(result)
+		}
+	}
+	return nt
+}
+
+func (nt *Network) constructSubnets() *Network {
+	for _, vpc := range nt.Vpcs {
+		if result, err := nt.manager.FetchSubnetsWithVpc(vpc.ID); err != nil {
+			nt.stackError(err)
+		} else {
+			vpc.Subnets = parseDescribeSubnetsOutputToSubnets(result)
+		}
+	}
+	return nt
+}
+
+func (nt *Network) associateRouteTableSubnet() *Network {
+	for _, vpc := range nt.Vpcs {
+		for _, sn := range vpc.Subnets {
+			for _, rt := range vpc.RouteTables {
+				for _, rtas := range rt.AssociationSubnets {
+					if rtas == sn.ID {
+						sn.AssociatedRouteTable = rt
+					}
+				}
+			}
+		}
+	}
+	return nt
+}
+
+func (nt *Network) convertPdf() {
+	pdf := gofpdf.New("P", "mm", "A4", "")
+	pdf.AddPage()
+	pdf.SetFont("Arial", "", 10)
+	for _, v := range nt.Vpcs {
+		pdf.CellFormat(0, 10, fmt.Sprintf("%s  %s", v.TagName, v.CidrBlock), "1", 0, "C", false, 0, "")
+		pdf.Ln(-1)
+		for _, rt := range v.RouteTables {
+			pdf.CellFormat(95, 10, fmt.Sprintf("%s", rt.TagName), "1", 0, "C", false, 0, "")
+			pdf.CellFormat(95, 10, "Association Subnets", "1", 0, "C", false, 0, "")
+			pdf.Ln(-1)
+			currentX, currentY := pdf.GetXY()
+			var rtHeight float64
+			for _, rtr := range rt.Routes {
+				pdf.MoveTo(currentX, currentY+rtHeight)
+				pdf.CellFormat(95, 10, fmt.Sprintf("%s %s", rtr.DestinationCidrBlock, rtr.Router), "RL", 0, "C", false, 0, "")
+				rtHeight += 10.0
+			}
+			var snHeight float64
+			for _, sn := range v.Subnets {
+				if sn.AssociatedRouteTable == rt {
+					pdf.MoveTo(currentX+95, currentY+snHeight)
+					pdf.CellFormat(95, 10, fmt.Sprintf("%s %s", sn.TagName, sn.CidrBlock), "RL", 0, "C", false, 0, "")
+					snHeight += 10.0
+				}
+			}
+			maxHeight := math.Max(snHeight, rtHeight)
+			pdf.MoveTo(currentX, currentY)
+			pdf.CellFormat(0, maxHeight, "", "1", 0, "C", false, 0, "")
+			pdf.Ln(-1)
+		}
+		pdf.CellFormat(0, 10, "No Association Subnets", "1", 0, "C", false, 0, "")
+		pdf.Ln(-1)
+		currentX, currentY := pdf.GetXY()
+		var noaSnHeight float64
+		for _, sn := range v.Subnets {
+			if sn.AssociatedRouteTable == nil {
+				pdf.CellFormat(0, 10, fmt.Sprintf("%s %s", sn.TagName, sn.CidrBlock), "LR", 0, "C", false, 0, "")
+				pdf.Ln(-1)
+				noaSnHeight += 10
+			}
+		}
+		pdf.MoveTo(currentX, currentY)
+		pdf.CellFormat(0, noaSnHeight, "", "1", 0, "C", false, 0, "")
+		pdf.AddPage()
+	}
+	if err := pdf.OutputFileAndClose("./network.pdf"); err != nil {
+		nt.stackError(err)
+	}
+}
+
+func (nt *Network) stackError(err error) *Network {
+	nt.Errs = append(nt.Errs, err)
+	return nt
+}
+
+func (nt *Network) flattenErrs() error {
+	if len(nt.Errs) == 0 {
+		return nil
+	}
+	var errStr string
+	for _, e := range nt.Errs {
+		errStr = errStr + e.Error() + "\n"
+	}
+	return fmt.Errorf(errStr)
+}
+
+func parseDescribeVpcsOutputToVpcs(output *ec2.DescribeVpcsOutput) []*Vpc {
+	vs := make([]*Vpc, 0)
+	for _, v := range output.Vpcs {
+		vpc := &Vpc{
+			ID:        *v.VpcId,
+			TagName:   extractTagName(v.Tags),
+			CidrBlock: *v.CidrBlock,
+		}
+		acbs := make([]string, 0)
+		for _, cbs := range v.CidrBlockAssociationSet {
+			acbs = append(acbs, *cbs.CidrBlock)
+		}
+		vpc.AssociatedCidrBlocks = acbs
+		vs = append(vs, vpc)
+	}
+	return vs
+}
+
+func parseDescribeRouteTablesOutputToRouteTables(output *ec2.DescribeRouteTablesOutput) []*RouteTable {
+	rts := make([]*RouteTable, 0)
+	for _, v := range output.RouteTables {
+		rt := &RouteTable{
+			ID:      *v.RouteTableId,
+			TagName: extractTagName(v.Tags),
+		}
+		rs := make([]*Route, 0)
+		for _, r := range v.Routes {
+			if r.DestinationCidrBlock == nil {
+				continue
+			}
+			rr := &Route{
+				DestinationCidrBlock: *r.DestinationCidrBlock,
+			}
+			var routerID string
+			if r.GatewayId != nil {
+				routerID = *r.GatewayId
+			}
+			if r.NatGatewayId != nil {
+				routerID = *r.NatGatewayId
+			}
+			if r.VpcPeeringConnectionId != nil {
+				routerID = *r.VpcPeeringConnectionId
+			}
+			rr.Router = routerID
+			rs = append(rs, rr)
+		}
+		rt.Routes = rs
+		asSubnets := make([]string, 0)
+		for _, as := range v.Associations {
+			if as.SubnetId != nil {
+				asSubnets = append(asSubnets, *as.SubnetId)
+			} else {
+				asSubnets = append(asSubnets, "implicit")
+			}
+		}
+		rt.AssociationSubnets = asSubnets
+		rts = append(rts, rt)
+	}
+	return rts
+}
+
+func parseDescribeSubnetsOutputToSubnets(output *ec2.DescribeSubnetsOutput) []*Subnet {
+	subnets := make([]*Subnet, 0)
+	for _, v := range output.Subnets {
+		sn := &Subnet{
+			ID:        *v.SubnetId,
+			TagName:   extractTagName(v.Tags),
+			CidrBlock: *v.CidrBlock,
+		}
+		subnets = append(subnets, sn)
+	}
+	return subnets
+}

cmd/network_model.go

@@ -0,0 +1,29 @@
+package cmd
+
+type Vpc struct {
+	ID                   string
+	TagName              string
+	CidrBlock            string
+	AssociatedCidrBlocks []string
+	RouteTables          []*RouteTable
+	Subnets              []*Subnet
+}
+
+type RouteTable struct {
+	ID                 string
+	TagName            string
+	Routes             []*Route
+	AssociationSubnets []string //subnet-id
+}
+
+type Route struct {
+	DestinationCidrBlock string
+	Router               string
+}
+
+type Subnet struct {
+	ID                   string
+	TagName              string
+	CidrBlock            string
+	AssociatedRouteTable *RouteTable
+}

cmd/util.go

@@ -0,0 +1,13 @@
+package cmd
+
+import "github.com/aws/aws-sdk-go/service/ec2"
+
+func extractTagName(tags []*ec2.Tag) string {
+	var name string
+	for _, tg := range tags {
+		if *tg.Key == "Name" {
+			name = *tg.Value
+		}
+	}
+	return name
+}

glide.lock

@@ -0,0 +1,8 @@
+package: github.com/atsushi-ishibashi/aws-state-report
+import:
+- package: github.com/jung-kurt/gofpdf
+  version: ~1.0.0
+- package: github.com/aws/aws-sdk-go
+  version: ~1.10.39
+- package: github.com/urfave/cli
+  version: ~1.20.0