Stardew_Valley_Mods/GeneralMods/BuildEndurance/BuildEndurance.cs

using System;
using System.IO;
using System.Linq;
using Omegasis.BuildEndurance.Framework;
using StardewModdingAPI;
using StardewModdingAPI.Events;
using StardewValley;

namespace Omegasis.BuildEndurance
{
    /// <summary>The mod entry point.</summary>
    public class BuildEndurance : Mod
    {
        /*********
        ** Properties
        *********/
        /// <summary>The relative path for the current player's data file.</summary>
        private string DataFilePath => Path.Combine("data", $"{Constants.SaveFolderName}.json");

        /// <summary>The absolute path for the current player's legacy data file.</summary>
        private string LegacyDataFilePath => Path.Combine(this.Helper.DirectoryPath, "PlayerData", $"BuildEndurance_data_{Game1.player.Name}.txt");

        /// <summary>The mod settings.</summary>
        private ModConfig Config;

        /// <summary>The data for the current player.</summary>
        private PlayerData PlayerData;

        /// <summary>Whether the player has been exhausted today.</summary>
        private bool WasExhausted;

        /// <summary>Whether the player has collapsed today.</summary>
        private bool WasCollapsed;

        /// <summary>Whether the player recently gained XP for tool use.</summary>
        private bool HasRecentToolExp;

        /// <summary>Whether the player was eating last time we checked.</summary>
        private bool WasEating;


        public IModHelper ModHelper;
        public IMonitor ModMonitor;

        /*********
        ** Public methods
        *********/
        /// <summary>The mod entry point, called after the mod is first loaded.</summary>
        /// <param name="helper">Provides simplified APIs for writing mods.</param>
        public override void Entry(IModHelper helper)
        {
            this.Config = helper.ReadConfig<ModConfig>();

            GameEvents.UpdateTick += this.GameEvents_UpdateTick;
            GameEvents.OneSecondTick += this.GameEvents_OneSecondTick;
            SaveEvents.AfterLoad += this.SaveEvents_AfterLoad;
            SaveEvents.BeforeSave += this.SaveEvents_BeforeSave;

            this.ModHelper = this.Helper;
            this.ModMonitor = this.Monitor;
        }


        /*********
        ** Private methods
        *********/
        /// <summary>The method invoked once per second during a game update.</summary>
        /// <param name="sender">The event sender.</param>
        /// <param name="e">The event data.</param>
        private void GameEvents_OneSecondTick(object sender, EventArgs e)
        {
            // nerf how quickly tool xp is gained (I hope)
            if (this.HasRecentToolExp)
                this.HasRecentToolExp = false;
        }

        /// <summary>The method invoked when the game updates (roughly 60 times per second).</summary>
        /// <param name="sender">The event sender.</param>
        /// <param name="e">The event data.</param>
        private void GameEvents_UpdateTick(object sender, EventArgs e)
        {
            if (!Context.IsWorldReady)
                return;

            // give XP when player finishes eating
            if (Game1.player.isEating)
                this.WasEating = true;
            else if (this.WasEating)
            {
                this.PlayerData.CurrentExp += this.Config.ExpForEating;
                this.WasEating = false;
            }

            // give XP when player uses tool
            if (!this.HasRecentToolExp && Game1.player.UsingTool)
            {
                this.PlayerData.CurrentExp += this.Config.ExpForToolUse;
                this.HasRecentToolExp = true;
            }

            // give XP when exhausted
            if (!this.WasExhausted && Game1.player.exhausted)
            {
                this.PlayerData.CurrentExp += this.Config.ExpForExhaustion;
                this.WasExhausted = true;
                //this.Monitor.Log("The player is exhausted");
            }

            // give XP when player stays up too late or collapses
            if (!this.WasCollapsed && Game1.farmerShouldPassOut)
            {
                this.PlayerData.CurrentExp += this.Config.ExpForCollapsing;
                this.WasCollapsed = true;
                //this.Monitor.Log("The player has collapsed!");
            }
        }

        /// <summary>The method invoked after the player loads a save.</summary>
        /// <param name="sender">The event sender.</param>
        /// <param name="e">The event data.</param>
        private void SaveEvents_AfterLoad(object sender, EventArgs e)
        {
            // reset state
            this.WasExhausted = false;
            this.WasCollapsed = false;
            this.HasRecentToolExp = false;
            this.WasEating = false;

            // load player data
            this.MigrateLegacyData();
            this.PlayerData = this.Helper.ReadJsonFile<PlayerData>(this.DataFilePath) ?? new PlayerData();
            if (this.PlayerData.OriginalMaxStamina == 0)
                this.PlayerData.OriginalMaxStamina = Game1.player.MaxStamina;

            // reset if needed
            if (this.PlayerData.ClearModEffects)
            {
                Game1.player.MaxStamina = this.PlayerData.OriginalMaxStamina;
                this.PlayerData.ExpToNextLevel = this.Config.ExpToNextLevel;
                this.PlayerData.CurrentExp = this.Config.CurrentExp;
                this.PlayerData.CurrentLevelStaminaBonus = 0;
                this.PlayerData.OriginalMaxStamina = Game1.player.MaxStamina;
                this.PlayerData.BaseStaminaBonus = 0;
                this.PlayerData.CurrentLevel = 0;
                this.PlayerData.ClearModEffects = false;
            }

            // else apply stamina
            else
            {
                Game1.player.MaxStamina = this.PlayerData.NightlyStamina <= 0
                    ? this.PlayerData.BaseStaminaBonus + this.PlayerData.CurrentLevelStaminaBonus + this.PlayerData.OriginalMaxStamina
                    : this.PlayerData.NightlyStamina;
            }
        }

        /// <summary>The method invoked just before the game is saved.</summary>
        /// <param name="sender">The event sender.</param>
        /// <param name="e">The event data.</param>
        private void SaveEvents_BeforeSave(object sender, EventArgs e)
        {
            // reset data
            this.WasExhausted = false;
            this.WasCollapsed = false;

            // update player data
            this.PlayerData.CurrentExp += this.Config.ExpForSleeping;
            if (this.PlayerData.OriginalMaxStamina == 0)
                this.PlayerData.OriginalMaxStamina = Game1.player.MaxStamina; //grab the initial stamina value

            if (this.PlayerData.CurrentLevel < this.Config.MaxLevel)
            {
                while (this.PlayerData.CurrentExp >= this.PlayerData.ExpToNextLevel)
                {
                    this.PlayerData.CurrentLevel += 1;
                    this.PlayerData.CurrentExp = this.PlayerData.CurrentExp - this.PlayerData.ExpToNextLevel;
                    this.PlayerData.ExpToNextLevel = (this.Config.ExpCurve * this.PlayerData.ExpToNextLevel);
                    Game1.player.MaxStamina += this.Config.StaminaIncreasePerLevel;
                    this.PlayerData.CurrentLevelStaminaBonus += this.Config.StaminaIncreasePerLevel;
                }
            }
            this.PlayerData.ClearModEffects = false;
            this.PlayerData.NightlyStamina = Game1.player.maxStamina;

            // save data
            this.Helper.WriteJsonFile(this.DataFilePath, this.PlayerData);
        }

        /// <summary>Migrate the legacy settings for the current player.</summary>
        private void MigrateLegacyData()
        {
            // skip if no legacy data or new data already exists
            if (!File.Exists(this.LegacyDataFilePath) || File.Exists(this.DataFilePath))
                return;

            // migrate to new file
            try
            {
                string[] text = File.ReadAllLines(this.LegacyDataFilePath);
                this.Helper.WriteJsonFile(this.DataFilePath, new PlayerData
                {
                    CurrentLevel = Convert.ToInt32(text[3]),
                    CurrentExp = Convert.ToDouble(text[5]),
                    ExpToNextLevel = Convert.ToDouble(text[7]),
                    BaseStaminaBonus = Convert.ToInt32(text[9]),
                    CurrentLevelStaminaBonus = Convert.ToInt32(text[11]),
                    ClearModEffects = Convert.ToBoolean(text[14]),
                    OriginalMaxStamina = Convert.ToInt32(text[16]),
                    NightlyStamina = Convert.ToInt32(text[18])
                });

                FileInfo file = new FileInfo(this.LegacyDataFilePath);
                file.Delete();
                if (!file.Directory.EnumerateFiles().Any())
                    file.Directory.Delete();
            }
            catch (Exception ex)
            {
                this.Monitor.Log($"Error migrating data from the legacy 'PlayerData' folder for the current player. Technical details:\n {ex}", LogLevel.Error);
            }
        }
    }
}