TheKingsRace/Assets/Scripts/Network/ConnectButtons.cs

using MLAPI;
using MLAPI.SceneManagement;
using MLAPI.Transports.UNET;
using System.Collections;
using System.Linq;
using UnityEngine;
using UnityEngine.UI;

public class ConnectButtons : MonoBehaviour {

    // The IP Used to connect to the dedicated server
    const string DEDICATED_SERVER_IP = "127.0.0.1";
    const int DEDICATED_SERVER_PORT = 7777;

    private UNetTransport transport;

    public GameObject ErrorPanel;
    public Text ErrorText;

    public InputField ipAddressField;

    private int connectionTimeoutTime = 5;

    void Start() {
        // Make sure the Error Panel is not enabled to start
        ErrorPanel.SetActive(false);

        // Find the UNetTransport object that is associated with the NetworkManager
        transport = NetworkManager.Singleton.GetComponent<UNetTransport>();
    }

    // Connect to the dedicated server
    public void ConnectDedicatedServer () {
        // Set the connection address to be equal to the ip address of the dedicated server
        transport.ConnectAddress = DEDICATED_SERVER_IP;
        transport.ConnectPort = DEDICATED_SERVER_PORT;

        // Set the password to connect with
        NetworkManager.Singleton.NetworkConfig.ConnectionData = System.Text.Encoding.ASCII.GetBytes("kingsrace");

        // Start the client on the address
        NetworkManager.Singleton.StartClient();

        // Run a coroutine to check if the client connects to the server
        StartCoroutine(checkIsConnectedClient());
    }

    // Host a private server
    public void HostPrivateServer() {
        NetworkManager.Singleton.StartHost();
        NetworkSceneManager.SwitchScene("Lobby");
    }

    // Connect to a private server via IP Address
    public void ConnectToPrivateServer() {
        string ipAddress = ipAddressField.text;
        int port = 7777;

        // Check if the IP Address is valid
        if (!ValidateIPv4(ipAddress)) {
            ThrowError("Invalid IP Address!  Please enter a valid IP Address and try again!");
            return;
        }

        // IP Address is valid - Attempt to connect

        // Set the connection address to be equal to the ip address entered into the input field
        transport.ConnectAddress = ipAddress;
        transport.ConnectPort = port;

        // Set the password to connect with
        NetworkManager.Singleton.NetworkConfig.ConnectionData = System.Text.Encoding.ASCII.GetBytes("kingsrace");

        // Start the client on the address
        NetworkManager.Singleton.StartClient();

        // Run a coroutine to check if the client connects to the server
        StartCoroutine(checkIsConnectedClient());
    }

    IEnumerator checkIsConnectedClient() {
        yield return new WaitForSecondsRealtime(connectionTimeoutTime);

        if (!NetworkManager.Singleton.IsConnectedClient) {
            // Failed to connect to the server
            NetworkManager.Singleton.StopClient();

            ThrowError("Could not connect to server!\nReason: Connection Timed Out");
        }
    }

    private void ThrowError(string errorMsg) {
        ErrorText.text = errorMsg;

        ErrorPanel.SetActive(true);
    }

    public void DismissError() {
        ErrorText.text = "";

        ErrorPanel.SetActive(false);
    }

    // Method by: Habib (Stack Overflow)
    // https://stackoverflow.com/questions/11412956/what-is-the-best-way-of-validating-an-ip-address
    private bool ValidateIPv4(string ipString) {
        if (string.IsNullOrWhiteSpace(ipString)) {
            return false;
        }

        string[] splitValues = ipString.Split('.');
        if (splitValues.Length != 4) {
            return false;
        }

        byte tempForParsing;

        return splitValues.All(r => byte.TryParse(r, out tempForParsing));
    }
}