mirror of
https://github.com/facundoolano/jorge.git
synced 2024-09-28 03:20:50 +02:00
82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
package markup
|
||
|
||
// Implements a naive version of smart quote replacement, see https://daringfireball.net/projects/smartypants/
|
||
// The quote replacement code was adapted from gojekyll's smartify filter to work on entire HTML documents
|
||
// https://github.com/osteele/gojekyll/blob/f1794a874890bfb601cae767a0cce15d672e9058/filters/smartify.go
|
||
// MIT License: https://github.com/osteele/gojekyll/blob/f1794a874890bfb601cae767a0cce15d672e9058/LICENSE
|
||
|
||
import (
|
||
"bytes"
|
||
"io"
|
||
"regexp"
|
||
"slices"
|
||
"strings"
|
||
|
||
"golang.org/x/net/html"
|
||
)
|
||
|
||
var SKIP_TAGS = []string{"pre", "code", "kbd", "script", "math"}
|
||
|
||
func Smartify(extension string, contentReader io.Reader) (io.Reader, error) {
|
||
if extension != ".html" {
|
||
return contentReader, nil
|
||
}
|
||
node, err := html.Parse(contentReader)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
smartifyHTMLNode(node)
|
||
var buf bytes.Buffer
|
||
html.Render(&buf, node)
|
||
|
||
return &buf, nil
|
||
}
|
||
|
||
func smartifyHTMLNode(node *html.Node) {
|
||
for node := node.FirstChild; node != nil; node = node.NextSibling {
|
||
if node.Type == html.ElementNode && slices.Contains(SKIP_TAGS, node.Data) {
|
||
continue
|
||
}
|
||
if node.Type == html.TextNode {
|
||
node.Data = smartifyString(node.Data)
|
||
} else {
|
||
smartifyHTMLNode(node)
|
||
}
|
||
}
|
||
}
|
||
|
||
var smartifyTransforms = []struct {
|
||
match *regexp.Regexp
|
||
repl string
|
||
}{
|
||
{regexp.MustCompile("(^|[^[:alnum:]])``(.+?)''"), "$1“$2”"},
|
||
{regexp.MustCompile(`(^|[^[:alnum:]])'`), "$1‘"},
|
||
{regexp.MustCompile(`'`), "’"},
|
||
{regexp.MustCompile(`(^|[^[:alnum:]?!\.])"`), "$1“"},
|
||
{regexp.MustCompile(`"($|[^[:alnum:]])`), "”$1"},
|
||
{regexp.MustCompile(`\\[‘’]`), "'"}, // undo backslashed replacements
|
||
{regexp.MustCompile(`\\[“”]`), `"`}, // undo backslashed replacements
|
||
}
|
||
|
||
var smartifyReplacer *strings.Replacer
|
||
|
||
func init() {
|
||
smartifyReplacer = strings.NewReplacer(
|
||
"...", "…",
|
||
"(c)", "©",
|
||
"(r)", "®",
|
||
"(tm)", "™",
|
||
// moving the dashes to straight replacements instead of regex (which weren't accurate)
|
||
// NOTE: go-org already does dash and ellipsis replacement, remove altogether if it causes issues
|
||
"---", "—",
|
||
"--", "–",
|
||
)
|
||
}
|
||
|
||
func smartifyString(s string) string {
|
||
for _, rule := range smartifyTransforms {
|
||
s = rule.match.ReplaceAllString(s, rule.repl)
|
||
}
|
||
return smartifyReplacer.Replace(s)
|
||
}
|