using System.Collections;
using System.Collections.Generic;
using System.Linq;
using UnityEngine;
using UnityEngine.Networking;
using FishNet.Transporting;
using System;
using System.Net;
using System.Net.Sockets;

namespace FishBait
{
    public partial class FishBaitTransport : Transport
    {
        public void RequestServerList(FishBaitRegions searchRegion = FishBaitRegions.Any)
        {
            if (_isAuthenticated && _connectedToRelay)
                StartCoroutine(GetServerList(searchRegion));
            else
                Debug.Log("You must be connected to Relay to request server list!");
        }

        IEnumerator RelayConnect()
        {
            string url = $"http://{loadBalancerAddress}:{loadBalancerPort}/api/join/";
            serverStatus = "Waiting for LLB...";
            using (UnityWebRequest webRequest = UnityWebRequest.Get(url))
            {
                // Request and wait for the desired page.
                webRequest.SetRequestHeader("x-Region", ((int)region).ToString());
                webRequest.SetRequestHeader("Access-Control-Allow-Credentials", "true");
                webRequest.SetRequestHeader("Access-Control-Allow-Headers", "Accept, X-Access-Token, X-Application-Name, X-Request-Sent-Time");
                webRequest.SetRequestHeader("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
                webRequest.SetRequestHeader("Access-Control-Allow-Origin", "*");

                yield return webRequest.SendWebRequest();
                var result = webRequest.downloadHandler.text;

#if UNITY_2020_1_OR_NEWER
                switch (webRequest.result)
                {
                    case UnityWebRequest.Result.ConnectionError:
                    case UnityWebRequest.Result.DataProcessingError:
                    case UnityWebRequest.Result.ProtocolError:
                        Debug.LogWarning("FishBait | Network Error while getting a relay to join from Load Balancer.");
                        break;
                    case UnityWebRequest.Result.Success:
                        var parsedAddress = JsonUtility.FromJson<RelayAddress>(result);
                        Connect(parsedAddress.address, parsedAddress.port);
                        endpointServerPort = parsedAddress.endpointPort;
                        break;
                }
#else
                if (webRequest.isNetworkError || webRequest.isHttpError)
                {
                    Debug.LogWarning("FishBait | Network Error while getting a relay to join from Load Balancer.");
                }
                else
                {
                    // join here
                    var parsedAddress = JsonUtility.FromJson<RelayAddress>(result);
                    Connect(parsedAddress.address, parsedAddress.port);
                    endpointServerPort = parsedAddress.endpointPort;
                }
#endif
            }
        }

        IEnumerator JoinOtherRelayAndMatch(Room? roomValue, string ID)
        {
            var room = new Room();

            // using load balancer, we NEED the server's relay address
            if (roomValue.HasValue)
                room = roomValue.Value;
            else
            {
                _serverListUpdated = false;
                RequestServerList();

                yield return new WaitUntil(() => _serverListUpdated);

                var foundRoom = GetServerForID(ID);

                if (foundRoom.HasValue)
                {
                    room = foundRoom.Value;
                }
                else
                {
                    Debug.LogWarning("FishBait | Client tried to join a server that does not exist!");
                    OnClientDisconnected?.Invoke();
                    yield break;
                }
            }

            // Wait for disconnection
            DisconnectFromRelay();

            while (IsAuthenticated())
                yield return null;

            endpointServerPort = room.relayInfo.endpointPort;
            Connect(room.relayInfo.address, room.relayInfo.port);

            while (!IsAuthenticated())
                yield return null;

            int pos = 0;
            _directConnected = false;
            _clientSendBuffer.WriteByte(ref pos, (byte)OpCodes.JoinServer);
            _clientSendBuffer.WriteString(ref pos, room.serverId);
            _clientSendBuffer.WriteBool(ref pos, _directConnectModule != null);

            string local = GetLocalIp();

            _clientSendBuffer.WriteString(ref pos, local ?? "0.0.0.0");

            _isClient = true;

            transport.SendToServer(0, new System.ArraySegment<byte>(_clientSendBuffer, 0, pos));

        }

        IEnumerator GetServerList(FishBaitRegions region)
        {
            if (!useLoadBalancer)
            {
                string uri = $"http://{serverIP}:{endpointServerPort}/api/compressed/servers";

                using (UnityWebRequest webRequest = UnityWebRequest.Get(uri))
                {
                    webRequest.SetRequestHeader("Access-Control-Allow-Credentials", "true");
                    webRequest.SetRequestHeader("Access-Control-Allow-Headers", "Accept, X-Access-Token, X-Application-Name, X-Request-Sent-Time");
                    webRequest.SetRequestHeader("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
                    webRequest.SetRequestHeader("Access-Control-Allow-Origin", "*");

                    // Request and wait for the desired page.
                    yield return webRequest.SendWebRequest();
                    var result = webRequest.downloadHandler.text;

#if UNITY_2020_1_OR_NEWER
                    switch (webRequest.result)
                    {
                        case UnityWebRequest.Result.ConnectionError:
                        case UnityWebRequest.Result.DataProcessingError:
                        case UnityWebRequest.Result.ProtocolError:
                            Debug.LogWarning("FishBait | Network Error while retreiving the server list!");
                            break;

                        case UnityWebRequest.Result.Success:
                            relayServerList?.Clear();
                            relayServerList = JsonUtilityHelper.FromJson<Room>(result.Decompress()).ToList();
                            serverListUpdated?.Invoke();
                            break;
                    }
#else
                    if (webRequest.isNetworkError || webRequest.isHttpError)
                    {
                        Debug.LogWarning("FishBait | Network Error while retreiving the server list!");
                    }
                    else
                    {
                        relayServerList?.Clear();
                        relayServerList = JsonUtilityHelper.FromJson<Room>(result.Decompress()).ToList();
                        serverListUpdated?.Invoke();
                    }
#endif
                }
            }
            else // get master list from load balancer
            {
                yield return StartCoroutine(RetrieveMasterServerListFromLoadBalancer(region));
            }

            /// <summary>
            /// Gets master list from the LB.
            /// This can be optimized but for now it is it's
            /// own separate method, so i can understand wtf is going on :D
            /// </summary>
            /// <returns></returns>
            IEnumerator RetrieveMasterServerListFromLoadBalancer(FishBaitRegions region)
            {
                string uri = $"http://{loadBalancerAddress}:{loadBalancerPort}/api/masterlist/";

                using (UnityWebRequest webRequest = UnityWebRequest.Get(uri))
                {
                    webRequest.SetRequestHeader("x-Region", ((int)region).ToString());
                    // Request and wait for the desired page.
                    yield return webRequest.SendWebRequest();
                    var result = webRequest.downloadHandler.text;

#if UNITY_2020_1_OR_NEWER
                    switch (webRequest.result)
                    { 
                        case UnityWebRequest.Result.ConnectionError:
                        case UnityWebRequest.Result.DataProcessingError:
                        case UnityWebRequest.Result.ProtocolError:
                            Debug.LogWarning("FishBait | Network Error while retreiving the server list!");
                            break;

                        case UnityWebRequest.Result.Success:
                            relayServerList?.Clear();
                            relayServerList = JsonUtilityHelper.FromJson<Room>(result).ToList();
                            serverListUpdated?.Invoke();
                            _serverListUpdated = true;
                            break;
                    }
#else
                if (webRequest.isNetworkError || webRequest.isHttpError)
                {
                    Debug.LogWarning("FishBait | Network Error while retreiving the server list!");
                }
                else
                {
                    relayServerList?.Clear();
                    relayServerList = JsonUtilityHelper.FromJson<Room>(result).ToList();
                    serverListUpdated?.Invoke();
                    _serverListUpdated = true;
                }
#endif
                }
            }
        }

        private static string GetLocalIp()
        {
            var host = Dns.GetHostEntry(Dns.GetHostName());
            foreach (var ip in host.AddressList)
            {
                if (ip.AddressFamily == AddressFamily.InterNetwork)
                {
                    return ip.ToString();
                }
            }

            return null;
        }
    }


}