aboutsummaryrefslogtreecommitdiffhomepage
path: root/markup/mmark/convert.go
blob: b4cc8b0961833ca1ffc826400733b925c347ede6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
// Copyright 2019 The Hugo Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Package mmark converts Markdown to HTML using MMark v1.
package mmark

import (
	"github.com/gohugoio/hugo/identity"
	"github.com/gohugoio/hugo/markup/blackfriday/blackfriday_config"
	"github.com/gohugoio/hugo/markup/converter"
	"github.com/miekg/mmark"
)

// Provider is the package entry point.
var Provider converter.ProviderProvider = provider{}

type provider struct {
}

func (p provider) New(cfg converter.ProviderConfig) (converter.Provider, error) {
	defaultBlackFriday := cfg.MarkupConfig.BlackFriday
	defaultExtensions := getMmarkExtensions(defaultBlackFriday)

	return converter.NewProvider("mmark", func(ctx converter.DocumentContext) (converter.Converter, error) {
		b := defaultBlackFriday
		extensions := defaultExtensions

		if ctx.ConfigOverrides != nil {
			var err error
			b, err = blackfriday_config.UpdateConfig(b, ctx.ConfigOverrides)
			if err != nil {
				return nil, err
			}
			extensions = getMmarkExtensions(b)
		}

		return &mmarkConverter{
			ctx:        ctx,
			b:          b,
			extensions: extensions,
			cfg:        cfg,
		}, nil
	}), nil
}

type mmarkConverter struct {
	ctx        converter.DocumentContext
	extensions int
	b          blackfriday_config.Config
	cfg        converter.ProviderConfig
}

func (c *mmarkConverter) Convert(ctx converter.RenderContext) (converter.Result, error) {
	r := getHTMLRenderer(c.ctx, c.b, c.cfg)
	return mmark.Parse(ctx.Src, r, c.extensions), nil
}

func (c *mmarkConverter) Supports(feature identity.Identity) bool {
	return false
}

func getHTMLRenderer(
	ctx converter.DocumentContext,
	cfg blackfriday_config.Config,
	pcfg converter.ProviderConfig) mmark.Renderer {
	var (
		flags      int
		documentID string
	)

	documentID = ctx.DocumentID

	renderParameters := mmark.HtmlRendererParameters{
		FootnoteAnchorPrefix:       cfg.FootnoteAnchorPrefix,
		FootnoteReturnLinkContents: cfg.FootnoteReturnLinkContents,
	}

	if documentID != "" && !cfg.PlainIDAnchors {
		renderParameters.FootnoteAnchorPrefix = documentID + ":" + renderParameters.FootnoteAnchorPrefix
	}

	htmlFlags := flags
	htmlFlags |= mmark.HTML_FOOTNOTE_RETURN_LINKS

	return &mmarkRenderer{
		BlackfridayConfig: cfg,
		Config:            pcfg,
		Renderer:          mmark.HtmlRendererWithParameters(htmlFlags, "", "", renderParameters),
	}
}

func getMmarkExtensions(cfg blackfriday_config.Config) int {
	flags := 0
	flags |= mmark.EXTENSION_TABLES
	flags |= mmark.EXTENSION_FENCED_CODE
	flags |= mmark.EXTENSION_AUTOLINK
	flags |= mmark.EXTENSION_SPACE_HEADERS
	flags |= mmark.EXTENSION_CITATION
	flags |= mmark.EXTENSION_TITLEBLOCK_TOML
	flags |= mmark.EXTENSION_HEADER_IDS
	flags |= mmark.EXTENSION_AUTO_HEADER_IDS
	flags |= mmark.EXTENSION_UNIQUE_HEADER_IDS
	flags |= mmark.EXTENSION_FOOTNOTES
	flags |= mmark.EXTENSION_SHORT_REF
	flags |= mmark.EXTENSION_NO_EMPTY_LINE_BEFORE_BLOCK
	flags |= mmark.EXTENSION_INCLUDE

	for _, extension := range cfg.Extensions {
		if flag, ok := mmarkExtensionMap[extension]; ok {
			flags |= flag
		}
	}
	return flags
}

var mmarkExtensionMap = map[string]int{
	"tables":                 mmark.EXTENSION_TABLES,
	"fencedCode":             mmark.EXTENSION_FENCED_CODE,
	"autolink":               mmark.EXTENSION_AUTOLINK,
	"laxHtmlBlocks":          mmark.EXTENSION_LAX_HTML_BLOCKS,
	"spaceHeaders":           mmark.EXTENSION_SPACE_HEADERS,
	"hardLineBreak":          mmark.EXTENSION_HARD_LINE_BREAK,
	"footnotes":              mmark.EXTENSION_FOOTNOTES,
	"noEmptyLineBeforeBlock": mmark.EXTENSION_NO_EMPTY_LINE_BEFORE_BLOCK,
	"headerIds":              mmark.EXTENSION_HEADER_IDS,
	"autoHeaderIds":          mmark.EXTENSION_AUTO_HEADER_IDS,
}