move removeOldestWtfFile to wtfbackup.go

lyledouglass · Mar 27, 2022 · ce1efb7 · ce1efb7
1 parent 8f7dd0b
commit ce1efb7
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 25 deletions.
diff --git a/cmd/removeOldestFile.go b/cmd/removeOldestFile.go
diff --git a/cmd/wtfBackup.go b/cmd/wtfBackup.go
@@ -3,6 +3,7 @@ package cmd
 import (
 	"fmt"
 	"log"
+	"os"
 	"time"
 
 	"wowtools/utilities"
@@ -16,7 +17,7 @@ func WtfBackup() {
 	wtfBackupDir := viper.GetString("backup_dir") + "WTF\\"
 	currentTime := time.Now()
 	folderName := currentTime.Format("2006-01-02")
-
+	removeOldElvuiZip()
 	fmt.Println("Beginning backup of WTF folder")
 	if err := utilities.ZipSource(wtfFolder, wtfBackupDir+folderName+".zip"); err != nil {
 		log.Fatal(err)
@@ -29,3 +30,17 @@ func WtfBackup() {
 	}
 	fmt.Println("Folder backup complete")
 }
+
+func removeOldestWtfZip() {
+	retentionRate := viper.GetInt("retention_rate")
+	wtfBackupDir := viper.GetString("backup_dir") + "WTF\\"
+	fileCount := utilities.GetFileCount(wtfBackupDir)
+	if fileCount > retentionRate {
+		oldestFile := utilities.GetOldestFolder(wtfBackupDir)
+		os.Chdir(wtfBackupDir)
+		removeFile := os.Remove(oldestFile)
+		if removeFile != nil {
+			log.Fatal()
+		}
+	}
+}
diff --git a/main.go b/main.go
@@ -19,7 +19,6 @@ func main() {
 	go utilities.VerifyFolders(viper.GetString("backup_dir")+"WTF", &wg)
 	wg.Wait()
 
-	cmd.RemoveOldestWtfZip()
 	cmd.WtfBackup()
 	cmd.UpdateElvUI()
 	utilities.OpenCurseforge()