From c8b583607ab57a8952c7838402d3f6817e6491ba Mon Sep 17 00:00:00 2001
From: Karmaa <bugattiguy527@gmail.com>
Date: Mon, 17 Mar 2025 21:39:39 -0500
Subject: [PATCH] Fix mongo downgrade loop

---
 docker/entrypoint.sh | 118 +++++++++++++++++++++++++++++--------------
 1 file changed, 79 insertions(+), 39 deletions(-)

diff --git a/docker/entrypoint.sh b/docker/entrypoint.sh
index 29c14e99..62d3580e 100644
--- a/docker/entrypoint.sh
+++ b/docker/entrypoint.sh
@@ -6,43 +6,79 @@ mkdir -p /data/db /var/log/mongodb /var/run/mongodb
 chown -R mongodb:mongodb /data/db /var/log/mongodb /var/run/mongodb
 chmod 755 /data/db /var/log/mongodb /var/run/mongodb
 
-# Check if we need to migrate from MongoDB 5.0
-if [ -f "/data/db/diagnostic.data/metrics.2" ] || [ -f "/data/db/WiredTiger.wt" ]; then
-    echo "Existing MongoDB data detected, checking version..."
-    
-    # Start MongoDB temporarily with --repair to handle version mismatch
-    gosu mongodb mongod --dbpath $MONGODB_DATA_DIR --logpath $MONGODB_LOG_DIR/mongodb.log --repair
-    
-    # Start MongoDB with specific options for migration
-    echo "Starting MongoDB for migration..."
-    gosu mongodb mongod --dbpath $MONGODB_DATA_DIR --logpath $MONGODB_LOG_DIR/mongodb.log --bind_ip 0.0.0.0 &
-    MONGO_PID=$!
-    
-    # Wait for MongoDB to be ready
-    echo "Waiting for MongoDB to start for migration..."
-    until gosu mongodb mongo --eval "print(\"waited for connection\")" > /dev/null 2>&1; do
-        sleep 1
-        if ! kill -0 $MONGO_PID 2>/dev/null; then
-            echo "MongoDB failed to start for migration. Checking logs:"
-            cat $MONGODB_LOG_DIR/mongodb.log
-            exit 1
-        fi
-    done
-    
-    echo "Setting featureCompatibilityVersion to 4.4..."
-    # Try to set featureCompatibilityVersion
-    if gosu mongodb mongo --eval 'db.adminCommand({setFeatureCompatibilityVersion: "4.4"})' > /dev/null 2>&1; then
-        echo "Successfully set featureCompatibilityVersion to 4.4"
-    else
-        echo "Failed to set featureCompatibilityVersion. Attempting repair..."
+# Function to check MongoDB version
+check_mongo_version() {
+    echo "Checking MongoDB version..."
+    if [ -f "/data/db/diagnostic.data/metrics.2" ] || [ -f "/data/db/WiredTiger.wt" ]; then
+        echo "Existing MongoDB data detected, attempting migration..."
+        
+        # Clear any existing mongod lock file
+        rm -f /tmp/mongodb-27017.sock
+        rm -f /data/db/mongod.lock
+        
+        # Run repair first
+        echo "Running initial repair..."
         gosu mongodb mongod --dbpath $MONGODB_DATA_DIR --repair
+        
+        # Start MongoDB in standalone mode for version check
+        echo "Starting MongoDB for version check..."
+        gosu mongodb mongod --dbpath $MONGODB_DATA_DIR --port 27017 --bind_ip 127.0.0.1 &
+        MONGO_PID=$!
+        
+        # Wait for MongoDB to start
+        echo "Waiting for MongoDB to start..."
+        MAX_TRIES=30
+        COUNT=0
+        while ! gosu mongodb mongo --quiet --eval "db.version()" > /dev/null 2>&1; do
+            sleep 2
+            COUNT=$((COUNT + 1))
+            if [ $COUNT -ge $MAX_TRIES ]; then
+                echo "Failed to start MongoDB after $MAX_TRIES attempts"
+                kill -9 $MONGO_PID 2>/dev/null || true
+                return 1
+            fi
+        done
+        
+        # Try to set compatibility version
+        echo "Attempting to set feature compatibility version..."
+        if gosu mongodb mongo --quiet --eval 'db.adminCommand({setFeatureCompatibilityVersion: "4.4"})'; then
+            echo "Successfully set feature compatibility version to 4.4"
+            kill $MONGO_PID
+            while kill -0 $MONGO_PID 2>/dev/null; do
+                sleep 1
+            done
+            return 0
+        else
+            echo "Failed to set feature compatibility version"
+            kill $MONGO_PID
+            while kill -0 $MONGO_PID 2>/dev/null; do
+                sleep 1
+            done
+            return 1
+        fi
     fi
-    
-    # Stop MongoDB after migration
-    kill $MONGO_PID
-    while kill -0 $MONGO_PID 2>/dev/null; do
-        sleep 1
-    done
+    return 0
+}
+
+# Try migration up to 3 times
+MAX_MIGRATION_ATTEMPTS=3
+MIGRATION_ATTEMPT=1
+
+while [ $MIGRATION_ATTEMPT -le $MAX_MIGRATION_ATTEMPTS ]; do
+    echo "Migration attempt $MIGRATION_ATTEMPT of $MAX_MIGRATION_ATTEMPTS"
+    if check_mongo_version; then
+        break
+    fi
+    MIGRATION_ATTEMPT=$((MIGRATION_ATTEMPT + 1))
+    if [ $MIGRATION_ATTEMPT -le $MAX_MIGRATION_ATTEMPTS ]; then
+        echo "Migration failed, waiting before retry..."
+        sleep 5
+    fi
+done
+
+if [ $MIGRATION_ATTEMPT -gt $MAX_MIGRATION_ATTEMPTS ]; then
+    echo "Migration failed after $MAX_MIGRATION_ATTEMPTS attempts"
+    exit 1
 fi
 
 # Start MongoDB normally
@@ -52,13 +88,17 @@ MONGO_PID=$!
 
 # Wait for MongoDB to be ready
 echo "Waiting for MongoDB to start..."
-until gosu mongodb mongo --eval "print(\"waited for connection\")" > /dev/null 2>&1; do
-    sleep 0.5
-    if ! kill -0 $MONGO_PID 2>/dev/null; then
-        echo "MongoDB failed to start. Checking logs:"
+MAX_TRIES=30
+COUNT=0
+while ! gosu mongodb mongo --quiet --eval "db.version()" > /dev/null 2>&1; do
+    sleep 2
+    COUNT=$((COUNT + 1))
+    if [ $COUNT -ge $MAX_TRIES ]; then
+        echo "Failed to start MongoDB. Checking logs:"
         cat $MONGODB_LOG_DIR/mongodb.log
         exit 1
     fi
+    echo "Waiting for MongoDB... (attempt $COUNT/$MAX_TRIES)"
 done
 echo "MongoDB started successfully"