r2mods/ilspy_dump/ror2_csproj/RoR2/BadClientEntityStateMachine...

100 lines
2.4 KiB
C#
Raw Normal View History

2024-10-04 07:26:37 +00:00
using System;
using System.Collections.Generic;
using EntityStates;
using UnityEngine;
namespace RoR2;
public static class BadClientEntityStateMachineFix
{
private class Watcher
{
public LocalUser localUser;
private TimerQueue.TimerHandle? currentTimerHandle;
public void OnInstall()
{
localUser.onBodyChanged += OnBodyChanged;
}
public void OnUninstall()
{
localUser.onBodyChanged -= OnBodyChanged;
if (currentTimerHandle.HasValue)
{
RoR2Application.fixedTimeTimers.RemoveTimer(currentTimerHandle.Value);
currentTimerHandle = null;
}
}
private void OnBodyChanged()
{
if (currentTimerHandle.HasValue)
{
RoR2Application.fixedTimeTimers.RemoveTimer(currentTimerHandle.Value);
currentTimerHandle = null;
}
if ((bool)localUser.cachedBody)
{
currentTimerHandle = RoR2Application.fixedTimeTimers.CreateTimer(1f, OnTimer);
}
}
private void OnTimer()
{
currentTimerHandle = null;
ForceInitializeEntityStateMachines(localUser.cachedBody);
}
private static void ForceInitializeEntityStateMachines(CharacterBody characterBody)
{
if (!characterBody)
{
return;
}
EntityStateMachine[] components = characterBody.GetComponents<EntityStateMachine>();
for (int i = 0; i < components.Length; i++)
{
EntityStateMachine entityStateMachine = components[i];
if (!entityStateMachine.HasPendingState() && entityStateMachine.state is Uninitialized)
{
EntityState entityState = EntityStateCatalog.InstantiateState(ref entityStateMachine.mainStateType);
if (entityState != null)
{
Debug.LogFormat("Setting {0} uninitialized state machine [{1}] next state to {2}", characterBody.name, i, entityState.GetType().Name);
entityStateMachine.SetInterruptState(entityState, InterruptPriority.Any);
}
}
}
}
}
private static readonly Dictionary<LocalUser, Watcher> watchers = new Dictionary<LocalUser, Watcher>();
[SystemInitializer(new Type[] { })]
private static void Init()
{
LocalUserManager.onUserSignIn += OnUserSignIn;
LocalUserManager.onUserSignOut += OnUserSignOut;
}
private static void OnUserSignIn(LocalUser localUser)
{
Watcher watcher = new Watcher
{
localUser = localUser
};
watchers[localUser] = watcher;
watcher.OnInstall();
}
private static void OnUserSignOut(LocalUser localUser)
{
Watcher watcher = watchers[localUser];
watcher.OnUninstall();
watcher.localUser = null;
watchers.Remove(localUser);
}
}