From 861ca5aa143c854529cbe9243a7ce0fdf05bc410 Mon Sep 17 00:00:00 2001
From: hhf <hhf@noreply.localhost>
Date: Fri, 6 Dec 2024 21:11:50 +0530
Subject: [PATCH] Add ptero-cleanup.sh

---
 ptero-cleanup.sh | 63 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 ptero-cleanup.sh

diff --git a/ptero-cleanup.sh b/ptero-cleanup.sh
new file mode 100644
index 0000000..4c680ad
--- /dev/null
+++ b/ptero-cleanup.sh
@@ -0,0 +1,63 @@
+#!/bin/bash
+
+# Log file setup
+LOG_FILE="/var/log/ptero-cleanup.log"
+DOCKER_PATH=$(which docker)
+
+# Function to log messages
+log_message() {
+    echo "[$(date '+%Y-%m-%d %H:%M:%S')] $1" | tee -a "$LOG_FILE"
+}
+
+# Check if docker is installed
+if [ -z "$DOCKER_PATH" ]; then
+    log_message "ERROR: Docker is not installed or not in PATH"
+    exit 1
+fi
+
+# Check if script is run as root
+if [ "$EUID" -ne 0 ]; then
+    log_message "ERROR: Please run as root"
+    exit 1
+fi
+
+log_message "Starting Pterodactyl image cleanup"
+
+# Get list of Pterodactyl-related images
+PTERO_IMAGES=$($DOCKER_PATH images --format "{{.Repository}}:{{.Tag}}" | grep -i "pterodactyl\|pterodactyl/yolks\|ghcr.io/pterodactyl")
+
+if [ -z "$PTERO_IMAGES" ]; then
+    log_message "No Pterodactyl images found"
+    exit 0
+fi
+
+# Counter for removed images
+REMOVED_COUNT=0
+FAILED_COUNT=0
+
+# Remove each unused Pterodactyl image
+echo "$PTERO_IMAGES" | while read -r image; do
+    # Check if image is being used by any container
+    if ! $DOCKER_PATH ps -a --format "{{.Image}}" | grep -q "^${image}$"; then
+        log_message "Attempting to remove image: $image"
+        if $DOCKER_PATH rmi "$image" > /dev/null 2>&1; then
+            log_message "Successfully removed: $image"
+            ((REMOVED_COUNT++))
+        else
+            log_message "Failed to remove: $image (might be in use)"
+            ((FAILED_COUNT++))
+        fi
+    else
+        log_message "Skipping $image - currently in use"
+    fi
+done
+
+# Display summary
+log_message "Cleanup completed"
+log_message "Images removed: $REMOVED_COUNT"
+log_message "Failed removals: $FAILED_COUNT"
+
+# Clean up old log files (keep last 5 days)
+find "$(dirname "$LOG_FILE")" -name "$(basename "$LOG_FILE")*" -mtime +5 -delete
+
+exit 0
\ No newline at end of file