mirror of
https://github.com/gohugoio/hugo.git
synced 2024-06-27 21:25:07 +00:00
7ff0a8ee9f
This is preparation for #6041. For historic reasons, the code for bulding the section tree and the taxonomies were very much separate. This works, but makes it hard to extend, maintain, and possibly not so fast as it could be. This simplification also introduces 3 slightly breaking changes, which I suspect most people will be pleased about. See referenced issues: This commit also switches the radix tree dependency to a mutable implementation: github.com/armon/go-radix. Fixes #6154 Fixes #6153 Fixes #6152
130 lines
2.7 KiB
Go
130 lines
2.7 KiB
Go
// 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 (
|
|
"github.com/gohugoio/hugo/common/types"
|
|
"github.com/gohugoio/hugo/helpers"
|
|
"github.com/gohugoio/hugo/resources/page"
|
|
)
|
|
|
|
type pageTree struct {
|
|
p *pageState
|
|
}
|
|
|
|
func (pt pageTree) IsAncestor(other interface{}) (bool, error) {
|
|
if pt.p == nil {
|
|
return false, nil
|
|
}
|
|
|
|
pp, err := unwrapPage(other)
|
|
if err != nil || pp == nil {
|
|
return false, err
|
|
}
|
|
|
|
if pt.p.Kind() == page.KindPage && len(pt.p.SectionsEntries()) == len(pp.SectionsEntries()) {
|
|
// A regular page is never its section's ancestor.
|
|
return false, nil
|
|
}
|
|
|
|
return helpers.HasStringsPrefix(pp.SectionsEntries(), pt.p.SectionsEntries()), nil
|
|
}
|
|
|
|
func (pt pageTree) CurrentSection() page.Page {
|
|
p := pt.p
|
|
|
|
if p.IsHome() || p.IsSection() {
|
|
return p
|
|
}
|
|
|
|
return p.Parent()
|
|
}
|
|
|
|
func (pt pageTree) IsDescendant(other interface{}) (bool, error) {
|
|
if pt.p == nil {
|
|
return false, nil
|
|
}
|
|
pp, err := unwrapPage(other)
|
|
if err != nil || pp == nil {
|
|
return false, err
|
|
}
|
|
|
|
if pp.Kind() == page.KindPage && len(pt.p.SectionsEntries()) == len(pp.SectionsEntries()) {
|
|
// A regular page is never its section's descendant.
|
|
return false, nil
|
|
}
|
|
return helpers.HasStringsPrefix(pt.p.SectionsEntries(), pp.SectionsEntries()), nil
|
|
}
|
|
|
|
func (pt pageTree) FirstSection() page.Page {
|
|
p := pt.p
|
|
|
|
parent := p.Parent()
|
|
|
|
if types.IsNil(parent) || parent.IsHome() {
|
|
return p
|
|
}
|
|
|
|
for {
|
|
current := parent
|
|
parent = parent.Parent()
|
|
if types.IsNil(parent) || parent.IsHome() {
|
|
return current
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
func (pt pageTree) InSection(other interface{}) (bool, error) {
|
|
if pt.p == nil || types.IsNil(other) {
|
|
return false, nil
|
|
}
|
|
|
|
pp, err := unwrapPage(other)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
if pp == nil {
|
|
return false, nil
|
|
}
|
|
|
|
return pp.CurrentSection().Eq(pt.p.CurrentSection()), nil
|
|
|
|
}
|
|
|
|
func (pt pageTree) Page() page.Page {
|
|
return pt.p
|
|
}
|
|
|
|
func (pt pageTree) Parent() page.Page {
|
|
if pt.p.parent != nil {
|
|
return pt.p.parent
|
|
}
|
|
|
|
if pt.p.bucket == nil || pt.p.bucket.parent == nil {
|
|
return nil
|
|
}
|
|
|
|
return pt.p.bucket.parent.owner
|
|
}
|
|
|
|
func (pt pageTree) Sections() page.Pages {
|
|
if pt.p.bucket == nil {
|
|
return nil
|
|
}
|
|
|
|
return pt.p.bucket.getSections()
|
|
}
|