From c48fa1124e2edc71e3fd3edd06955233c149633b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Oliver=20H=C3=B6hn?= <oliverphiliphohn@gmail.com>
Date: Sat, 26 Dec 2020 13:07:06 +0000
Subject: [PATCH] Allow moving of files instead of copy

This prevents loss of meta-data (e.g. btime, birth time) as copy would create a new file with metadata set to now, instead of matching the metadata of the original file
---
 organizemyfiles/main.go | 40 +++++++++++++++++++++++++++++++++++++---
 1 file changed, 37 insertions(+), 3 deletions(-)

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
+}