/* 
  *  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
  
  // This script runs the connection tests we need to run.
  
  private var natCapable : ConnectionTesterStatus = ConnectionTesterStatus.Undetermined;
  static var filterNATHosts = false;
  private var probingPublicIP = false;
  private var doneTestingNAT = false;
  private var timer : float = 0.0;
  
  private var hideTest = false;
  private var testMessage = "Undetermined NAT capabilities";
  
  function Start(){
                  
          // Start connection test
          natCapable = Network.TestConnection();
          
          yield;
          
          if (Network.HavePublicAddress()){
                  Debug.Log("This machine has a public IP address");
          }else{
                  Debug.Log("This machine has a private IP address");
          }        
  }
  
  private var testing : boolean = false;
  
  function Update(){
          if (!doneTestingNAT && !testing) {
                  TestConnection();
          }
  
  }
  
  function TestConnection() {
          // Start/Poll the connection test, report the results in a label and react to the results accordingly
          testing=true;
          
          natCapable = Network.TestConnection();
          yield WaitForSeconds(0.5);        
  
          switch (natCapable) {
                  case ConnectionTesterStatus.Error: 
                          testMessage = "Problem determining NAT capabilities";
                          doneTestingNAT = true;
                          break;
                          
                  case ConnectionTesterStatus.Undetermined: 
                          testMessage = "Undetermined NAT capabilities";
                          doneTestingNAT = false;
                          break;
                          
                  case ConnectionTesterStatus.PrivateIPNoNATPunchthrough: 
                          testMessage = "Cannot do NAT punchthrough, filtering NAT enabled hosts for client connections, local LAN games only.";
                          filterNATHosts = true;
                          Network.useNat = true;
                          doneTestingNAT = true;
                          break;
                          
                  case ConnectionTesterStatus.PrivateIPHasNATPunchThrough:
                          if (probingPublicIP)
                                  testMessage = "Non-connectable public IP address (port  blocked), NAT punchthrough can circumvent the firewall.";
                          else
                                  testMessage = "NAT punchthrough capable. Enabling NAT punchthrough functionality.";
                          // NAT functionality is enabled in case a server is started,
                          // clients should enable this based on if the host requires it
                          Network.useNat = true;
                          doneTestingNAT = true;
                          break;
                          
                  case ConnectionTesterStatus.PublicIPIsConnectable:
                          testMessage = "Directly connectable public IP address.";
                          Network.useNat = false;
                          doneTestingNAT = true;
                          break;
                          
                  // This case is a bit special as we now need to check if we can 
                  // cicrumvent the blocking by using NAT punchthrough
                  case ConnectionTesterStatus.PublicIPPortBlocked:
                          testMessage = "Non-connectble public IP address (port  blocked), running a server is impossible.";
                          Network.useNat = false;
                          // If no NAT punchthrough test has been performed on this public IP, force a test
                          if (!probingPublicIP)
                          {
                                  Debug.Log("Testing if firewall can be circumnvented");
                                  natCapable = Network.TestConnectionNAT();
                                  probingPublicIP = true;
                                  timer = Time.time + 10;
                          }
                          // NAT punchthrough test was performed but we still get blocked
                          else if (Time.time > timer)
                          {
                                  probingPublicIP = false;                 // reset
                                  Network.useNat = true;
                                  doneTestingNAT = true;
                          }
                          break;
                  case ConnectionTesterStatus.PublicIPNoServerStarted:
                          testMessage = "Public IP address but server not initialized, it must be started to check server accessibility. Restart connection test when ready.";
                          break;
                  default: 
                          testMessage = "Error in test routine, got " + natCapable;
          }
          Debug.Log(testMessage);
  
          
          if(doneTestingNAT){        
                  officialNATstatus=Network.useNat;
                  Debug.Log("TestConn:"+testMessage);
                  Debug.Log("TestConn:"+natCapable + " " + probingPublicIP + " " + doneTestingNAT);
          }
          testing=false;
  }