diff options
Diffstat (limited to 'hugolib/shortcode_page.go')
-rw-r--r-- | hugolib/shortcode_page.go | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/hugolib/shortcode_page.go b/hugolib/shortcode_page.go new file mode 100644 index 000000000..e8a3a37e1 --- /dev/null +++ b/hugolib/shortcode_page.go @@ -0,0 +1,56 @@ +// 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 hugolib + +import ( + "html/template" + + "github.com/gohugoio/hugo/resources/page" +) + +var tocShortcodePlaceholder = createShortcodePlaceholder("TOC", 0) + +// This is sent to the shortcodes. They cannot access the content +// they're a part of. It would cause an infinite regress. +// +// Go doesn't support virtual methods, so this careful dance is currently (I think) +// the best we can do. +type pageForShortcode struct { + page.PageWithoutContent + page.ContentProvider + + // We need to replace it after we have rendered it, so provide a + // temporary placeholder. + toc template.HTML + + p *pageState +} + +func newPageForShortcode(p *pageState) page.Page { + return &pageForShortcode{ + PageWithoutContent: p, + ContentProvider: page.NopPage, + toc: template.HTML(tocShortcodePlaceholder), + p: p, + } +} + +func (p *pageForShortcode) page() page.Page { + return p.PageWithoutContent.(page.Page) +} + +func (p *pageForShortcode) TableOfContents() template.HTML { + p.p.enablePlaceholders() + return p.toc +} |