diff options
Diffstat (limited to 'hugolib/page__ref.go')
-rw-r--r-- | hugolib/page__ref.go | 117 |
1 files changed, 117 insertions, 0 deletions
diff --git a/hugolib/page__ref.go b/hugolib/page__ref.go new file mode 100644 index 000000000..41bd527db --- /dev/null +++ b/hugolib/page__ref.go @@ -0,0 +1,117 @@ +// 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 ( + "fmt" + + "github.com/gohugoio/hugo/common/text" + + "github.com/mitchellh/mapstructure" + "github.com/pkg/errors" +) + +func newPageRef(p *pageState) pageRef { + return pageRef{p: p} +} + +type pageRef struct { + p *pageState +} + +func (p pageRef) Ref(argsm map[string]interface{}) (string, error) { + return p.ref(argsm, p.p) +} + +func (p pageRef) RefFrom(argsm map[string]interface{}, source interface{}) (string, error) { + return p.ref(argsm, source) +} + +func (p pageRef) RelRef(argsm map[string]interface{}) (string, error) { + return p.relRef(argsm, p.p) +} + +func (p pageRef) RelRefFrom(argsm map[string]interface{}, source interface{}) (string, error) { + return p.relRef(argsm, source) +} + +func (p pageRef) decodeRefArgs(args map[string]interface{}) (refArgs, *Site, error) { + var ra refArgs + err := mapstructure.WeakDecode(args, &ra) + if err != nil { + return ra, nil, nil + } + + s := p.p.s + + if ra.Lang != "" && ra.Lang != p.p.s.Language().Lang { + // Find correct site + found := false + for _, ss := range p.p.s.h.Sites { + if ss.Lang() == ra.Lang { + found = true + s = ss + } + } + + if !found { + p.p.s.siteRefLinker.logNotFound(ra.Path, fmt.Sprintf("no site found with lang %q", ra.Lang), nil, text.Position{}) + return ra, nil, nil + } + } + + return ra, s, nil +} + +func (p pageRef) ref(argsm map[string]interface{}, source interface{}) (string, error) { + args, s, err := p.decodeRefArgs(argsm) + if err != nil { + return "", errors.Wrap(err, "invalid arguments to Ref") + } + + if s == nil { + return p.p.s.siteRefLinker.notFoundURL, nil + } + + if args.Path == "" { + return "", nil + } + + return s.refLink(args.Path, source, false, args.OutputFormat) + +} + +func (p pageRef) relRef(argsm map[string]interface{}, source interface{}) (string, error) { + args, s, err := p.decodeRefArgs(argsm) + if err != nil { + return "", errors.Wrap(err, "invalid arguments to Ref") + } + + if s == nil { + return p.p.s.siteRefLinker.notFoundURL, nil + } + + if args.Path == "" { + return "", nil + } + + return s.refLink(args.Path, source, true, args.OutputFormat) + +} + +type refArgs struct { + Path string + Lang string + OutputFormat string +} |