worldbox-observer-mod/IdleSpectator/Story/LifeSagaSessionPatches.cs

using System;
using System.Reflection;
using HarmonyLib;

namespace IdleSpectator;

/// <summary>Reset life-saga session when the live map finishes loading or is cleared.</summary>
[HarmonyPatch]
public static class LifeSagaSessionPatches
{
    [HarmonyPatch(typeof(MapBox), nameof(MapBox.finishingUpLoading))]
    [HarmonyPostfix]
    public static void PostfixFinishingUpLoading()
    {
        try
        {
            LifeSagaSession.OnWorldLoadFinished();
        }
        catch
        {
            // Load path must not throw into the game.
        }
    }

    [HarmonyPatch(typeof(MapBox), nameof(MapBox.clearWorld))]
    [HarmonyPrefix]
    public static void PrefixClearWorld()
    {
        try
        {
            LifeSagaSession.ResetForWorldChange("clear-world");
        }
        catch
        {
            // Teardown path must not throw into the game.
        }
    }
}

/// <summary>Promote narrative recovery state only after WorldBox confirms a save completed.</summary>
[HarmonyPatch]
public static class NarrativeSaveCompletionPatch
{
    public static MethodBase TargetMethod()
    {
        Type[] types = typeof(MapBox).Assembly.GetTypes();
        for (int i = 0; i < types.Length; i++)
        {
            MethodInfo method = AccessTools.Method(types[i], "finishSaving");
            if (method != null)
            {
                return method;
            }
        }

        return null;
    }

    [HarmonyPostfix]
    public static void Postfix()
    {
        try
        {
            NarrativePersistence.SaveCommittedWorld();
        }
        catch
        {
            // Saving the base world must never depend on the narrative sidecar.
        }
    }
}