1
1
mirror of https://github.com/adammck/terraform-inventory synced 2024-11-23 04:22:08 +01:00
terraform-inventory/parser.go
2015-08-11 16:24:11 +01:00

99 lines
2.0 KiB
Go

package main
import (
"encoding/json"
"io"
"io/ioutil"
"regexp"
)
type state struct {
Modules []moduleState `json:"modules"`
}
// keyNames contains the names of the keys to check for in each resource in the
// state file. This allows us to support multiple types of resource without too
// much fuss.
var keyNames []string
func init() {
keyNames = []string{
"ipv4_address", // DO
"public_ip", // AWS
"private_ip", // AWS
"ipaddress", // CS
"ip_address", // VMware
}
}
// read populates the state object from a statefile.
func (s *state) read(stateFile io.Reader) error {
// read statefile contents
b, err := ioutil.ReadAll(stateFile)
if err != nil {
return err
}
// parse into struct
err = json.Unmarshal(b, s)
if err != nil {
return err
}
return nil
}
// resources returns a map of name to resourceState, for any supported resources
// found in the statefile.
func (s *state) resources() map[string]resourceState {
typeRemover := regexp.MustCompile(`^[\w_]+\.`)
inst := make(map[string]resourceState)
for _, m := range s.Modules {
for k, r := range m.Resources {
if r.isSupported() {
name := typeRemover.ReplaceAllString(k, "")
inst[name] = r
}
}
}
return inst
}
type moduleState struct {
Resources map[string]resourceState `json:"resources"`
}
type resourceState struct {
Type string `json:"type"`
Primary instanceState `json:"primary"`
}
// isSupported returns true if terraform-inventory supports this resource.
func (s resourceState) isSupported() bool {
return s.Address() != ""
}
// Address returns the IP address of this resource.
func (s resourceState) Address() string {
for _, key := range keyNames {
if ip := s.Primary.Attributes[key]; ip != "" {
return ip
}
}
return ""
}
// Attributes returns a map containing everything we know about this resource.
func (s resourceState) Attributes() map[string]string {
return s.Primary.Attributes
}
type instanceState struct {
ID string `json:"id"`
Attributes map[string]string `json:"attributes,omitempty"`
}