add peek and pop operations

.gitignore

@@ -3,4 +3,8 @@ bin
 pkg
 
 # IDK where this comes from
-src/uri.txt
+uri.txt
+
+# VIM
+*~
+*.swp

src/azq/main.go

@@ -16,6 +16,8 @@ Usage:
 	azq tp [ -F configFile ] [ -e environment ] <queueName>
 	azq profile [ -F configFile ] [ -e environment ] <queueName> [<duration>]
 	azq put [ -F configFile ] [ -e environment ] <queueName> <message>
+	azq peek [ -F configFile ] [ -e environment ] <queueName>
+	azq pop [ -F configFile ] [ -e environment ] <queueName>
 Arguments:
 	queueName    The name(s) of one or more queues
 	queuePrefix  A prefix for filtering which queues to show
@@ -87,6 +89,16 @@ func doIt(dict map[string]interface{}) {
 		message := dict["<message>"].(string)
 		engineroom.Put(queueName, message)
 	}
+
+	if dict["pop"].(bool) {
+		queueName := dict["<queueName>"].(string)
+		engineroom.Pop(queueName)
+	}
+
+	if dict["peek"].(bool) {
+		queueName := dict["<queueName>"].(string)
+		engineroom.Peek(queueName)
+	}
 }
 
 //To do: get a better handle on what docopt.Parse arguments do.

src/engineroom/peek.go

@@ -0,0 +1,35 @@
+package engineroom
+
+import (
+	"encoding/base64"
+	"fmt"
+	"github.com/nadroz/azure-sdk-for-go/storage"
+)
+
+func Peek(queueName string) {
+	client := getStorageClient()
+	params := storage.PeekMessagesParameters{
+		NumOfMessages: 1,
+	}
+
+	messages, err := client.PeekMessages(queueName, params)
+	if err != nil {
+		fmt.Printf("Failed to peek messages: %s\n", err)
+		return
+	}
+
+	if len(messages.QueueMessagesList) != 1 {
+		return
+	}
+
+	msg := messages.QueueMessagesList[0]
+
+	txt, err := base64.StdEncoding.DecodeString(msg.MessageText)
+
+	if err != nil {
+		fmt.Printf("Failed to decode message: %s\n", err)
+		return
+	}
+
+	fmt.Printf("%s\n", txt)
+}

src/engineroom/pop.go

@@ -0,0 +1,42 @@
+package engineroom
+
+import (
+	"encoding/base64"
+	"fmt"
+	"github.com/nadroz/azure-sdk-for-go/storage"
+)
+
+func Pop(queueName string) {
+	client := getStorageClient()
+	params := storage.GetMessagesParameters{
+		NumOfMessages:     1,
+		VisibilityTimeout: 30,
+	}
+
+	messages, err := client.GetMessages(queueName, params)
+	if err != nil {
+		fmt.Printf("Failed to pop message: %s\n", err)
+		return
+	}
+
+	if len(messages.QueueMessagesList) != 1 {
+		return
+	}
+
+	msg := messages.QueueMessagesList[0]
+
+	err = client.DeleteMessage(queueName, msg.MessageID, msg.PopReceipt)
+
+	if err != nil {
+		fmt.Printf("Failed to pop message: %s\n", err)
+	}
+
+	txt, err := base64.StdEncoding.DecodeString(msg.MessageText)
+
+	if err != nil {
+		fmt.Printf("Failed to decode message: %s\n", err)
+		return
+	}
+
+	fmt.Printf("%s\n", txt)
+}