Added replace helper

helpers.go

@@ -26,6 +26,23 @@ func lastPart(splitter, input interface{}) interface{} {
 	return parts[len(parts)-1]
 }
 
+func replace(old, new, input interface{}) interface{} {
+	newString := realString(new)
+	if newString == "" {
+		return input
+	}
+	oldString := realString(old)
+	if oldString == "" {
+		return input
+	}
+	inputString := realString(input)
+	if inputString == "" {
+		return input
+	}
+
+	return strings.Replace(inputString, oldString, newString, -1)
+}
+
 func splitCammelJoin(splitter, input interface{}) interface{} {
 	inputString := realString(input)
 	if inputString == "" {

helpers_test.go

@@ -31,6 +31,29 @@ func TestSplitHelper(t *testing.T) {
 	}
 }
 
+func TestReplaceHelper(t *testing.T) {
+
+	tests := map[string]string{
+		`{{ "protocol.buffers" | replace "." "/" }}`:                          "protocol/buffers",
+		`{{ "goodbye.world" | replace "goodbye" "hello" | replace "." " " }}`: "hello world",
+	}
+
+	for templateData, expected := range tests {
+		funcMap := template.FuncMap{
+			"replace": replace,
+		}
+
+		tpl := template.Must(template.New("test").Funcs(funcMap).Parse(templateData))
+		var output bytes.Buffer
+		if err := tpl.Execute(&output, nil); err != nil {
+			t.Fatalf("%v", err)
+		}
+		if output.String() != expected {
+			t.Fatalf("Expected '%s' but got '%s'", expected, output.String())
+		}
+	}
+}
+
 func TestSplitCammelJoin(t *testing.T) {
 	tests := map[string]string{
 		`{{ "nick.gerakines" | splitCammelJoin "." }}`:          "NickGerakines",

plugin.go

@@ -54,6 +54,7 @@ func RunPlugin(request *plugin_go.CodeGeneratorRequest) (*plugin_go.CodeGenerato
 		"lastPart":        lastPart,
 		"splitCammelJoin": splitCammelJoin,
 		"hasServices":     hasServices,
+		"replace":         replace,
 	}
 
 	resp := new(plugin_go.CodeGeneratorResponse)