class Discharger::SetupRunner::Commands::PgToolsCommand

def pg_dump_wrapper_content
  <<~BASH
    #!/usr/bin/env bash
    set -e
    CONTAINER="#{container_name}"
    # Docker first: use container's pg_dump if running
    if docker ps --format '{{.Names}}' 2>/dev/null | grep -q "^${CONTAINER}$"; then
      echo "Using pg_dump from Docker container: $CONTAINER" >&2
      # Parse arguments to handle --file option specially
      # When running in Docker, we need to output to stdout and redirect locally
      OUTPUT_FILE=""
      HAS_USER=""
      ARGS=()
      while [[ $# -gt 0 ]]; do
        case $1 in
          --file)
            OUTPUT_FILE="$2"
            shift 2
            ;;
          --file=*)
            OUTPUT_FILE="${1#*=}"
            shift
            ;;
          -f)
            OUTPUT_FILE="$2"
            shift 2
            ;;
          -U|--username|--username=*)
            HAS_USER="1"
            ARGS+=("$1")
            shift
            ;;
          *)
            ARGS+=("$1")
            shift
            ;;
        esac
      done
      # Default to postgres user if not specified (container runs as root)
      if [[ -z "$HAS_USER" ]]; then
        ARGS=("-U" "postgres" "${ARGS[@]}")
      fi
      if [[ -n "$OUTPUT_FILE" ]]; then
        # Run pg_dump in container, output to stdout, redirect to local file
        docker exec -i "$CONTAINER" pg_dump "${ARGS[@]}" > "$OUTPUT_FILE"
      else
        # No file output, just exec normally
        exec docker exec -i "$CONTAINER" pg_dump "${ARGS[@]}"
      fi
      exit 0
    fi
    # Fallback to system pg_dump (skip this wrapper in PATH to avoid infinite loop)
    SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
    CLEAN_PATH="${PATH//$SCRIPT_DIR:/}"
    CLEAN_PATH="${CLEAN_PATH%:$SCRIPT_DIR}"
    FALLBACK=$(PATH="$CLEAN_PATH" command -v pg_dump 2>/dev/null)
    if [[ -n "$FALLBACK" ]]; then
      exec "$FALLBACK" "$@"
    fi
    echo "Error: pg_dump not found. Start Docker or install PostgreSQL client tools." >&2
    exit 1
  BASH
end