133 lines
3.6 KiB
Go
133 lines
3.6 KiB
Go
package scanner
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/navidrome/navidrome/consts"
|
|
"github.com/navidrome/navidrome/log"
|
|
"github.com/navidrome/navidrome/utils"
|
|
)
|
|
|
|
type (
|
|
dirStats struct {
|
|
Path string
|
|
ModTime time.Time
|
|
HasImages bool
|
|
HasPlaylist bool
|
|
AudioFilesCount uint32
|
|
}
|
|
walkResults = chan dirStats
|
|
)
|
|
|
|
func walkDirTree(ctx context.Context, rootFolder string, results walkResults) error {
|
|
err := walkFolder(ctx, rootFolder, rootFolder, results)
|
|
if err != nil {
|
|
log.Error(ctx, "Error loading directory tree", err)
|
|
}
|
|
close(results)
|
|
return err
|
|
}
|
|
|
|
func walkFolder(ctx context.Context, rootPath string, currentFolder string, results walkResults) error {
|
|
children, stats, err := loadDir(ctx, currentFolder)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, c := range children {
|
|
err := walkFolder(ctx, rootPath, c, results)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
dir := filepath.Clean(currentFolder)
|
|
log.Trace(ctx, "Found directory", "dir", dir, "audioCount", stats.AudioFilesCount,
|
|
"hasImages", stats.HasImages, "hasPlaylist", stats.HasPlaylist)
|
|
stats.Path = dir
|
|
results <- stats
|
|
|
|
return nil
|
|
}
|
|
|
|
func loadDir(ctx context.Context, dirPath string) (children []string, stats dirStats, err error) {
|
|
dirInfo, err := os.Stat(dirPath)
|
|
if err != nil {
|
|
log.Error(ctx, "Error stating dir", "path", dirPath, err)
|
|
return
|
|
}
|
|
stats.ModTime = dirInfo.ModTime()
|
|
|
|
files, err := ioutil.ReadDir(dirPath)
|
|
if err != nil {
|
|
log.Error(ctx, "Error reading dir", "path", dirPath, err)
|
|
return
|
|
}
|
|
for _, f := range files {
|
|
isDir, err := isDirOrSymlinkToDir(dirPath, f)
|
|
// Skip invalid symlinks
|
|
if err != nil {
|
|
log.Error(ctx, "Invalid symlink", "dir", dirPath)
|
|
continue
|
|
}
|
|
if isDir && !isDirIgnored(dirPath, f) && isDirReadable(dirPath, f) {
|
|
children = append(children, filepath.Join(dirPath, f.Name()))
|
|
} else {
|
|
if f.ModTime().After(stats.ModTime) {
|
|
stats.ModTime = f.ModTime()
|
|
}
|
|
if utils.IsAudioFile(f.Name()) {
|
|
stats.AudioFilesCount++
|
|
} else {
|
|
stats.HasPlaylist = stats.HasPlaylist || utils.IsPlaylist(f.Name())
|
|
stats.HasImages = stats.HasImages || utils.IsImageFile(f.Name())
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// isDirOrSymlinkToDir returns true if and only if the dirInfo represents a file
|
|
// system directory, or a symbolic link to a directory. Note that if the dirInfo
|
|
// is not a directory but is a symbolic link, this method will resolve by
|
|
// sending a request to the operating system to follow the symbolic link.
|
|
// Copied from github.com/karrick/godirwalk
|
|
func isDirOrSymlinkToDir(baseDir string, dirInfo os.FileInfo) (bool, error) {
|
|
if dirInfo.IsDir() {
|
|
return true, nil
|
|
}
|
|
if dirInfo.Mode()&os.ModeSymlink == 0 {
|
|
return false, nil
|
|
}
|
|
// Does this symlink point to a directory?
|
|
dirInfo, err := os.Stat(filepath.Join(baseDir, dirInfo.Name()))
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return dirInfo.IsDir(), nil
|
|
}
|
|
|
|
// isDirIgnored returns true if the directory represented by dirInfo contains an
|
|
// `ignore` file (named after consts.SkipScanFile)
|
|
func isDirIgnored(baseDir string, dirInfo os.FileInfo) bool {
|
|
if strings.HasPrefix(dirInfo.Name(), ".") {
|
|
return true
|
|
}
|
|
_, err := os.Stat(filepath.Join(baseDir, dirInfo.Name(), consts.SkipScanFile))
|
|
return err == nil
|
|
}
|
|
|
|
// isDirReadable returns true if the directory represented by dirInfo is readable
|
|
func isDirReadable(baseDir string, dirInfo os.FileInfo) bool {
|
|
path := filepath.Join(baseDir, dirInfo.Name())
|
|
res, err := utils.IsDirReadable(path)
|
|
if !res {
|
|
log.Debug("Warning: Skipping unreadable directory", "path", path, err)
|
|
}
|
|
return res
|
|
}
|