90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
package files
|
|
|
|
import (
|
|
"path/filepath"
|
|
|
|
"github.com/gtsteffaniak/filebrowser/backend/settings"
|
|
)
|
|
|
|
// UpdateFileMetadata updates the FileInfo for the specified directory in the index.
|
|
func (si *Index) UpdateMetadata(info *FileInfo) bool {
|
|
si.mu.Lock()
|
|
defer si.mu.Unlock()
|
|
si.Directories[info.Path] = info
|
|
return true
|
|
}
|
|
|
|
// GetMetadataInfo retrieves the FileInfo from the specified directory in the index.
|
|
func (si *Index) GetReducedMetadata(target string, isDir bool) (*FileInfo, bool) {
|
|
si.mu.Lock()
|
|
defer si.mu.Unlock()
|
|
checkDir := si.makeIndexPath(target)
|
|
if !isDir {
|
|
checkDir = si.makeIndexPath(filepath.Dir(target))
|
|
}
|
|
dir, exists := si.Directories[checkDir]
|
|
if !exists {
|
|
return nil, false
|
|
}
|
|
|
|
if isDir {
|
|
return dir, true
|
|
}
|
|
// handle file
|
|
if checkDir == "/" {
|
|
checkDir = ""
|
|
}
|
|
baseName := filepath.Base(target)
|
|
for _, item := range dir.Files {
|
|
if item.Name == baseName {
|
|
return &FileInfo{
|
|
Path: checkDir + "/" + item.Name,
|
|
ItemInfo: item,
|
|
}, true
|
|
}
|
|
}
|
|
return nil, false
|
|
|
|
}
|
|
|
|
// GetMetadataInfo retrieves the FileInfo from the specified directory in the index.
|
|
func (si *Index) GetMetadataInfo(target string, isDir bool) (*FileInfo, bool) {
|
|
si.mu.RLock()
|
|
defer si.mu.RUnlock()
|
|
checkDir := si.makeIndexPath(target)
|
|
if !isDir {
|
|
checkDir = si.makeIndexPath(filepath.Dir(target))
|
|
}
|
|
dir, exists := si.Directories[checkDir]
|
|
return dir, exists
|
|
}
|
|
|
|
func (si *Index) RemoveDirectory(path string) {
|
|
si.mu.Lock()
|
|
defer si.mu.Unlock()
|
|
si.NumDeleted++
|
|
delete(si.Directories, path)
|
|
}
|
|
|
|
func GetIndex(root string) *Index {
|
|
for _, index := range indexes {
|
|
if index.Root == root {
|
|
return index
|
|
}
|
|
}
|
|
if settings.Config.Server.Root != "" {
|
|
rootPath = settings.Config.Server.Root
|
|
}
|
|
newIndex := &Index{
|
|
Root: rootPath,
|
|
Directories: map[string]*FileInfo{},
|
|
NumDirs: 0,
|
|
NumFiles: 0,
|
|
}
|
|
newIndex.Directories["/"] = &FileInfo{}
|
|
indexesMutex.Lock()
|
|
indexes = append(indexes, newIndex)
|
|
indexesMutex.Unlock()
|
|
return newIndex
|
|
}
|