mirror of https://github.com/stashapp/stash.git
49 lines
841 B
Go
49 lines
841 B
Go
package api
|
|
|
|
import (
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"golang.org/x/text/collate"
|
|
)
|
|
|
|
type dirLister []fs.DirEntry
|
|
|
|
func (s dirLister) Len() int {
|
|
return len(s)
|
|
}
|
|
|
|
func (s dirLister) Swap(i, j int) {
|
|
s[j], s[i] = s[i], s[j]
|
|
}
|
|
|
|
func (s dirLister) Bytes(i int) []byte {
|
|
return []byte(s[i].Name())
|
|
}
|
|
|
|
// listDir will return the contents of a given directory path as a string slice
|
|
func listDir(col *collate.Collator, path string) ([]string, error) {
|
|
var dirPaths []string
|
|
files, err := os.ReadDir(path)
|
|
if err != nil {
|
|
path = filepath.Dir(path)
|
|
files, err = os.ReadDir(path)
|
|
if err != nil {
|
|
return dirPaths, err
|
|
}
|
|
}
|
|
|
|
if col != nil {
|
|
col.Sort(dirLister(files))
|
|
}
|
|
|
|
for _, file := range files {
|
|
if !file.IsDir() {
|
|
continue
|
|
}
|
|
dirPaths = append(dirPaths, filepath.Join(path, file.Name()))
|
|
}
|
|
return dirPaths, nil
|
|
}
|