Added dynamic configuration for search results fields

Author: pattonjp

README.md

@@ -2,17 +2,21 @@
 
 Shell into any server in your aws account.
 
-#Search
+Search
+====
 calling the search command will look up all ec2 instances then filter the results by the term entered. if only a single result the app will ssh directly. If multiple results returned the app will display a numbered list and prompt for which to connect.
 
+
 ```
 aws-ssh search {{search term}}
 ```
 
-#Connect
+Connect
+=====
 connect allows you to connect directly to a known server. (team member running windows asked for this.)
 
-#Configuration
+Configuration
+======
 
 you  will need to specify aws credentials. see [AWS Credentials](http://blogs.aws.amazon.com/security/post/Tx3D6U6WSFGOK2H/A-New-and-Standardized-Way-to-Manage-Credentials-in-the-AWS-SDKs) for details.
 
@@ -24,5 +28,18 @@ aws_access_key_id = ACCESS_KEY
 aws_secret_access_key = SECRET_KEY
 ```
 
+#### Config File
+by setting a `~/.aws-ssh.yml` file you can set additional fields to the search
+results.
+
+example `.aws-ssh.yml` file:
+```markdown
+tags:
+  - env
+  - name
+  - role
+```
+
+
 ##PEM keys
 if setting a key add the name only expected to be at ~/.ssh/{{KEY_NAME}}.pem

cmd/versionCmd.go

@@ -10,7 +10,7 @@ import (
 	"github.com/spf13/cobra"
 )
 
-const Version = "0.0.10"
+const Version = "0.0.11"
 
 // versionCmd represents the version command
 var versionCmd = &cobra.Command{

modules/instanceCollection.go

@@ -10,6 +10,7 @@ import (
 	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/aws/aws-sdk-go/service/ec2"
 	"github.com/olekukonko/tablewriter"
+	"github.com/spf13/viper"
 )
 
 //Instances Instance Collection type
@@ -106,17 +107,24 @@ func (instances Instances) FilterByName(name string) Instances {
 //CreateTable creates a formated table of instances
 func (instances Instances) CreateTable(writer io.Writer) {
 	table := tablewriter.NewWriter(writer)
-	table.SetHeader([]string{"Num", "Id", "State", "Public", "Private", "env", "role"})
+	tags := viper.GetStringSlice("tags")
+	fmt.Println(tags)
+	headers := []string{"Num", "Id", "State", "Public", "Private"}
+	headers = append(headers, tags...)
+	table.SetHeader(headers)
 
 	for pos, inst := range instances {
-		table.Append([]string{strconv.Itoa(pos),
+		row := []string{strconv.Itoa(pos),
 			inst.Id,
 			inst.State,
 			inst.Host,
 			inst.PrivateIp,
-			inst.GetKey("env"),
-			inst.GetKey("role"),
-		})
+		}
+		for _, tag := range tags {
+			row = append(row, inst.GetKey(tag))
+		}
+		// row = append(row, tags...)
+		table.Append(row)
 		// fmt.Println(pos, "  ", inst.Name, " ", inst.State, " ", inst.Host)
 	}
 	table.Render()