Refactor helper assignment in PrefabManager to use Main.helper
This commit is contained in:
234
Main.cs
234
Main.cs
@@ -1,4 +1,4 @@
|
|||||||
using Assets.Code;
|
using Assets.Code;
|
||||||
using Assets.Code.UI;
|
using Assets.Code.UI;
|
||||||
using Assets.Interface;
|
using Assets.Interface;
|
||||||
using Harmony;
|
using Harmony;
|
||||||
@@ -2034,238 +2034,6 @@ namespace KCM
|
|||||||
|
|
||||||
#endregion
|
#endregion
|
||||||
|
|
||||||
/**
|
|
||||||
*
|
|
||||||
* Find all Player.inst references and reconstruct method with references to client planyer
|
|
||||||
*
|
|
||||||
* Instantiating main player object and setting landmass teamid in KCPLayer
|
|
||||||
*
|
|
||||||
* E.G instead of Player.inst, it should be Main.kCPlayers[Client].player for example, and the rest of the code is the same
|
|
||||||
*
|
|
||||||
* Prefix that sets Player.inst to the right client instance and then calls that instances method?
|
|
||||||
*
|
|
||||||
*/
|
|
||||||
|
|
||||||
[HarmonyPatch]
|
|
||||||
public class PlayerReferencePatch
|
|
||||||
{
|
|
||||||
static IEnumerable<MethodBase> TargetMethods()
|
|
||||||
{
|
|
||||||
Assembly assembly = typeof(Player).Assembly;
|
|
||||||
|
|
||||||
Type[] types = new Type[] { typeof(Player)/*, typeof(World), typeof(LandmassOwner), typeof(Keep), typeof(Villager), typeof(DragonSpawn), typeof(DragonController), typeof(Dragon)*/ };
|
|
||||||
|
|
||||||
var methodsInNamespace = types
|
|
||||||
.SelectMany(t => t.GetMethods(BindingFlags.NonPublic | BindingFlags.Public | BindingFlags.Instance | BindingFlags.DeclaredOnly).Where(m => !m.IsAbstract))
|
|
||||||
.ToList();
|
|
||||||
|
|
||||||
helper.Log("Methods in namespace: " + methodsInNamespace.Count);
|
|
||||||
|
|
||||||
return methodsInNamespace.ToArray().Cast<MethodBase>();
|
|
||||||
}
|
|
||||||
|
|
||||||
static IEnumerable<CodeInstruction> Transpiler(MethodBase method, IEnumerable<CodeInstruction> instructions)
|
|
||||||
{
|
|
||||||
int PlayerInstCount = 0;
|
|
||||||
|
|
||||||
var codes = new List<CodeInstruction>(instructions);
|
|
||||||
for (var i = 0; i < codes.Count; i++)
|
|
||||||
{
|
|
||||||
if (codes[i].opcode == OpCodes.Ldsfld && codes[i].operand.ToString() == "Player inst")
|
|
||||||
{
|
|
||||||
PlayerInstCount++;
|
|
||||||
|
|
||||||
codes[i].opcode = (OpCodes.Ldarg_0); // Replace all instance methods static ref with "this" instead of Player.inst
|
|
||||||
|
|
||||||
// Replace ldsfld Player::inst with the sequence to load from Main.kCPlayers
|
|
||||||
// Step 1: Load Main.kCPlayers onto the evaluation stack.
|
|
||||||
//codes[i] = new CodeInstruction(OpCodes.Ldsfld, typeof(Main).GetField("kCPlayers"));
|
|
||||||
|
|
||||||
// Step 2: Load the value of Main.PlayerSteamID onto the evaluation stack as the key
|
|
||||||
//codes.Insert(++i, new CodeInstruction(OpCodes.Ldsfld, typeof(Main).GetField("PlayerSteamID")));
|
|
||||||
|
|
||||||
// Step 3: Call Dictionary<TKey, TValue>.get_Item(TKey key) to get the Player instance.
|
|
||||||
//codes.Insert(++i, new CodeInstruction(OpCodes.Callvirt, typeof(Dictionary<string, KCPlayer>).GetMethod("get_Item")));
|
|
||||||
|
|
||||||
// Now, access the 'inst' field of the fetched Player instance, if necessary.
|
|
||||||
//codes.Insert(++i, new CodeInstruction(OpCodes.Ldfld, typeof(KCPlayer).GetField("inst")));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (PlayerInstCount > 0)
|
|
||||||
Main.helper.Log($"Found {PlayerInstCount} static Player.inst references in {method.Name}");
|
|
||||||
|
|
||||||
return codes.AsEnumerable();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
[HarmonyPatch]
|
|
||||||
public class BuildingPlayerReferencePatch
|
|
||||||
{
|
|
||||||
static IEnumerable<MethodBase> TargetMethods()
|
|
||||||
{
|
|
||||||
Assembly assembly = typeof(Building).Assembly;
|
|
||||||
|
|
||||||
Type[] types = new Type[] { typeof(Building) };
|
|
||||||
|
|
||||||
var methodsInNamespace = types
|
|
||||||
.SelectMany(t => t.GetMethods(BindingFlags.NonPublic | BindingFlags.Public | BindingFlags.Instance | BindingFlags.DeclaredOnly).Where(m => !m.IsAbstract))
|
|
||||||
.ToList();
|
|
||||||
|
|
||||||
helper.Log("Methods in namespace: " + methodsInNamespace.Count);
|
|
||||||
|
|
||||||
return methodsInNamespace.ToArray().Cast<MethodBase>();
|
|
||||||
}
|
|
||||||
|
|
||||||
static IEnumerable<CodeInstruction> Transpiler(MethodBase method, IEnumerable<CodeInstruction> instructions)
|
|
||||||
{
|
|
||||||
int PlayerInstCount = 0;
|
|
||||||
|
|
||||||
var codes = new List<CodeInstruction>(instructions);
|
|
||||||
MethodInfo getPlayerByBuildingMethodInfo = typeof(Main).GetMethod("GetPlayerByBuilding", BindingFlags.Static | BindingFlags.Public);
|
|
||||||
|
|
||||||
for (var i = 0; i < codes.Count; i++)
|
|
||||||
{
|
|
||||||
if (codes[i].opcode == OpCodes.Ldsfld && codes[i].operand.ToString() == "Player inst")
|
|
||||||
{
|
|
||||||
PlayerInstCount++;
|
|
||||||
|
|
||||||
// Check if the current instruction is ldsfld Player.inst
|
|
||||||
if (codes[i].opcode == OpCodes.Ldsfld && codes[i].operand.ToString().Contains("Player inst"))
|
|
||||||
{
|
|
||||||
// Replace the instruction sequence
|
|
||||||
// Step 1: Load 'this' for the Building instance
|
|
||||||
codes[i].opcode = OpCodes.Ldarg_0;
|
|
||||||
|
|
||||||
// Step 2: Call GetPlayerByBuilding(Building instance) static method in Main
|
|
||||||
var callTeamID = new CodeInstruction(OpCodes.Call, getPlayerByBuildingMethodInfo);
|
|
||||||
codes.Insert(++i, callTeamID);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (PlayerInstCount > 0)
|
|
||||||
Main.helper.Log($"Found {PlayerInstCount} static building Player.inst references in {method.Name}");
|
|
||||||
|
|
||||||
return codes.AsEnumerable();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
[HarmonyPatch]
|
|
||||||
public class PlayerPatch
|
|
||||||
{
|
|
||||||
static IEnumerable<MethodBase> TargetMethods()
|
|
||||||
{
|
|
||||||
var meth = typeof(Player).GetMethods(BindingFlags.NonPublic | BindingFlags.Public | BindingFlags.Instance | BindingFlags.DeclaredOnly);
|
|
||||||
return meth.Cast<MethodBase>();
|
|
||||||
}
|
|
||||||
|
|
||||||
public static bool Prefix(MethodBase __originalMethod, Player __instance)
|
|
||||||
{
|
|
||||||
if (__originalMethod.Name.Equals("Awake") && (KCServer.IsRunning || KCClient.client.IsConnected))
|
|
||||||
{
|
|
||||||
helper.Log("Awake run on player instance while server is running");
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (__originalMethod.Name.Equals("Awake") && __instance.gameObject.name.Contains("Client Player"))
|
|
||||||
{
|
|
||||||
helper.Log("Awake run on client instance");
|
|
||||||
try
|
|
||||||
{
|
|
||||||
//___defaultEnabledFlags = new bool[38];
|
|
||||||
//for (int i = 0; i < ___defaultEnabledFlags.Length; i++)
|
|
||||||
//{
|
|
||||||
// ___defaultEnabledFlags[i] = true;
|
|
||||||
//}
|
|
||||||
//__instance.PlayerLandmassOwner = __instance.gameObject.AddComponent<LandmassOwner>();
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
//helper.Log(__instance.PlayerLandmassOwner.ToString());
|
|
||||||
}
|
|
||||||
catch (Exception e)
|
|
||||||
{
|
|
||||||
helper.Log(e.ToString());
|
|
||||||
helper.Log(e.Message);
|
|
||||||
helper.Log(e.StackTrace);
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (__originalMethod.Name.Equals("Update") && __instance.gameObject.name.Contains("Client Player"))
|
|
||||||
{
|
|
||||||
//helper.Log("Update run on client instance");
|
|
||||||
try
|
|
||||||
{
|
|
||||||
//___defaultEnabledFlags = new bool[38];
|
|
||||||
//for (int i = 0; i < ___defaultEnabledFlags.Length; i++)
|
|
||||||
//{
|
|
||||||
// ___defaultEnabledFlags[i] = true;
|
|
||||||
//}
|
|
||||||
//__instance.PlayerLandmassOwner = __instance.gameObject.AddComponent<LandmassOwner>();
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
//helper.Log(__instance.PlayerLandmassOwner.ToString());
|
|
||||||
}
|
|
||||||
catch (Exception e)
|
|
||||||
{
|
|
||||||
helper.Log(e.ToString());
|
|
||||||
helper.Log(e.Message);
|
|
||||||
helper.Log(e.StackTrace);
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (__originalMethod.Name.Equals("Update"))
|
|
||||||
{
|
|
||||||
//helper.Log($"Update called for: {__instance.gameObject.name}");
|
|
||||||
|
|
||||||
try
|
|
||||||
{
|
|
||||||
if (KCClient.client.IsConnected && !__instance.gameObject.name.Contains("Client Player"))
|
|
||||||
{
|
|
||||||
StateObserver.RegisterObserver(__instance, new string[] {
|
|
||||||
"bannerIdx", "kingdomHappiness", "landMassHappiness", "landMassIntegrity", "bDidFirstFire", "CurrYear",
|
|
||||||
"timeAtFailHappiness", "hasUsedCheats", "nameForOldAgeDeath", "deathsThisYear", /*"poorHealthGracePeriod",*/
|
|
||||||
});
|
|
||||||
|
|
||||||
//StateObserver.Update(__instance);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
catch (Exception e)
|
|
||||||
{
|
|
||||||
helper.Log(e.ToString());
|
|
||||||
helper.Log(e.Message);
|
|
||||||
helper.Log(e.StackTrace);
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void Postfix(MethodBase __originalMethod, Player __instance)
|
|
||||||
{
|
|
||||||
if (__originalMethod.Name.Equals("Update"))
|
|
||||||
{
|
|
||||||
//helper.Log($"Update called for: {__instance.gameObject.name} in POSTFIX");
|
|
||||||
|
|
||||||
|
|
||||||
//helper.Log("CHECKING ALL COMPONENTS IN UPDATE: ");
|
|
||||||
//Component[] components = __instance.gameObject.GetComponents<Component>();
|
|
||||||
|
|
||||||
//foreach (Component component in components)
|
|
||||||
//{
|
|
||||||
// helper.Log("--- " + component.GetType().kingdomName);
|
|
||||||
//}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#region "Unity Log Hooks"
|
#region "Unity Log Hooks"
|
||||||
|
|
||||||
[HarmonyPatch(typeof(UnityEngine.Debug), "Log", new Type[] { typeof(object) })]
|
[HarmonyPatch(typeof(UnityEngine.Debug), "Log", new Type[] { typeof(object) })]
|
||||||
|
|||||||
@@ -26,9 +26,9 @@ namespace KCM
|
|||||||
{
|
{
|
||||||
helper = _helper;
|
helper = _helper;
|
||||||
}
|
}
|
||||||
else if (Main.instance != null)
|
else if (Main.helper != null)
|
||||||
{
|
{
|
||||||
helper = Main.instance.helper;
|
helper = Main.helper;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (helper == null)
|
if (helper == null)
|
||||||
|
|||||||
Reference in New Issue
Block a user