# -*- mode: sh; -*-
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
	if [[ ! -p /dev/fd/1 ]]; then
		set_title "file://${server}/"
		command ssh "$@"
		retval=$?
		set_title
		return $retval
	else
		command ssh "$@"
		return $retval
	fi
}
mosh() {
	command mosh "$@"
	retval=$?
	set_title
	return $retval
}