¿Continuar con los comandos de Shell una vez conectado a Ssh?

No estoy seguro si la pregunta que estoy haciendo es correcta, pero básicamente quería automatizar este proceso

  scp ~/.ssh/id_rsa.pub [email protected]:~/
  ssh [email protected]
  mkdir .ssh
  cat id_rsa.pub >> .ssh/authorized_keys
  rm id_rsa.pub
  chmod go-w ~
  chmod 700 ~/.ssh
  chmod 600 ~/.ssh/authorized_keys

Así que pensé que podría crear un script de shell y ponerlo en mi .bash_profile así:function

setup_ssh () {
  scp ~/.ssh/id_rsa.pub $1:~/
  ssh $1
  #the following is happens when connected to the server using ssh
  mkdir .ssh
  cat id_rsa.pub >> .ssh/authorized_keys
  rm id_rsa.pub
  chmod go-w ~
  chmod 700 ~/.ssh
  chmod 600 ~/.ssh/authorized_keys
}

Pero, por supuesto, esto no funciona porque no continúa con los comandos una vez conectado al servidor. ¿Hay alguna forma de continuar con los comandos una vez que está conectado al servidor a través de ssh?

Respuesta aceptada:

Agrega su identidad al servidor remoto, también crea el árbol ~/.ssh con los permisos apropiados si es necesario.

ssh-copy-id -i ~/.ssh/id_rsa.pub [email protected]