heimoshuiyu/gonic
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add support for adding mockfs item by glob pattern

Browse Source
This commit is contained in:
sentriz
2023-08-23 18:51:44 +01:00
parent 623d5c3709
commit 77fa858531
1 changed files with 27 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
mockfs/mockfs.go
View File

@@ -98,20 +98,26 @@ func (m *MockFS) ResetDates() {
}
}
func (m *MockFS) AddItems() { m.addItems("", false) }
func (m *MockFS) AddItemsPrefix(prefix string) { m.addItems(prefix, false) }
func (m *MockFS) AddItemsWithCovers() { m.addItems("", true) }
func (m *MockFS) AddItemsPrefixWithCovers(prefix string) { m.addItems(prefix, true) }
func (m *MockFS) AddItems() { m.addItems("", "", false) }
func (m *MockFS) AddItemsGlob(onlyGlob string) { m.addItems("", onlyGlob, false) }
func (m *MockFS) AddItemsPrefix(prefix string) { m.addItems(prefix, "", false) }
func (m *MockFS) AddItemsWithCovers() { m.addItems("", "", true) }
func (m *MockFS) AddItemsPrefixWithCovers(prefix string) { m.addItems(prefix, "", true) }
func (m *MockFS) addItems(prefix string, covers bool) {
func (m *MockFS) addItems(prefix string, onlyGlob string, covers bool) {
p := func(format string, a ...interface{}) string {
return filepath.Join(prefix, fmt.Sprintf(format, a...))
}
for ar := 0; ar < 3; ar++ {
for al := 0; al < 3; al++ {
for tr := 0; tr < 3; tr++ {
m.AddTrack(p("artist-%d/album-%d/track-%d.flac", ar, al, tr))
m.SetTags(p("artist-%d/album-%d/track-%d.flac", ar, al, tr), func(tags *Tags) error {
path := p("artist-%d/album-%d/track-%d.flac", ar, al, tr)
if !match(onlyGlob, path) {
continue
}
m.AddTrack(path)
m.SetTags(path, func(tags *Tags) error {
tags.RawArtist = fmt.Sprintf("artist-%d", ar)
tags.RawAlbumArtist = fmt.Sprintf("artist-%d", ar)
tags.RawAlbum = fmt.Sprintf("album-%d", al)
@@ -120,7 +126,12 @@ func (m *MockFS) addItems(prefix string, covers bool) {
})
}
if covers {
m.AddCover(p("artist-%d/album-%d/cover.png", ar, al))
path := p("artist-%d/album-%d/cover.png", ar, al)
if !match(onlyGlob, path) {
continue
}
m.AddCover(path)
}
}
}
@@ -396,3 +407,11 @@ func firstInt(or int, ints ...int) int {
}
return or
}
func match(pattern, name string) bool {
if pattern == "" {
return true
}
ok, _ := filepath.Match(pattern, name)
return ok
}