Mac OS X 使用交流

sh -c "$(curl -fsSL https://raw.githubusercontent.com/diseng/quickLogin/master/quickLogin-install.sh)"

#!/bin/zsh

function show()
{
	grep -e "Host [a-zA-Z1-9].*" ~/.ssh/config|awk '{print $2}'|cat -n
}

function init()
{
	softwareCheck
	if [ ! -d ~/.ssh ]; then
		echo "create ~/.ssh directory"
		mkdir ~/.ssh
		echo "done"
	else
		echo "~/.ssh directory already exist"
	fi
	if [ ! -e ~/.ssh/config ]; then
		echo "create ~/.ssh/config file"
		touch ~/.ssh/config
		echo "done"
	else
		echo "~/.ssh/config file already exist"
	fi
	if [ ! -e ~/.ssh/id_rsa.pub ]; then
		echo "create ~/.ssh/id_rsa.pub file"
		ssh-keygen -t rsa -f ~/.ssh/id_rsa -N '' > /dev/null
		echo "done"
	else
		echo "~/.ssh/id_rsa.pub file already exist"
	fi
	if [ `grep "ServerAliveInterval" ~/.ssh/config|wc -l` -eq 0 ]; then
		echo -e "ServerAliveInterval 60\nHost *\nControlMaster auto\nControlPath ~/.ssh/master-%r@%h:%p" > /tmp/sshconfig
		cat ~/.ssh/config >> /tmp/sshconfig
		mv ~/.ssh/config ~/.ssh/config.bak
		mv /tmp/sshconfig ~/.ssh/config
	fi
	echo "环境初始化成功，执行s help查看使用帮助"
}

function softwareCheck()
{
	type ssh-keygen >/dev/null 2>&1 || { echo >&2 "注意：使用此脚本，请确保机器上含有ssh-keygen和ssh-copy-id，请先安装ssh-keygen"; exit 1; }
	type ssh-copy-id >/dev/null 2>&1 || { echo >&2 "注意：使用此脚本，请确保机器上含有ssh-keygen和ssh-copy-id，请先安装ssh-copy-id(brew install ssh-copy-id)"; exit 1; }
}

function login()
{
#	gbk
	var=$(echo $1 | bc 2>/dev/null)
	if [ "$var" = "$1"  ]; then
		host=$(grep -e "Host [a-zA-Z1-9].*" ~/.ssh/config|awk '{print $2}'|sed -n "$var"p)
		ssh $host
	else
		ssh $1
	fi
#	utf8
}

function add()
{	
	if [ $# = 3 ]; then
		ssh-copy-id -i ~/.ssh/id_rsa.pub `whoami`@$3
	elif [ $# = 4 ]; then
		ssh-copy-id -i ~/.ssh/id_rsa.pub $4@$3
	elif [ $# = 5 ]; then
		ssh-copy-id -i ~/.ssh/id_rsa.pub -p $5 $4@$3
	fi
	echo "Host $2" >> ~/.ssh/config
	echo "HostName $3" >> ~/.ssh/config
	if [ $# = 3 ]; then
		echo "User `whoami`" >> ~/.ssh/config
	elif [ $# = 4 ]; then
		echo "User $4" >> ~/.ssh/config	
	elif [ $# = 5 ]; then
		echo "Port $5" >> ~/.ssh/config
	fi
	echo "" >> ~/.ssh/config
	echo "add done"
}

function help()
{
	echo "注意：使用此脚本，请确保机器上含有ssh-keygen和ssh-copy-id"
	echo "1) s init 脚本环境初始化(第一次使用时请执行)"
	echo "2) s add Host HostName [User] [Port] 添加登陆机器的信息;a)Host为自定义的机器tag b)HostName为机器IP或者机器名 c)User和Port为机器用户名和SSH端口号 选填，默认为当前用户名和22端口"
	echo "3) s 显示可登陆列表"
	echo "4) s number|Host 使用number或者Host快速登录到机器，其中number为执行s是显示的机器序号，Host为自定义机器tag"
	echo "5) s help 显示帮助"
}


if [ $# = 0 ]; then
	show
elif [ $# = 1 ]; then
	if [ $1 = "init" ]; then
		init
	elif [ $1 = "help" ]; then
		help
	else
		login $1
	fi
elif [ $# = 3 ] || [ $# = 4 ] || [ $# = 5 ]; then
	if [ $1 = "add" ]; then
		add $@
	fi
else
	help
fi