#!/bin/bash

# Configuration
REMOTE_HOST="${SSH_HOST:-root@alpha.mintel.me}"
ACTION=$1
ENV=$2

# Help
if [ -z "$ACTION" ] || [ -z "$ENV" ]; then
    echo "Usage: ./scripts/sync-directus.sh [push|pull] [testing|staging|production]"
    echo ""
    echo "Commands:"
    echo "  push   Sync LOCAL data -> REMOTE"
    echo "  pull   Sync REMOTE data -> LOCAL"
    echo ""
    echo "Environments:"
    echo "  testing, staging, production"
    exit 1
fi

# Project Configuration (extracted from package.json and aligned with deploy.yml)
PRJ_ID=$(jq -r .name package.json | sed 's/@mintel\///' | sed 's/\.com$//')
REMOTE_DIR="/home/deploy/sites/${PRJ_ID}.com"

case $ENV in
  testing)    PROJECT_NAME="${PRJ_ID}-testing"; ENV_FILE=".env.testing" ;;
  staging)    PROJECT_NAME="${PRJ_ID}-staging"; ENV_FILE=".env.staging" ;;
  production) PROJECT_NAME="${PRJ_ID}-production"; ENV_FILE=".env.prod"    ;;
  *) echo "❌ Invalid environment: $ENV"; exit 1 ;;
esac

# DB Details (matching docker-compose defaults)
DB_USER="directus"
DB_NAME="directus"

echo "🔍 Detecting local database..."
LOCAL_DB_CONTAINER=$(docker compose ps -q directus-db)

if [ -z "$LOCAL_DB_CONTAINER" ]; then
    # Check if it exists but is stopped
    LOCAL_DB_EXISTS=$(docker compose ps -a -q directus-db)
    if [ -n "$LOCAL_DB_EXISTS" ]; then
        echo "⏳ Local directus-db is stopped. Starting it..."
        docker compose up -d directus-db
        # Wait a few seconds for PG to be ready
        sleep 2
        LOCAL_DB_CONTAINER=$(docker compose ps -q directus-db)
    fi
fi

if [ -z "$LOCAL_DB_CONTAINER" ]; then
    echo "❌ Local directus-db container not found. Is it defined in docker-compose.yaml?"
    exit 1
fi

if [ "$ACTION" == "push" ]; then
    echo "🚀 Pushing LOCAL -> $ENV ($PROJECT_NAME)..."
    
    # 1. DB Dump
    echo "📦 Dumping local database..."
    docker exec "$LOCAL_DB_CONTAINER" pg_dump -U "$DB_USER" --clean --if-exists --no-owner --no-privileges "$DB_NAME" > dump.sql
    
    # 2. Upload Dump
    echo "📤 Uploading dump to remote server..."
    scp dump.sql "$REMOTE_HOST:$REMOTE_DIR/dump.sql"
    
    # 3. Restore on Remote
    echo "🔄 Restoring dump on $ENV..."
    REMOTE_DB_CONTAINER=$(ssh "$REMOTE_HOST" "cd $REMOTE_DIR && docker compose -p $PROJECT_NAME ps -q directus-db")
    
    if [ -z "$REMOTE_DB_CONTAINER" ]; then
        echo "❌ Remote $ENV-db container not found!"
        exit 1
    fi

    echo "🧹 Wiping remote database schema..."
    ssh "$REMOTE_HOST" "docker exec $REMOTE_DB_CONTAINER psql -U $DB_USER $DB_NAME -c 'DROP SCHEMA public CASCADE; CREATE SCHEMA public;'"
    
    echo "⚡ Restoring database..."
    ssh "$REMOTE_HOST" "docker exec -i $REMOTE_DB_CONTAINER psql -U $DB_USER $DB_NAME < $REMOTE_DIR/dump.sql"
    
    # 4. Sync Uploads
    echo "📁 Syncing uploads (Local -> $ENV)..."
    rsync -avz --progress ./directus/uploads/ "$REMOTE_HOST:$REMOTE_DIR/directus/uploads/"
    
    # Clean up
    rm dump.sql
    ssh "$REMOTE_HOST" "rm $REMOTE_DIR/dump.sql"
    
    # 5. Restart Directus to trigger migrations and refresh schema cache
    echo "🔄 Restarting remote Directus to apply migrations..."
    ssh "$REMOTE_HOST" "cd $REMOTE_DIR && docker compose -p $PROJECT_NAME restart directus"
    
    echo "✨ Push to $ENV complete!"

elif [ "$ACTION" == "pull" ]; then
    echo "📥 Pulling $ENV Data -> LOCAL..."
    
    # 1. DB Dump on Remote
    echo "📦 Dumping remote database ($ENV)..."
    REMOTE_DB_CONTAINER=$(ssh "$REMOTE_HOST" "cd $REMOTE_DIR && docker compose -p $PROJECT_NAME ps -q directus-db")
    
    if [ -z "$REMOTE_DB_CONTAINER" ]; then
        echo "❌ Remote $ENV-db container not found!"
        exit 1
    fi
    
    ssh "$REMOTE_HOST" "docker exec $REMOTE_DB_CONTAINER pg_dump -U $DB_USER --clean --if-exists --no-owner --no-privileges $DB_NAME > $REMOTE_DIR/dump.sql"
    
    # 2. Download Dump
    echo "📥 Downloading dump..."
    scp "$REMOTE_HOST:$REMOTE_DIR/dump.sql" dump.sql
    
    # 3. Restore Locally
    echo "🧹 Wiping local database schema..."
    docker exec "$LOCAL_DB_CONTAINER" psql -U "$DB_USER" "$DB_NAME" -c 'DROP SCHEMA public CASCADE; CREATE SCHEMA public;'
    
    echo "⚡ Restoring database locally..."
    docker exec -i "$LOCAL_DB_CONTAINER" psql -U "$DB_USER" "$DB_NAME" < dump.sql
    
    # 4. Sync Uploads
    echo "📁 Syncing uploads ($ENV -> Local)..."
    rsync -avz --progress "$REMOTE_HOST:$REMOTE_DIR/directus/uploads/" ./directus/uploads/
    
    # Clean up
    rm dump.sql
    ssh "$REMOTE_HOST" "rm $REMOTE_DIR/dump.sql"
    
    echo "✨ Pull to Local complete!"
fi