50 lines
912 B
Go
50 lines
912 B
Go
package localFileStructure
|
|
|
|
import (
|
|
"github.com/sirupsen/logrus"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
func ScanLocalFileStructure(path string) (map[string]*FilesystemNode, error) {
|
|
fileMap := make(map[string]*FilesystemNode)
|
|
|
|
relativeRoot := filepath.Base(path) + "/"
|
|
|
|
numberOfDirectories := 0
|
|
numberOfImages := 0
|
|
|
|
err := filepath.Walk(path, func(p string, info os.FileInfo, err error) error {
|
|
if path == p {
|
|
return nil
|
|
}
|
|
|
|
//TODO: Only allow jpg and png files here
|
|
|
|
key := strings.Replace(p, relativeRoot, "", 1)
|
|
|
|
fileMap[p] = &FilesystemNode{
|
|
Key: key,
|
|
Name: info.Name(),
|
|
IsDir: info.IsDir(),
|
|
}
|
|
|
|
if info.IsDir() {
|
|
numberOfDirectories += 1
|
|
} else {
|
|
numberOfImages += 1
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
logrus.Infof("Found %d directories and %d images on the local filesystem", numberOfDirectories, numberOfImages)
|
|
|
|
return fileMap, nil
|
|
}
|