2014-04-11 02:20:58 +08:00
|
|
|
// Copyright 2014 The Gogs Authors. All rights reserved.
|
2018-10-31 06:26:28 +08:00
|
|
|
// Copyright 2018 The Gitea Authors. All rights reserved.
|
2014-04-11 02:20:58 +08:00
|
|
|
// Use of this source code is governed by a MIT-style
|
|
|
|
// license that can be found in the LICENSE file.
|
|
|
|
|
2016-02-21 06:10:05 +08:00
|
|
|
package markdown
|
2014-04-11 02:20:58 +08:00
|
|
|
|
|
|
|
import (
|
2021-03-16 07:20:05 +08:00
|
|
|
"fmt"
|
|
|
|
"io"
|
2020-08-05 03:56:37 +08:00
|
|
|
"strings"
|
2019-12-31 09:53:28 +08:00
|
|
|
"sync"
|
2014-04-11 02:20:58 +08:00
|
|
|
|
2019-12-31 09:53:28 +08:00
|
|
|
"code.gitea.io/gitea/modules/log"
|
2017-04-21 15:01:08 +08:00
|
|
|
"code.gitea.io/gitea/modules/markup"
|
2019-12-31 09:53:28 +08:00
|
|
|
"code.gitea.io/gitea/modules/markup/common"
|
2016-11-11 00:24:48 +08:00
|
|
|
"code.gitea.io/gitea/modules/setting"
|
2019-12-31 09:53:28 +08:00
|
|
|
giteautil "code.gitea.io/gitea/modules/util"
|
|
|
|
|
2020-07-01 05:34:03 +08:00
|
|
|
chromahtml "github.com/alecthomas/chroma/formatters/html"
|
2019-12-31 09:53:28 +08:00
|
|
|
"github.com/yuin/goldmark"
|
2021-03-16 07:20:05 +08:00
|
|
|
highlighting "github.com/yuin/goldmark-highlighting"
|
2020-04-09 18:54:50 +08:00
|
|
|
meta "github.com/yuin/goldmark-meta"
|
2019-12-31 09:53:28 +08:00
|
|
|
"github.com/yuin/goldmark/extension"
|
|
|
|
"github.com/yuin/goldmark/parser"
|
|
|
|
"github.com/yuin/goldmark/renderer"
|
|
|
|
"github.com/yuin/goldmark/renderer/html"
|
|
|
|
"github.com/yuin/goldmark/util"
|
2014-04-11 02:20:58 +08:00
|
|
|
)
|
|
|
|
|
2022-01-21 01:46:10 +08:00
|
|
|
var (
|
|
|
|
converter goldmark.Markdown
|
|
|
|
once = sync.Once{}
|
|
|
|
)
|
2014-04-11 02:20:58 +08:00
|
|
|
|
2022-01-21 01:46:10 +08:00
|
|
|
var (
|
|
|
|
urlPrefixKey = parser.NewContextKey()
|
|
|
|
isWikiKey = parser.NewContextKey()
|
|
|
|
renderMetasKey = parser.NewContextKey()
|
|
|
|
)
|
2018-02-27 15:09:18 +08:00
|
|
|
|
2021-03-16 07:20:05 +08:00
|
|
|
type limitWriter struct {
|
2021-11-19 18:46:47 +08:00
|
|
|
w io.Writer
|
2021-03-16 07:20:05 +08:00
|
|
|
sum int64
|
|
|
|
limit int64
|
|
|
|
}
|
|
|
|
|
|
|
|
// Write implements the standard Write interface:
|
|
|
|
func (l *limitWriter) Write(data []byte) (int, error) {
|
|
|
|
leftToWrite := l.limit - l.sum
|
|
|
|
if leftToWrite < int64(len(data)) {
|
|
|
|
n, err := l.w.Write(data[:leftToWrite])
|
|
|
|
l.sum += int64(n)
|
|
|
|
if err != nil {
|
|
|
|
return n, err
|
|
|
|
}
|
|
|
|
return n, fmt.Errorf("Rendered content too large - truncating render")
|
|
|
|
}
|
|
|
|
n, err := l.w.Write(data)
|
|
|
|
l.sum += int64(n)
|
|
|
|
return n, err
|
|
|
|
}
|
|
|
|
|
2021-04-20 06:25:08 +08:00
|
|
|
// newParserContext creates a parser.Context with the render context set
|
|
|
|
func newParserContext(ctx *markup.RenderContext) parser.Context {
|
2019-12-31 09:53:28 +08:00
|
|
|
pc := parser.NewContext(parser.WithIDs(newPrefixedIDs()))
|
2021-04-20 06:25:08 +08:00
|
|
|
pc.Set(urlPrefixKey, ctx.URLPrefix)
|
|
|
|
pc.Set(isWikiKey, ctx.IsWiki)
|
|
|
|
pc.Set(renderMetasKey, ctx.Metas)
|
2019-12-31 09:53:28 +08:00
|
|
|
return pc
|
2016-01-09 10:59:04 +08:00
|
|
|
}
|
|
|
|
|
2021-03-16 07:20:05 +08:00
|
|
|
// actualRender renders Markdown to HTML without handling special links.
|
2021-04-20 06:25:08 +08:00
|
|
|
func actualRender(ctx *markup.RenderContext, input io.Reader, output io.Writer) error {
|
2019-12-31 09:53:28 +08:00
|
|
|
once.Do(func() {
|
|
|
|
converter = goldmark.New(
|
2021-08-03 00:11:18 +08:00
|
|
|
goldmark.WithExtensions(
|
|
|
|
extension.NewTable(
|
|
|
|
extension.WithTableCellAlignMethod(extension.TableCellAlignAttribute)),
|
2019-12-31 09:53:28 +08:00
|
|
|
extension.Strikethrough,
|
|
|
|
extension.TaskList,
|
|
|
|
extension.DefinitionList,
|
|
|
|
common.FootnoteExtension,
|
2020-07-01 05:34:03 +08:00
|
|
|
highlighting.NewHighlighting(
|
|
|
|
highlighting.WithFormatOptions(
|
|
|
|
chromahtml.WithClasses(true),
|
|
|
|
chromahtml.PreventSurroundingPre(true),
|
|
|
|
),
|
|
|
|
highlighting.WithWrapperRenderer(func(w util.BufWriter, c highlighting.CodeBlockContext, entering bool) {
|
|
|
|
if entering {
|
2020-08-05 03:56:37 +08:00
|
|
|
language, _ := c.Language()
|
|
|
|
if language == nil {
|
|
|
|
language = []byte("text")
|
|
|
|
}
|
|
|
|
|
|
|
|
languageStr := string(language)
|
|
|
|
|
2021-11-16 16:16:05 +08:00
|
|
|
preClasses := []string{"code-block"}
|
2020-08-05 03:56:37 +08:00
|
|
|
if languageStr == "mermaid" {
|
|
|
|
preClasses = append(preClasses, "is-loading")
|
|
|
|
}
|
|
|
|
|
2021-11-16 16:16:05 +08:00
|
|
|
_, err := w.WriteString(`<pre class="` + strings.Join(preClasses, " ") + `">`)
|
|
|
|
if err != nil {
|
|
|
|
return
|
2020-08-05 03:56:37 +08:00
|
|
|
}
|
|
|
|
|
2020-07-01 05:34:03 +08:00
|
|
|
// include language-x class as part of commonmark spec
|
2021-11-16 16:16:05 +08:00
|
|
|
_, err = w.WriteString(`<code class="chroma language-` + string(language) + `">`)
|
2020-07-01 05:34:03 +08:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
_, err := w.WriteString("</code></pre>")
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}),
|
|
|
|
),
|
2020-04-24 21:22:36 +08:00
|
|
|
meta.Meta,
|
2019-12-31 09:53:28 +08:00
|
|
|
),
|
|
|
|
goldmark.WithParserOptions(
|
|
|
|
parser.WithAttribute(),
|
|
|
|
parser.WithAutoHeadingID(),
|
|
|
|
parser.WithASTTransformers(
|
2020-04-24 21:22:36 +08:00
|
|
|
util.Prioritized(&ASTTransformer{}, 10000),
|
2019-12-31 09:53:28 +08:00
|
|
|
),
|
|
|
|
),
|
|
|
|
goldmark.WithRendererOptions(
|
|
|
|
html.WithUnsafe(),
|
|
|
|
),
|
|
|
|
)
|
|
|
|
|
|
|
|
// Override the original Tasklist renderer!
|
|
|
|
converter.Renderer().AddOptions(
|
|
|
|
renderer.WithNodeRenderers(
|
2020-04-24 21:22:36 +08:00
|
|
|
util.Prioritized(NewHTMLRenderer(), 10),
|
2019-12-31 09:53:28 +08:00
|
|
|
),
|
|
|
|
)
|
|
|
|
})
|
2014-04-11 02:20:58 +08:00
|
|
|
|
2021-03-16 07:20:05 +08:00
|
|
|
lw := &limitWriter{
|
2021-11-19 18:46:47 +08:00
|
|
|
w: output,
|
2021-03-16 07:20:05 +08:00
|
|
|
limit: setting.UI.MaxDisplayFileSize * 3,
|
2015-09-01 20:32:02 +08:00
|
|
|
}
|
2021-03-16 07:20:05 +08:00
|
|
|
|
2021-11-19 18:46:47 +08:00
|
|
|
// FIXME: should we include a timeout to abort the renderer if it takes too long?
|
|
|
|
defer func() {
|
|
|
|
err := recover()
|
|
|
|
if err == nil {
|
2021-04-20 06:25:08 +08:00
|
|
|
return
|
|
|
|
}
|
2021-11-19 18:46:47 +08:00
|
|
|
|
|
|
|
log.Warn("Unable to render markdown due to panic in goldmark: %v", err)
|
|
|
|
if log.IsDebug() {
|
|
|
|
log.Debug("Panic in markdown: %v\n%s", err, string(log.Stack(2)))
|
2021-03-16 07:20:05 +08:00
|
|
|
}
|
|
|
|
}()
|
2021-11-19 18:46:47 +08:00
|
|
|
|
|
|
|
// FIXME: Don't read all to memory, but goldmark doesn't support
|
|
|
|
pc := newParserContext(ctx)
|
|
|
|
buf, err := io.ReadAll(input)
|
|
|
|
if err != nil {
|
|
|
|
log.Error("Unable to ReadAll: %v", err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if err := converter.Convert(giteautil.NormalizeEOL(buf), lw, parser.WithContext(pc)); err != nil {
|
|
|
|
log.Error("Unable to render: %v", err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
2021-03-16 07:20:05 +08:00
|
|
|
}
|
|
|
|
|
2021-11-19 18:46:47 +08:00
|
|
|
// Note: The output of this method must get sanitized.
|
2021-04-20 06:25:08 +08:00
|
|
|
func render(ctx *markup.RenderContext, input io.Reader, output io.Writer) error {
|
2021-03-16 07:20:05 +08:00
|
|
|
defer func() {
|
|
|
|
err := recover()
|
|
|
|
if err == nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-11-19 18:46:47 +08:00
|
|
|
log.Warn("Unable to render markdown due to panic in goldmark - will return raw bytes")
|
2021-03-16 07:20:05 +08:00
|
|
|
if log.IsDebug() {
|
|
|
|
log.Debug("Panic in markdown: %v\n%s", err, string(log.Stack(2)))
|
|
|
|
}
|
2021-11-19 18:46:47 +08:00
|
|
|
_, err = io.Copy(output, input)
|
2021-04-20 06:25:08 +08:00
|
|
|
if err != nil {
|
2021-11-19 18:46:47 +08:00
|
|
|
log.Error("io.Copy failed: %v", err)
|
2021-04-20 06:25:08 +08:00
|
|
|
}
|
2021-03-16 07:20:05 +08:00
|
|
|
}()
|
2021-04-20 06:25:08 +08:00
|
|
|
return actualRender(ctx, input, output)
|
2014-05-06 01:08:01 +08:00
|
|
|
}
|
|
|
|
|
2022-01-21 01:46:10 +08:00
|
|
|
// MarkupName describes markup's name
|
|
|
|
var MarkupName = "markdown"
|
2017-04-21 15:01:08 +08:00
|
|
|
|
|
|
|
func init() {
|
2021-04-20 06:25:08 +08:00
|
|
|
markup.RegisterRenderer(Renderer{})
|
2017-04-21 15:01:08 +08:00
|
|
|
}
|
|
|
|
|
2021-04-20 06:25:08 +08:00
|
|
|
// Renderer implements markup.Renderer
|
|
|
|
type Renderer struct{}
|
2017-04-21 15:01:08 +08:00
|
|
|
|
2021-04-20 06:25:08 +08:00
|
|
|
// Name implements markup.Renderer
|
|
|
|
func (Renderer) Name() string {
|
2017-04-21 15:01:08 +08:00
|
|
|
return MarkupName
|
|
|
|
}
|
|
|
|
|
2021-04-20 06:25:08 +08:00
|
|
|
// NeedPostProcess implements markup.Renderer
|
|
|
|
func (Renderer) NeedPostProcess() bool { return true }
|
2021-04-13 15:06:31 +08:00
|
|
|
|
2021-04-20 06:25:08 +08:00
|
|
|
// Extensions implements markup.Renderer
|
|
|
|
func (Renderer) Extensions() []string {
|
2017-04-21 15:01:08 +08:00
|
|
|
return setting.Markdown.FileExtensions
|
|
|
|
}
|
|
|
|
|
2021-06-24 05:09:51 +08:00
|
|
|
// SanitizerRules implements markup.Renderer
|
|
|
|
func (Renderer) SanitizerRules() []setting.MarkupSanitizerRule {
|
|
|
|
return []setting.MarkupSanitizerRule{}
|
|
|
|
}
|
|
|
|
|
2021-04-20 06:25:08 +08:00
|
|
|
// Render implements markup.Renderer
|
|
|
|
func (Renderer) Render(ctx *markup.RenderContext, input io.Reader, output io.Writer) error {
|
|
|
|
return render(ctx, input, output)
|
2017-09-17 01:17:57 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Render renders Markdown to HTML with all specific handling stuff.
|
2021-04-20 06:25:08 +08:00
|
|
|
func Render(ctx *markup.RenderContext, input io.Reader, output io.Writer) error {
|
2021-11-19 18:46:47 +08:00
|
|
|
if ctx.Type == "" {
|
|
|
|
ctx.Type = MarkupName
|
2021-04-20 06:25:08 +08:00
|
|
|
}
|
|
|
|
return markup.Render(ctx, input, output)
|
2017-09-17 01:17:57 +08:00
|
|
|
}
|
|
|
|
|
2021-04-20 06:25:08 +08:00
|
|
|
// RenderString renders Markdown string to HTML with all specific handling stuff and return string
|
|
|
|
func RenderString(ctx *markup.RenderContext, content string) (string, error) {
|
|
|
|
var buf strings.Builder
|
|
|
|
if err := Render(ctx, strings.NewReader(content), &buf); err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
return buf.String(), nil
|
2020-05-24 16:14:26 +08:00
|
|
|
}
|
|
|
|
|
2021-04-20 06:25:08 +08:00
|
|
|
// RenderRaw renders Markdown to HTML without handling special links.
|
|
|
|
func RenderRaw(ctx *markup.RenderContext, input io.Reader, output io.Writer) error {
|
2021-11-19 18:46:47 +08:00
|
|
|
rd, wr := io.Pipe()
|
|
|
|
defer func() {
|
|
|
|
_ = rd.Close()
|
|
|
|
_ = wr.Close()
|
|
|
|
}()
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
if err := render(ctx, input, wr); err != nil {
|
|
|
|
_ = wr.CloseWithError(err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
_ = wr.Close()
|
|
|
|
}()
|
|
|
|
|
|
|
|
return markup.SanitizeReader(rd, "", output)
|
2017-09-17 01:17:57 +08:00
|
|
|
}
|
|
|
|
|
2021-04-20 06:25:08 +08:00
|
|
|
// RenderRawString renders Markdown to HTML without handling special links and return string
|
|
|
|
func RenderRawString(ctx *markup.RenderContext, content string) (string, error) {
|
|
|
|
var buf strings.Builder
|
|
|
|
if err := RenderRaw(ctx, strings.NewReader(content), &buf); err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
return buf.String(), nil
|
2017-09-17 01:17:57 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// IsMarkdownFile reports whether name looks like a Markdown file
|
|
|
|
// based on its extension.
|
|
|
|
func IsMarkdownFile(name string) bool {
|
|
|
|
return markup.IsMarkupFile(name, MarkupName)
|
2017-04-21 15:01:08 +08:00
|
|
|
}
|