From 21f81935e251459f5de97ebfb390c147a1f284f3 Mon Sep 17 00:00:00 2001
From: Eric Tune <etune@google.com>
Date: Mon, 7 Dec 2015 14:16:51 -0800
Subject: [PATCH] examples munger allows any file extension.

---
 cmd/mungedocs/example_syncer.go      | 11 +++++++++--
 cmd/mungedocs/example_syncer_test.go |  6 ++++++
 cmd/mungedocs/testdata/example.txt   |  1 +
 3 files changed, 16 insertions(+), 2 deletions(-)
 create mode 100644 cmd/mungedocs/testdata/example.txt

diff --git a/cmd/mungedocs/example_syncer.go b/cmd/mungedocs/example_syncer.go
index ba153b3c46d..bc755356f07 100644
--- a/cmd/mungedocs/example_syncer.go
+++ b/cmd/mungedocs/example_syncer.go
@@ -27,7 +27,7 @@ const exampleToken = "EXAMPLE"
 
 const exampleLineStart = "<!-- BEGIN MUNGE: EXAMPLE"
 
-var exampleMungeTagRE = regexp.MustCompile(beginMungeTag(fmt.Sprintf("%s %s", exampleToken, `(([^ ])*.(yaml|json))`)))
+var exampleMungeTagRE = regexp.MustCompile(beginMungeTag(fmt.Sprintf("%s %s", exampleToken, `(([^ ])*[.]([^.]*))`)))
 
 // syncExamples updates all examples in markdown file.
 //
@@ -77,7 +77,14 @@ func syncExamples(filePath string, mlines mungeLines) (mungeLines, error) {
 	}
 	// update all example Tags
 	for _, tag := range exampleTags {
-		example, err := exampleContent(filePath, tag.linkText, tag.fileType)
+		ft := ""
+		if tag.fileType == "json" {
+			ft = "json"
+		}
+		if tag.fileType == "yaml" {
+			ft = "yaml"
+		}
+		example, err := exampleContent(filePath, tag.linkText, ft)
 		if err != nil {
 			return mlines, err
 		}
diff --git a/cmd/mungedocs/example_syncer_test.go b/cmd/mungedocs/example_syncer_test.go
index 72c1514a1ff..9b53a6d0f7d 100644
--- a/cmd/mungedocs/example_syncer_test.go
+++ b/cmd/mungedocs/example_syncer_test.go
@@ -33,6 +33,8 @@ spec:
     image: nginx
     ports:
     - containerPort: 80
+`
+	var textExample = `some text
 `
 	var cases = []struct {
 		in       string
@@ -47,6 +49,10 @@ spec:
 			"<!-- BEGIN MUNGE: EXAMPLE ../mungedocs/testdata/pod.yaml -->\n<!-- END MUNGE: EXAMPLE ../mungedocs/testdata/pod.yaml -->\n",
 			"<!-- BEGIN MUNGE: EXAMPLE ../mungedocs/testdata/pod.yaml -->\n\n```yaml\n" + podExample + "```\n\n[Download example](../mungedocs/testdata/pod.yaml?raw=true)\n<!-- END MUNGE: EXAMPLE ../mungedocs/testdata/pod.yaml -->\n",
 		},
+		{
+			"<!-- BEGIN MUNGE: EXAMPLE testdata/example.txt -->\n<!-- END MUNGE: EXAMPLE testdata/example.txt -->\n",
+			"<!-- BEGIN MUNGE: EXAMPLE testdata/example.txt -->\n\n```\n" + textExample + "```\n\n[Download example](testdata/example.txt?raw=true)\n<!-- END MUNGE: EXAMPLE testdata/example.txt -->\n",
+		},
 	}
 	repoRoot = ""
 	for _, c := range cases {
diff --git a/cmd/mungedocs/testdata/example.txt b/cmd/mungedocs/testdata/example.txt
new file mode 100644
index 00000000000..7b57bd29ea8
--- /dev/null
+++ b/cmd/mungedocs/testdata/example.txt
@@ -0,0 +1 @@
+some text