#!/bin/bash

export LANG="zh_CN.UTF-8"
export LC_CTYPE="en_US"
debug=true

mydir=$(cd $(dirname $0);pwd)
me=$mydir/$(basename $0)
homedir=$(dirname $mydir)

usage() {
   echo "usage: command ip index"
   echo "       command help"
   echo "ip index list:"
   grep -P '^\d+\)' $me | awk '{print $1,$3}' 
}

exp=$mydir/mylogin.exp
[ "$debug" = "true" -o ! -f $exp ] && {
cat /dev/null >$exp
cat >>$exp <<EOF
#!/user/bin/expect
set timeout -1
proc timeoutFunc {} {
    send_user "except timeout\n";
    exit;
}
proc eofFunc {} {
    send_user "process is stoped\n";
    exit;
}
set ssh_ip [lindex \$argv 0]
spawn ssh \$ssh_ip
expect {
    "Last login:" {}
    timeout timeoutFunc
    eof eofFunc
}
expect {
    "app@VM" {}
    "feinenxiang@VM" { send "sudo su - app\r" }
    timeout timeoutFunc
    eof eofFunc
}
interact
EOF
}

loginFunction() {
    loginIp=$1
    echo "expect file is $(basename $exp)"
    expect $exp $loginIp
}

case "$1" in
1)  loginFunction 10.107.98.1
	;;
2)  loginFunction 10.107.98.2
    ;;
3)  loginFunction 10.107.98.3
    ;;
4)  loginFunction 10.107.98.4
    ;;
*)  usage
esac