From f0814d4781946707fc04aab5199ad232e6ec265d Mon Sep 17 00:00:00 2001
From: Ted Unangst <tedu@tedunangst.com>
Date: Tue, 26 May 2020 01:42:34 -0400
Subject: [PATCH] pull in updates from flak

---
 markitzero.go      | 12 ++++++++++--
 markitzero_test.go |  6 +++---
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/markitzero.go b/markitzero.go
index dfe263d..25da646 100644
--- a/markitzero.go
+++ b/markitzero.go
@@ -35,10 +35,12 @@ var re_zerolink = regexp.MustCompile(`\[([^]]*)\]\(([^)]*\)?)\)`)
 var re_imgfix = regexp.MustCompile(`<img ([^>]*)>`)
 var re_lister = regexp.MustCompile(`((^|\n)(\+|-).*)+\n?`)
 var re_tabler = regexp.MustCompile(`((^|\n)\|.*)+\n?`)
-var re_header = regexp.MustCompile(`(^|\n)(#+) (.*)`)
+var re_header = regexp.MustCompile(`(^|\n)(#+) (.*)\n?`)
 
 var lighter = synlight.New(synlight.Options{Format: synlight.HTML})
 
+var allowInlineHtml = false
+
 func markitzero(s string) string {
 	// prepare the string
 	s = strings.TrimSpace(s)
@@ -137,6 +139,7 @@ func markitzero(s string) string {
 		return r.String()
 	})
 	s = re_header.ReplaceAllStringFunc(s, func(s string) string {
+		s = strings.TrimSpace(s)
 		m := re_header.FindStringSubmatch(s)
 		num := len(m[2])
 		return fmt.Sprintf("<h%d>%s</h%d><p>", num, m[3], num)
@@ -150,6 +153,9 @@ func markitzero(s string) string {
 		return img
 	})
 
+	s = strings.Replace(s, "\n\n", "<p>", -1)
+	s = strings.Replace(s, "\n", "<br>", -1)
+
 	// now restore the code blocks
 	s = re_coder.ReplaceAllStringFunc(s, func(string) string {
 		code := lilcodes[0]
@@ -158,6 +164,9 @@ func markitzero(s string) string {
 			code := bigcodes[0]
 			bigcodes = bigcodes[1:]
 			m := re_bigcoder.FindStringSubmatch(code)
+			if allowInlineHtml && m[1] == "inlinehtml" {
+				return m[2]
+			}
 			return "<pre><code>" + lighter.HighlightString(m[2], m[1]) + "</code></pre><p>"
 		}
 		code = html.EscapeString(code)
@@ -167,7 +176,6 @@ func markitzero(s string) string {
 	s = re_coder.ReplaceAllString(s, "<code>$1</code>")
 
 	// some final fixups
-	s = strings.Replace(s, "\n", "<br>", -1)
 	s = strings.Replace(s, "<br><blockquote>", "<blockquote>", -1)
 	s = strings.Replace(s, "<br><cite></cite>", "", -1)
 	s = strings.Replace(s, "<br><pre>", "<pre>", -1)
diff --git a/markitzero_test.go b/markitzero_test.go
index 4d49ff6..fcdf2f7 100644
--- a/markitzero_test.go
+++ b/markitzero_test.go
@@ -32,19 +32,19 @@ func TestLinebreak1(t *testing.T) {
 
 func TestLinebreak2(t *testing.T) {
 	input := "hello\n\n> a quote\n\na comment"
-	output := "hello<br><blockquote>a quote</blockquote><p>a comment"
+	output := "hello<p><blockquote>a quote</blockquote><p>a comment"
 	doonezerotest(t, input, output)
 }
 
 func TestLinebreak3(t *testing.T) {
 	input := "hello\n\n```\nfunc(s string)\n```\n\ndoes it go?"
-	output := "hello<br><pre><code>func(s string)</code></pre><p>does it go?"
+	output := "hello<p><pre><code>func(s string)</code></pre><p>does it go?"
 	doonezerotest(t, input, output)
 }
 
 func TestCodeStyles(t *testing.T) {
 	input := "hello\n\n```go\nfunc(s string)\n```\n\ndoes it go?"
-	output := "hello<br><pre><code><span class=kw>func</span><span class=op>(</span>s <span class=tp>string</span><span class=op>)</span></code></pre><p>does it go?"
+	output := "hello<p><pre><code><span class=kw>func</span><span class=op>(</span>s <span class=tp>string</span><span class=op>)</span></code></pre><p>does it go?"
 	doonezerotest(t, input, output)
 }