set_title () {
  print -Pn "\e]0;$1\7\n"
}
ssh () {
  local arg server retval
  for arg
  do
    if [[ "${arg#-}" == "$arg" ]]
    then
      server=$arg
      break
    fi
  done
  set_title "file://${server}/"
  command ssh "$@"
  retval=$?
  set_title
  return $retval
}
mosh () {
  command mosh "$@"
  retval=$?
  set_title
  return $retval
}