78 lines
1.3 KiB
Go
78 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
type Entry interface {
|
|
fmt.Stringer
|
|
stringRecursive(depth, maxDepth int, b *strings.Builder)
|
|
StringRecursive(depth int) string
|
|
|
|
Name() string
|
|
Path() string
|
|
|
|
Size() uint64
|
|
SizeModified() uint64
|
|
|
|
FileCount() uint64
|
|
FileCountModified() uint64
|
|
|
|
Entries() []Entry
|
|
Entry(path string) Entry
|
|
|
|
SetRemovalMark(mark bool)
|
|
RemovalMark() bool
|
|
RemovalStats(stats *RemovalStats)
|
|
|
|
modify(size uint64, fileCount uint64)
|
|
|
|
IsDir() bool
|
|
|
|
Parent() Entry
|
|
|
|
Scan(ch chan<- string, mounts map[string]struct{})
|
|
}
|
|
|
|
func NewEntryFromDirEntry(parentEntry Entry, dirEntry fs.DirEntry) Entry {
|
|
if dirEntry.IsDir() {
|
|
return &DirectoryEntry{
|
|
path: filepath.Join(parentEntry.Path(), dirEntry.Name()),
|
|
parent: parentEntry,
|
|
}
|
|
} else {
|
|
size := uint64(0)
|
|
noPerm := false
|
|
if fi, err := dirEntry.Info(); err == nil {
|
|
size = uint64(fi.Size())
|
|
} else {
|
|
noPerm = true
|
|
}
|
|
return &FileEntry{
|
|
path: filepath.Join(parentEntry.Path(), dirEntry.Name()),
|
|
size: &size,
|
|
noPerm: noPerm,
|
|
parent: parentEntry,
|
|
}
|
|
}
|
|
}
|
|
|
|
func NewEntry(path string) (Entry, error) {
|
|
path = filepath.Clean(path)
|
|
|
|
fi, err := os.Lstat(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if fi.IsDir() {
|
|
return &DirectoryEntry{path: path}, nil
|
|
} else {
|
|
return &FileEntry{path: path}, nil
|
|
}
|
|
}
|