Escape spaces and hyphens when generating cli filenames

badges · Aug 11, 2013 · 8e6e740 · 8e6e740
1 parent a34e872
commit 8e6e740
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -16,6 +16,7 @@ import (
 
 var (
 	wsReplacer = strings.NewReplacer("__", "_", "_", " ")
+	revWsReplacer = strings.NewReplacer(" ", "_", "_", "__", "-", "--")
 
 	// set last modifed to server startup. close enough to release.
 	lastModified    = time.Now()
@@ -137,8 +138,8 @@ func cliMode(vendor string, status string, color string, args []string) {
 		}
 		d := Data{vendor, status, c}
 
-		// XXX could escape here
-		name := vendor + "-" + status + "-" + color + ".png"
+		name := fmt.Sprintf("%s-%s-%s.png", revWsReplacer.Replace(vendor),
+			revWsReplacer.Replace(status), color);
 
 		if len(args) > 1 {
 			fatal("You can only specify one output file name")