|
@@ -2,41 +2,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
-package base
|
|
|
+package markdown
|
|
|
|
|
|
import (
|
|
|
"bytes"
|
|
|
"fmt"
|
|
|
"io"
|
|
|
- "net/http"
|
|
|
"path"
|
|
|
"path/filepath"
|
|
|
"regexp"
|
|
|
"strings"
|
|
|
|
|
|
"github.com/Unknwon/com"
|
|
|
+ "github.com/microcosm-cc/bluemonday"
|
|
|
"github.com/russross/blackfriday"
|
|
|
"golang.org/x/net/html"
|
|
|
|
|
|
+ "github.com/gogits/gogs/modules/base"
|
|
|
"github.com/gogits/gogs/modules/setting"
|
|
|
)
|
|
|
|
|
|
-
|
|
|
+var Sanitizer = bluemonday.UGCPolicy()
|
|
|
|
|
|
-func isletter(c byte) bool {
|
|
|
- return (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z')
|
|
|
-}
|
|
|
+
|
|
|
+
|
|
|
+func BuildSanitizer() {
|
|
|
+
|
|
|
+ Sanitizer.AllowAttrs("class").Matching(regexp.MustCompile(`[\p{L}\p{N}\s\-_',:\[\]!\./\\\(\)&]*`)).OnElements("code")
|
|
|
+
|
|
|
+
|
|
|
+ Sanitizer.AllowAttrs("type").Matching(regexp.MustCompile(`^checkbox$`)).OnElements("input")
|
|
|
+ Sanitizer.AllowAttrs("checked", "disabled").OnElements("input")
|
|
|
|
|
|
-func isalnum(c byte) bool {
|
|
|
- return (c >= '0' && c <= '9') || isletter(c)
|
|
|
+
|
|
|
+ Sanitizer.AllowURLSchemes(setting.Markdown.CustomURLSchemes...)
|
|
|
}
|
|
|
|
|
|
var validLinksPattern = regexp.MustCompile(`^[a-z][\w-]+://`)
|
|
|
|
|
|
+
|
|
|
func isLink(link []byte) bool {
|
|
|
return validLinksPattern.Match(link)
|
|
|
}
|
|
|
|
|
|
+
|
|
|
+
|
|
|
func IsMarkdownFile(name string) bool {
|
|
|
name = strings.ToLower(name)
|
|
|
switch filepath.Ext(name) {
|
|
@@ -46,57 +56,46 @@ func IsMarkdownFile(name string) bool {
|
|
|
return false
|
|
|
}
|
|
|
|
|
|
-func IsTextFile(data []byte) (string, bool) {
|
|
|
- contentType := http.DetectContentType(data)
|
|
|
- if strings.Index(contentType, "text/") != -1 {
|
|
|
- return contentType, true
|
|
|
- }
|
|
|
- return contentType, false
|
|
|
-}
|
|
|
-
|
|
|
-func IsImageFile(data []byte) (string, bool) {
|
|
|
- contentType := http.DetectContentType(data)
|
|
|
- if strings.Index(contentType, "image/") != -1 {
|
|
|
- return contentType, true
|
|
|
- }
|
|
|
- return contentType, false
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
+
|
|
|
+
|
|
|
func IsReadmeFile(name string) bool {
|
|
|
name = strings.ToLower(name)
|
|
|
if len(name) < 6 {
|
|
|
return false
|
|
|
} else if len(name) == 6 {
|
|
|
- if name == "readme" {
|
|
|
- return true
|
|
|
- }
|
|
|
- return false
|
|
|
- }
|
|
|
- if name[:7] == "readme." {
|
|
|
- return true
|
|
|
+ return name == "readme"
|
|
|
}
|
|
|
- return false
|
|
|
+ return name[:7] == "readme."
|
|
|
}
|
|
|
|
|
|
var (
|
|
|
- MentionPattern = regexp.MustCompile(`(\s|^)@[0-9a-zA-Z_\.]+`)
|
|
|
- commitPattern = regexp.MustCompile(`(\s|^)https?.*commit/[0-9a-zA-Z]+(#+[0-9a-zA-Z-]*)?`)
|
|
|
- issueFullPattern = regexp.MustCompile(`(\s|^)https?.*issues/[0-9]+(#+[0-9a-zA-Z-]*)?`)
|
|
|
- issueIndexPattern = regexp.MustCompile(`( |^|\()#[0-9]+\b`)
|
|
|
- sha1CurrentPattern = regexp.MustCompile(`\b[0-9a-f]{40}\b`)
|
|
|
+
|
|
|
+ MentionPattern = regexp.MustCompile(`(\s|^)@[0-9a-zA-Z_\.]+`)
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ CommitPattern = regexp.MustCompile(`(\s|^)https?.*commit/[0-9a-zA-Z]+(#+[0-9a-zA-Z-]*)?`)
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ IssueFullPattern = regexp.MustCompile(`(\s|^)https?.*issues/[0-9]+(#+[0-9a-zA-Z-]*)?`)
|
|
|
+
|
|
|
+ IssueIndexPattern = regexp.MustCompile(`( |^|\()#[0-9]+\b`)
|
|
|
+
|
|
|
+
|
|
|
+ Sha1CurrentPattern = regexp.MustCompile(`\b[0-9a-f]{40}\b`)
|
|
|
)
|
|
|
|
|
|
-type CustomRender struct {
|
|
|
+
|
|
|
+type Renderer struct {
|
|
|
blackfriday.Renderer
|
|
|
urlPrefix string
|
|
|
}
|
|
|
|
|
|
-func (r *CustomRender) Link(out *bytes.Buffer, link []byte, title []byte, content []byte) {
|
|
|
+
|
|
|
+func (r *Renderer) Link(out *bytes.Buffer, link []byte, title []byte, content []byte) {
|
|
|
if len(link) > 0 && !isLink(link) {
|
|
|
- if link[0] == '#' {
|
|
|
-
|
|
|
- } else {
|
|
|
+ if link[0] != '#' {
|
|
|
link = []byte(path.Join(r.urlPrefix, string(link)))
|
|
|
}
|
|
|
}
|
|
@@ -104,14 +103,17 @@ func (r *CustomRender) Link(out *bytes.Buffer, link []byte, title []byte, conten
|
|
|
r.Renderer.Link(out, link, title, content)
|
|
|
}
|
|
|
|
|
|
-func (r *CustomRender) AutoLink(out *bytes.Buffer, link []byte, kind int) {
|
|
|
- if kind != 1 {
|
|
|
+
|
|
|
+
|
|
|
+func (r *Renderer) AutoLink(out *bytes.Buffer, link []byte, kind int) {
|
|
|
+ if kind != blackfriday.LINK_TYPE_NORMAL {
|
|
|
r.Renderer.AutoLink(out, link, kind)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
-
|
|
|
- m := commitPattern.Find(link)
|
|
|
+
|
|
|
+
|
|
|
+ m := CommitPattern.Find(link)
|
|
|
if m != nil {
|
|
|
m = bytes.TrimSpace(m)
|
|
|
i := strings.Index(string(m), "commit/")
|
|
@@ -119,11 +121,11 @@ func (r *CustomRender) AutoLink(out *bytes.Buffer, link []byte, kind int) {
|
|
|
if j == -1 {
|
|
|
j = len(m)
|
|
|
}
|
|
|
- out.WriteString(fmt.Sprintf(` <code><a href="%s">%s</a></code>`, m, ShortSha(string(m[i+7:j]))))
|
|
|
+ out.WriteString(fmt.Sprintf(` <code><a href="%s">%s</a></code>`, m, base.ShortSha(string(m[i+7:j]))))
|
|
|
return
|
|
|
}
|
|
|
|
|
|
- m = issueFullPattern.Find(link)
|
|
|
+ m = IssueFullPattern.Find(link)
|
|
|
if m != nil {
|
|
|
m = bytes.TrimSpace(m)
|
|
|
i := strings.Index(string(m), "issues/")
|
|
@@ -131,14 +133,16 @@ func (r *CustomRender) AutoLink(out *bytes.Buffer, link []byte, kind int) {
|
|
|
if j == -1 {
|
|
|
j = len(m)
|
|
|
}
|
|
|
- out.WriteString(fmt.Sprintf(` <a href="%s">#%s</a>`, m, ShortSha(string(m[i+7:j]))))
|
|
|
+ out.WriteString(fmt.Sprintf(` <a href="%s">#%s</a>`, m, base.ShortSha(string(m[i+7:j]))))
|
|
|
return
|
|
|
}
|
|
|
|
|
|
r.Renderer.AutoLink(out, link, kind)
|
|
|
}
|
|
|
|
|
|
-func (options *CustomRender) ListItem(out *bytes.Buffer, text []byte, flags int) {
|
|
|
+
|
|
|
+func (options *Renderer) ListItem(out *bytes.Buffer, text []byte, flags int) {
|
|
|
+
|
|
|
switch {
|
|
|
case bytes.HasPrefix(text, []byte("[ ] ")):
|
|
|
text = append([]byte(`<input type="checkbox" disabled="" />`), text[3:]...)
|
|
@@ -148,6 +152,8 @@ func (options *CustomRender) ListItem(out *bytes.Buffer, text []byte, flags int)
|
|
|
options.Renderer.ListItem(out, text, flags)
|
|
|
}
|
|
|
|
|
|
+
|
|
|
+
|
|
|
var (
|
|
|
svgSuffix = []byte(".svg")
|
|
|
svgSuffixWithMark = []byte(".svg?")
|
|
@@ -155,11 +161,13 @@ var (
|
|
|
spaceEncodedBytes = []byte("%20")
|
|
|
)
|
|
|
|
|
|
-func (r *CustomRender) Image(out *bytes.Buffer, link []byte, title []byte, alt []byte) {
|
|
|
+
|
|
|
+func (r *Renderer) Image(out *bytes.Buffer, link []byte, title []byte, alt []byte) {
|
|
|
prefix := strings.Replace(r.urlPrefix, "/src/", "/raw/", 1)
|
|
|
if len(link) > 0 {
|
|
|
if isLink(link) {
|
|
|
|
|
|
+
|
|
|
if bytes.HasSuffix(link, svgSuffix) || bytes.Contains(link, svgSuffixWithMark) {
|
|
|
r.Renderer.Image(out, link, title, alt)
|
|
|
return
|
|
@@ -180,6 +188,8 @@ func (r *CustomRender) Image(out *bytes.Buffer, link []byte, title []byte, alt [
|
|
|
out.WriteString("</a>")
|
|
|
}
|
|
|
|
|
|
+
|
|
|
+
|
|
|
func cutoutVerbosePrefix(prefix string) string {
|
|
|
count := 0
|
|
|
for i := 0; i < len(prefix); i++ {
|
|
@@ -193,29 +203,40 @@ func cutoutVerbosePrefix(prefix string) string {
|
|
|
return prefix
|
|
|
}
|
|
|
|
|
|
+
|
|
|
func RenderIssueIndexPattern(rawBytes []byte, urlPrefix string, metas map[string]string) []byte {
|
|
|
urlPrefix = cutoutVerbosePrefix(urlPrefix)
|
|
|
- ms := issueIndexPattern.FindAll(rawBytes, -1)
|
|
|
+ ms := IssueIndexPattern.FindAll(rawBytes, -1)
|
|
|
for _, m := range ms {
|
|
|
var space string
|
|
|
- m2 := m
|
|
|
- if m2[0] != '#' {
|
|
|
- space = string(m2[0])
|
|
|
- m2 = m2[1:]
|
|
|
+ if m[0] != '#' {
|
|
|
+ space = string(m[0])
|
|
|
+ m = m[1:]
|
|
|
}
|
|
|
if metas == nil {
|
|
|
rawBytes = bytes.Replace(rawBytes, m, []byte(fmt.Sprintf(`%s<a href="%s/issues/%s">%s</a>`,
|
|
|
- space, urlPrefix, m2[1:], m2)), 1)
|
|
|
+ space, urlPrefix, m[1:], m)), 1)
|
|
|
} else {
|
|
|
|
|
|
- metas["index"] = string(m2[1:])
|
|
|
+ metas["index"] = string(m[1:])
|
|
|
rawBytes = bytes.Replace(rawBytes, m, []byte(fmt.Sprintf(`%s<a href="%s">%s</a>`,
|
|
|
- space, com.Expand(metas["format"], metas), m2)), 1)
|
|
|
+ space, com.Expand(metas["format"], metas), m)), 1)
|
|
|
}
|
|
|
}
|
|
|
return rawBytes
|
|
|
}
|
|
|
|
|
|
+
|
|
|
+func RenderSha1CurrentPattern(rawBytes []byte, urlPrefix string) []byte {
|
|
|
+ ms := Sha1CurrentPattern.FindAll(rawBytes, -1)
|
|
|
+ for _, m := range ms {
|
|
|
+ rawBytes = bytes.Replace(rawBytes, m, []byte(fmt.Sprintf(
|
|
|
+ `<a href="%s/commit/%s"><code>%s</code></a>`, urlPrefix, m, base.ShortSha(string(m)))), -1)
|
|
|
+ }
|
|
|
+ return rawBytes
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
func RenderSpecialLink(rawBytes []byte, urlPrefix string, metas map[string]string) []byte {
|
|
|
ms := MentionPattern.FindAll(rawBytes, -1)
|
|
|
for _, m := range ms {
|
|
@@ -229,20 +250,12 @@ func RenderSpecialLink(rawBytes []byte, urlPrefix string, metas map[string]strin
|
|
|
return rawBytes
|
|
|
}
|
|
|
|
|
|
-func RenderSha1CurrentPattern(rawBytes []byte, urlPrefix string) []byte {
|
|
|
- ms := sha1CurrentPattern.FindAll(rawBytes, -1)
|
|
|
- for _, m := range ms {
|
|
|
- rawBytes = bytes.Replace(rawBytes, m, []byte(fmt.Sprintf(
|
|
|
- `<a href="%s/commit/%s"><code>%s</code></a>`, urlPrefix, m, ShortSha(string(m)))), -1)
|
|
|
- }
|
|
|
- return rawBytes
|
|
|
-}
|
|
|
-
|
|
|
-func RenderRawMarkdown(body []byte, urlPrefix string) []byte {
|
|
|
+
|
|
|
+func RenderRaw(body []byte, urlPrefix string) []byte {
|
|
|
htmlFlags := 0
|
|
|
htmlFlags |= blackfriday.HTML_SKIP_STYLE
|
|
|
htmlFlags |= blackfriday.HTML_OMIT_CONTENTS
|
|
|
- renderer := &CustomRender{
|
|
|
+ renderer := &Renderer{
|
|
|
Renderer: blackfriday.HtmlRenderer(htmlFlags, "", ""),
|
|
|
urlPrefix: urlPrefix,
|
|
|
}
|
|
@@ -272,9 +285,9 @@ var (
|
|
|
|
|
|
var noEndTags = []string{"img", "input", "br", "hr"}
|
|
|
|
|
|
-
|
|
|
+
|
|
|
|
|
|
-func PostProcessMarkdown(rawHtml []byte, urlPrefix string, metas map[string]string) []byte {
|
|
|
+func PostProcess(rawHtml []byte, urlPrefix string, metas map[string]string) []byte {
|
|
|
startTags := make([]string, 0, 5)
|
|
|
var buf bytes.Buffer
|
|
|
tokenizer := html.NewTokenizer(bytes.NewReader(rawHtml))
|
|
@@ -342,13 +355,15 @@ OUTER_LOOP:
|
|
|
return rawHtml
|
|
|
}
|
|
|
|
|
|
-func RenderMarkdown(rawBytes []byte, urlPrefix string, metas map[string]string) []byte {
|
|
|
- result := RenderRawMarkdown(rawBytes, urlPrefix)
|
|
|
- result = PostProcessMarkdown(result, urlPrefix, metas)
|
|
|
+
|
|
|
+func Render(rawBytes []byte, urlPrefix string, metas map[string]string) []byte {
|
|
|
+ result := RenderRaw(rawBytes, urlPrefix)
|
|
|
+ result = PostProcess(result, urlPrefix, metas)
|
|
|
result = Sanitizer.SanitizeBytes(result)
|
|
|
return result
|
|
|
}
|
|
|
|
|
|
-func RenderMarkdownString(raw, urlPrefix string, metas map[string]string) string {
|
|
|
- return string(RenderMarkdown([]byte(raw), urlPrefix, metas))
|
|
|
+
|
|
|
+func RenderString(raw, urlPrefix string, metas map[string]string) string {
|
|
|
+ return string(Render([]byte(raw), urlPrefix, metas))
|
|
|
}
|