github.com/linchen2chris/hugo@v0.0.0-20230307053224-cec209389705/tpl/path/init.go (about)

     1  // Copyright 2018 The Hugo Authors. All rights reserved.
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  // http://www.apache.org/licenses/LICENSE-2.0
     7  //
     8  // Unless required by applicable law or agreed to in writing, software
     9  // distributed under the License is distributed on an "AS IS" BASIS,
    10  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    11  // See the License for the specific language governing permissions and
    12  // limitations under the License.
    13  
    14  package path
    15  
    16  import (
    17  	"context"
    18  	"fmt"
    19  	"path/filepath"
    20  
    21  	"github.com/gohugoio/hugo/deps"
    22  	"github.com/gohugoio/hugo/tpl/internal"
    23  )
    24  
    25  const name = "path"
    26  
    27  func init() {
    28  	f := func(d *deps.Deps) *internal.TemplateFuncsNamespace {
    29  		ctx := New(d)
    30  
    31  		ns := &internal.TemplateFuncsNamespace{
    32  			Name:    name,
    33  			Context: func(cctx context.Context, args ...any) (any, error) { return ctx, nil },
    34  		}
    35  
    36  		ns.AddMethodMapping(ctx.Split,
    37  			nil,
    38  			[][2]string{
    39  				{`{{ "/my/path/filename.txt" | path.Split }}`, `/my/path/|filename.txt`},
    40  				{fmt.Sprintf(`{{ %q | path.Split }}`, filepath.FromSlash("/my/path/filename.txt")), `/my/path/|filename.txt`},
    41  			},
    42  		)
    43  
    44  		testDir := filepath.Join("my", "path")
    45  		testFile := filepath.Join(testDir, "filename.txt")
    46  
    47  		ns.AddMethodMapping(ctx.Join,
    48  			nil,
    49  			[][2]string{
    50  				{fmt.Sprintf(`{{ slice %q "filename.txt" | path.Join }}`, testDir), `my/path/filename.txt`},
    51  				{`{{ path.Join "my" "path" "filename.txt" }}`, `my/path/filename.txt`},
    52  				{fmt.Sprintf(`{{ %q | path.Ext }}`, testFile), `.txt`},
    53  				{fmt.Sprintf(`{{ %q | path.Base }}`, testFile), `filename.txt`},
    54  				{fmt.Sprintf(`{{ %q | path.Dir }}`, testFile), `my/path`},
    55  			},
    56  		)
    57  
    58  		return ns
    59  	}
    60  	internal.AddTemplateFuncsNamespace(f)
    61  }