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)