/* 
*  This file is part of the Unity networking tutorial by M2H (http://www.M2H.nl)
*  The original author of this code is Mike Hergaarden, even though some small parts 
*  are copied from the Unity tutorials/manuals.
*  Feel free to use this code for your own projects, drop us a line if you made something exciting! 
*/
#pragma strict

var skin : GUISkin;

var gameName = "Example3_Lobby";
var serverPort = 45671;

private var timeoutHostList = 0.0;
private var lastHostListRequest = -1000.0;
private var hostListRefreshTimeout = 60.0;

private var windowRect1;
private var windowRect2;

static var quickplayMode : boolean = false;
static var quickplayModeStarted  : float = 0.0;
private var tryingToConnectquickplay : boolean = false;
private var tryingToConnectquickplayNumber : int = 0;
private var readyToConnect : boolean = false;

private var remotePort : int[] = new int[3];
private var remoteIP : String[] = new String[3];
private var connectionInfo : String = "";

private var lastMSConnectionAttemptForcedNat : boolean= false;
private var NAToptionWasSwitchedForTesting : boolean = false;
private var officialNATstatus : boolean = Network.useNat;
private var errorMessage : String = "";
private var lastquickplayConnectionTime : float;


var customButton : GUIStyle;

private var showMenu : boolean = false;

private var mainMenuScript : MainMenu;

/////////////////////////////

function Awake(){
	windowRect1 = Rect (Screen.width/2-305,Screen.height/2-140,380,280);
  	windowRect2 = Rect (Screen.width/2+110,Screen.height/2-140,220,100);
}

function Start(){
	mainMenuScript =  MainMenu.SP;
}


function EnableMenu(quickplay : boolean){
	showMenu=true;
	
	tryingToConnectquickplayModeNumber=0;
	tryingToConnectquickplay=false;
	
	quickplayMode=quickplay;
	
	if(quickplay){		
		quickplayModeStarted=Time.time;
	}
		
	MasterServer.RequestHostList (gameName);
	lastHostListRequest = Time.realtimeSinceStartup;
	
	remoteIP[0]="127.0.0.1";
	remotePort[0]=serverPort;
}


function OnFailedToConnectToMasterServer(info: NetworkConnectionError)
{
	//Dont request the MS list the next 60 sec
	lastHostListRequest = Time.realtimeSinceStartup + 60;
}

function OnFailedToConnect(info: NetworkConnectionError)
{
	Debug.Log("Failed To Connect info: "+info);
	
	var invalidPass : boolean = false;
	if(info==NetworkConnectionError.InvalidPassword){
		invalidPass=true;
	}
	FailedConnRetry(invalidPass);	
}

function FailedConnRetry(invalidPassword : boolean){
	if(tryingToConnectquickplay){
		//Try again without NAT if we used NAT
		if(Network.useNat && lastMSConnectionAttemptForcedNat){
			Debug.Log("Failed 1A");
		
			lastMSConnectionAttemptForcedNat=false;
			Network.useNat=false;
			remotePort[0]=serverPort;//Fall back to default server port
			Network.Connect(remoteIP, remotePort[0]);
			lastquickplayConnectionTime=Time.time;
		}else{
			Debug.Log("Failed 1B");
		
			//Reset NAT to org value
			Network.useNat=officialNATstatus;
						
			tryingToConnectquickplayNumber++;
			tryingToConnectquickplay=false;
		}
	}else{
		
		//Direct connect or host list
		connectionInfo="Failed to connect!";
		
		if(Network.useNat && lastMSConnectionAttemptForcedNat){
			Debug.Log("Failed 2A");
		
			Network.useNat=false;
			Network.Connect(remoteIP, remotePort[0]);
			lastquickplayConnectionTime=Time.time;
		}else{
			Debug.Log("Failed 2B");
		
			errorMessage="Failed to connect to "+remoteIP[0]+":"+remotePort[0];
			if(invalidPassword){
				errorMessage+="\nYou used the wrong password (or you didn't need to enter one!).";
			}
		//reset default port
			remotePort[0]=serverPort;
			
			//Reset nat to tested value
			Network.useNat=officialNATstatus;
			
		}
	}	
}


function OnConnectedToServer(){
	Debug.Log("Connected to lobby!");
	showMenu=false;
	var gameLobbyScript : GameLobby = GetComponent(GameLobby);
	gameLobbyScript.EnableLobby();
	
}

function OnGUI ()
{		
	if(!showMenu){
		return;
	}


	//Back to main menu
	if(GUI.Button(Rect(40,10,150,20), "Back to main menu")){
		showMenu=false;
		mainMenuScript.OpenMenu("multiplayer");
	}

	
	if(errorMessage && errorMessage!=""){	
		GUI.Box(Rect(Screen.width/2-100,Screen.height/2-30,200,60), "Error");
		GUI.Label(Rect(Screen.width/2-90,Screen.height/2-15,180,50), errorMessage);
		if(GUI.Button(Rect(Screen.width/2+40,Screen.height/2+5,50,20), "Close")){
			errorMessage="";
		}
	}
	
	if(quickplayMode){
		quickplayFunction();
	}else{
		if(!errorMessage || errorMessage==""){ //Hide windows on error
			windowRect1 = GUILayout.Window (0, windowRect1, listGUI, "Join a game via the list");	
			windowRect2 = GUILayout.Window (1, windowRect2, directConnectGUIWindow, "Directly join a game via an IP");	
			//windowRect3 = GUILayout.Window (2, windowRect3, hostGUI, "Host a game");
		}	
	}
		
	
}


function quickplayFunction(){
	
		GUI.Box(Rect(Screen.width/4+180,Screen.height/2-30,280,50), "");
		
		var i : int=0;
		var data : HostData[] = MasterServer.PollHostList();
		for (var element in data)
		{
			// Do not display NAT enabled games if we cannot do NAT punchthrough
			if ( !(natTester.filterNATHosts && element.useNat) && !element.passwordProtected )
			{
				aHost=1;
								
				if(element.connectedPlayers<element.playerLimit)
				{					
					if(tryingToConnectquickplay){
						var natText;
						if(Network.useNat){
							natText=" with option 1/2";
						}else{
							natText=" with option 2/2";
						}
						GUI.Label(Rect(Screen.width/4+200,Screen.height/2-25,285,50), "Trying to connect to host "+(tryingToConnectquickplayNumber+1)+"/"+data.length+" "+natText);
						if(!Network.useNat && lastquickplayConnectionTime+0.75<=Time.time || lastquickplayConnectionTime+5.00<=Time.time){
							FailedConnRetry(false);							
						}
						return;	
					}		
					if(!tryingToConnectquickplay && tryingToConnectquickplayNumber<=i){
						Debug.Log("Trying to connect to game nr "+i+" & "+tryingToConnectquickplayNumber);
						tryingToConnectquickplay=true;
						tryingToConnectquickplayNumber=i;
						// Enable NAT functionality based on what the hosts if configured to do
						lastMSConnectionAttemptForcedNat=element.useNat;
						remoteIP=element.ip;
						remotePort[0]=element.port;
						Network.useNat = element.useNat;
						var connectPort : int=element.port;
						if (Network.useNat){
							print("Using Nat punchthrough to connect");
						}else{
							print("Connecting directly to host");
						}
						Debug.Log("connecting to "+element.gameName+" "+element.ip+":"+connectPort);
						Network.Connect(element.ip, connectPort);	
						lastquickplayConnectionTime=Time.time;
					}
					i++;		
				}
			}
			
		}
		
		//List is empty, d ont give up yet: Give MS 5 seconds to feed the list
		if(Time.time<quickplayModeStarted+5){
			//Debug.Log("PlayNow: delay up to 3 sec (no hosts yet)" );		
			GUI.Label(Rect(Screen.width/4+200,Screen.height/2-25,280,50), "Waiting for masterserver..."+Mathf.Ceil((quickplayModeStarted+5)-Time.time) );
			return;	
		}
		
		if(!tryingToConnectquickplay){
			Debug.Log("PlayNow: No games hosted, so hosting one ourselves");			
				showMenu=false;
				var gameLobbyScript : GameLobby = GetComponent(GameLobby);
				gameLobbyScript.EnableLobby();	
		}
		
	
}

var password : String = "";
function directConnectGUIWindow(id : int){

	GUILayout.BeginVertical();
	GUILayout.Space(5);
	GUILayout.EndVertical();
	GUILayout.Label(connectionInfo);
		
	
	GUILayout.BeginHorizontal();
	GUILayout.Space(10);
		remoteIP[0] = GUILayout.TextField(remoteIP[0], GUILayout.MinWidth(70));
		remotePort[0] = parseInt(GUILayout.TextField(remotePort[0]+""));
	GUILayout.Space(10);
	GUILayout.EndHorizontal();

	GUILayout.BeginHorizontal();
	GUILayout.Space(10);
	GUILayout.Label("Password");	
	password = GUILayout.TextField(password, GUILayout.MinWidth(50));
	GUILayout.Space(10);
	GUILayout.EndHorizontal();
	
	GUILayout.BeginHorizontal();
	GUILayout.Space(10);
	Network.useNat = GUILayout.Toggle(Network.useNat, "Advanced: Use NAT");
	GUILayout.EndHorizontal();
	
	
	GUILayout.BeginHorizontal();
	GUILayout.Space(10);
	GUILayout.FlexibleSpace();
	if (GUILayout.Button ("Connect"))
	{
		lastMSConnectionAttemptForcedNat=Network.useNat;
		connectionInfo="";
		Network.Connect(remoteIP, remotePort[0], password);
	}		
	GUILayout.FlexibleSpace();
	GUILayout.EndHorizontal();
	
}

var scrollPosition : Vector2;

function listGUI (id : int) {
	
		GUILayout.BeginVertical();
		GUILayout.Space(6);
		GUILayout.EndVertical();
	
		
		GUILayout.BeginHorizontal();
		GUILayout.Space(200);

		// Refresh hosts
		if (GUILayout.Button ("Refresh available Servers") || Time.realtimeSinceStartup > lastHostListRequest + hostListRefreshTimeout)
		{
			if(Time.realtimeSinceStartup > lastHostListRequest + 5){//max once per 5 sec
				MasterServer.RequestHostList (gameName);
				lastHostListRequest = Time.realtimeSinceStartup;
			}
		}
		
		GUILayout.FlexibleSpace();
		
		GUILayout.EndHorizontal();

		
		
		scrollPosition = GUILayout.BeginScrollView (scrollPosition);

		var aHost =0;
		
		var data : HostData[] = MasterServer.PollHostList();
		for (var element in data)
		{
			
			GUILayout.BeginHorizontal();

			// Do not display NAT enabled games if we cannot do NAT punchthrough
			if ( !(natTester.filterNATHosts && element.useNat) )
			{
				aHost=1;
				var name = element.gameName + " ";
				
				GUILayout.Label(name);	
				GUILayout.FlexibleSpace();
				GUILayout.Label(element.connectedPlayers + "/" + element.playerLimit);
				
				if(element.useNat){
					GUILayout.Label(".");
				}
				GUILayout.FlexibleSpace();
				GUILayout.Label("[" + element.ip[0] + ":" + element.port + "]");	
				GUILayout.FlexibleSpace();
				
				if(element.passwordProtected){
					GUILayout.Label("PASSWORD");
					if (GUILayout.Button("Connect"))
					{
						if(password==""){
							errorMessage="You must enter a password if you want to join this game!";
							return;	
						}// Enable NAT functionality based on what the hosts if configured to do
						Network.useNat = element.useNat;
						lastMSConnectionAttemptForcedNat = element.useNat;
						if (Network.useNat){
							print("Using Nat punchthrough to connect");
						}else{
							print("Connecting directly to host");
						}
						remoteIP=element.ip;
						remotePort[0]=element.port;
						var connectPort2 : int=element.port;
						Network.Connect(element.ip, connectPort2, password);			
					}

				}else{
					if (GUILayout.Button("Connect"))
					{
						// Enable NAT functionality based on what the hosts if configured to do
						Network.useNat = element.useNat;
						lastMSConnectionAttemptForcedNat = element.useNat;
						if (Network.useNat){
							print("Using Nat punchthrough to connect");
						}else{
							print("Connecting directly to host");
						}
						remoteIP=element.ip;
						remotePort[0]=element.port;
						var connectPort : int=element.port;	
						Network.Connect(element.ip, connectPort);			
					}
				}
				GUILayout.Space(15);
			}
			GUILayout.EndHorizontal();	
		}
		if(aHost==0){
			GUILayout.Label("No games hosted at the moment..");
		}
		
		GUILayout.EndScrollView ();
}