COM3D2.MeidoPhotoStudio/COM3D2.MeidoPhotoStudio.Plugin/MeidoPhotoStudio/MenuFileCache.cs

using System;
using System.IO;
using System.Collections.Generic;

/*
    All of this is pretty much stolen from COM3D2.CacheEditMenu. Thanks Mr. Horsington.
    https://git.coder.horse/ghorsington/COM3D2.CacheEditMenu
*/
namespace COM3D2.MeidoPhotoStudio.Plugin
{
    using static MenuFileUtility;
    internal static class BinaryExtensions
    {
        public static string ReadNullableString(this BinaryReader binaryReader)
        {
            return binaryReader.ReadBoolean() ? binaryReader.ReadString() : null;
        }

        public static void WriteNullableString(this BinaryWriter binaryWriter, string str)
        {
            binaryWriter.Write(str != null);
            if (str != null) binaryWriter.Write(str);
        }
    }

    internal class MenuFileCache
    {
        private const int cacheVersion = 765;
        public static readonly string cachePath = Path.Combine(Constants.configPath, "cache.dat");
        private Dictionary<string, ModItem> modItems;
        private bool rebuild = false;
        public ModItem this[string menu]
        {
            get => modItems[menu];
            set
            {
                if (!modItems.ContainsKey(menu))
                {
                    rebuild = true;
                    modItems[menu] = value;
                }
            }
        }

        public MenuFileCache()
        {
            modItems = new Dictionary<string, ModItem>();
            if (File.Exists(cachePath)) Deserialize();
        }

        public bool Has(string menuFileName)
        {
            return modItems.ContainsKey(menuFileName);
        }

        private void Deserialize()
        {
            using (BinaryReader binaryReader = new BinaryReader(File.OpenRead(cachePath)))
            {
                if (binaryReader.ReadInt32() != cacheVersion)
                {
                    UnityEngine.Debug.Log($"Cache version out of date. Rebuilding");
                    return;
                }
                while (binaryReader.BaseStream.Position < binaryReader.BaseStream.Length)
                {
                    ModItem item = ModItem.Deserialize(binaryReader);
                    modItems[item.MenuFile] = item;
                }
            }
        }

        public void Serialize()
        {
            if (!rebuild) return;
            using (BinaryWriter binaryWriter = new BinaryWriter(File.OpenWrite(cachePath)))
            {
                binaryWriter.Write(cacheVersion);
                foreach (ModItem item in modItems.Values)
                {
                    item.Serialize(binaryWriter);
                }
            }
        }
    }
}