mirror/terraform-inventory
1
1
Fork 0
mirror of https://github.com/adammck/terraform-inventory synced 2024-11-23 04:22:08 +01:00
Code Issues Releases Activity

Merge pull request #28 from anarcher/master

Browse Source 
Add ENV TERRAFORM_INVENTORY_KEYNAME to specify which IP to return
This commit is contained in:
Adam Mckaig 2016-02-19 22:35:56 -05:00
commit 35e77604ba
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
resource.go
View File

@ -2,6 +2,7 @@ package main
import ( import (
"fmt" "fmt"
"os"
"regexp" "regexp"
"strconv" "strconv"
"strings" "strings"
@ -133,10 +134,16 @@ func (r Resource) NameWithCounter() string {
// Address returns the IP address of this resource. // Address returns the IP address of this resource.
func (r Resource) Address() string { func (r Resource) Address() string {
for _, key := range keyNames { if keyName := os.Getenv("TF_KEY_NAME"); keyName != "" {
if ip := r.State.Primary.Attributes[key]; ip != "" { if ip := r.State.Primary.Attributes[keyName]; ip != "" {
return ip return ip
} }
} else {
for _, key := range keyNames {
if ip := r.State.Primary.Attributes[key]; ip != "" {
return ip
}
}
} }
return "" return ""