#! /usr/bin/ruby
#
#   2
#  / \
# 1-5-4
#  \ / \
#   3---6
#
# 1: dga
# 3: dwen
# 4: srini
# 5: vyass
# 6: gnychis
#

## SKELETON STOLEN FROM http://www.bigbold.com/snippets/posts/show/1785
require 'socket'
require 'rdaemon'
require 'irc'
#$SAFE = 1

$total_points = 0
$total_tests = 0

##
# The main program.  Tests are listed below this point.  All tests
# should call the "result" function to report if they pass or fail.
##

def write_config()
    File.open("final_grading1.conf", "w") { |afile|
	afile.puts "1 127.0.0.1 34100 34101 34102"
	afile.puts "2 127.0.0.1 34103 34104 34105"
	afile.puts "3 127.0.0.1 34106 34107 34108"
	afile.puts "5 127.0.0.1 34112 34113 34114"
    }

    File.open("final_grading2.conf", "w") { |afile|
	afile.puts "1 127.0.0.1 34100 34101 34102"
	afile.puts "2 127.0.0.1 34103 34104 34105"
	afile.puts "4 127.0.0.1 34109 34110 34111"
    }

    File.open("final_grading3.conf", "w") { |afile|
	afile.puts "1 127.0.0.1 34100 34101 34102"
	afile.puts "3 127.0.0.1 34106 34107 34108"
	afile.puts "4 127.0.0.1 34109 34110 34111"
	afile.puts "6 127.0.0.1 34115 34116 34117"
    }

    File.open("final_grading4.conf", "w") { |afile|
	afile.puts "4 127.0.0.1 34109 34110 34111"
	afile.puts "2 127.0.0.1 34103 34104 34105"
	afile.puts "3 127.0.0.1 34106 34107 34108"
	afile.puts "5 127.0.0.1 34112 34113 34114"
	afile.puts "6 127.0.0.1 34115 34116 34117"
    }
    
		File.open("final_grading5.conf", "w") { |afile|
	afile.puts "1 127.0.0.1 34100 34101 34102"
	afile.puts "4 127.0.0.1 34109 34110 34111"
	afile.puts "5 127.0.0.1 34112 34113 34114"
    }
    
		File.open("final_grading6.conf", "w") { |afile|
	afile.puts "4 127.0.0.1 34109 34110 34111"
	afile.puts "3 127.0.0.1 34106 34107 34108"
	afile.puts "6 127.0.0.1 34115 34116 34117"
    }
end

# Go through the config file, looking for the "id" passed,
# and spawn it
def spawn_daemon(file,id)
afile = File.open(file, "r")
afile.each_line do |line|
	line.chomp!
	(nodeid, ip, lport, dport, sport)=line.split
	if(nodeid.to_i == id)  # dont forget to convert to int
		puts "./srouted -i #{id} -c #{file} -a 1 &"
	  system("./srouted -i #{id} -c #{file} -a 1 &> /dev/null &")
		return 1 
	end
end
end

def conn_daemon(file,id)
afile = File.open(file, "r")
afile.each_line do |line|
	line.chomp!
	(nodeid, ip, lport, dport, sport)=line.split
	if(nodeid.to_i == id)  # dont forget to convert to int
		rdaemon = RDAEMON.new(ip, dport.to_i, '', '')
		rdaemon.connect()
		return rdaemon 
	end
end
end

def spawn_server(config,id)
afile = File.open(config, "r")
afile.each_line do |line|
	line.chomp!
	(nodeid, ip, lport, dport, sport)=line.split
	if(nodeid.to_i == id)  # dont forget to convert to int
		puts "./sircd #{id} #{config} &> /dev/null &"
		system("./sircd #{id} #{config} &> /dev/null &")
		return 1 
	end
end
end

def conn_server(config,id)
afile = File.open(config, "r")
afile.each_line do |line|
	line.chomp!
	(nodeid, ip, lport, dport, sport)=line.split
	if(nodeid.to_i == id)  # dont forget to convert to int
		server = IRC.new(ip, sport.to_i, '', '')
		server.connect()
		return server
	end
end
end
		
def test_name(n)
    puts "////// #{n} \\\\\\\\\\\\"
    return n
end

def result(n, passed_exp, failed_exp, passed, points)
    explanation = nil
		$total_tests += 1
    if (passed)
	puts "(+) #{n} passed"
	$total_points += points
	explanation = passed_exp
    else
	puts "(-) #{n} failed"
	explanation = failed_exp
    end

    if (explanation)
	puts ": #{explanation}"
    else
	puts ""
    end
end

def eval_test(n, passed_exp, failed_exp, passed, points = 1)
    result(n, passed_exp, failed_exp, passed, points)
#   exit(0) if !passed
end

#### WRITE CONFIG FILE
#
# YOU mAY EDIT THE PORT NUMBERS HERE ONLY
# (to avoid port collisions while testing on andrew)
write_config()

begin

# Spawn the daemons.  This runs command lines of
# ./srouted -i ## -c cprnode##.conf  (where ## == 1 through 4)
spawn_daemon("final_grading1.conf", 1)
spawn_daemon("final_grading2.conf", 2)
spawn_daemon("final_grading3.conf", 3)
spawn_daemon("final_grading4.conf", 4)
spawn_daemon("final_grading5.conf", 5)
spawn_daemon("final_grading6.conf", 6)

puts "Letting the daemons settle."
sleep(2)

spawn_server("final_grading1.conf", 1)
spawn_server("final_grading2.conf", 2)
spawn_server("final_grading3.conf", 3)
spawn_server("final_grading4.conf", 4)
spawn_server("final_grading5.conf", 5)
spawn_server("final_grading6.conf", 6)

# Let the servers settle
puts "Letting the servers settle."
sleep(3)


irc1 = conn_server("final_grading1.conf", 1)
irc2 = conn_server("final_grading2.conf", 2)
irc3 = conn_server("final_grading3.conf", 3)
irc4 = conn_server("final_grading4.conf", 4)
irc5 = conn_server("final_grading5.conf", 5)
irc6 = conn_server("final_grading6.conf", 6)

# Let everything settle
puts "Letting everything settle."
sleep(2)


################## USER PRIVMSG TEST #################
#
	irc1.send_nick("dga")
  irc1.send_user("please give me :The MOTD")
	irc1.join_chan("#15441")
	irc3.send_nick("dwen")
  irc3.send_user("please give me :The MOTD")
	irc3.join_chan("#15441")
	irc4.send_nick("srini")
  irc4.send_user("please give me :The MOTD")
	irc4.join_chan("#15441")
	irc5.send_nick("vyass")
  irc5.send_user("please give me :The MOTD")
	irc5.join_chan("#15441")
	irc6.send_nick("gnychis")
  irc6.send_user("please give me :The MOTD")
	irc6.join_chan("#15441")
	irc1.ignore_reply()
	irc3.ignore_reply()
	irc4.ignore_reply()
	irc5.ignore_reply()
	irc6.ignore_reply()

	puts "Letting user routing tables settle"
	sleep(5)

	tn = test_name("dwen --PRIVMSG--> #15441")
	irc3.send_privmsg("#15441", "dump trucks run the Internet")
	eval_test(tn, nil, nil, irc1.checkmsg("dwen", "#15441", "dump trucks run the Internet"))
	eval_test(tn, nil, nil, irc4.checkmsg("dwen", "#15441", "dump trucks run the Internet"))
	eval_test(tn, nil, nil, irc5.checkmsg("dwen", "#15441", "dump trucks run the Internet"))
	eval_test(tn, nil, nil, irc6.checkmsg("dwen", "#15441", "dump trucks run the Internet"))
	
	tn = test_name("gnychis --PRIVMSG--> #15441")
	irc6.send_privmsg("#15441", "totally, I see them every day walking to school")
	eval_test(tn, nil, nil, irc1.checkmsg("gnychis", "#15441", "totally, I see them every day walking to school"))
	eval_test(tn, nil, nil, irc4.checkmsg("gnychis", "#15441", "totally, I see them every day walking to school"))
	eval_test(tn, nil, nil, irc5.checkmsg("gnychis", "#15441", "totally, I see them every day walking to school"))
	eval_test(tn, nil, nil, irc3.checkmsg("gnychis", "#15441", "totally, I see them every day walking to school"))
	
	tn = test_name("vyass --PRIVMSG--> #15441")
	irc5.send_privmsg("#15441", "I drive one of them")
	eval_test(tn, nil, nil, irc1.checkmsg("vyass", "#15441", "I drive one of them"))
	eval_test(tn, nil, nil, irc4.checkmsg("vyass", "#15441", "I drive one of them"))
	eval_test(tn, nil, nil, irc6.checkmsg("vyass", "#15441", "I drive one of them"))
	eval_test(tn, nil, nil, irc3.checkmsg("vyass", "#15441", "I drive one of them"))
	
	# Test the ability to join another channel which should part the current channel
	irc1.join_chan("#private")
	irc1.ignore_reply()
	irc4.join_chan("#private")
	irc4.ignore_reply()

	# Test communication in the new private channel, make sure the TA's don't get it
	tn = test_name("srini --PRIVMSG--> #private")
	irc4.send_privmsg("#private", "If they seriously think dump trucks run the Internet, maybe we should give them time off")
	eval_test(tn, nil, nil, irc1.checkmsg("srini", "#private", "If they seriously think dump trucks run the Internet, maybe we should give them time off"))
	
	tn = test_name("dga --PRIVMSG--> #private")
	irc1.send_privmsg("#private", "Lets just make them code more ruby")
	eval_test(tn, nil, nil, irc4.checkmsg("dga", "#private", "Lets just make them code more ruby"))

	# Make sure gnychis, dwen, and vyass didn't get the #private messages
	tn = test_name("gnychis silence")
	eval_test(tn, nil, nil, irc6.test_silence(1))
	tn = test_name("dwen silence")
	eval_test(tn, nil, nil, irc3.test_silence(1))
	tn = test_name("vyass silence")
	eval_test(tn, nil, nil, irc5.test_silence(1))
	
	# Send another message in #15441 and make sure dga and srini don't hear it
	tn = test_name("gnychis --PRIVMSG--> #15441")
	irc6.send_privmsg("#15441", "I think we scared them away")
	eval_test(tn, nil, nil, irc5.checkmsg("gnychis", "#15441", "I think we scared them away"))
	eval_test(tn, nil, nil, irc3.checkmsg("gnychis", "#15441", "I think we scared them away"))

	# test the silence
	tn = test_name("dga silence")
	eval_test(tn, nil, nil, irc1.test_silence(1))
	tn = test_name("srini silence")
	eval_test(tn, nil, nil, irc4.test_silence(1))

rescue Interrupt
rescue Exception => detail
    puts detail.message()
    print detail.backtrace.join("\n")
ensure
#    rdaemon.disconnect()
		system("killall srouted")
		system("killall sircd")
		puts "\nTests passed: #{$total_points} / #{$total_tests}\n"
end