From 17be4dfd2db77a32edeca18a5f27b061360a3920 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aure=CC=81lie=20Delhaie?= <roboform@hotmail.fr>
Date: Wed, 13 Apr 2022 23:24:06 +0200
Subject: [PATCH] Fix dst folder creation

---
 main.go | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index 3a9ba27..500884d 100644
--- a/main.go
+++ b/main.go
@@ -18,14 +18,20 @@ func main() {
 	flag.Parse()
 	if len(*path) == 0 {
 		fmt.Println("missing --path argument")
-		os.Exit(2)
+		os.Exit(1)
 	}
 	if *count < 1 {
 		fmt.Println("--count must be greater than 0")
+		os.Exit(2)
+	}
+	// If file or folder does not exist, stop the script
+	if _, err := os.Stat(*path); err != nil {
+		fmt.Println(err)
 		os.Exit(3)
 	}
-	if _, err := os.Stat(*path); err != nil {
-		err = os.MkdirAll(*path, 0750)
+	// Create archive folder if not exist
+	if _, err := os.Stat(*backupFolder); err != nil {
+		err = os.MkdirAll(*backupFolder, 0750)
 		if err != nil {
 			fmt.Println(err)
 			os.Exit(4)
@@ -39,7 +45,7 @@ func main() {
 	absDstPath, err := filepath.Abs(*backupFolder)
 	if err != nil {
 		fmt.Println(err)
-		os.Exit(5)
+		os.Exit(6)
 	}
 	if err := makeBackup(absSourcePath, absDstPath, *count); err != nil {
 		fmt.Println(err)