#!/bin/bash

SSH_TARGET=""
SSH_USER=""
SSH_HOST=""

while [[ $# -gt 0 ]]; do
    case $1 in
        -l)
            SSH_USER="$2"
            shift 2
            ;;
        -p)
            SSH_PORT="$2"
            shift 2
            ;;
        -*)
            shift
            ;;
        *)
            if [[ -z "$SSH_TARGET" ]]; then
                SSH_TARGET="$1"
                break
            fi
            shift
            ;;
    esac
done

if [[ "$SSH_TARGET" =~ ^([^@]+)@(.+)$ ]]; then
    SSH_USER="${BASH_REMATCH[1]}"
    SSH_HOST="${BASH_REMATCH[2]}"
    FORMATTED_TARGET="${SSH_USER}@${SSH_HOST}"
elif [[ -n "$SSH_USER" && -n "$SSH_TARGET" ]]; then
    SSH_HOST="$SSH_TARGET"
    FORMATTED_TARGET="${SSH_USER}@${SSH_HOST}"
else
    FORMATTED_TARGET="$SSH_TARGET"
fi

if [[ -n "$SSH_PORT" ]]; then
    FORMATTED_TARGET="${FORMATTED_TARGET}:${SSH_PORT}"
fi

export PKG_TARGET_HOST="$FORMATTED_TARGET"
export LD_PRELOAD="/usr/lib/x86_64-linux-gnu/libssl1.1.so.1:$LD_PRELOAD"

exec /usr/bin/ssh "$@"