hugo/tpl/urls/urls.go

188 lines
4.7 KiB
Go
Raw Normal View History

// Copyright 2017 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 urls provides template functions to deal with URLs.
package urls
import (
"errors"
"fmt"
"html/template"
"net/url"
"github.com/gohugoio/hugo/common/urls"
"github.com/gohugoio/hugo/deps"
"github.com/spf13/cast"
)
// New returns a new instance of the urls-namespaced template functions.
func New(deps *deps.Deps) *Namespace {
return &Namespace{
deps: deps,
multihost: deps.Cfg.GetBool("multihost"),
}
}
// Namespace provides template functions for the "urls" namespace.
type Namespace struct {
deps *deps.Deps
multihost bool
}
2022-04-30 16:12:08 +00:00
// AbsURL takes the string s and converts it to an absolute URL.
func (ns *Namespace) AbsURL(s any) (template.HTML, error) {
ss, err := cast.ToStringE(s)
if err != nil {
return "", nil
}
2022-04-30 16:12:08 +00:00
return template.HTML(ns.deps.PathSpec.AbsURL(ss, false)), nil
}
// Parse parses rawurl into a URL structure. The rawurl may be relative or
// absolute.
func (ns *Namespace) Parse(rawurl any) (*url.URL, error) {
s, err := cast.ToStringE(rawurl)
if err != nil {
return nil, fmt.Errorf("Error in Parse: %w", err)
}
return url.Parse(s)
}
2022-04-30 16:12:08 +00:00
// RelURL takes the string s and prepends the relative path according to a
// page's position in the project directory structure.
2022-04-30 16:12:08 +00:00
func (ns *Namespace) RelURL(s any) (template.HTML, error) {
ss, err := cast.ToStringE(s)
if err != nil {
return "", nil
}
2022-04-30 16:12:08 +00:00
return template.HTML(ns.deps.PathSpec.RelURL(ss, false)), nil
}
2022-04-30 16:12:08 +00:00
// URLize returns the the strings s formatted as an URL.
func (ns *Namespace) URLize(s any) (string, error) {
ss, err := cast.ToStringE(s)
if err != nil {
return "", nil
}
2022-04-30 16:12:08 +00:00
return ns.deps.PathSpec.URLize(ss), nil
}
2022-04-30 16:12:08 +00:00
// Anchorize creates sanitized anchor name version of the string s that is compatible
// with how your configured markdown renderer does it.
func (ns *Namespace) Anchorize(s any) (string, error) {
ss, err := cast.ToStringE(s)
2018-04-15 21:17:50 +00:00
if err != nil {
return "", nil
}
2022-04-30 16:12:08 +00:00
return ns.deps.ContentSpec.SanitizeAnchorName(ss), nil
2018-04-15 21:17:50 +00:00
}
2022-04-30 16:12:08 +00:00
// Ref returns the absolute URL path to a given content item from Page p.
func (ns *Namespace) Ref(p any, args any) (template.HTML, error) {
pp, ok := p.(urls.RefLinker)
if !ok {
return "", errors.New("invalid Page received in Ref")
}
argsm, err := ns.refArgsToMap(args)
if err != nil {
return "", err
}
2022-04-30 16:12:08 +00:00
s, err := pp.Ref(argsm)
return template.HTML(s), err
}
2022-04-30 16:12:08 +00:00
// RelRef returns the relative URL path to a given content item from Page p.
func (ns *Namespace) RelRef(p any, args any) (template.HTML, error) {
pp, ok := p.(urls.RefLinker)
if !ok {
return "", errors.New("invalid Page received in RelRef")
}
argsm, err := ns.refArgsToMap(args)
if err != nil {
return "", err
}
2022-04-30 16:12:08 +00:00
s, err := pp.RelRef(argsm)
return template.HTML(s), err
}
func (ns *Namespace) refArgsToMap(args any) (map[string]any, error) {
var (
s string
of string
)
v := args
if _, ok := v.([]any); ok {
v = cast.ToStringSlice(v)
}
switch v := v.(type) {
case map[string]any:
return v, nil
case map[string]string:
m := make(map[string]any)
for k, v := range v {
m[k] = v
}
return m, nil
case []string:
if len(v) == 0 || len(v) > 2 {
2020-12-16 11:11:32 +00:00
return nil, fmt.Errorf("invalid number of arguments to ref")
}
// These where the options before we introduced the map type:
s = v[0]
if len(v) == 2 {
of = v[1]
}
default:
var err error
s, err = cast.ToStringE(args)
if err != nil {
return nil, err
}
}
return map[string]any{
"path": s,
"outputFormat": of,
}, nil
}
2022-04-30 16:12:08 +00:00
// RelLangURL takes the string s and prepends the relative path according to a
// page's position in the project directory structure and the current language.
2022-04-30 16:12:08 +00:00
func (ns *Namespace) RelLangURL(s any) (template.HTML, error) {
ss, err := cast.ToStringE(s)
if err != nil {
return "", err
}
2022-04-30 16:12:08 +00:00
return template.HTML(ns.deps.PathSpec.RelURL(ss, !ns.multihost)), nil
}
2022-04-30 16:12:08 +00:00
// AbsLangURL the string s and converts it to an absolute URL according
// to a page's position in the project directory structure and the current
// language.
2022-04-30 16:12:08 +00:00
func (ns *Namespace) AbsLangURL(s any) (template.HTML, error) {
ss, err := cast.ToStringE(s)
if err != nil {
return "", err
}
2022-04-30 16:12:08 +00:00
return template.HTML(ns.deps.PathSpec.AbsURL(ss, !ns.multihost)), nil
}