From 020a6fbd7f6996ed84d80ba6c37fe0d8c2536806 Mon Sep 17 00:00:00 2001 From: Niklas Fasching Date: Thu, 3 Oct 2019 23:27:51 +0200 Subject: Add BaseFs to RenderingContext The org mode renderer supports including other files [1]. We don't want to allow reading of arbitrary files (go-org defaults to ioutil.ReadFile [2]) but want to make use of the FileSystem abstractions hugo provides. For starters we will allow reading from the content directory only [1]: e.g. `#+INCLUDE: ./foo.py src python` includes `foo.py` as a python source block. --- helpers/content.go | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'helpers') diff --git a/helpers/content.go b/helpers/content.go index f6576c04f..591403cb0 100644 --- a/helpers/content.go +++ b/helpers/content.go @@ -27,6 +27,7 @@ import ( "unicode/utf8" "github.com/gohugoio/hugo/common/maps" + "github.com/gohugoio/hugo/hugolib/filesystems" "github.com/niklasfasching/go-org/org" bp "github.com/gohugoio/hugo/bufferpool" @@ -34,6 +35,7 @@ import ( "github.com/miekg/mmark" "github.com/mitchellh/mapstructure" "github.com/russross/blackfriday" + "github.com/spf13/afero" jww "github.com/spf13/jwalterweatherman" "strings" @@ -466,6 +468,7 @@ func ExtractTOC(content []byte) (newcontent []byte, toc []byte) { // for a given content rendering. // By creating you must set the Config, otherwise it will panic. type RenderingContext struct { + BaseFs *filesystems.BaseFs Content []byte PageFmt string DocumentID string @@ -752,6 +755,9 @@ func getPandocContent(ctx *RenderingContext) []byte { func orgRender(ctx *RenderingContext, c ContentSpec) []byte { config := org.New() config.Log = jww.WARN + config.ReadFile = func(filename string) ([]byte, error) { + return afero.ReadFile(ctx.BaseFs.Content.Fs, filename) + } writer := org.NewHTMLWriter() writer.HighlightCodeBlock = func(source, lang string) string { highlightedSource, err := c.Highlight(source, lang, "") -- cgit v1.2.3