diff options
Diffstat (limited to 'fileop/fileop.go')
-rw-r--r-- | fileop/fileop.go | 46 |
1 files changed, 36 insertions, 10 deletions
diff --git a/fileop/fileop.go b/fileop/fileop.go index d08cb82..1be3f2e 100644 --- a/fileop/fileop.go +++ b/fileop/fileop.go @@ -8,7 +8,15 @@ import ( "path/filepath" ) -func WalkFolder(folder string) <-chan string { +type WalkMode int + +const ( + Folder WalkMode = iota + File + FileFolder +) + +func WalkFolder(folder string, walkMode WalkMode) <-chan string { c := make(chan string) go func(folder string, c chan string) { @@ -16,7 +24,17 @@ func WalkFolder(folder string) <-chan string { file, _ := os.Open(path) defer file.Close() fileInfo, _ := file.Stat() - if !fileInfo.IsDir() { + + switch walkMode { + case Folder: + if fileInfo.IsDir() { + c <- path + } + case File: + if !fileInfo.IsDir() { + c <- path + } + case FileFolder: c <- path } return nil @@ -28,15 +46,9 @@ func WalkFolder(folder string) <-chan string { return c } -func CalculateSHA256(file string) (string, error) { - f, err := os.Open(file) - if err != nil { - return "", err - } - defer f.Close() - +func CalculateSHA256(r io.Reader) (string, error) { h := sha256.New() - if _, err := io.Copy(h, f); err != nil { + if _, err := io.Copy(h, r); err != nil { return "", err } @@ -64,3 +76,17 @@ func Move() chan<- *MoveCommand { return c } + +func IsEmpty(name string) (bool, error) { + f, err := os.Open(name) + if err != nil { + return false, err + } + defer f.Close() + + _, err = f.Readdirnames(1) + if err == io.EOF { + return true, nil + } + return false, err +} |