mirror of
https://github.com/gohugoio/hugo.git
synced 2025-08-30 22:39:58 +02:00
committed by
Bjørn Erik Pedersen
parent
b6a30283f0
commit
2818878994
@@ -48,6 +48,13 @@ func init() {
|
||||
},
|
||||
)
|
||||
|
||||
ns.AddMethodMapping(ctx.FileExists,
|
||||
[]string{"fileExists"},
|
||||
[][2]string{
|
||||
{`{{ fileExists "foo.txt" }}`, `false`},
|
||||
},
|
||||
)
|
||||
|
||||
return ns
|
||||
|
||||
}
|
||||
|
19
tpl/os/os.go
19
tpl/os/os.go
@@ -96,3 +96,22 @@ func (ns *Namespace) ReadDir(i interface{}) ([]_os.FileInfo, error) {
|
||||
|
||||
return list, nil
|
||||
}
|
||||
|
||||
// FileExists checks whether a file exists under the given path.
|
||||
func (ns *Namespace) FileExists(i interface{}) (bool, error) {
|
||||
path, err := cast.ToStringE(i)
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
|
||||
if path == "" {
|
||||
return false, errors.New("fileExists needs a path to a file")
|
||||
}
|
||||
|
||||
status, err := afero.Exists(ns.deps.Fs.WorkingDir, path)
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
|
||||
return status, nil
|
||||
}
|
||||
|
@@ -63,3 +63,39 @@ func TestReadFile(t *testing.T) {
|
||||
assert.Equal(t, test.expect, result, errMsg)
|
||||
}
|
||||
}
|
||||
|
||||
func TestFileExists(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
workingDir := "/home/hugo"
|
||||
|
||||
v := viper.New()
|
||||
v.Set("workingDir", workingDir)
|
||||
|
||||
ns := New(&deps.Deps{Fs: hugofs.NewMem(v)})
|
||||
|
||||
afero.WriteFile(ns.deps.Fs.Source, filepath.Join(workingDir, "/f/f1.txt"), []byte("f1-content"), 0755)
|
||||
afero.WriteFile(ns.deps.Fs.Source, filepath.Join("/home", "f2.txt"), []byte("f2-content"), 0755)
|
||||
|
||||
for i, test := range []struct {
|
||||
filename string
|
||||
expect interface{}
|
||||
}{
|
||||
{filepath.FromSlash("/f/f1.txt"), true},
|
||||
{filepath.FromSlash("f/f1.txt"), true},
|
||||
{filepath.FromSlash("../f2.txt"), false},
|
||||
{"b", false},
|
||||
{"", nil},
|
||||
} {
|
||||
errMsg := fmt.Sprintf("[%d] %v", i, test)
|
||||
result, err := ns.FileExists(test.filename)
|
||||
|
||||
if test.expect == nil {
|
||||
require.Error(t, err, errMsg)
|
||||
continue
|
||||
}
|
||||
|
||||
require.NoError(t, err, errMsg)
|
||||
assert.Equal(t, test.expect, result, errMsg)
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user