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
+}