SubModule.cs

/*
 * Author: Thor Tronrud
 * SubModule.cs:
 * 
 * Bannerlord's required SubModule class. Provides access to
 * the virtual methods we want to implement to load the mod into
 * the game.
 * 
 * This mod does not change save files - and only activates on game start/load
 * so it can be turned on and off as desired.
 * In the starting methods I also add several event listeners based on values from
 * the settings file, so if they're deactivated they aren't even invoked.
 */

using System;
using System.IO;
using System.Xml;
using System.Xml.Serialization;
using TaleWorlds.CampaignSystem;
using TaleWorlds.Core;
using TaleWorlds.Library;
using TaleWorlds.MountAndBlade;
//using HarmonyLib;
using TaleWorlds.CampaignSystem.MapEvents;
using System.Reflection;

namespace DistinguishedService
{
    public class dsSubModule : MBSubModuleBase
    {
        private Settings CurrentSettings { get; set; }
        public static dsSubModule instance;
        public bool gamestarted = false;
        private static PromotionManager _pm = null;

        protected override void OnGameStart(Game game, IGameStarter gameStarterObject)
        {
            if (!(Game.Current.GameType is Campaign))
                return; //OnCampaignStart apparently doesn't always mean it's a campaign game -_-

            try
            {
                //Try to see if the OG modules file exists, if so preferentially use that
                string path = Path.Combine(TaleWorlds.ModuleManager.ModuleHelper.GetModuleFullPath("DistinguishedService110"), "Settings.xml");
                if (File.Exists(Path.Combine(BasePath.Name, "Modules", "DistinguishedService110", "Settings.xml")))
                {
                    path = Path.Combine(BasePath.Name, "Modules", "DistinguishedService110", "Settings.xml");
                }
                this.DeserializeObject(path);
            }
            catch (Exception ex)
            {
                InformationManager.DisplayMessage(new InformationMessage("Could not serialize Settings.xml: " + ex.Message.ToString() + " Using default values!", Color.FromUint(4282569842U)));
                this.CurrentSettings = new Settings();
            }

            try
            {
                if (PromotionManager.__instance == null)
                {
                    _pm = new PromotionManager();
                }
                else
                {
                    _pm = PromotionManager.__instance;
                }

                ((CampaignGameStarter)gameStarterObject).AddBehavior((CampaignBehaviorBase)new DSBattleBehavior());

                //ai gaining companions
                if (CurrentSettings.ai_promotion_chance > 0)
                {
                    try
                    {
                        InformationManager.DisplayMessage(new InformationMessage("AI Companions have been disabled for now!", Color.FromUint(4282569842U)));
                        //CampaignEvents.MapEventEnded.AddNonSerializedListener((object)this, new Action<MapEvent>(_pm.MapEventEnded));
                    }
                    catch(Exception e)
                    {
                        InformationManager.DisplayMessage(new InformationMessage("Serialized Listener for MapEventEnded could not be created. AI Promotion chance reset to 0.", Color.FromUint(4282569842U)));
                        CurrentSettings.ai_promotion_chance = 0;
                    }
                }

                //for alternative nomination behaviour
                if (CurrentSettings.upgrade_to_hero)
                {
                    CampaignEvents.PlayerUpgradedTroopsEvent.AddNonSerializedListener((object)this, new Action<CharacterObject, CharacterObject, int>(_pm.upgrade_to_hero));
                    CampaignEvents.OnUnitRecruitedEvent.AddNonSerializedListener((object)this, new Action<CharacterObject, int>(_pm.recruit_to_hero));

                }

                InformationManager.DisplayMessage(new InformationMessage("Distinguished Service loaded successfully", Colors.Blue));
            }
            catch (Exception ex)
            {
                InformationManager.DisplayMessage(new InformationMessage("There was a problem:\n" + ex.ToString(), Colors.Blue));
            }
            gamestarted = true;
        }
        
        public override void OnGameLoaded(Game game, object initializerObject)
        {
            if (!(game.GameType is Campaign))
                return;
            bool reload = false;
            try
            {
                //Try to see if the OG modules file exists, if so preferentially use that
                string path = Path.Combine(TaleWorlds.ModuleManager.ModuleHelper.GetModuleFullPath("DistinguishedService110"), "Settings.xml");
                if (File.Exists(Path.Combine(BasePath.Name, "Modules", "DistinguishedService110", "Settings.xml")))
                {
                    path = Path.Combine(BasePath.Name, "Modules", "DistinguishedService110", "Settings.xml");
                }
                this.DeserializeObject(path);
            }
            catch (Exception ex)
            {
                InformationManager.DisplayMessage(new InformationMessage("Could not serialize Settings.xml: " + ex.Message.ToString() + " Using default values!", Color.FromUint(4282569842U)));
                this.CurrentSettings = new Settings();
            }

            try
            {
                if(PromotionManager.__instance == null)
                {
                    _pm = new PromotionManager();
                }
                else
                {
                    _pm = PromotionManager.__instance;
                    reload = true;
                }
                
                if(!gamestarted)
                    ((CampaignGameStarter)initializerObject).AddBehavior((CampaignBehaviorBase)new DSBattleBehavior());

                //ai gaining companions
                if(CurrentSettings.ai_promotion_chance > 0)
                {
                    try
                    {
                        InformationManager.DisplayMessage(new InformationMessage("AI Companions have been disabled for now!", Color.FromUint(4282569842U)));
                        //CampaignEvents.MapEventEnded.AddNonSerializedListener((object)this, new Action<MapEvent>(_pm.MapEventEnded));
                    }
                    catch (Exception e)
                    {
                        InformationManager.DisplayMessage(new InformationMessage("Serialized Listener for MapEventEnded could not be created. AI Promotion chance reset to 0.", Color.FromUint(4282569842U)));
                        CurrentSettings.ai_promotion_chance = 0;
                    }
                }

                //for alternative nomination behaviour
                if (CurrentSettings.upgrade_to_hero)
                {
                    CampaignEvents.PlayerUpgradedTroopsEvent.AddNonSerializedListener((object)this, new Action<CharacterObject, CharacterObject, int>(_pm.upgrade_to_hero));
                    CampaignEvents.OnUnitRecruitedEvent.AddNonSerializedListener((object)this, new Action<CharacterObject, int>(_pm.recruit_to_hero));
                }

                if(!reload)
                    InformationManager.DisplayMessage(new InformationMessage("Distinguished Service loaded successfully", Colors.Blue));
            }
            catch (Exception ex)
            {
                InformationManager.DisplayMessage(new InformationMessage("There was a problem:\n" + ex.ToString(), Colors.Blue));
            }
            gamestarted = false;
        }

        //Once the game's been fully loaded, we check whether MLWB is loaded
        //and set the state accordingly
        public override void OnAfterGameInitializationFinished(Game game, object starterObject)
        {
            PromotionManager.MyLittleWarbandLoaded = CheckMLWBLoaded();
            gamestarted = false;
        }
        //stupid function to determine if my little warband is loaded
        //because the default, random equipment selection will catch all the empty
        //equipment slots because the MLWB team decided to fuck with things.
        //It's ok, I'm not salty that *I'm* the one making a fix.
        public bool CheckMLWBLoaded()
        {
            foreach (Assembly assem in AppDomain.CurrentDomain.GetAssemblies())
            {
                if (assem.FullName.Contains("MyLittleWarband"))
                    return true;
            }
            return false;
        }

        //Serialization
        private void DeserializeObject(string filename)
        {
            Settings settings;
            using (Stream stream = (Stream)new FileStream(filename, FileMode.Open))
                settings = (Settings)new XmlSerializer(typeof(Settings)).Deserialize(stream);
            this.CurrentSettings = settings;
        }

        private void SerializeObject(string filename)
        {
            Console.WriteLine("Writing With XmlTextWriter");
            XmlSerializer xmlSerializer = new XmlSerializer(typeof(Settings));
            Settings settings = new Settings();
            XmlWriter xmlWriter = XmlWriter.Create((Stream)new FileStream(filename, FileMode.Create), new XmlWriterSettings()
            {
                Indent = true,
                IndentChars = "\t",
                OmitXmlDeclaration = true
            });
            xmlSerializer.Serialize(xmlWriter, (object)settings);
            xmlWriter.Close();
        }
    }
}