Facepunch.Steamworks/Facepunch.Steamworks/SteamNetworkingSockets.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading.Tasks;
using Steamworks.Data;

namespace Steamworks
{
	public class SteamNetworkingSockets : SteamClass
	{
		internal static ISteamNetworkingSockets Internal;
		internal override SteamInterface Interface => Internal;

		internal override void InitializeInterface( bool server )
		{
			Internal = new ISteamNetworkingSockets( server );

			SocketInterfaces = new Dictionary<uint, SocketInterface>();
			ConnectionInterfaces = new Dictionary<uint, ConnectionInterface>();

			InstallEvents();
		}

#region SocketInterface

		static Dictionary<uint, SocketInterface> SocketInterfaces;

		internal static SocketInterface GetSocketInterface( uint id )
		{
			if ( SocketInterfaces == null ) return null;
			if ( id == 0 ) throw new System.ArgumentException( "Invalid Socket" );

			if ( SocketInterfaces.TryGetValue( id, out var isocket ) )
				return isocket;

			return null;
		}

		internal static void SetSocketInterface( uint id, SocketInterface iface )
		{
			if ( id == 0 ) throw new System.ArgumentException( "Invalid Socket" );
			SocketInterfaces[id] = iface;
		}
#endregion

#region ConnectionInterface
		static Dictionary<uint, ConnectionInterface> ConnectionInterfaces;


		internal static ConnectionInterface GetConnectionInterface( uint id )
		{
			if ( ConnectionInterfaces == null ) return null;
			if ( id == 0 ) return null;

			if ( ConnectionInterfaces.TryGetValue( id, out var iconnection ) )
				return iconnection;

			return null;
		}

		internal static void SetConnectionInterface( uint id, ConnectionInterface iface )
		{
			if ( id == 0 ) throw new System.ArgumentException( "Invalid Connection" );
			ConnectionInterfaces[id] = iface;
		}
#endregion



		internal static void InstallEvents( bool server = false )
		{
			Dispatch.Install<SteamNetConnectionStatusChangedCallback_t>( ConnectionStatusChanged, server );
		}


		private static void ConnectionStatusChanged( SteamNetConnectionStatusChangedCallback_t data )
		{
			//
			// This is a message from/to a listen socket
			//
			if ( data.Nfo.listenSocket.Id > 0 )
			{
				var iface = GetSocketInterface( data.Nfo.listenSocket.Id );
				iface?.OnConnectionChanged( data.Conn, data.Nfo );
			}
			else
			{
				var iface = GetConnectionInterface( data.Conn.Id );
				iface?.OnConnectionChanged( data.Nfo );
			}

			OnConnectionStatusChanged?.Invoke( data.Conn, data.Nfo );
		}

		public static event Action<Connection, ConnectionInfo> OnConnectionStatusChanged;


		/// <summary>
		/// Creates a "server" socket that listens for clients to connect to by calling
		/// Connect, over ordinary UDP (IPv4 or IPv6)
		/// </summary>
		public static T CreateNormalSocket<T>( NetAddress address ) where T : SocketInterface, new()
		{
			var t = new T();
			var options = new NetKeyValue[0];
			t.Socket = Internal.CreateListenSocketIP( ref address, options.Length, options );
			t.Initialize();

			SetSocketInterface( t.Socket.Id, t );
			return t;
		}

		/// <summary>
		/// Connect to a socket created via <method>CreateListenSocketIP</method>
		/// </summary>
		public static T ConnectNormal<T>( NetAddress address ) where T : ConnectionInterface, new()
		{
			var t = new T();
			var options = new NetKeyValue[0];
			t.Connection = Internal.ConnectByIPAddress( ref address, options.Length, options );
			SetConnectionInterface( t.Connection.Id, t );
			return t;
		}

		/// <summary>
		/// Creates a server that will be relayed via Valve's network (hiding the IP and improving ping)
		/// </summary>
		public static T CreateRelaySocket<T>( int virtualport = 0 ) where T : SocketInterface, new()
		{
			var t = new T();
			var options = new NetKeyValue[0];
			t.Socket = Internal.CreateListenSocketP2P( virtualport, options.Length, options );
			SetSocketInterface( t.Socket.Id, t );
			return t;
		}

		/// <summary>
		/// Connect to a relay server
		/// </summary>
		public static T ConnectRelay<T>( SteamId serverId, int virtualport = 0 ) where T : ConnectionInterface, new()
		{
			var t = new T();
			NetIdentity identity = serverId;
			var options = new NetKeyValue[0];
			t.Connection = Internal.ConnectP2P( ref identity, virtualport, options.Length, options );
			SetConnectionInterface( t.Connection.Id, t );
			return t;
		}
	}
}