class Kitsune::Kit::Commands::SetupPostgresDocker

def perform_rollback(ssh, postgres_defaults)
  output = ssh.exec! <<~EOH
    set -e
    BASE_DIR="$HOME/docker/postgres"
    BACKUP_DIR="/usr/local/backups"
    SCRIPT_ID="setup_postgres_docker"
    AFTER_FILE="${BACKUP_DIR}/${SCRIPT_ID}.after"
    if [ -f "$AFTER_FILE" ]; then
      echo "🔁 Stopping and removing docker containers..."
      cd "$BASE_DIR"
      docker compose down -v || true
      echo "🧹 Cleaning up files..."
      rm -rf "$BASE_DIR"
      sudo rm -f "$AFTER_FILE"
      if command -v ufw >/dev/null; then
        echo "🛡️ Removing PostgreSQL port from firewall..."
        sudo ufw delete allow #{postgres_defaults[:postgres_port]} || true
      fi
    else
      echo "💡 Nothing to rollback"
    fi
    echo "✅ Rollback completed"
  EOH
  say output
end