#!/bin/bash
# toggle_site.sh
# Usage: ./toggle_site.sh down|up <sftp_user> <site_path>

ACTION=$1
SFTP_USER=$2
SITE_PATH=$3
HTACCESS_FILE="$SITE_PATH/public_html/.htaccess"
HTACCESS_BACKUP="$SITE_PATH/.htaccess.bak"
MAINTENANCE_HTACCESS="$SITE_PATH/public_html/.htaccess.maintenance"
MAINTENANCE_HTML="$SITE_PATH/public_html/maintenance.html"
MASTER_TEMPLATE="/var/www/maintenance/maintenance_template.html"

if [ -z "$ACTION" ] || [ -z "$SFTP_USER" ] || [ -z "$SITE_PATH" ]; then
    echo "Usage: $0 down|up <sftp_user> <site_path>"
    exit 1
fi

case $ACTION in
    down)
        echo ">>> Putting site DOWN..."

        # 1. Disable SFTP user (lock account)
        echo " - Locking SFTP user: $SFTP_USER"
        sudo usermod -L "$SFTP_USER"

        # 2. Backup .htaccess
        if [ -f "$HTACCESS_FILE" ]; then
            echo " - Backing up .htaccess"
            cp "$HTACCESS_FILE" "$HTACCESS_BACKUP"
        fi

        # 3. Create maintenance .htaccess dynamically
        echo " - Creating maintenance .htaccess"
        cat > "$MAINTENANCE_HTACCESS" <<EOL
# Maintenance Mode
<IfModule mod_rewrite.c>
RewriteEngine On
RewriteCond %{REQUEST_URI} !/maintenance.html$
RewriteRule ^.*$ /maintenance.html [R=302,L]
</IfModule>
EOL

        mv "$MAINTENANCE_HTACCESS" "$HTACCESS_FILE"

        # 4. Copy global maintenance template
        if [ -f "$MASTER_TEMPLATE" ]; then
            echo " - Copying global maintenance template"
            cp "$MASTER_TEMPLATE" "$MAINTENANCE_HTML"
        else
            echo " ! Master template not found at $MASTER_TEMPLATE"
        fi

        echo ">>> Site is now in maintenance mode."
        ;;

    up)
        echo ">>> Bringing site UP..."

        # 1. Re-enable SFTP user
        echo " - Unlocking SFTP user: $SFTP_USER"
        sudo usermod -U "$SFTP_USER"

        # 2. Restore .htaccess
        if [ -f "$HTACCESS_BACKUP" ]; then
            echo " - Restoring .htaccess"
            mv "$HTACCESS_BACKUP" "$HTACCESS_FILE"
        else
            echo " ! No .htaccess backup found"
        fi

        # 3. Remove maintenance.html
        if [ -f "$MAINTENANCE_HTML" ]; then
            echo " - Removing maintenance.html"
            rm -f "$MAINTENANCE_HTML"
        fi

        echo ">>> Site is now live."
        ;;

    *)
        echo "Invalid action: $ACTION"
        echo "Usage: $0 down|up <sftp_user> <site_path>"
        exit 1
        ;;
esac