mirror of
https://github.com/Steffo99/better-tee.git
synced 2024-11-26 00:34:18 +00:00
118 lines
2.9 KiB
C#
118 lines
2.9 KiB
C#
#if UNITY_WEBGL && !UNITY_EDITOR
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using System.Net.Sockets;
|
|
using System.Net.WebSockets;
|
|
using System.Runtime.InteropServices;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using AOT;
|
|
using Ninja.WebSockets;
|
|
using UnityEngine;
|
|
|
|
namespace Mirror.Websocket
|
|
{
|
|
// this is the client implementation used by browsers
|
|
public class Client
|
|
{
|
|
static int idGenerator = 0;
|
|
static readonly Dictionary<int, Client> clients = new Dictionary<int, Client>();
|
|
|
|
public bool NoDelay = true;
|
|
|
|
public event Action Connected;
|
|
public event Action<ArraySegment<byte>> ReceivedData;
|
|
public event Action Disconnected;
|
|
public event Action<Exception> ReceivedError;
|
|
|
|
public bool Connecting { get; set; }
|
|
public bool IsConnected
|
|
{
|
|
get
|
|
{
|
|
return SocketState(m_NativeRef) != 0;
|
|
}
|
|
}
|
|
|
|
int m_NativeRef = 0;
|
|
readonly int id;
|
|
|
|
public Client()
|
|
{
|
|
id = Interlocked.Increment(ref idGenerator);
|
|
}
|
|
|
|
public void Connect(Uri uri)
|
|
{
|
|
clients[id] = this;
|
|
|
|
Connecting = true;
|
|
|
|
m_NativeRef = SocketCreate(uri.ToString(), id, OnOpen, OnData, OnClose);
|
|
}
|
|
|
|
public void Disconnect()
|
|
{
|
|
SocketClose(m_NativeRef);
|
|
}
|
|
|
|
// send the data or throw exception
|
|
public void Send(byte[] data)
|
|
{
|
|
SocketSend(m_NativeRef, data, data.Length);
|
|
}
|
|
|
|
|
|
#region Javascript native functions
|
|
[DllImport("__Internal")]
|
|
static extern int SocketCreate(
|
|
string url,
|
|
int id,
|
|
Action<int> onpen,
|
|
Action<int, IntPtr, int> ondata,
|
|
Action<int> onclose);
|
|
|
|
[DllImport("__Internal")]
|
|
static extern int SocketState(int socketInstance);
|
|
|
|
[DllImport("__Internal")]
|
|
static extern void SocketSend(int socketInstance, byte[] ptr, int length);
|
|
|
|
[DllImport("__Internal")]
|
|
static extern void SocketClose(int socketInstance);
|
|
|
|
#endregion
|
|
|
|
#region Javascript callbacks
|
|
|
|
[MonoPInvokeCallback(typeof(Action))]
|
|
public static void OnOpen(int id)
|
|
{
|
|
clients[id].Connecting = false;
|
|
clients[id].Connected?.Invoke();
|
|
}
|
|
|
|
[MonoPInvokeCallback(typeof(Action))]
|
|
public static void OnClose(int id)
|
|
{
|
|
clients[id].Connecting = false;
|
|
clients[id].Disconnected?.Invoke();
|
|
clients.Remove(id);
|
|
}
|
|
|
|
[MonoPInvokeCallback(typeof(Action))]
|
|
public static void OnData(int id, IntPtr ptr, int length)
|
|
{
|
|
byte[] data = new byte[length];
|
|
Marshal.Copy(ptr, data, 0, length);
|
|
|
|
clients[id].ReceivedData(new ArraySegment<byte>(data));
|
|
}
|
|
#endregion
|
|
}
|
|
}
|
|
|
|
#endif
|