forgejo/modules/base/markdown.go

170 lines
4.5 KiB
Go
Raw Normal View History

2014-03-17 11:46:54 +01:00
// Copyright 2014 The Gogs Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package base
import (
2014-03-17 16:22:27 +01:00
"bytes"
2014-04-07 06:47:19 +02:00
"fmt"
2014-03-24 16:56:32 +01:00
"net/http"
2014-03-17 16:22:27 +01:00
"path"
2014-03-20 06:31:24 +01:00
"path/filepath"
2014-04-07 06:47:19 +02:00
"regexp"
2014-03-20 06:31:24 +01:00
"strings"
2014-03-17 16:22:27 +01:00
"github.com/gogits/gfm"
2014-03-17 11:46:54 +01:00
)
2014-03-17 16:22:27 +01:00
func isletter(c byte) bool {
return (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z')
}
func isalnum(c byte) bool {
return (c >= '0' && c <= '9') || isletter(c)
}
var validLinks = [][]byte{[]byte("http://"), []byte("https://"), []byte("ftp://"), []byte("mailto://")}
func isLink(link []byte) bool {
for _, prefix := range validLinks {
if len(link) > len(prefix) && bytes.Equal(bytes.ToLower(link[:len(prefix)]), prefix) && isalnum(link[len(prefix)]) {
return true
}
}
return false
}
2014-03-20 06:31:24 +01:00
func IsMarkdownFile(name string) bool {
name = strings.ToLower(name)
switch filepath.Ext(name) {
2014-03-20 14:11:48 +01:00
case ".md", ".markdown", ".mdown":
2014-03-20 06:31:24 +01:00
return true
}
return false
}
2014-03-24 16:56:32 +01:00
func IsTextFile(data []byte) (string, bool) {
contentType := http.DetectContentType(data)
if strings.Index(contentType, "text/") != -1 {
return contentType, true
}
return contentType, false
}
2014-03-27 17:30:20 +01:00
func IsImageFile(data []byte) (string, bool) {
contentType := http.DetectContentType(data)
2014-03-27 17:50:13 +01:00
if strings.Index(contentType, "image/") != -1 {
2014-03-27 17:30:20 +01:00
return contentType, true
}
return contentType, false
}
2014-03-20 06:31:24 +01:00
func IsReadmeFile(name string) bool {
name = strings.ToLower(name)
if len(name) < 6 {
return false
}
if name[:6] == "readme" {
return true
}
return false
}
2014-03-17 16:22:27 +01:00
type CustomRender struct {
gfm.Renderer
urlPrefix string
}
func (options *CustomRender) Link(out *bytes.Buffer, link []byte, title []byte, content []byte) {
if len(link) > 0 && !isLink(link) {
if link[0] == '#' {
2014-03-20 10:31:18 +01:00
// link = append([]byte(options.urlPrefix), link...)
2014-03-17 16:22:27 +01:00
} else {
link = []byte(path.Join(options.urlPrefix, string(link)))
}
}
options.Renderer.Link(out, link, title, content)
}
2014-04-07 06:47:19 +02:00
var (
mentionPattern = regexp.MustCompile(`@[0-9a-zA-Z_]{1,}`)
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(`(\s|^)#[0-9]+`)
2014-04-07 06:47:19 +02:00
)
func RenderSpecialLink(rawBytes []byte, urlPrefix string) []byte {
ms := mentionPattern.FindAll(rawBytes, -1)
for _, m := range ms {
rawBytes = bytes.Replace(rawBytes, m,
[]byte(fmt.Sprintf(`<a href="/user/%s">%s</a>`, m[1:], m)), -1)
}
ms = commitPattern.FindAll(rawBytes, -1)
for _, m := range ms {
m = bytes.TrimPrefix(m, []byte(" "))
i := strings.Index(string(m), "commit/")
j := strings.Index(string(m), "#")
if j == -1 {
j = len(m)
}
rawBytes = bytes.Replace(rawBytes, m, []byte(fmt.Sprintf(
` <code><a href="%s">%s</a></code>`, m, ShortSha(string(m[i+7:j])))), -1)
}
ms = issueFullPattern.FindAll(rawBytes, -1)
for _, m := range ms {
m = bytes.TrimPrefix(m, []byte(" "))
i := strings.Index(string(m), "issues/")
j := strings.Index(string(m), "#")
if j == -1 {
j = len(m)
}
rawBytes = bytes.Replace(rawBytes, m, []byte(fmt.Sprintf(
` <a href="%s">#%s</a>`, m, ShortSha(string(m[i+7:j])))), -1)
}
ms = issueIndexPattern.FindAll(rawBytes, -1)
for _, m := range ms {
m = bytes.TrimPrefix(m, []byte(" "))
rawBytes = bytes.Replace(rawBytes, m, []byte(fmt.Sprintf(
` <a href="%s/issues/%s">%s</a>`, urlPrefix, m[1:], m)), -1)
2014-04-07 06:47:19 +02:00
}
return rawBytes
}
2014-03-17 16:22:27 +01:00
func RenderMarkdown(rawBytes []byte, urlPrefix string) []byte {
2014-04-07 06:47:19 +02:00
body := RenderSpecialLink(rawBytes, urlPrefix)
fmt.Println(string(body))
2014-03-17 11:46:54 +01:00
htmlFlags := 0
2014-03-23 09:40:51 +01:00
// htmlFlags |= gfm.HTML_USE_XHTML
2014-03-17 16:22:27 +01:00
// htmlFlags |= gfm.HTML_USE_SMARTYPANTS
// htmlFlags |= gfm.HTML_SMARTYPANTS_FRACTIONS
// htmlFlags |= gfm.HTML_SMARTYPANTS_LATEX_DASHES
htmlFlags |= gfm.HTML_SKIP_HTML
htmlFlags |= gfm.HTML_SKIP_STYLE
htmlFlags |= gfm.HTML_SKIP_SCRIPT
htmlFlags |= gfm.HTML_GITHUB_BLOCKCODE
htmlFlags |= gfm.HTML_OMIT_CONTENTS
2014-03-23 09:40:51 +01:00
// htmlFlags |= gfm.HTML_COMPLETE_PAGE
// renderer := &CustomRender{
// Renderer: gfm.HtmlRenderer(htmlFlags, "", ""),
// urlPrefix: urlPrefix,
// }
2014-03-17 11:46:54 +01:00
// set up the parser
extensions := 0
2014-03-17 16:22:27 +01:00
extensions |= gfm.EXTENSION_NO_INTRA_EMPHASIS
extensions |= gfm.EXTENSION_TABLES
extensions |= gfm.EXTENSION_FENCED_CODE
extensions |= gfm.EXTENSION_AUTOLINK
extensions |= gfm.EXTENSION_STRIKETHROUGH
extensions |= gfm.EXTENSION_HARD_LINE_BREAK
extensions |= gfm.EXTENSION_SPACE_HEADERS
extensions |= gfm.EXTENSION_NO_EMPTY_LINE_BEFORE_BLOCK
// body = gfm.Markdown(body, renderer, extensions)
// fmt.Println(string(body))
2014-03-17 11:46:54 +01:00
return body
}