#!/usr/bin/expect
#
# $Id: system_installer.sh 2804 2012-03-22 12:57:42Z pleblanc $
#
# Install RPM and custom OS files on system
# Argument 1 - Remote Module Name
# Argument 2 - Remote Server Host Name or IP address
# Argument 3 - Root Password of remote server
# Argument 4 - Package name being installed
# Argument 5 - Install Type, "initial" or "upgrade"
# Argument 6 - Debug flag 1 for on, 0 for off
set timeout 30
set USERNAME root
set MODULE [lindex $argv 0]
set SERVER [lindex $argv 1]
set PASSWORD [lindex $argv 2]
set RPMPACKAGE1 [lindex $argv 3]
set RPMPACKAGE2 [lindex $argv 4]
set RPMPACKAGE3 [lindex $argv 5]
set INSTALLTYPE [lindex $argv 6]
set DEBUG [lindex $argv 7]
log_user $DEBUG
spawn -noecho /bin/bash
#
if { $INSTALLTYPE == "initial" || $INSTALLTYPE == "uninstall" } {
	# 
	# erase package
	#
	send_user "Erase InfiniDB Packages on Module                 "
	expect -re {[$#] }
	send "ssh $USERNAME@$SERVER 'rpm -e --nodeps --allmatches calpont >/dev/null 2>&1; rpm -e --nodeps --allmatches infinidb-enterprise >/dev/null 2>&1; rpm -e --nodeps --allmatches infinidb-libs infinidb-platform'\n"
	expect {
		"Host key verification failed" { send_user "FAILED: Host key verification failed\n" ; exit }
		"service not known" { send_user "FAILED: Invalid Host\n" ; exit }
		"authenticity" { send "yes\n" 
							expect {
								"word: " { send "$PASSWORD\n" }
							}
		}
		"word: " { send "$PASSWORD\n" }
	}
	# password for ssh
	send "$PASSWORD\n"
	expect {
		-re {[$#] }                  { send_user "DONE" }
		"uninstall completed" { send_user "DONE" }
		"ERROR dependencies" { send_user "ERROR: ERROR dependencies\n" ; exit -1 }
		"Permission denied, please try again"   { send_user "ERROR: Invalid password\n" ; exit -1 }
		"error: package"       { send_user "INFO: Package not installed" }
		"not installed"       { send_user "INFO: Package not installed" }
	}
	send_user "\n"
}
if { $INSTALLTYPE == "uninstall" } { exit 0 }

# 
# send the package
#
expect -re {[$#] }
send_user "Copy New InfiniDB Packages to Module              "
send "ssh $USERNAME@$SERVER 'rm -f /root/infinidb-*.rpm'\n"
expect "word: "
# password for ssh
send "$PASSWORD\n"
# check return
expect {
	-re {[$#] } { }
}
send "scp $RPMPACKAGE1 $RPMPACKAGE2 $RPMPACKAGE3 $USERNAME@$SERVER:.\n"
expect "word: "
# send the password
send "$PASSWORD\n"
# check return
expect {
	"100%" 				{ send_user "DONE" }
	"scp"  				{ send_user "ERROR\n" ; 
				 			send_user "\n*** Installation ERROR\n" ; 
							exit -1 }
	"Permission denied, please try again"         { send_user "ERROR: Invalid password\n" ; exit -1 }
	"No such file or directory" { send_user "ERROR: Invalid package\n" ; exit -1 }
}
send_user "\n"
#
set timeout 60
if { $INSTALLTYPE == "initial"} {
	#
	# install package
	#
	send_user "Install InfiniDB Packages on Module               "
	send "ssh $USERNAME@$SERVER 'rpm -ivh $RPMPACKAGE1 $RPMPACKAGE2 $RPMPACKAGE3'\n"
	expect "word: "
	# password for ssh
	send "$PASSWORD\n"
	# check return
	expect {
		-re {[$#] } 		  		  { send_user "DONE" }
		"completed" 		  { send_user "DONE" }
		"ERROR dependencies" { send_user "ERROR: ERROR dependencies\n" ; 
									send_user "\n*** Installation ERROR\n" ; 
										exit -1 }
		"Permission denied, please try again"   { send_user "ERROR: Invalid password\n" ; exit -1 }
	}
} else {
	#
	# upgrade package
	#
	send_user "Upgrade InfiniDB Packages on Module               "
	send "ssh $USERNAME@$SERVER 'rpm -Uvh --noscripts $RPMPACKAGE1 $RPMPACKAGE2 $RPMPACKAGE3'\n"
	expect "word: "
	# password for ssh
	send "$PASSWORD\n"
	# check return
	expect {
		-re {[$#] } 		  		  { send_user "DONE" }
		"completed" 		  { send_user "DONE" }
		"already installed"   { send_user "INFO: Already Installed\n" ; exit -1 }
		"ERROR dependencies" { send_user "ERROR: ERROR dependencies\n" ; 
									send_user "\n*** Installation ERROR\n" ; 
										exit -1 }
		"Permission denied, please try again"   { send_user "ERROR: Invalid password\n" ; exit -1 }
	}
}
expect -re {[$#] }
send_user "\n"
set timeout 30
#
if { $INSTALLTYPE == "initial"} {
	#
	# copy over InfiniDB OS files
	#
	send_user "Copy InfiniDB OS files to Module                 "
	send "scp /usr/local/MariaDB/Columnstore/local/etc/$MODULE/*  $USERNAME@$SERVER:/usr/local/MariaDB/Columnstore/local/.\n"
	expect "word: "
	# send the password
	send "$PASSWORD\n"
	expect {
		-re {[$#] } 		  		  { send_user "DONE" }
		"Permission denied, please try again"   { send_user "ERROR: Invalid password\n" ; exit -1 }
	}
	#
	send_user "\n"
	set timeout 120
	#
	# Start module installer to setup Customer OS files
	#
	send_user "Run Module Installer                            "
	send "ssh $USERNAME@$SERVER '/usr/local/MariaDB/Columnstore/bin/module_installer.sh'\n"
	expect "word: "
	# send the password
	send "$PASSWORD\n"
	expect {
		"!!!Module" 				  			{ send_user "DONE" }
		"Permission denied, please try again"   { send_user "ERROR: Invalid password\n" ; exit -1 }
		"FAILED"   								{ send_user "ERROR: missing OS file\n" ; exit -1 }
	}
	send_user "\n"
}
#
send_user "\nInstallation Successfully Completed on '$MODULE'\n"
exit 0