r2mods/ilspy_dump/ror2_csproj/RoR2.Networking/NetworkMessageHandlerAttrib...

88 lines
2.6 KiB
C#

using System;
using System.Collections.Generic;
using System.Reflection;
using HG.Reflection;
using JetBrains.Annotations;
using UnityEngine;
using UnityEngine.Networking;
namespace RoR2.Networking;
[MeansImplicitUse]
[AttributeUsage(AttributeTargets.Method, AllowMultiple = false, Inherited = false)]
public class NetworkMessageHandlerAttribute : HG.Reflection.SearchableAttribute
{
public short msgType;
public bool server;
public bool client;
private NetworkMessageDelegate messageHandler;
private static List<NetworkMessageHandlerAttribute> clientMessageHandlers = new List<NetworkMessageHandlerAttribute>();
private static List<NetworkMessageHandlerAttribute> serverMessageHandlers = new List<NetworkMessageHandlerAttribute>();
[InitDuringStartup]
private static void CollectHandlers()
{
clientMessageHandlers.Clear();
serverMessageHandlers.Clear();
HashSet<short> hashSet = new HashSet<short>();
foreach (NetworkMessageHandlerAttribute instance in HG.Reflection.SearchableAttribute.GetInstances<NetworkMessageHandlerAttribute>())
{
MethodInfo methodInfo = instance.target as MethodInfo;
if (methodInfo == null || !methodInfo.IsStatic)
{
continue;
}
instance.messageHandler = (NetworkMessageDelegate)Delegate.CreateDelegate(typeof(NetworkMessageDelegate), methodInfo);
if (instance.messageHandler != null)
{
if (instance.client)
{
clientMessageHandlers.Add(instance);
hashSet.Add(instance.msgType);
}
if (instance.server)
{
serverMessageHandlers.Add(instance);
hashSet.Add(instance.msgType);
}
}
if (instance.messageHandler == null)
{
Debug.LogWarningFormat("Could not register message handler for {0}. The function signature is likely incorrect.", methodInfo.Name);
}
if (!instance.client && !instance.server)
{
Debug.LogWarningFormat("Could not register message handler for {0}. It is marked as neither server nor client.", methodInfo.Name);
}
}
for (short num = 48; num < 80; num++)
{
if (!hashSet.Contains(num))
{
Debug.LogWarningFormat("Network message MsgType.Highest + {0} is unregistered.", num - 47);
}
}
}
public static void RegisterServerMessages()
{
foreach (NetworkMessageHandlerAttribute serverMessageHandler in serverMessageHandlers)
{
NetworkServer.RegisterHandler(serverMessageHandler.msgType, serverMessageHandler.messageHandler);
}
}
public static void RegisterClientMessages(NetworkClient client)
{
foreach (NetworkMessageHandlerAttribute clientMessageHandler in clientMessageHandlers)
{
client.RegisterHandler(clientMessageHandler.msgType, clientMessageHandler.messageHandler);
}
}
}