revert for now, this is probably not the slow point for anything

master
Ted Unangst 5 years ago
parent 95dfba8d14
commit 2668c04b8e

@ -16,11 +16,11 @@
package main
import (
"bytes"
"fmt"
"regexp"
"strings"
"golang.org/x/net/html"
"humungus.tedunangst.com/r/webs/synlight"
)
@ -37,8 +37,27 @@ var re_lister = regexp.MustCompile(`((^|\n)(\+|-).*)+\n?`)
var lighter = synlight.New(synlight.Options{Format: synlight.HTML})
// fewer side effects than html.EscapeString
func fasterescaper(s []byte) []byte {
func markitzero(s string) string {
// prepare the string
s = strings.TrimSpace(s)
s = strings.Replace(s, "\r", "", -1)
// 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 "``````"
})
s = re_coder.ReplaceAllStringFunc(s, func(code string) string {
lilcodes = append(lilcodes, code)
return "`x`"
})
s = re_imgfix.ReplaceAllStringFunc(s, func(img string) string {
images = append(images, img)
return "<img x>"
})
// fewer side effects than html.EscapeString
buf := make([]byte, 0, len(s))
for _, c := range []byte(s) {
switch c {
@ -52,130 +71,64 @@ func fasterescaper(s []byte) []byte {
buf = append(buf, c)
}
}
return buf
}
func replaceifmatch(re *regexp.Regexp, input []byte, repl []byte) []byte {
if !re.Match(input) {
return input
}
return re.ReplaceAll(input, repl)
}
func replaceifmatchfn(re *regexp.Regexp, input []byte, repl func([]byte) []byte) []byte {
if !re.Match(input) {
return input
}
return re.ReplaceAllFunc(input, repl)
}
func replacenocopy(input []byte, pat []byte, repl []byte) []byte {
if !bytes.Contains(input, pat) {
return input
}
return bytes.Replace(input, pat, repl, -1)
}
func markitzero(ss string) string {
s := []byte(ss)
// prepare the string
s = bytes.TrimSpace(s)
s = replacenocopy(s, []byte("\r"), []byte(""))
hascode := bytes.Contains(s, []byte("`"))
// save away the code blocks so we don't mess them up further
var bigcodes, lilcodes, images [][]byte
if hascode {
s = replaceifmatchfn(re_bigcoder, s, func(code []byte) []byte {
bigcodes = append(bigcodes, code)
return []byte("``````")
})
s = replaceifmatchfn(re_coder, s, func(code []byte) []byte {
lilcodes = append(lilcodes, code)
return []byte("`x`")
})
}
s = replaceifmatchfn(re_imgfix, s, func(img []byte) []byte {
images = append(images, img)
return []byte("<img x>")
})
s = fasterescaper(s)
s = string(buf)
// mark it zero
if bytes.Contains(s, []byte("http")) {
s = replaceifmatchfn(re_link, s, linkreplacer)
}
s = replaceifmatch(re_zerolink, s, []byte(`<a href="$2">$1</a>`))
if bytes.Contains(s, []byte("**")) {
s = replaceifmatch(re_bolder, s, []byte("$1<b>$2</b>$3"))
}
if bytes.Contains(s, []byte("*")) {
s = replaceifmatch(re_italicer, s, []byte("$1<i>$2</i>$3"))
}
if bytes.Contains(s, []byte("&gt; ")) {
s = replaceifmatch(re_quoter, s, []byte("<blockquote>$1<br><cite>$3</cite></blockquote><p>"))
s = replaceifmatch(re_reciter, s, []byte("$1$2$3"))
}
s = replacenocopy(s, []byte("\n---\n"), []byte("<hr><p>"))
if bytes.Contains(s, []byte("\n+")) || bytes.Contains(s, []byte("\n-")) {
s = replaceifmatchfn(re_lister, s, func(m []byte) []byte {
m = bytes.Trim(m, "\n")
items := bytes.Split(m, []byte("\n"))
r := []byte("<ul>")
for _, item := range items {
r = append(r, []byte("<li>")...)
r = append(r, bytes.Trim(item[1:], " ")...)
}
r = append(r, []byte("</ul><p>")...)
return r
})
}
s = re_link.ReplaceAllStringFunc(s, linkreplacer)
s = re_zerolink.ReplaceAllString(s, `<a href="$2">$1</a>`)
s = re_bolder.ReplaceAllString(s, "$1<b>$2</b>$3")
s = re_italicer.ReplaceAllString(s, "$1<i>$2</i>$3")
s = re_quoter.ReplaceAllString(s, "<blockquote>$1<br><cite>$3</cite></blockquote><p>")
s = re_reciter.ReplaceAllString(s, "$1$2$3")
s = strings.Replace(s, "\n---\n", "<hr><p>", -1)
s = re_lister.ReplaceAllStringFunc(s, func(m string) string {
m = strings.Trim(m, "\n")
items := strings.Split(m, "\n")
r := "<ul>"
for _, item := range items {
r += "<li>" + strings.Trim(item[1:], " ")
}
r += "</ul><p>"
return r
})
// restore images
s = replacenocopy(s, []byte("&lt;img x&gt;"), []byte("<img x>"))
s = replaceifmatchfn(re_imgfix, s, func([]byte) []byte {
s = strings.Replace(s, "&lt;img x&gt;", "<img x>", -1)
s = re_imgfix.ReplaceAllStringFunc(s, func(string) string {
img := images[0]
images = images[1:]
return img
})
// now restore the code blocks
if hascode {
s = replaceifmatchfn(re_coder, s, func([]byte) []byte {
code := lilcodes[0]
lilcodes = lilcodes[1:]
return fasterescaper(code)
})
s = replaceifmatchfn(re_bigcoder, s, func([]byte) []byte {
code := bigcodes[0]
bigcodes = bigcodes[1:]
m := re_bigcoder.FindSubmatch(code)
var buf bytes.Buffer
buf.WriteString("<pre><code>")
lighter.Highlight(m[2], string(m[1]), &buf)
buf.WriteString("</code></pre><p>")
return buf.Bytes()
})
s = replaceifmatch(re_coder, s, []byte("<code>$1</code>"))
}
s = re_coder.ReplaceAllStringFunc(s, func(string) string {
code := lilcodes[0]
lilcodes = lilcodes[1:]
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
s = replacenocopy(s, []byte("\n"), []byte("<br>"))
s = replacenocopy(s, []byte("<br><blockquote>"), []byte("<blockquote>"))
s = replacenocopy(s, []byte("<br><cite></cite>"), []byte(""))
s = replacenocopy(s, []byte("<br><pre>"), []byte("<pre>"))
s = replacenocopy(s, []byte("<br><ul>"), []byte("<ul>"))
s = replacenocopy(s, []byte("<p><br>"), []byte("<p>"))
return string(s)
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)
s = strings.Replace(s, "<br><ul>", "<ul>", -1)
s = strings.Replace(s, "<p><br>", "<p>", -1)
return s
}
func linkreplacer(burl []byte) []byte {
url := string(burl)
func linkreplacer(url string) string {
if url[0:2] == "](" {
return burl
return url
}
prefix := ""
for !strings.HasPrefix(url, "http") {
@ -199,5 +152,5 @@ func linkreplacer(burl []byte) []byte {
if addparen {
url += ")"
}
return []byte(prefix + url)
return prefix + url
}

Loading…
Cancel
Save