diff --git a/organizemyfiles/main.go b/organizemyfiles/main.go index 9bd4561..7c642e0 100644 --- a/organizemyfiles/main.go +++ b/organizemyfiles/main.go @@ -16,6 +16,7 @@ import ( var inDir = flag.String("in", "", "Path to directory with files to organize") var outDir = flag.String("out", "", "Path to directory where to place the organized files") var dryrun = flag.Bool("dryrun", false, "Enable to only print how the files would be organised, instead of actually organizing them in the out directory.") +var opType = flag.String("op_type", "copy", "Type of operation to use when organizing. Allowed values are: copy, or move.") type File struct { Path string @@ -110,10 +111,19 @@ func main() { return nil } - err = copy(f, fileIndex) - if err != nil { - return err + switch *opType { + case "copy": + if err := copy(f, fileIndex); err != nil { + return err + } + case "move": + if err := move(f, fileIndex); err != nil { + return err + } + default: + log.Fatalf("unknown op_type: %s", *opType) } + fileIndex++ } @@ -174,3 +184,27 @@ func copy(f *File, fileIndex uint64) error { return nil } + +func move(f *File, fileIndex uint64) error { + outPrefix := filepath.Join(*outDir, f.GetYear(), f.GetQuarter()) + // Suffix filename with an index to avoid clashes for similarly named files + outFilename := fmt.Sprintf("%s_%d%s", f.GetName(), fileIndex, f.GetExt()) + out := filepath.Join(outPrefix, outFilename) + + if *dryrun { + fmt.Printf("(dryrun) Move %s to %s\n", f.Path, out) + return nil + } + + // 0777 = Read, Write, Execute for Users, Groups, and Other + if err := os.MkdirAll(outPrefix, 0777); err != nil { + return fmt.Errorf("unable to create output directory: %s, due to: %w", outPrefix, err) + } + + if err := os.Rename(f.Path, out); err != nil { + return fmt.Errorf("unable to move: %s, to: %s, due to: %w", f.Path, out, err) + } + fmt.Printf("Moved %s to %s\n", f.Path, out) + + return nil +}