Put this in your .bash_profile(s) to open VS Code on your local machine while SSH'd into a remote machine.

.bash_profile

# this needs to be on the *remote* machine.

if [ ! -z "$SSH_CLIENT" ]
then
  function code() {
    local ssh_client_host=$(echo $SSH_CLIENT | cut -d ' ' -f1)

    if [ -z "$1" ]
    then
      local argpath="."
    else
      local argpath="$1"
    fi

    # ssh-remote requires a directory
    local resolved=$(realpath -m $argpath)
    if [ -f "$resolved" ]; then
      local resolved=$(dirname $resolved)
    fi
    ssh $ssh_client_host code \
      --folder-uri=vscode-remote://ssh-remote+$(hostname)$resolved
  }
fi

neat!

i'd prefer wrapping the code() definition itself in the if statement checking for SSH_CLIENT. that way the init file simply doesn't define the code function unless it's been started over ssh. this way there's no complication in local logins mode =D

also using local variables removes the need to unset afterwards