SungjinYoo · February 16, 2016 13:54
diff --git a/gwk.sh b/gwk.sh
 #!/bin/bash
 #==============================================================================
 # GW-Kit

 # 	@author : (origin) yunsang.choi([email protected])
 #       @author : (forked) jinkwon([email protected])
 #	@src : (origin) https://gist.github.com/gists/3115022
 #	@src : (forked) https://github.com/Jinkwon/naver-gw-kit/

 #-------
 #	v1.3.4
 #		- minor change (display text)
 #	v1.3.3
 #		- allow '-' character in hostname
 #	v1.3.2
 #		- add feature : manage host description.
 #	v1.3.1
 #		- rearrange key
 #	v1.3.0
 #		- support interactive kinit
 #	v1.2.2
 #		- display message when rlogin
 #	v1.2.1
 #		- fix tab auto-completion
 #	v1.2.0
 #		- support bash key map
 #		- improve tab auto-completion
 #		- thanx to @worwornf for feedback
 #==============================================================================

 #========================================
 # Constants for configuration
 #========================================
 HOST_LIST_FILE=~/.known_hosts
 HOST_DISPLAY_COUNT=15
 KINIT_PW_FILE=~/.kinit_passwd
 USER_LIST_FOR_RLOGIN=(irteam irteamsu )

 VERSION="1.3.4"
 #========================================
 # kinit
 #========================================
 # run direct
 kinit -f

 #========================================
 # Read key input
 #	see : http://www.linuxquestions.org/questions/programming-9/bash-case-with-arrow-keys-and-del-backspace-etc-523441/
 #========================================
 SpecialKeyCodes="
 0027 0091 0051 0126 0;delete
 0027 0091 0065 0;up
 0027 0091 0066 0;down
 0027 0091 0067 0;right
 0027 0091 0068 0;left
 0027 0091 0090 0;shift-tab
 "

 ControlCharacters=(
 	[0x00]="enter"			# "Null character"
 	[0x09]="tab"
 	[0x7F]="backspace"
 	[0x08]="ctrl-h"
 	[0x15]="ctrl-u"
 	[0x01]="ctrl-a"
 	[0x05]="ctrl-e"
 	[0x02]="ctrl-b"
 	[0x06]="ctrl-f"
 	[0x0b]="ctrl-k"
 	[0x04]="ctrl-d"
 	[0x17]="ctrl-w"
 	[0x0e]="ctrl-n"
 	[0x12]="ctrl-r"
 )

 function read1 {
 	IFS='' read  -sn1 "${@}" scancode
 }
 function read2 {
 	# Captues Ctrl-C
 	local -i ECode=0
 	stty -echo raw
 	scancode=$(dd bs=1 count=1 2>/dev/null || true )
 	stty echo -raw
 }
 function read_key {
 	local scancode
 	local scancode_f
 	local d
 	if read1 ${1:-}; then
 		d=$(printf '%04d' "'${scancode}")
 		case "${scancode}" in
 			[^[:cntrl:]])
 				echo "${scancode}"
 				;;
 			$'\e')
 				local -i icnt=1
 				local oscancode Match=
 				scancode_f="${d}"
 				while read1 -t1 && [ ${icnt} -lt 9 ]; do
 					let icnt+=1
 					scancode_f="${scancode_f} $(printf '%04d' "'${scancode}")"
 					if  [ ${icnt} -eq 2 ] ; then
 						case "${scancode}" in
 							[[O]) continue;;
 							[^[:cntrl:]]) Match="alt-${scancode}"; break ;;
 						esac
 					fi
 					Match="$(echo "${SpecialKeyCodes}" | grep "^${scancode_f} 0;")"
 					if [ -n "${Match:-}" ]; then
 						Match="${Match#*;}"
 						Match="${Match%;*}"
 						break
 					fi
 					oscancode="${scancode}"
 				done
 				if [ -z "${Match}" ]; then
 					case "${scancode_f}" in
 						'0027') Match="escape" ;;
 						'0027 0079') Match="alt-O" ;;
 						'0027 0091') Match="alt-[" ;;
 					esac
 				fi
 				if [ -n "${Match}" ]; then
 					echo "${Match}"
 				else
 					# print any unrecognised codes
 					echo "${scancode_f} 0;"
 				fi
 				;;
 			*)
 				d=$(printf '%d' "'${scancode}")
 				if [ -n "${ControlCharacters[${d}]:-}" ]; then
 					echo "${ControlCharacters[${d}]}"
 				else
 					# print any unrecognised control characters
 					printf 'Unknown %02x\n' "'${scancode}"
 				fi
 				;;
 		esac
 	fi
 }


 #========================================
 # Control screen
 #========================================
 black="\x1b[30m"
 red="\x1b[31m"
 green="\x1b[32m"
 yellow="\x1b[33m"
 blue="\x1b[34m"
 magenta="\x1b[35m"
 cyan="\x1b[36m"
 white="\x1b[37m"
 reset="\x1b[39m"

 bg_black="\x1b[40m"
 bg_red="\x1b[41m"
 bg_green="\x1b[42m"
 bg_yellow="\x1b[43m"
 bg_blue="\x1b[44m"
 bg_magenta="\x1b[45m"
 bg_cyan="\x1b[46m"
 bg_white="\x1b[47m"
 bg_reset="\x1b[49m"

 bold="\x1b[1m"
 bold_off="\x1b[22m"
 underline="\x1b[4m"
 underline_off="\x1b[24m"

 save_cursor() {
 	tput sc
 }
 restore_cursor() {
 	tput rc
 }
 move_cursor() {
 	tput cup $1 $2
 }

 #========================================
 # Control user list
 #========================================
 # var: user list
 user_list=(${USER_LIST_FOR_RLOGIN[@]})
 # var: selected user
 user=${user_list[0]}
 # func: change user
 next_user() {
 	local index=0
 	local cnt=${#user_list[@]}
 	while true;do
 		if [ "$user" == "${user_list[index]}" ]; then
 			user=${user_list[(index+1) % cnt]}
 			return
 		fi
 		let "index=index+1 % cnt"
 	done
 }

 #========================================
 # Control host list
 #========================================
 host_list=()
 host_descs=()
 hostname_filter=
 filtered_host_list=
 selected_host=
 cursor_offset=0
 init_host_list() {
 	if [ ! -f $HOST_LIST_FILE ];then
 		touch $HOST_LIST_FILE
 	fi
 	while read line;do
 		local len=${#host_list[@]}
 		if [ -n "$line" ];then
 			host_list[$len]=`echo "$line" | sed -r 's/[ \t].+//'`
 		fi
 	done < $HOST_LIST_FILE
 	host_list=($(printf "%s\n" ${host_list[@]} | sort -u ))
 	host_descs=($(
 		for i in ${host_list[@]};do
 			echo "$(host_description $i)"
 		done))
 	filter_host_list
 }

 modify_host_description() {
 	if [ ! -z "$selected_host" ];then
 #		sed -i~ -r "s/^${selected_host}[ \t]?.*/$selected_host/"  $HOST_LIST_FILE
 		set_host_description "$selected_host"
 	else
 		message="select a host to modify."
 	fi
 }

 set_host_description() {
 	local host=$1
 	local old_desc="$(host_description $host)"

 	clear
 	print_signature
       	echo -e "Enter description for ${yellow}$host${reset}: "
 	if [ ! -z "$old_desc" ];then
 		echo -e "(current description : ${magenta}$old_desc${reset})${white}"
 	fi
       	read -e desc
 	echo -e "${reset}"
 	if [ -z "$desc" ];then desc="$old_desc"; fi
 	if [ -z "$desc" ];then desc=""; fi

 	if [ -z "`cat $HOST_LIST_FILE | grep -E "^$host"`" ];then
 		echo "$host     $desc" >> $HOST_LIST_FILE
 	else
 		sed -i~ -r "s/^${host}[ \t]?.*/$host   $desc/"  $HOST_LIST_FILE
 	fi
 }

 add_host_to_list() {
 	local host=$1
 	if [ -z "`cat $HOST_LIST_FILE | grep -E "^$host"`" ];then
 		host_list=("${host_list[@]}" "$host")
 		host_list=($(printf "%s\n" ${host_list[*]} | sort -u ))
 		set_host_description "$host"
 	fi
 }

 filter_host_list() {
 	local len=${#host_list[@]}
 	filtered_host_list=($(
 		for (( i = 0; i < $len; i++ )) do
 			if [ -n "`echo ${host_list[i]} | grep "$hostname_filter"`" ] || [ -n "`echo ${host_descs[i]} | grep "$hostname_filter"`" ]; then
 				echo ${host_list[i]}
 			fi
 		done
 	))
 	if [ -z `echo $selected_host | grep "$hostname_filter"` ];then
 		selected_host=
 	fi
 }

 host_index_of() {
 	local host=$1
 	local index=0
 	while [ $index -lt ${#filtered_host_list[@]} ];do
 		if [ "${filtered_host_list[index]}" == "$host" ];then
 			echo $index
 			return
 		fi
 		let "index++"
 	done
 	echo ""
 }

 select_next_host() {
 	local cycle=${1:-false}
 	local index
 	local cnt=${#filtered_host_list[@]}
 	if [ $cnt == 0 ];then
 		selected_host=
 		return
 	elif [ -z "$selected_host" ];then
 		selected_host="${filtered_host_list[0]}"
 	else
 		index=$(host_index_of $selected_host)
 		let "index++"
 		if [ $index -lt "${#filtered_host_list[@]}" ];then
 			selected_host="${filtered_host_list[index]}"
 		elif [ $cycle == true ];then
 			selected_host="${filtered_host_list[0]}"
 		fi
 	fi
 }

 select_prev_host() {
 	local cycle=${1:-false}
 	local index
 	local cnt=${#filtered_host_list[@]}
 	if [ $cnt == 0 ];then
 		selected_host=
 		return
 	elif [ -z "$selected_host" ];then
 		local cnt=${#filtered_host_list[@]}
 		selected_host="${filtered_host_list[cnt-1]}"
 	else
 		index=$(host_index_of $selected_host)
 		let "index--"
 		if [ $index -ge 0 ];then
 			selected_host="${filtered_host_list[index]}"
 		elif [ $cycle == true ];then
 			selected_host="${filtered_host_list[cnt-1]}"
 		fi
 	fi
 }

 complete_hostname() {
 	local common_matched
 	local matched

 	while true;do
 		for hostname in "${filtered_host_list[@]}" ;do
 			matched="`echo $hostname | grep -o "${hostname_filter}." | head -n 1`"
 			common_matched="${common_matched:-$matched}"
 			if [ "$matched" != "$common_matched" ];then
 				return
 			fi
 		done
 		common_matched="${matched}"
 		if [ -z "$common_matched" ]; then
 			break
 		fi
 		if [ "$hostname_filter" == "$common_matched" ];then
 			break
 		fi
 		hostname_filter="$common_matched"
 		common_matched=
 		filter_host_list
 	done
        if [ ${#filtered_host_list[@]} -eq 1 ];then
                hostname_filter="${filtered_host_list[0]}"
                filter_host_list
        fi
 }

 append_hostname_filter() {
 	local ch=$1
 	local len=${#hostname_filter}
 	local prefix=${hostname_filter:0:($len+$cursor_offset)}
 	local postfix=${hostname_filter:($cursor_offset)}
 	if [ $cursor_offset -eq 0 ];then
 		postfix=""
 	fi
 	hostname_filter=$prefix$ch$postfix
 	filter_host_list
 }

 delete_prev_ch_hostname_filter() {
 	local len=${#hostname_filter}
 	local prefix=${hostname_filter:0:($len+$cursor_offset)}
 	local postfix=${hostname_filter:($cursor_offset)}
 	if [ $cursor_offset -eq 0 ];then
 		postfix=""
 	fi
 	local prefix_len=${#prefix}
 	if [ $len -gt 0 -a $prefix_len -gt 0 ];then
 		prefix=${prefix:0:($prefix_len-1)}
 		hostname_filter=${prefix}${postfix}
 		filter_host_list
 	fi
 }

 delete_next_ch_hostname_filter() {
 	local len=${#hostname_filter}
 	local prefix=${hostname_filter:0:($len+$cursor_offset)}
 	local postfix=${hostname_filter:($cursor_offset)}
 	if [ $cursor_offset -eq 0 ];then
 		postfix=""
 		return
 	fi
 	if [ $len -gt 0 ];then
 		hostname_filter=${prefix}${postfix:1}
 		cursor_move_right
 		filter_host_list
 	fi
 }

 delete_prev_all_hostname_filter() {
 	local len=${#hostname_filter}
 	local prefix=${hostname_filter:0:($len+$cursor_offset)}
 	local postfix=${hostname_filter:($cursor_offset)}
 	if [ $cursor_offset -eq 0 ];then
 		postfix=""
 	fi
 	local prefix_len=${#prefix}
 	if [ $len -gt 0 -a $prefix_len -gt 0 ];then
 		hostname_filter=${postfix}
 		filter_host_list
 	fi
 }

 delete_next_all_hostname_filter() {
 	local len=${#hostname_filter}
 	local prefix=${hostname_filter:0:($len+$cursor_offset)}
 	local postfix=${hostname_filter:($cursor_offset)}
 	if [ $cursor_offset -eq 0 ];then
 		postfix=""
 		return
 	fi
 	if [ $len -gt 0 ];then
 		hostname_filter=${prefix}
 		cursor_move_end
 		filter_host_list
 	fi
 }


 clear_hostname_filter() {
 	hostname_filter=
 	cursor_offset=0
 	filter_host_list
 }

 cursor_move_left() {
 	let "cursor_offset--"
 	local len=${#hostname_filter}
 	if (( len < (0-$cursor_offset) ));then
 		let "cursor_offset=0-len"
 	fi
 }

 cursor_move_right() {
 	let "cursor_offset++"
 	if [ $cursor_offset -gt 0 ];then
 		cursor_offset=0
 	fi
 }

 cursor_move_begin() {
 	local len="${#hostname_filter}"
 	let "cursor_offset=0-len"
 }

 cursor_move_end() {
 	cursor_offset=0
 }


 #========================================
 # Print host list
 #========================================
 begin_index=0
 adjust_begin_index() {
 	local selected=$(host_index_of $selected_host)
 	local last_index
 	let last_index=begin_index+HOST_DISPLAY_COUNT-1
 	if [ -z $selected ];then
 		selected=0
 	fi
 	if [ $selected -lt $begin_index ];then
 		begin_index=$selected
 	fi
 	if [ $selected -gt $last_index ];then
 		let begin_index=selected-HOST_DISPLAY_COUNT+1
 	fi
 }

 host_description() {
 	local hostname=$1
 	echo "`cat $HOST_LIST_FILE | grep \"$hostname\" | sed -r -e 's/^[^ \t]+[ \t]*//' -e 's/[ \t]+$//'`"
 }

 print_host_list() {
 	adjust_begin_index
 	echo "Known hosts (${#filtered_host_list[@]}/${#host_list[@]})"
 	local last_index
 	let last_index=begin_index+HOST_DISPLAY_COUNT-1
 	for (( i= $begin_index; i <= $last_index; i++ ));do
 		if [ $i -ge ${#filtered_host_list[@]} ];then
 			break
 		fi
 		local host="${filtered_host_list[i]}"
 		local printmsg=" - $host"
 		if [ -n "$hostname_filter" ];then
 			local matched=`echo $host | grep -o "$hostname_filter" | head -n 1`
 			printmsg=${printmsg/$matched/${underline}$matched${underline_off}}
 		fi
 		if [ "$selected_host" == "$host" ];then
 			printmsg=${printmsg/ -/>>}
 			printmsg=${bg_yellow}${black}$printmsg${reset}${bg_reset}
 		else
 			printmsg=${white}$printmsg${reset}
 		fi
 			echo -en "$printmsg"
 		local blank=30
 		local hostlen=${#host}
 		let "blank=blank-hostlen"
 		for (( k=0; k<$blank; k++));do echo -n " " ;done
 		local description=$(host_description $host)
 		echo -e "$description"
 	done
 }
 #========================================
 # Logo And Help
 #========================================
 print_logo(){
 	echo -en "${cyan}"
 	echo -en "${reset}"
 	echo -e "==============================================================================="

 }

 print_help() {
 	echo -en " [/]: change user to rlogin        "
 	echo -en " [ctrl-n]: clear hostname          "
 	echo -e  " [ctrl-c]: quit"
 	echo -e  " [=]: modify description for the selected host"
 	echo -e  " - hostname will be autosaved if it is valid.($HOST_LIST_FILE)"
 }

 print_command() {
 	local command
 	command="${command}${magenta}`whoami`@gw-kit> ${reset}"
 	command="${command}${white}rlogin -l ${bold}$user ${yellow}$hostname_filter${reset}${bold_off}"

 	echo "-------------------------------------------------------------------------------"
 	if [ -n "$message" ];then
 		echo -e "${red}${bold}[!] $message${bold_off}${reset}"
 		message=
 	else
 		echo
 	fi
 	echo -en $command
 	save_cursor
 	echo
 	echo "-------------------------------------------------------------------------------"
 }
 set_cursor() {
 	for (( i=0; i > $cursor_offset; i-- ));do
 		tput cub1
 	done
 }

 print_screen() {
 	clear
 	print_logo
 	print_help
 	print_command
 	print_host_list
 	restore_cursor
 	set_cursor
 }

 print_signature(){
 	echo -e "${cyan}Gateway-kit${reset} ${blue}ver$VERSION${reset}"
 }

 #========================================
 # execute rlogin
 #========================================
 do_execute() {
 	if [ -n "$selected_host" ];then
 		hostname_filter=$selected_host
 		selected_host=
 		filter_host_list
 		return
 	fi
 	if [ -z "$hostname_filter" ];then
 		message="you should select a host to rlogin!"
 		return
 	fi
 	clear
 	local tmpfile="$(basename $0).$$.tmp"
 	print_signature
 	echo -e "Try to rlogin to ${yellow}$hostname_filter${reset} as ${magenta}$user${reset} ..."
 	echo
 	`rlogin -l $user $hostname_filter 2> $tmpfile > /dev/tty`
 	if [ $? == 0 ];then
 		clear
 		add_host_to_list $hostname_filter
 	else
 		message=$( cat $tmpfile | head -n 1)
 	fi
 	filter_host_list
 	clear
 }

 good_bye(){
 	tput cud1
 	tput cud1
 	tput ed
 	echo -e "${reset}${red}Happy Hacking!${reset}"
 	echo
 	stty echo
 	rm ~/gwk.sh.*.tmp &> /dev/null
 	exit 0;
 }

 #==============================================================================
 # Initialize
 #==============================================================================
 init_host_list
 trap good_bye INT # trap ctrl+c

 #==============================================================================
 # Main Loop
 #==============================================================================
 while true;do
 	print_screen
 	k=$(read_key)
 	case $k in
 		# bash key bind
 		ctrl-a ) cursor_move_begin;;
 		ctrl-e ) cursor_move_end;;
 		ctrl-b ) cursor_move_left;;
 		ctrl-f ) cursor_move_right;;
 		ctrl-d ) delete_next_ch_hostname_filter;;
 		ctrl-h ) delete_prev_ch_hostname_filter;;
 		ctrl-k ) delete_next_all_hostname_filter;;
 		ctrl-u ) delete_prev_all_hostname_filter;;
 		ctrl-w ) delete_prev_all_hostname_filter;;
 		# common
 		backspace ) delete_prev_ch_hostname_filter;;
 		up ) select_prev_host;;
 		down ) select_next_host;;
 		left ) cursor_move_left;;
 		right ) cursor_move_right;;
 		tab ) complete_hostname;;
 		/ ) next_user;;
 		delete ) delete_next_ch_hostname_filter;;
 		# special
 		ctrl-n ) clear_hostname_filter;;
 		= ) modify_host_description;;
 		# inputhost name filter
 		[-a-zA-Z0-9_\.] ) append_hostname_filter $k;;
 		enter ) do_execute;;
 	esac
 done
	#!/bin/bash
	#==============================================================================
	# GW-Kit

	# @author : (origin) yunsang.choi([email protected])
	# @author : (forked) jinkwon([email protected])
	# @src : (origin) https://gist.github.com/gists/3115022
	# @src : (forked) https://github.com/Jinkwon/naver-gw-kit/

	#-------
	# v1.3.4
	# - minor change (display text)
	# v1.3.3
	# - allow '-' character in hostname
	# v1.3.2
	# - add feature : manage host description.
	# v1.3.1
	# - rearrange key
	# v1.3.0
	# - support interactive kinit
	# v1.2.2
	# - display message when rlogin
	# v1.2.1
	# - fix tab auto-completion
	# v1.2.0
	# - support bash key map
	# - improve tab auto-completion
	# - thanx to @worwornf for feedback
	#==============================================================================

	#========================================
	# Constants for configuration
	#========================================
	HOST_LIST_FILE=~/.known_hosts
	HOST_DISPLAY_COUNT=15
	KINIT_PW_FILE=~/.kinit_passwd
	USER_LIST_FOR_RLOGIN=(irteam irteamsu )

	VERSION="1.3.4"
	#========================================
	# kinit
	#========================================
	# run direct
	kinit -f

	#========================================
	# Read key input
	# see : http://www.linuxquestions.org/questions/programming-9/bash-case-with-arrow-keys-and-del-backspace-etc-523441/
	#========================================
	SpecialKeyCodes="
	0027 0091 0051 0126 0;delete
	0027 0091 0065 0;up
	0027 0091 0066 0;down
	0027 0091 0067 0;right
	0027 0091 0068 0;left
	0027 0091 0090 0;shift-tab
	"

	ControlCharacters=(
	[0x00]="enter" # "Null character"
	[0x09]="tab"
	[0x7F]="backspace"
	[0x08]="ctrl-h"
	[0x15]="ctrl-u"
	[0x01]="ctrl-a"
	[0x05]="ctrl-e"
	[0x02]="ctrl-b"
	[0x06]="ctrl-f"
	[0x0b]="ctrl-k"
	[0x04]="ctrl-d"
	[0x17]="ctrl-w"
	[0x0e]="ctrl-n"
	[0x12]="ctrl-r"
	)

	function read1 {
	IFS='' read -sn1 "${@}" scancode
	}
	function read2 {
	# Captues Ctrl-C
	local -i ECode=0
	stty -echo raw
	scancode=$(dd bs=1 count=1 2>/dev/null \|\| true )
	stty echo -raw
	}
	function read_key {
	local scancode
	local scancode_f
	local d
	if read1 ${1:-}; then
	d=$(printf '%04d' "'${scancode}")
	case "${scancode}" in
	[^[:cntrl:]])
	echo "${scancode}"
	;;
	$'\e')
	local -i icnt=1
	local oscancode Match=
	scancode_f="${d}"
	while read1 -t1 && [ ${icnt} -lt 9 ]; do
	let icnt+=1
	scancode_f="${scancode_f} $(printf '%04d' "'${scancode}")"
	if [ ${icnt} -eq 2 ] ; then
	case "${scancode}" in
	[[O]) continue;;
	[^[:cntrl:]]) Match="alt-${scancode}"; break ;;
	esac
	fi
	Match="$(echo "${SpecialKeyCodes}" \| grep "^${scancode_f} 0;")"
	if [ -n "${Match:-}" ]; then
	Match="${Match#*;}"
	Match="${Match%;*}"
	break
	fi
	oscancode="${scancode}"
	done
	if [ -z "${Match}" ]; then
	case "${scancode_f}" in
	'0027') Match="escape" ;;
	'0027 0079') Match="alt-O" ;;
	'0027 0091') Match="alt-[" ;;
	esac
	fi
	if [ -n "${Match}" ]; then
	echo "${Match}"
	else
	# print any unrecognised codes
	echo "${scancode_f} 0;"
	fi
	;;
	*)
	d=$(printf '%d' "'${scancode}")
	if [ -n "${ControlCharacters[${d}]:-}" ]; then
	echo "${ControlCharacters[${d}]}"
	else
	# print any unrecognised control characters
	printf 'Unknown %02x\n' "'${scancode}"
	fi
	;;
	esac
	fi
	}


	#========================================
	# Control screen
	#========================================
	black="\x1b[30m"
	red="\x1b[31m"
	green="\x1b[32m"
	yellow="\x1b[33m"
	blue="\x1b[34m"
	magenta="\x1b[35m"
	cyan="\x1b[36m"
	white="\x1b[37m"
	reset="\x1b[39m"

	bg_black="\x1b[40m"
	bg_red="\x1b[41m"
	bg_green="\x1b[42m"
	bg_yellow="\x1b[43m"
	bg_blue="\x1b[44m"
	bg_magenta="\x1b[45m"
	bg_cyan="\x1b[46m"
	bg_white="\x1b[47m"
	bg_reset="\x1b[49m"

	bold="\x1b[1m"
	bold_off="\x1b[22m"
	underline="\x1b[4m"
	underline_off="\x1b[24m"

	save_cursor() {
	tput sc
	}
	restore_cursor() {
	tput rc
	}
	move_cursor() {
	tput cup $1 $2
	}

	#========================================
	# Control user list
	#========================================
	# var: user list
	user_list=(${USER_LIST_FOR_RLOGIN[@]})
	# var: selected user
	user=${user_list[0]}
	# func: change user
	next_user() {
	local index=0
	local cnt=${#user_list[@]}
	while true;do
	if [ "$user" == "${user_list[index]}" ]; then
	user=${user_list[(index+1) % cnt]}
	return
	fi
	let "index=index+1 % cnt"
	done
	}

	#========================================
	# Control host list
	#========================================
	host_list=()
	host_descs=()
	hostname_filter=
	filtered_host_list=
	selected_host=
	cursor_offset=0
	init_host_list() {
	if [ ! -f $HOST_LIST_FILE ];then
	touch $HOST_LIST_FILE
	fi
	while read line;do
	local len=${#host_list[@]}
	if [ -n "$line" ];then
	host_list[$len]=`echo "$line" \| sed -r 's/[ \t].+//'`
	fi
	done < $HOST_LIST_FILE
	host_list=($(printf "%s\n" ${host_list[@]} \| sort -u ))
	host_descs=($(
	for i in ${host_list[@]};do
	echo "$(host_description $i)"
	done))
	filter_host_list
	}

	modify_host_description() {
	if [ ! -z "$selected_host" ];then
	# sed -i~ -r "s/^${selected_host}[ \t]?.*/$selected_host/" $HOST_LIST_FILE
	set_host_description "$selected_host"
	else
	message="select a host to modify."
	fi
	}

	set_host_description() {
	local host=$1
	local old_desc="$(host_description $host)"

	clear
	print_signature
	echo -e "Enter description for ${yellow}$host${reset}: "
	if [ ! -z "$old_desc" ];then
	echo -e "(current description : ${magenta}$old_desc${reset})${white}"
	fi
	read -e desc
	echo -e "${reset}"
	if [ -z "$desc" ];then desc="$old_desc"; fi
	if [ -z "$desc" ];then desc=""; fi

	if [ -z "`cat $HOST_LIST_FILE \| grep -E "^$host"`" ];then
	echo "$host $desc" >> $HOST_LIST_FILE
	else
	sed -i~ -r "s/^${host}[ \t]?.*/$host $desc/" $HOST_LIST_FILE
	fi
	}

	add_host_to_list() {
	local host=$1
	if [ -z "`cat $HOST_LIST_FILE \| grep -E "^$host"`" ];then
	host_list=("${host_list[@]}" "$host")
	host_list=($(printf "%s\n" ${host_list[*]} \| sort -u ))
	set_host_description "$host"
	fi
	}

	filter_host_list() {
	local len=${#host_list[@]}
	filtered_host_list=($(
	for (( i = 0; i < $len; i++ )) do
	if [ -n "`echo ${host_list[i]} \| grep "$hostname_filter"`" ] \|\| [ -n "`echo ${host_descs[i]} \| grep "$hostname_filter"`" ]; then
	echo ${host_list[i]}
	fi
	done
	))
	if [ -z `echo $selected_host \| grep "$hostname_filter"` ];then
	selected_host=
	fi
	}

	host_index_of() {
	local host=$1
	local index=0
	while [ $index -lt ${#filtered_host_list[@]} ];do
	if [ "${filtered_host_list[index]}" == "$host" ];then
	echo $index
	return
	fi
	let "index++"
	done
	echo ""
	}

	select_next_host() {
	local cycle=${1:-false}
	local index
	local cnt=${#filtered_host_list[@]}
	if [ $cnt == 0 ];then
	selected_host=
	return
	elif [ -z "$selected_host" ];then
	selected_host="${filtered_host_list[0]}"
	else
	index=$(host_index_of $selected_host)
	let "index++"
	if [ $index -lt "${#filtered_host_list[@]}" ];then
	selected_host="${filtered_host_list[index]}"
	elif [ $cycle == true ];then
	selected_host="${filtered_host_list[0]}"
	fi
	fi
	}

	select_prev_host() {
	local cycle=${1:-false}
	local index
	local cnt=${#filtered_host_list[@]}
	if [ $cnt == 0 ];then
	selected_host=
	return
	elif [ -z "$selected_host" ];then
	local cnt=${#filtered_host_list[@]}
	selected_host="${filtered_host_list[cnt-1]}"
	else
	index=$(host_index_of $selected_host)
	let "index--"
	if [ $index -ge 0 ];then
	selected_host="${filtered_host_list[index]}"
	elif [ $cycle == true ];then
	selected_host="${filtered_host_list[cnt-1]}"
	fi
	fi
	}

	complete_hostname() {
	local common_matched
	local matched

	while true;do
	for hostname in "${filtered_host_list[@]}" ;do
	matched="`echo $hostname \| grep -o "${hostname_filter}." \| head -n 1`"
	common_matched="${common_matched:-$matched}"
	if [ "$matched" != "$common_matched" ];then
	return
	fi
	done
	common_matched="${matched}"
	if [ -z "$common_matched" ]; then
	break
	fi
	if [ "$hostname_filter" == "$common_matched" ];then
	break
	fi
	hostname_filter="$common_matched"
	common_matched=
	filter_host_list
	done
	if [ ${#filtered_host_list[@]} -eq 1 ];then
	hostname_filter="${filtered_host_list[0]}"
	filter_host_list
	fi
	}

	append_hostname_filter() {
	local ch=$1
	local len=${#hostname_filter}
	local prefix=${hostname_filter:0:($len+$cursor_offset)}
	local postfix=${hostname_filter:($cursor_offset)}
	if [ $cursor_offset -eq 0 ];then
	postfix=""
	fi
	hostname_filter=$prefix$ch$postfix
	filter_host_list
	}

	delete_prev_ch_hostname_filter() {
	local len=${#hostname_filter}
	local prefix=${hostname_filter:0:($len+$cursor_offset)}
	local postfix=${hostname_filter:($cursor_offset)}
	if [ $cursor_offset -eq 0 ];then
	postfix=""
	fi
	local prefix_len=${#prefix}
	if [ $len -gt 0 -a $prefix_len -gt 0 ];then
	prefix=${prefix:0:($prefix_len-1)}
	hostname_filter=${prefix}${postfix}
	filter_host_list
	fi
	}

	delete_next_ch_hostname_filter() {
	local len=${#hostname_filter}
	local prefix=${hostname_filter:0:($len+$cursor_offset)}
	local postfix=${hostname_filter:($cursor_offset)}
	if [ $cursor_offset -eq 0 ];then
	postfix=""
	return
	fi
	if [ $len -gt 0 ];then
	hostname_filter=${prefix}${postfix:1}
	cursor_move_right
	filter_host_list
	fi
	}

	delete_prev_all_hostname_filter() {
	local len=${#hostname_filter}
	local prefix=${hostname_filter:0:($len+$cursor_offset)}
	local postfix=${hostname_filter:($cursor_offset)}
	if [ $cursor_offset -eq 0 ];then
	postfix=""
	fi
	local prefix_len=${#prefix}
	if [ $len -gt 0 -a $prefix_len -gt 0 ];then
	hostname_filter=${postfix}
	filter_host_list
	fi
	}

	delete_next_all_hostname_filter() {
	local len=${#hostname_filter}
	local prefix=${hostname_filter:0:($len+$cursor_offset)}
	local postfix=${hostname_filter:($cursor_offset)}
	if [ $cursor_offset -eq 0 ];then
	postfix=""
	return
	fi
	if [ $len -gt 0 ];then
	hostname_filter=${prefix}
	cursor_move_end
	filter_host_list
	fi
	}


	clear_hostname_filter() {
	hostname_filter=
	cursor_offset=0
	filter_host_list
	}

	cursor_move_left() {
	let "cursor_offset--"
	local len=${#hostname_filter}
	if (( len < (0-$cursor_offset) ));then
	let "cursor_offset=0-len"
	fi
	}

	cursor_move_right() {
	let "cursor_offset++"
	if [ $cursor_offset -gt 0 ];then
	cursor_offset=0
	fi
	}

	cursor_move_begin() {
	local len="${#hostname_filter}"
	let "cursor_offset=0-len"
	}

	cursor_move_end() {
	cursor_offset=0
	}


	#========================================
	# Print host list
	#========================================
	begin_index=0
	adjust_begin_index() {
	local selected=$(host_index_of $selected_host)
	local last_index
	let last_index=begin_index+HOST_DISPLAY_COUNT-1
	if [ -z $selected ];then
	selected=0
	fi
	if [ $selected -lt $begin_index ];then
	begin_index=$selected
	fi
	if [ $selected -gt $last_index ];then
	let begin_index=selected-HOST_DISPLAY_COUNT+1
	fi
	}

	host_description() {
	local hostname=$1
	echo "`cat $HOST_LIST_FILE \| grep \"$hostname\" \| sed -r -e 's/^[^ \t]+[ \t]*//' -e 's/[ \t]+$//'`"
	}

	print_host_list() {
	adjust_begin_index
	echo "Known hosts (${#filtered_host_list[@]}/${#host_list[@]})"
	local last_index
	let last_index=begin_index+HOST_DISPLAY_COUNT-1
	for (( i= $begin_index; i <= $last_index; i++ ));do
	if [ $i -ge ${#filtered_host_list[@]} ];then
	break
	fi
	local host="${filtered_host_list[i]}"
	local printmsg=" - $host"
	if [ -n "$hostname_filter" ];then
	local matched=`echo $host \| grep -o "$hostname_filter" \| head -n 1`
	printmsg=${printmsg/$matched/${underline}$matched${underline_off}}
	fi
	if [ "$selected_host" == "$host" ];then
	printmsg=${printmsg/ -/>>}
	printmsg=${bg_yellow}${black}$printmsg${reset}${bg_reset}
	else
	printmsg=${white}$printmsg${reset}
	fi
	echo -en "$printmsg"
	local blank=30
	local hostlen=${#host}
	let "blank=blank-hostlen"
	for (( k=0; k<$blank; k++));do echo -n " " ;done
	local description=$(host_description $host)
	echo -e "$description"
	done
	}
	#========================================
	# Logo And Help
	#========================================
	print_logo(){
	echo -en "${cyan}"
	echo -en "${reset}"
	echo -e "==============================================================================="

	}

	print_help() {
	echo -en " [/]: change user to rlogin "
	echo -en " [ctrl-n]: clear hostname "
	echo -e " [ctrl-c]: quit"
	echo -e " [=]: modify description for the selected host"
	echo -e " - hostname will be autosaved if it is valid.($HOST_LIST_FILE)"
	}

	print_command() {
	local command
	command="${command}${magenta}`whoami`@gw-kit> ${reset}"
	command="${command}${white}rlogin -l ${bold}$user ${yellow}$hostname_filter${reset}${bold_off}"

	echo "-------------------------------------------------------------------------------"
	if [ -n "$message" ];then
	echo -e "${red}${bold}[!] $message${bold_off}${reset}"
	message=
	else
	echo
	fi
	echo -en $command
	save_cursor
	echo
	echo "-------------------------------------------------------------------------------"
	}
	set_cursor() {
	for (( i=0; i > $cursor_offset; i-- ));do
	tput cub1
	done
	}

	print_screen() {
	clear
	print_logo
	print_help
	print_command
	print_host_list
	restore_cursor
	set_cursor
	}

	print_signature(){
	echo -e "${cyan}Gateway-kit${reset} ${blue}ver$VERSION${reset}"
	}

	#========================================
	# execute rlogin
	#========================================
	do_execute() {
	if [ -n "$selected_host" ];then
	hostname_filter=$selected_host
	selected_host=
	filter_host_list
	return
	fi
	if [ -z "$hostname_filter" ];then
	message="you should select a host to rlogin!"
	return
	fi
	clear
	local tmpfile="$(basename $0).$$.tmp"
	print_signature
	echo -e "Try to rlogin to ${yellow}$hostname_filter${reset} as ${magenta}$user${reset} ..."
	echo
	`rlogin -l $user $hostname_filter 2> $tmpfile > /dev/tty`
	if [ $? == 0 ];then
	clear
	add_host_to_list $hostname_filter
	else
	message=$( cat $tmpfile \| head -n 1)
	fi
	filter_host_list
	clear
	}

	good_bye(){
	tput cud1
	tput cud1
	tput ed
	echo -e "${reset}${red}Happy Hacking!${reset}"
	echo
	stty echo
	rm ~/gwk.sh.*.tmp &> /dev/null
	exit 0;
	}

	#==============================================================================
	# Initialize
	#==============================================================================
	init_host_list
	trap good_bye INT # trap ctrl+c

	#==============================================================================
	# Main Loop
	#==============================================================================
	while true;do
	print_screen
	k=$(read_key)
	case $k in
	# bash key bind
	ctrl-a ) cursor_move_begin;;
	ctrl-e ) cursor_move_end;;
	ctrl-b ) cursor_move_left;;
	ctrl-f ) cursor_move_right;;
	ctrl-d ) delete_next_ch_hostname_filter;;
	ctrl-h ) delete_prev_ch_hostname_filter;;
	ctrl-k ) delete_next_all_hostname_filter;;
	ctrl-u ) delete_prev_all_hostname_filter;;
	ctrl-w ) delete_prev_all_hostname_filter;;
	# common
	backspace ) delete_prev_ch_hostname_filter;;
	up ) select_prev_host;;
	down ) select_next_host;;
	left ) cursor_move_left;;
	right ) cursor_move_right;;
	tab ) complete_hostname;;
	/ ) next_user;;
	delete ) delete_next_ch_hostname_filter;;
	# special
	ctrl-n ) clear_hostname_filter;;
	= ) modify_host_description;;
	# inputhost name filter
	[-a-zA-Z0-9_\.] ) append_hostname_filter $k;;
	enter ) do_execute;;
	esac
	done