#!/usr/bin/expect -f
# Original code from StackOverflow.com - Modified by Steve Hajek
# Liquidware Labs - 10/16/2015

set server [lindex $argv 0]
set ntp1 [lindex $argv 1]
set ntp2 [lindex $argv 2]

if { [ string length $server ] == 0 } {
	return [puts "Do not execute directly, run ./ntpch.sh ntp1.domain.local ntp2.domain.local instead."]
}
if { [ string length $ntp2 ] == 0 } {
spawn ssh -to StrictHostKeychecking=no friend@$server su
expect "assword:"
send "sspassword\r"
expect "assword:"
send "sspassword\r"
expect "friend]#"
send "cp /etc/ntp.conf /etc/ntp.conf.save\r"
send "sed -ie 's/server /#server /' /etc/ntp.conf; sed -ie 's/#server  127/server  127/' /etc/ntp.conf; sed -ie 's/#server 127/server 127/' /etc/ntp.conf\r"
send "echo server $ntp1 >> /etc/ntp.conf\r"
expect "friend]#"
send "/sbin/service ntpd restart\r"
expect "friend]#"
send "exit\r"
expect eof
	return [puts "---Added 1 NTP server.---"]
}

spawn ssh -to StrictHostKeychecking=no friend@$server su
expect "assword:"
send "sspassword\r"
expect "assword:"
send "sspassword\r"
expect "friend]#"
send "cp /etc/ntp.conf /etc/ntp.conf.save\r"
send "sed -ie 's/server /#server /' /etc/ntp.conf; sed -ie 's/#server  127/server  127/' /etc/ntp.conf; sed -ie 's/#server 127/server 127/' /etc/ntp.conf\r"
send "echo server $ntp1 >> /etc/ntp.conf\r"
expect "friend]#"
send "echo server $ntp2 >> /etc/ntp.conf\r"
expect "friend]#"
send "/sbin/service ntpd restart\r"
expect "friend]#"
send "exit\r"
expect eof
return [puts "---Added 2 NTP servers.---"]
