Fix single mount rename panic

Fixes #12141
This commit is contained in:
Bjørn Erik Pedersen 2024-02-28 09:12:41 +01:00
parent 6bc0d745a5
commit a322282e70
6 changed files with 119 additions and 97 deletions

View file

@ -73,7 +73,9 @@ type FileMeta struct {
InclusionFilter *glob.FilenameFilter InclusionFilter *glob.FilenameFilter
// Rename the name part of the file (not the directory). // Rename the name part of the file (not the directory).
Rename func(name string, toFrom bool) string // Returns the new name and a boolean indicating if the file
// should be included.
Rename func(name string, toFrom bool) (string, bool)
} }
func (m *FileMeta) Copy() *FileMeta { func (m *FileMeta) Copy() *FileMeta {

View file

@ -1,4 +1,4 @@
// Copyright 2019 The Hugo Authors. All rights reserved. // Copyright 2024 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -80,42 +80,36 @@ func NewRootMappingFs(fs afero.Fs, rms ...RootMapping) (*RootMappingFs, error) {
if !fi.IsDir() { if !fi.IsDir() {
// We do allow single file mounts. // We do allow single file mounts.
// However, the file system logic will be much simpler with just directories. // However, the file system logic will be much simpler with just directories.
// So, convert this mount into a directory mount with a nameTo filter and renamer. // So, convert this mount into a directory mount with a renamer,
// which will tell the caller if name should be included.
dirFrom, nameFrom := filepath.Split(rm.From) dirFrom, nameFrom := filepath.Split(rm.From)
dirTo, nameTo := filepath.Split(rm.To) dirTo, nameTo := filepath.Split(rm.To)
dirFrom, dirTo = strings.TrimSuffix(dirFrom, filepathSeparator), strings.TrimSuffix(dirTo, filepathSeparator) dirFrom, dirTo = strings.TrimSuffix(dirFrom, filepathSeparator), strings.TrimSuffix(dirTo, filepathSeparator)
rm.From = dirFrom rm.From = dirFrom
singleFileMeta := rm.Meta.Copy()
fi, err = fs.Stat(rm.To) singleFileMeta.Name = nameFrom
if err != nil { rm.fiSingleFile = NewFileMetaInfo(fi, singleFileMeta)
if herrors.IsNotExist(err) {
continue
}
return nil, err
}
rm.fiSingleFile = NewFileMetaInfo(fi, rm.Meta.Copy())
rm.To = dirTo rm.To = dirTo
rm.Meta.Rename = func(name string, toFrom bool) string { rm.Meta.Rename = func(name string, toFrom bool) (string, bool) {
if toFrom { if toFrom {
if name == nameTo { if name == nameTo {
return nameFrom return nameFrom, true
} }
return name return "", false
} }
if name == nameFrom { if name == nameFrom {
return nameTo return nameTo, true
} }
return name return "", false
} }
nameToFilename := filepathSeparator + nameTo nameToFilename := filepathSeparator + nameTo
rm.Meta.InclusionFilter = rm.Meta.InclusionFilter.Append(glob.NewFilenameFilterForInclusionFunc( rm.Meta.InclusionFilter = rm.Meta.InclusionFilter.Append(glob.NewFilenameFilterForInclusionFunc(
func(filename string) bool { func(filename string) bool {
return strings.HasPrefix(nameToFilename, filename) return nameToFilename == filename
}, },
)) ))
@ -252,13 +246,11 @@ func (fs *RootMappingFs) Mounts(base string) ([]FileMetaInfo, error) {
fss := make([]FileMetaInfo, len(roots)) fss := make([]FileMetaInfo, len(roots))
for i, r := range roots { for i, r := range roots {
if r.fiSingleFile != nil { if r.fiSingleFile != nil {
// A single file mount. // A single file mount.
fss[i] = r.fiSingleFile fss[i] = r.fiSingleFile
continue continue
} }
bfs := NewBasePathFs(fs.Fs, r.To) bfs := NewBasePathFs(fs.Fs, r.To)
fs := bfs fs := bfs
if r.Meta.InclusionFilter != nil { if r.Meta.InclusionFilter != nil {
@ -269,11 +261,6 @@ func (fs *RootMappingFs) Mounts(base string) ([]FileMetaInfo, error) {
if err != nil { if err != nil {
return nil, fmt.Errorf("RootMappingFs.Dirs: %w", err) return nil, fmt.Errorf("RootMappingFs.Dirs: %w", err)
} }
if !fi.IsDir() {
fi.(FileMetaInfo).Meta().Merge(r.Meta)
}
fss[i] = fi.(FileMetaInfo) fss[i] = fi.(FileMetaInfo)
} }
@ -365,12 +352,18 @@ func (fs *RootMappingFs) ReverseLookupComponent(component, filename string, chec
if component != "" && first.FromBase != component { if component != "" && first.FromBase != component {
continue continue
} }
if first.Meta.Rename != nil {
name = first.Meta.Rename(name, true)
}
var filename string
if first.Meta.Rename != nil {
// Single file mount.
if newname, ok := first.Meta.Rename(name, true); ok {
filename = filepathSeparator + filepath.Join(first.path, dir, newname)
} else {
continue
}
} else {
// Now we know that this file _could_ be in this fs. // Now we know that this file _could_ be in this fs.
filename := filepathSeparator + filepath.Join(first.path, dir, name) filename = filepathSeparator + filepath.Join(first.path, dir, name)
if checkExists { if checkExists {
// Confirm that it exists. // Confirm that it exists.
@ -379,6 +372,7 @@ func (fs *RootMappingFs) ReverseLookupComponent(component, filename string, chec
continue continue
} }
} }
}
cps = append(cps, ComponentPath{ cps = append(cps, ComponentPath{
Component: first.FromBase, Component: first.FromBase,
@ -570,9 +564,11 @@ func (rfs *RootMappingFs) collectDirEntries(prefix string) ([]iofs.DirEntry, err
} }
fi = newDirNameOnlyFileInfo(name, meta, opener) fi = newDirNameOnlyFileInfo(name, meta, opener)
} else if rm.Meta.Rename != nil { } else if rm.Meta.Rename != nil {
if n := rm.Meta.Rename(fi.Name(), true); n != fi.Name() { n, ok := rm.Meta.Rename(fi.Name(), true)
fi.(MetaProvider).Meta().Name = n if !ok {
continue
} }
fi.(MetaProvider).Meta().Name = n
} }
fis = append(fis, fi) fis = append(fis, fi)
} }
@ -617,25 +613,16 @@ func (rfs *RootMappingFs) collectDirEntries(prefix string) ([]iofs.DirEntry, err
rms := v.([]RootMapping) rms := v.([]RootMapping)
for _, rm := range rms { for _, rm := range rms {
if !rm.fi.IsDir() {
// A single file mount
fis = append(fis, rm.fi)
continue
}
name := filepath.Base(rm.From) name := filepath.Base(rm.From)
if seen[name] { if seen[name] {
continue continue
} }
seen[name] = true seen[name] = true
opener := func() (afero.File, error) { opener := func() (afero.File, error) {
return rfs.Open(rm.From) return rfs.Open(rm.From)
} }
fi := newDirNameOnlyFileInfo(name, rm.Meta, opener) fi := newDirNameOnlyFileInfo(name, rm.Meta, opener)
fis = append(fis, fi) fis = append(fis, fi)
} }
return false return false
@ -701,48 +688,20 @@ func (fs *RootMappingFs) doStat(name string) ([]FileMetaInfo, error) {
return nil, err return nil, err
} }
fileCount := 0
var wasFiltered bool
for _, root := range roots {
meta := root.fi.Meta()
if !meta.InclusionFilter.Match(strings.TrimPrefix(meta.Filename, meta.SourceRoot), root.fi.IsDir()) {
wasFiltered = true
continue
}
if !root.fi.IsDir() {
fileCount++
}
if fileCount > 1 {
break
}
}
if fileCount == 0 {
if wasFiltered {
return nil, os.ErrNotExist
}
// Dir only.
return []FileMetaInfo{newDirNameOnlyFileInfo(name, roots[0].Meta, fs.virtualDirOpener(name))}, nil return []FileMetaInfo{newDirNameOnlyFileInfo(name, roots[0].Meta, fs.virtualDirOpener(name))}, nil
}
if fileCount > 1 {
// Not supported by this filesystem.
return nil, fmt.Errorf("found multiple files with name %q, use .Readdir or the source filesystem directly", name)
}
return []FileMetaInfo{roots[0].fi}, nil
} }
func (fs *RootMappingFs) statRoot(root RootMapping, filename string) (FileMetaInfo, error) { func (fs *RootMappingFs) statRoot(root RootMapping, filename string) (FileMetaInfo, error) {
dir, name := filepath.Split(filename) dir, name := filepath.Split(filename)
if root.Meta.Rename != nil { if root.Meta.Rename != nil {
if n := root.Meta.Rename(name, false); n != name { n, ok := root.Meta.Rename(name, false)
if !ok {
return nil, os.ErrNotExist
}
filename = filepath.Join(dir, n) filename = filepath.Join(dir, n)
} }
}
if !root.Meta.InclusionFilter.Match(root.trimFrom(filename), root.fi.IsDir()) { if !root.Meta.InclusionFilter.Match(root.trimFrom(filename), true) {
return nil, os.ErrNotExist return nil, os.ErrNotExist
} }
@ -753,24 +712,27 @@ func (fs *RootMappingFs) statRoot(root RootMapping, filename string) (FileMetaIn
} }
var opener func() (afero.File, error) var opener func() (afero.File, error)
if fi.IsDir() { if !fi.IsDir() {
// Make sure metadata gets applied in ReadDir. // Open the file directly.
opener = fs.realDirOpener(filename, root.Meta)
} else {
if root.Meta.Rename != nil {
if n := root.Meta.Rename(fi.Name(), true); n != fi.Name() {
meta := fi.(MetaProvider).Meta()
meta.Name = n
}
}
// Opens the real file directly. // Opens the real file directly.
opener = func() (afero.File, error) { opener = func() (afero.File, error) {
return fs.Fs.Open(filename) return fs.Fs.Open(filename)
} }
} else if root.Meta.Rename != nil {
// A single file mount where we have mounted the containing directory.
n, ok := root.Meta.Rename(fi.Name(), true)
if !ok {
return nil, os.ErrNotExist
}
meta := fi.(MetaProvider).Meta()
meta.Name = n
// Opens the real file directly.
opener = func() (afero.File, error) {
return fs.Fs.Open(filename)
}
} else {
// Make sure metadata gets applied in ReadDir.
opener = fs.realDirOpener(filename, root.Meta)
} }
fim := decorateFileInfo(fi, opener, "", root.Meta) fim := decorateFileInfo(fi, opener, "", root.Meta)

View file

@ -280,6 +280,11 @@ func TestRootMappingFsMount(t *testing.T) {
c.Assert(err, qt.IsNil) c.Assert(err, qt.IsNil)
c.Assert(cps, qt.DeepEquals, []ComponentPath{ c.Assert(cps, qt.DeepEquals, []ComponentPath{
{Component: "content", Path: "singles/p1.md", Lang: "no"}, {Component: "content", Path: "singles/p1.md", Lang: "no"},
})
cps, err = rfs.ReverseLookup(filepath.FromSlash("singlefiles/sv.txt"), true)
c.Assert(err, qt.IsNil)
c.Assert(cps, qt.DeepEquals, []ComponentPath{
{Component: "content", Path: "singles/p1.md", Lang: "sv"}, {Component: "content", Path: "singles/p1.md", Lang: "sv"},
}) })

View file

@ -1,4 +1,4 @@
// Copyright 2018 The Hugo Authors. All rights reserved. // Copyright 2024 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.

View file

@ -234,6 +234,7 @@ foo
b := hugolib.Test(t, files) b := hugolib.Test(t, files)
bfs := b.H.BaseFs bfs := b.H.BaseFs
watchFilenames := bfs.WatchFilenames() watchFilenames := bfs.WatchFilenames()
// []string{"/hugo_stats.json", "/content", "/content2", "/themes/t1/layouts", "/themes/t1/layouts/_default", "/themes/t1/static"}
b.Assert(watchFilenames, qt.HasLen, 6) b.Assert(watchFilenames, qt.HasLen, 6)
} }
@ -543,6 +544,30 @@ files/f2.txt false
`) `)
} }
func TestMountIssue12141(t *testing.T) {
files := `
-- hugo.toml --
disableKinds = ["taxonomy", "term"]
[module]
[[module.mounts]]
source = "myfiles"
target = "static"
[[module.mounts]]
source = "myfiles/f1.txt"
target = "static/f2.txt"
-- myfiles/f1.txt --
f1
`
b := hugolib.Test(t, files)
fs := b.H.BaseFs.StaticFs("")
b.AssertFs(fs, `
. true
f1.txt false
f2.txt false
`)
}
func checkFileCount(fs afero.Fs, dirname string, c *qt.C, expected int) { func checkFileCount(fs afero.Fs, dirname string, c *qt.C, expected int) {
c.Helper() c.Helper()
count, names, err := countFilesAndGetFilenames(fs, dirname) count, names, err := countFilesAndGetFilenames(fs, dirname)

View file

@ -282,7 +282,7 @@ func (s *IntegrationTestBuilder) AssertPublishDir(matches ...string) {
func (s *IntegrationTestBuilder) AssertFs(fs afero.Fs, matches ...string) { func (s *IntegrationTestBuilder) AssertFs(fs afero.Fs, matches ...string) {
s.Helper() s.Helper()
var buff bytes.Buffer var buff bytes.Buffer
helpers.PrintFs(fs, "", &buff) s.Assert(s.printAndCheckFs(fs, "", &buff), qt.IsNil)
printFsLines := strings.Split(buff.String(), "\n") printFsLines := strings.Split(buff.String(), "\n")
sort.Strings(printFsLines) sort.Strings(printFsLines)
content := strings.TrimSpace((strings.Join(printFsLines, "\n"))) content := strings.TrimSpace((strings.Join(printFsLines, "\n")))
@ -305,6 +305,34 @@ func (s *IntegrationTestBuilder) AssertFs(fs afero.Fs, matches ...string) {
} }
} }
func (s *IntegrationTestBuilder) printAndCheckFs(fs afero.Fs, path string, w io.Writer) error {
if fs == nil {
return nil
}
return afero.Walk(fs, path, func(path string, info os.FileInfo, err error) error {
if err != nil {
return fmt.Errorf("error: path %q: %s", path, err)
}
path = filepath.ToSlash(path)
if path == "" {
path = "."
}
if !info.IsDir() {
f, err := fs.Open(path)
if err != nil {
return fmt.Errorf("error: path %q: %s", path, err)
}
defer f.Close()
// This will panic if the file is a directory.
var buf [1]byte
io.ReadFull(f, buf[:])
}
fmt.Fprintln(w, path, info.IsDir())
return nil
})
}
func (s *IntegrationTestBuilder) AssertFileExists(filename string, b bool) { func (s *IntegrationTestBuilder) AssertFileExists(filename string, b bool) {
checker := qt.IsNil checker := qt.IsNil
if !b { if !b {