using MareSynchronos.API; using MareSynchronos.FileCacheDB; using MareSynchronos.Models; using MareSynchronos.Utils; using System; using System.Collections.Generic; using System.Linq; namespace MareSynchronos.Managers { public delegate void TransientResourceLoadedEvent(IntPtr drawObject); public class TransientResourceManager : IDisposable { private readonly IpcManager manager; private readonly DalamudUtil dalamudUtil; public event TransientResourceLoadedEvent? TransientResourceLoaded; private Dictionary> TransientResources { get; } = new(); private Dictionary> SemiTransientResources { get; } = new(); public TransientResourceManager(IpcManager manager, DalamudUtil dalamudUtil) { manager.PenumbraResourceLoadEvent += Manager_PenumbraResourceLoadEvent; this.manager = manager; this.dalamudUtil = dalamudUtil; dalamudUtil.FrameworkUpdate += DalamudUtil_FrameworkUpdate; dalamudUtil.ClassJobChanged += DalamudUtil_ClassJobChanged; } private void DalamudUtil_ClassJobChanged() { if (SemiTransientResources.ContainsKey(ObjectKind.Pet)) { SemiTransientResources[ObjectKind.Pet].Clear(); } } private void DalamudUtil_FrameworkUpdate() { foreach (var item in TransientResources.ToList()) { if (!dalamudUtil.IsGameObjectPresent(item.Key)) { Logger.Debug("Object not present anymore: " + item.Key); TransientResources.Remove(item.Key); } } } public void CleanSemiTransientResources(ObjectKind objectKind) { if (SemiTransientResources.ContainsKey(objectKind)) { SemiTransientResources[objectKind].Clear(); } } public List GetTransientResources(IntPtr gameObject) { if (TransientResources.TryGetValue(gameObject, out var result)) { return result.ToList(); } return new List(); } public List GetSemiTransientResources(ObjectKind objectKind) { if (SemiTransientResources.TryGetValue(objectKind, out var result)) { return result.ToList(); } return new List(); } private void Manager_PenumbraResourceLoadEvent(IntPtr gameObject, string gamePath, string filePath) { if (!TransientResources.ContainsKey(gameObject)) { TransientResources[gameObject] = new(); } if (filePath.StartsWith("|")) { filePath = filePath.Split("|")[2]; } var newPath = filePath.ToLowerInvariant().Replace("\\", "/"); if (filePath != gamePath && !TransientResources[gameObject].Contains(newPath) && !SemiTransientResources.Any(r => r.Value.Any(f => f.ResolvedPath.ToLowerInvariant() == newPath.ToLowerInvariant()))) { TransientResources[gameObject].Add(newPath); Logger.Debug($"Adding {filePath.ToLowerInvariant().Replace("\\", "/")} for {gameObject}"); TransientResourceLoaded?.Invoke(gameObject); } } public void RemoveTransientResource(IntPtr gameObject, FileReplacement fileReplacement) { if (TransientResources.ContainsKey(gameObject)) { TransientResources[gameObject].RemoveWhere(f => fileReplacement.ResolvedPath == f); } } public void PersistTransientResources(IntPtr gameObject, ObjectKind objectKind, Func createFileReplacement) { if (!SemiTransientResources.ContainsKey(objectKind)) { SemiTransientResources[objectKind] = new HashSet(); } if (!TransientResources.TryGetValue(gameObject, out var resources)) { return; } var transientResources = resources.ToList(); Logger.Debug("Persisting " + transientResources.Count + " transient resources"); foreach (var item in transientResources) { if (!SemiTransientResources[objectKind].Any(f => f.ResolvedPath.ToLowerInvariant() == item.ToLowerInvariant())) { SemiTransientResources[objectKind].Add(createFileReplacement(item.ToLowerInvariant(), false)); } } TransientResources[gameObject].Clear(); } public void Dispose() { dalamudUtil.FrameworkUpdate -= DalamudUtil_FrameworkUpdate; manager.PenumbraResourceLoadEvent -= Manager_PenumbraResourceLoadEvent; dalamudUtil.ClassJobChanged -= DalamudUtil_ClassJobChanged; TransientResources.Clear(); } internal void AddSemiTransientResource(ObjectKind objectKind, FileReplacement item) { if (!SemiTransientResources.ContainsKey(objectKind)) { SemiTransientResources[objectKind] = new HashSet(); } if (!SemiTransientResources[objectKind].Any(f => f.ResolvedPath.ToLowerInvariant() == item.ResolvedPath.ToLowerInvariant())) { SemiTransientResources[objectKind].Add(item); } } } }