photoprism/internal/query/folders.go

33 lines
759 B
Go
Raw Normal View History

package query
import (
"path/filepath"
"github.com/photoprism/photoprism/internal/entity"
"github.com/photoprism/photoprism/pkg/fs"
)
type Folders []entity.Folder
// FoldersByPath returns a slice of folders in a given directory incl sub directories in recursive mode.
func FoldersByPath(root, rootPath, path string, recursive bool) (folders Folders, err error) {
dirs, err := fs.Dirs(filepath.Join(rootPath, path), recursive)
if err != nil {
return folders, err
}
folders = make(Folders, len(dirs))
for i, dir := range dirs {
folder := entity.NewFolder(root, filepath.Join(path, dir), nil)
if err := entity.FirstOrCreateFolder(&folder); err != nil {
log.Errorf("folders: %s", err)
}
folders[i] = folder
}
return folders, nil
}