DataStudy/Packages/SocketIOUnity/Runtime/SocketIOUnity.cs

//
//  SocketIOUnity.cs
//  SocketIOUnity
//
//  Created by itisnajim on 10/30/2021.
//  Copyright (c) 2021 itisnajim. All rights reserved.
//

using System;
using System.Threading;
using System.Threading.Tasks;
using SocketIOClient;
using SocketIOClient.Messages;

public class SocketIOUnity : SocketIO
{
    public enum UnityThreadScope
    {
        Update,
        LateUpdate,
        FixedUpdate
    }
    public UnityThreadScope unityThreadScope = UnityThreadScope.Update;

    public SocketIOUnity(string uri, UnityThreadScope unityThreadScope = UnityThreadScope.Update) : base(uri)
    {
        CommonInit(unityThreadScope);
    }

    public SocketIOUnity(Uri uri, UnityThreadScope unityThreadScope = UnityThreadScope.Update) : base(uri)
    {
        CommonInit(unityThreadScope);
    }

    public SocketIOUnity(string uri, SocketIOOptions options, UnityThreadScope unityThreadScope = UnityThreadScope.Update) : base(uri, options)
    {
        CommonInit(unityThreadScope);
    }

    public SocketIOUnity(Uri uri, SocketIOOptions options, UnityThreadScope unityThreadScope = UnityThreadScope.Update) : base(uri, options)
    {
        CommonInit(unityThreadScope);
    }

    private void CommonInit(UnityThreadScope unityThreadScope)
    {
        UnityThread.initUnityThread();
        this.unityThreadScope = unityThreadScope;
    }

    /// <summary>
    /// Register a new handler for the given event.
    /// </summary>
    /// <param name="eventName"></param>
    /// <param name="callback"></param>
    public void OnUnityThread(string eventName, Action<SocketIOResponse> callback)
    {
        On(eventName, res =>
        {
            ExecuteInUnityThread(() => callback(res));
        });

    }

    public void OnAnyInUnityThread(OnAnyHandler handler)
    {
        OnAny((name, response) =>
        {
            ExecuteInUnityThread(() => handler(name, response));
        });
    }

    /// <summary>
    /// Emits an event to the socket
    /// </summary>
    /// <param name="eventName"></param>
    /// <param name="data">Any other parameters can be included. All serializable datastructures are supported, including byte[]</param>
    /// <returns></returns>
    public void Emit(string eventName, params object[] data)
    {
        EmitAsync(eventName, data).ContinueWith(t => {});
    }

    public void Emit(string eventName, Action<SocketIOResponse> ack, params object[] data)
    {
        EmitAsync(eventName, CancellationToken.None, ack, data).ContinueWith(t => {});
    }


    public async Task EmitStringAsJSONAsync(string eventName, string json)
    {

        var msg = new EventMessage
        {
            Namespace = Namespace,
            Event = eventName,
        };
        if (!string.IsNullOrEmpty(json))
        {
            msg.Json = "["+json+"]";
        }
        await _transport.SendAsync(msg, CancellationToken.None).ConfigureAwait(false);
    }

    public void EmitStringAsJSON(string eventName, string json)
    {
        EmitStringAsJSONAsync(eventName, json).ContinueWith(t => { });
    }

    public void Connect()
    {
        ConnectAsync().ContinueWith(t => {});
    }

    public void Disconnect()
    {
        DisconnectAsync().ContinueWith(t => {});
    }

    private void ExecuteInUnityThread(Action action)
    {
        switch (unityThreadScope)
        {
            case UnityThreadScope.LateUpdate :
                UnityThread.executeInLateUpdate(action);
                break;
            case UnityThreadScope.FixedUpdate :
                UnityThread.executeInFixedUpdate(action);
                break;
            default :
                UnityThread.executeInUpdate(action);
                break;
        }
    }

}