From 38348b15330bb34dedd30f1ccc979a4cf1b940a6 Mon Sep 17 00:00:00 2001
From: Ted Unangst <tedu@tedunangst.com>
Date: Sat, 28 Mar 2020 22:49:34 -0400
Subject: [PATCH] fix some bugs involving multiple code blocks

---
 markitzero.go | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/markitzero.go b/markitzero.go
index c524794..36a50b8 100644
--- a/markitzero.go
+++ b/markitzero.go
@@ -43,11 +43,13 @@ func markitzero(s string) string {
 	s = strings.TrimSpace(s)
 	s = strings.Replace(s, "\r", "", -1)
 
+	codeword := "`elided big code`"
+
 	// save away the code blocks so we don't mess them up further
 	var bigcodes, lilcodes, images []string
 	s = re_bigcoder.ReplaceAllStringFunc(s, func(code string) string {
 		bigcodes = append(bigcodes, code)
-		return "``````"
+		return codeword
 	})
 	s = re_coder.ReplaceAllStringFunc(s, func(code string) string {
 		lilcodes = append(lilcodes, code)
@@ -146,15 +148,16 @@ func markitzero(s string) string {
 	s = re_coder.ReplaceAllStringFunc(s, func(string) string {
 		code := lilcodes[0]
 		lilcodes = lilcodes[1:]
+		if code == codeword && len(bigcodes) > 0 {
+			code := bigcodes[0]
+			bigcodes = bigcodes[1:]
+			m := re_bigcoder.FindStringSubmatch(code)
+			return "<pre><code>" + lighter.HighlightString(m[2], m[1]) + "</code></pre><p>"
+		}
 		code = html.EscapeString(code)
 		return code
 	})
-	s = re_bigcoder.ReplaceAllStringFunc(s, func(string) string {
-		code := bigcodes[0]
-		bigcodes = bigcodes[1:]
-		m := re_bigcoder.FindStringSubmatch(code)
-		return "<pre><code>" + lighter.HighlightString(m[2], m[1]) + "</code></pre><p>"
-	})
+
 	s = re_coder.ReplaceAllString(s, "<code>$1</code>")
 
 	// some final fixups