1
1
mirror of https://github.com/adammck/terraform-inventory synced 2024-11-22 20:01:58 +01:00

Merge pull request #28 from anarcher/master

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

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