mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-21 20:46:30 -05:00
b08d9f26ae
Currently a `[]byte` copy is returned. In most cases this is the safe thing to do, but we should just modify/grow the slice as needed. This is faster and consumes less memory: ``` benchmark old ns/op new ns/op delta BenchmarkReplaceShortcodeTokens-4 7350 4419 -39.88% benchmark old allocs new allocs delta BenchmarkReplaceShortcodeTokens-4 5 1 -80.00% benchmark old bytes new bytes delta BenchmarkReplaceShortcodeTokens-4 4816 1152 -76.08% ``` This commit is aso a small spring cleaning of duplicated code in the different `PageConvert` methods. Fixes #1516
133 lines
3.7 KiB
Go
133 lines
3.7 KiB
Go
// Copyright © 2014 Steve Francia <spf@spf13.com>.
|
|
//
|
|
// Licensed under the Simple Public 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://opensource.org/licenses/Simple-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 hugolib
|
|
|
|
import (
|
|
"github.com/spf13/hugo/helpers"
|
|
"github.com/spf13/hugo/source"
|
|
"github.com/spf13/hugo/tpl"
|
|
jww "github.com/spf13/jwalterweatherman"
|
|
)
|
|
|
|
func init() {
|
|
RegisterHandler(new(markdownHandler))
|
|
RegisterHandler(new(htmlHandler))
|
|
RegisterHandler(new(asciidocHandler))
|
|
RegisterHandler(new(rstHandler))
|
|
RegisterHandler(new(mmarkHandler))
|
|
}
|
|
|
|
type basicPageHandler Handle
|
|
|
|
func (b basicPageHandler) Read(f *source.File, s *Site) HandledResult {
|
|
page, err := NewPage(f.Path())
|
|
if err != nil {
|
|
return HandledResult{file: f, err: err}
|
|
}
|
|
|
|
if _, err := page.ReadFrom(f.Contents); err != nil {
|
|
return HandledResult{file: f, err: err}
|
|
}
|
|
|
|
page.Site = &s.Info
|
|
page.Tmpl = s.Tmpl
|
|
|
|
return HandledResult{file: f, page: page, err: err}
|
|
}
|
|
|
|
func (b basicPageHandler) FileConvert(*source.File, *Site) HandledResult {
|
|
return HandledResult{}
|
|
}
|
|
|
|
type markdownHandler struct {
|
|
basicPageHandler
|
|
}
|
|
|
|
func (h markdownHandler) Extensions() []string { return []string{"mdown", "markdown", "md"} }
|
|
func (h markdownHandler) PageConvert(p *Page, t tpl.Template) HandledResult {
|
|
return commonConvert(p, t)
|
|
}
|
|
|
|
type htmlHandler struct {
|
|
basicPageHandler
|
|
}
|
|
|
|
func (h htmlHandler) Extensions() []string { return []string{"html", "htm"} }
|
|
func (h htmlHandler) PageConvert(p *Page, t tpl.Template) HandledResult {
|
|
p.ProcessShortcodes(t)
|
|
var err error
|
|
|
|
if len(p.contentShortCodes) > 0 {
|
|
p.rawContent, err = replaceShortcodeTokens(p.rawContent, shortcodePlaceholderPrefix, p.contentShortCodes)
|
|
|
|
if err != nil {
|
|
jww.FATAL.Printf("Failed to replace short code tokens in %s:\n%s", p.BaseFileName(), err.Error())
|
|
return HandledResult{err: err}
|
|
}
|
|
}
|
|
|
|
p.Content = helpers.BytesToHTML(p.rawContent)
|
|
return HandledResult{err: nil}
|
|
}
|
|
|
|
type asciidocHandler struct {
|
|
basicPageHandler
|
|
}
|
|
|
|
func (h asciidocHandler) Extensions() []string { return []string{"asciidoc", "adoc", "ad"} }
|
|
func (h asciidocHandler) PageConvert(p *Page, t tpl.Template) HandledResult {
|
|
return commonConvert(p, t)
|
|
}
|
|
|
|
type rstHandler struct {
|
|
basicPageHandler
|
|
}
|
|
|
|
func (h rstHandler) Extensions() []string { return []string{"rest", "rst"} }
|
|
func (h rstHandler) PageConvert(p *Page, t tpl.Template) HandledResult {
|
|
return commonConvert(p, t)
|
|
}
|
|
|
|
type mmarkHandler struct {
|
|
basicPageHandler
|
|
}
|
|
|
|
func (h mmarkHandler) Extensions() []string { return []string{"mmark"} }
|
|
func (h mmarkHandler) PageConvert(p *Page, t tpl.Template) HandledResult {
|
|
return commonConvert(p, t)
|
|
}
|
|
|
|
func commonConvert(p *Page, t tpl.Template) HandledResult {
|
|
p.ProcessShortcodes(t)
|
|
|
|
var err error
|
|
|
|
renderedContent := p.renderContent(helpers.RemoveSummaryDivider(p.rawContent))
|
|
|
|
if len(p.contentShortCodes) > 0 {
|
|
renderedContent, err = replaceShortcodeTokens(renderedContent, shortcodePlaceholderPrefix, p.contentShortCodes)
|
|
|
|
if err != nil {
|
|
jww.FATAL.Printf("Failed to replace shortcode tokens in %s:\n%s", p.BaseFileName(), err.Error())
|
|
return HandledResult{err: err}
|
|
}
|
|
}
|
|
|
|
tmpContent, tmpTableOfContents := helpers.ExtractTOC(renderedContent)
|
|
|
|
p.Content = helpers.BytesToHTML(tmpContent)
|
|
p.TableOfContents = helpers.BytesToHTML(tmpTableOfContents)
|
|
|
|
return HandledResult{err: nil}
|
|
}
|