Merge pull request #77 from sivchari/add-quote

add quote

flow/flow.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"os"
+	"strconv"
 	"strings"
 
 	"github.com/sakajunquality/cloud-pubsub-events/gcrevent"
@@ -32,7 +33,7 @@ func New(c *Config) (*Flow, error) {
 	return f, nil
 }
 
-func (f *Flow) ProcessGCREvent(ctx context.Context, e gcrevent.Event) error {
+func (f *Flow) ProcessGCREvent(ctx context.Context, e gcrevent.Event, quote bool) error {
 	if e.Action != gcrevent.ActionInsert {
 		return nil
 	}
@@ -51,5 +52,9 @@ func (f *Flow) ProcessGCREvent(ctx context.Context, e gcrevent.Event) error {
 		return fmt.Errorf("image format invalid: %s", *e.Tag)
 	}
 
+	if quote {
+		version = strconv.Quote(version)
+	}
+
 	return f.processImage(ctx, image, version)
 }

main.go

@@ -3,6 +3,7 @@ package main
 import (
 	"context"
 	"encoding/json"
+	"flag"
 	"fmt"
 	"io"
 	"log"
@@ -38,7 +39,14 @@ var (
 	f *flow.Flow
 )
 
+var (
+	q *bool
+)
+
 func main() {
+	q = flag.Bool("q", true, "if true, the version is quoted. Default is true.")
+	flag.Parse()
+
 	cfg, err := getConfig()
 	if err != nil {
 		fmt.Fprintf(os.Stderr, "Cloud not read the file:%s.\n", err)
@@ -101,7 +109,7 @@ func handlePubSubMessage(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	err = f.ProcessGCREvent(ctx, event)
+	err = f.ProcessGCREvent(ctx, event, *q)
 	log.Printf("process: %s", err)
 
 	res := &Response{