forked from mirrors/gotosocial
105 lines
2 KiB
Go
105 lines
2 KiB
Go
package util
|
|
|
|
import (
|
|
"io/fs"
|
|
"os"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"codeberg.org/gruf/go-fastpath"
|
|
)
|
|
|
|
var dotdot = "../"
|
|
|
|
// CountDotdots returns the number of "dot-dots" (../) in a cleaned filesystem path
|
|
func CountDotdots(path string) int {
|
|
if !strings.HasSuffix(path, dotdot) {
|
|
return 0
|
|
}
|
|
return strings.Count(path, dotdot)
|
|
}
|
|
|
|
// WalkDir traverses the dir tree of the supplied path, performing the supplied walkFn on each entry
|
|
func WalkDir(pb *fastpath.Builder, path string, walkFn func(string, fs.DirEntry)) error {
|
|
// Read supplied dir path
|
|
dirEntries, err := os.ReadDir(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Iter entries
|
|
for _, entry := range dirEntries {
|
|
// Pass to walk fn
|
|
walkFn(path, entry)
|
|
|
|
// Recurse dir entries
|
|
if entry.IsDir() {
|
|
err = WalkDir(pb, pb.Join(path, entry.Name()), walkFn)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// CleanDirs traverses the dir tree of the supplied path, removing any folders with zero children
|
|
func CleanDirs(path string) error {
|
|
// Acquire builder
|
|
pb := GetPathBuilder()
|
|
defer PutPathBuilder(pb)
|
|
|
|
// Get dir entries
|
|
entries, err := os.ReadDir(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Recurse dirs
|
|
for _, entry := range entries {
|
|
if entry.IsDir() {
|
|
err := cleanDirs(pb, pb.Join(path, entry.Name()))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// cleanDirs performs the actual dir cleaning logic for the exported version
|
|
func cleanDirs(pb *fastpath.Builder, path string) error {
|
|
// Get dir entries
|
|
entries, err := os.ReadDir(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// If no entries, delete
|
|
if len(entries) < 1 {
|
|
return os.Remove(path)
|
|
}
|
|
|
|
// Recurse dirs
|
|
for _, entry := range entries {
|
|
if entry.IsDir() {
|
|
err := cleanDirs(pb, pb.Join(path, entry.Name()))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// RetryOnEINTR is a low-level filesystem function for retrying syscalls on O_EINTR received
|
|
func RetryOnEINTR(do func() error) error {
|
|
for {
|
|
err := do()
|
|
if err == syscall.EINTR {
|
|
continue
|
|
}
|
|
return err
|
|
}
|
|
}
|