Merge pull request #2 from marcinkoziej/feature/ssl-option

Add -S --ssl flag to use SSL (amqps)

cmd/consume.go

@@ -3,7 +3,6 @@ package cmd
 import (
 	"fmt"
 	"reflect"
-	"strconv"
 	"strings"
 
 	"github.com/oleiade/reflections"
@@ -36,8 +35,8 @@ Use comma-separated values for binding the same queue with multiple routing keys
 		cmd.SilenceUsage = true
 		cmd.SilenceErrors = true
 
+		uri := getUri()
 		// Dial amqp server
-		uri := "amqp://" + username + ":" + password + "@" + host + ":" + strconv.Itoa(port) + vhost
 		conn, err := amqp.Dial(uri)
 		if err != nil {
 			return fmt.Errorf("connection.open: %v", err)

cmd/produce.go

@@ -5,7 +5,6 @@ import (
 	"io/ioutil"
 	"os"
 	"reflect"
-	"strconv"
 	"time"
 
 	"github.com/spf13/cobra"
@@ -46,7 +45,7 @@ To pass headers and properites, use '--headers' & '--properties' any number of t
 			message = string(bytes)
 		}
 
-		uri := "amqp://" + username + ":" + password + "@" + host + ":" + strconv.Itoa(port) + vhost
+		uri := getUri()
 		conn, err := amqp.Dial(uri)
 		if err != nil {
 			return fmt.Errorf("connection.open: %v", err)

cmd/root.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"os"
 	"strings"
+	"strconv"
 
 	"github.com/spf13/cobra"
 	"github.com/spf13/pflag"
@@ -18,6 +19,7 @@ var (
 	vhost    string
 	username string
 	password string
+    ssl      bool
 
 	// exchange options
 	exchange        string
@@ -34,6 +36,20 @@ var (
 	durableQueue bool
 )
 
+func getUri() string {
+	var proto string = "amqp://"
+	if (ssl) {
+		proto = "amqps://"
+		// Change the port from amqp default to amqps default.
+		// not sure how to check if -P flag was given by the user
+		// so the perverse situtation where amqps runs on port 5672 would not work sorry
+		if (port == 5672) {
+			port = 5671
+		}
+	}
+	return proto + username + ":" + password + "@" + host + ":" + strconv.Itoa(port) + vhost
+}
+
 var valid_properties = map[string]string{
 	"content-type":     "ContentType",
 	"content-encoding": "ContentEncoding",
@@ -93,6 +109,7 @@ func commonFlagSet() *pflag.FlagSet {
 	fs.StringVarP(&vhost, "vhost", "v", "/", "specify vhost")
 	fs.StringVarP(&username, "username", "u", "guest", "specify username")
 	fs.StringVarP(&password, "password", "p", "guest", "specify password")
+	fs.BoolVarP(&ssl, "ssl", "S", false, "use amqps")
 
 	fs.StringVarP(&exchange, "exchange", "e", "", `exchange name (default "")`)
 	fs.StringVarP(&routingkey, "key", "k", "", `routing key (default "")`)