From 14668fd4d85e384d582f598579da6de59f392bfe Mon Sep 17 00:00:00 2001
From: Daniel Legt <contact@danlegt.com>
Date: Thu, 18 Jan 2024 22:18:03 +0200
Subject: [PATCH] Gitea google drive implementation

---
 gitea/gitea_backup.sh | 67 +++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 65 insertions(+), 2 deletions(-)

diff --git a/gitea/gitea_backup.sh b/gitea/gitea_backup.sh
index aa9cd7e..32d85c4 100755
--- a/gitea/gitea_backup.sh
+++ b/gitea/gitea_backup.sh
@@ -38,6 +38,10 @@ backup_from_docker() {
         exit 1
     fi
 
+    log "Backup file $BACKUP_FILE created."
+}
+
+delete_old_local_backups() {
     # Keep only the last $KEEP_BACKUP_COUNT backups
     if [ -n "$KEEP_BACKUP_COUNT" ]; then
         # Get the number of backup files
@@ -45,23 +49,82 @@ backup_from_docker() {
 
         # Check if the current number of backups exceeds the limit
         if [ "$BACKUP_COUNT" -gt "$KEEP_BACKUP_COUNT" ]; then
+            log "Removing old backups."
             # Remove the oldest files
             ls -1tr $BACKUP_SOURCE_PATH/gitea-dump-*.zip | head -n -$KEEP_BACKUP_COUNT | xargs -d '\n' rm -f
         fi
     fi
-
-    log "Backup file $BACKUP_FILE created."
 }
 
+# Google Drive
+
+upload_to_google_drive() {
+    if [ -n "$GOOGLE_DRIVE_FOLDER_ID" ] && [ -n "$BACKUP_FILE" ]; then
+        log "Uploading backup zip to google drive."
+
+        gdrive files upload --parent $GOOGLE_DRIVE_FOLDER_ID $BACKUP_FILE
+        if [ $? -eq 0 ]; then
+            log "Backup file uploaded to Google Drive."
+            send_discord_notification "Backup file uploaded to Google Drive." "65280"
+        else
+            error "Failed to upload backup file to Google Drive."
+            send_discord_notification "Failed to upload backup file to Google Drive." "16711680"
+            exit 1
+        fi
+    else
+        log "Google Drive upload not configured. Skipping."
+    fi
+}
+
+delete_old_gdrive_backups() {
+    if [ -n "$GOOGLE_DRIVE_FOLDER_ID" ] && [ -n "$GOOGLE_DRIVE_KEEP_BACKUP_COUNT" ]; then
+        GCLOUD_BACKUPS=$(gdrive files list --parent $GOOGLE_DRIVE_FOLDER_ID --skip-header --order-by "createdTime asc" | awk '{print $1}')
+        BACKUP_COUNT=$(echo "$GCLOUD_BACKUPS" | wc -l)
+
+        log "Deleting old gdrive backups."
+
+        if [ "$BACKUP_COUNT" -gt "$GOOGLE_DRIVE_KEEP_BACKUP_COUNT" ]; then
+            DELETE_COUNT=$((BACKUP_COUNT - GOOGLE_DRIVE_KEEP_BACKUP_COUNT))
+            DELETE_BACKUPS=$(echo "$GCLOUD_BACKUPS" | head -n "$DELETE_COUNT")
+
+            echo "$DELETE_BACKUPS" | while read -r BACKUP_ID; do
+                if [ -n "$BACKUP_ID" ]; then
+                    gdrive files delete $BACKUP_ID
+                    if [ $? -eq 0 ]; then
+                        local MSG_X="Deleted old backup with ID $BACKUP_ID from Google Drive."
+                        log "$MSG_X"
+                        send_discord_notification "$MSG_X" "16776960"
+                    else
+                        local MSG_X="Failed to delete backup with ID $BACKUP_ID from Google Drive."
+                        log "$MSG_X"
+                        send_discord_notification "$MSG_X" "16711680"
+                    fi
+                fi
+            done
+        else
+            log "No old backups to delete."
+        fi
+    else
+        log "Google Drive cleanup not configured. Skipping."
+    fi
+}
+
+
 ### [ Main ] ###
 echo -e "${YELLOW}Starting ${PROGRAM_NAME}...${NC}"
 send_discord_notification "Starting Vaultwarden backup..." "16776960" # Yellow color
 
 check_required_programs "$SCRIPT_DIR/required_programs.txt"
 
+# Run the main backup process
 backup_from_docker
 
 # Run google drive backup
+upload_to_google_drive
+
+# Delete old files
+delete_old_local_backups
+delete_old_gdrive_backups
 
 # Final message
 log "Backup process completed."