.

GlamourerOld/Designs/DesignManager.cs

@@ -0,0 +1,371 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using Dalamud.Plugin;
+using Dalamud.Utility;
+using Glamourer.Customization;
+using Glamourer.Services;
+using Newtonsoft.Json;
+using Newtonsoft.Json.Linq;
+using OtterGui;
+using Penumbra.GameData.Enums;
+using Penumbra.GameData.Structs;
+
+namespace Glamourer.Designs;
+
+public class DesignManager
+{
+    public const    string DesignFolderName = "designs";
+    public readonly string DesignFolder;
+
+    private readonly ItemManager  _items;
+    private readonly SaveService  _saveService;
+    private readonly List<Design> _designs = new();
+
+    public enum DesignChangeType
+    {
+        Created,
+        Deleted,
+        ReloadedAll,
+        Renamed,
+        ChangedDescription,
+        AddedTag,
+        RemovedTag,
+        ChangedTag,
+        Customize,
+        Equip,
+        Weapon,
+        Stain,
+        ApplyCustomize,
+        ApplyEquip,
+        Other,
+    }
+
+    public delegate void DesignChangeDelegate(DesignChangeType type, Design design, object? changeData = null);
+
+    public event DesignChangeDelegate? DesignChange;
+
+    public IReadOnlyList<Design> Designs
+        => _designs;
+
+    public DesignManager(DalamudPluginInterface pi, SaveService saveService, ItemManager items)
+    {
+        _saveService = saveService;
+        _items       = items;
+        DesignFolder = SetDesignFolder(pi);
+        LoadDesigns();
+        MigrateOldDesigns(pi, Path.Combine(new DirectoryInfo(DesignFolder).Parent!.FullName, "Designs.json"));
+    }
+
+    private static string SetDesignFolder(DalamudPluginInterface pi)
+    {
+        var ret = Path.Combine(pi.GetPluginConfigDirectory(), DesignFolderName);
+        if (Directory.Exists(ret))
+            return ret;
+
+        try
+        {
+            Directory.CreateDirectory(ret);
+        }
+        catch (Exception ex)
+        {
+            Glamourer.Log.Error($"Could not create design folder directory at {ret}:\n{ex}");
+        }
+
+        return ret;
+    }
+
+    public void LoadDesigns()
+    {
+        _designs.Clear();
+        List<(Design, string)> invalidNames = new();
+        var                    skipped      = 0;
+        foreach (var file in new DirectoryInfo(DesignFolder).EnumerateFiles("*.json", SearchOption.TopDirectoryOnly))
+        {
+            try
+            {
+                var text   = File.ReadAllText(file.FullName);
+                var data   = JObject.Parse(text);
+                var design = Design.LoadDesign(_items, data, out var changes);
+                if (design.Identifier.ToString() != Path.GetFileNameWithoutExtension(file.Name))
+                    invalidNames.Add((design, file.FullName));
+                if (_designs.Any(f => f.Identifier == design.Identifier))
+                    throw new Exception($"Identifier {design.Identifier} was not unique.");
+
+                // TODO something when changed?
+                design.Index = _designs.Count;
+                _designs.Add(design);
+            }
+            catch (Exception ex)
+            {
+                Glamourer.Log.Error($"Could not load design, skipped:\n{ex}");
+                ++skipped;
+            }
+        }
+
+        var failed = 0;
+        foreach (var (design, name) in invalidNames)
+        {
+            try
+            {
+                var correctName = _saveService.FileNames.DesignFile(design);
+                File.Move(name, correctName, false);
+                Glamourer.Log.Information($"Moved invalid design file from {Path.GetFileName(name)} to {Path.GetFileName(correctName)}.");
+            }
+            catch (Exception ex)
+            {
+                ++failed;
+                Glamourer.Log.Error($"Failed to move invalid design file from {Path.GetFileName(name)}:\n{ex}");
+            }
+        }
+
+        if (invalidNames.Count > 0)
+            Glamourer.Log.Information(
+                $"Moved {invalidNames.Count - failed} designs to correct names.{(failed > 0 ? $" Failed to move {failed} designs to correct names." : string.Empty)}");
+
+        Glamourer.Log.Information(
+            $"Loaded {_designs.Count} designs.{(skipped > 0 ? $" Skipped loading {skipped} designs due to errors." : string.Empty)}");
+        DesignChange?.Invoke(DesignChangeType.ReloadedAll, null!);
+    }
+
+    public Design Create(string name)
+    {
+        var design = new Design(_items)
+        {
+            CreationDate = DateTimeOffset.UtcNow,
+            Identifier   = CreateNewGuid(),
+            Index        = _designs.Count,
+            Name         = name,
+        };
+        _designs.Add(design);
+        Glamourer.Log.Debug($"Added new design {design.Identifier}.");
+        _saveService.ImmediateSave(design);
+        DesignChange?.Invoke(DesignChangeType.Created, design);
+        return design;
+    }
+
+    public void Delete(Design design)
+    {
+        _designs.RemoveAt(design.Index);
+        foreach (var d in _designs.Skip(design.Index + 1))
+            --d.Index;
+        _saveService.ImmediateDelete(design);
+    }
+
+    public void Rename(Design design, string newName)
+    {
+        var oldName = design.Name.Text;
+        _saveService.ImmediateDelete(design);
+        design.Name = newName;
+        Glamourer.Log.Debug($"Renamed design {design.Identifier}.");
+        _saveService.ImmediateSave(design);
+        DesignChange?.Invoke(DesignChangeType.Renamed, design, oldName);
+    }
+
+    public void ChangeDescription(Design design, string description)
+    {
+        design.Description = description;
+        Glamourer.Log.Debug($"Changed description of design {design.Identifier}.");
+        _saveService.QueueSave(design);
+        DesignChange?.Invoke(DesignChangeType.ChangedDescription, design);
+    }
+
+    public void AddTag(Design design, string tag)
+    {
+        if (design.Tags.Contains(tag))
+            return;
+
+        design.Tags = design.Tags.Append(tag).OrderBy(t => t).ToArray();
+        var idx = design.Tags.IndexOf(tag);
+        Glamourer.Log.Debug($"Added tag {tag} at {idx} to design {design.Identifier}.");
+        _saveService.QueueSave(design);
+        DesignChange?.Invoke(DesignChangeType.AddedTag, design);
+    }
+
+    public void RemoveTag(Design design, string tag)
+    {
+        var idx = design.Tags.IndexOf(tag);
+        if (idx >= 0)
+            RemoveTag(design, idx);
+    }
+
+    public void RemoveTag(Design design, int tagIdx)
+    {
+        var oldTag = design.Tags[tagIdx];
+        design.Tags = design.Tags.Take(tagIdx).Concat(design.Tags.Skip(tagIdx + 1)).ToArray();
+        Glamourer.Log.Debug($"Removed tag {oldTag} at {tagIdx} from design {design.Identifier}.");
+        _saveService.QueueSave(design);
+        DesignChange?.Invoke(DesignChangeType.RemovedTag, design);
+    }
+
+
+    public void RenameTag(Design design, int tagIdx, string newTag)
+    {
+        var oldTag = design.Tags[tagIdx];
+        if (oldTag == newTag)
+            return;
+
+        design.Tags[tagIdx] = newTag;
+        Array.Sort(design.Tags);
+        Glamourer.Log.Debug($"Renamed tag {oldTag} at {tagIdx} to {newTag} in design {design.Identifier} and reordered tags.");
+        _saveService.QueueSave(design);
+        DesignChange?.Invoke(DesignChangeType.ChangedTag, design);
+    }
+
+    public void ChangeCustomize(Design design, CustomizeIndex idx, CustomizeValue value)
+    {
+        var old = design.GetCustomize(idx);
+        if (!design.SetCustomize(idx, value))
+            return;
+
+        Glamourer.Log.Debug($"Changed customize {idx} in design {design.Identifier} from {old.Value} to {value.Value}");
+        _saveService.QueueSave(design);
+        DesignChange?.Invoke(DesignChangeType.Customize, design, idx);
+    }
+
+    public void ChangeApplyCustomize(Design design, CustomizeIndex idx, bool value)
+    {
+        if (!design.SetApplyCustomize(idx, value))
+            return;
+
+        Glamourer.Log.Debug($"Set applying of customization {idx} to {value}.");
+        _saveService.QueueSave(design);
+        DesignChange?.Invoke(DesignChangeType.ApplyCustomize, design, idx);
+    }
+
+    public void ChangeEquip(Design design, EquipSlot slot, uint itemId, Lumina.Excel.GeneratedSheets.Item? item = null)
+    {
+        var old = design.Armor(slot);
+        if (!design.SetArmor(_items, slot, itemId, item))
+            return;
+
+        var n = design.Armor(slot);
+        Glamourer.Log.Debug(
+            $"Set {slot} equipment piece in design {design.Identifier} from {old.Name} ({old.ItemId}) to {n.Name} ({n.ItemId}).");
+        _saveService.QueueSave(design);
+        DesignChange?.Invoke(DesignChangeType.Equip, design, slot);
+    }
+
+    public void ChangeWeapon(Design design, uint itemId, EquipSlot offhand, Lumina.Excel.GeneratedSheets.Item? item = null)
+    {
+        var (old, change, n) = offhand == EquipSlot.OffHand
+            ? (design.WeaponOff, design.SetOffhand(_items, itemId, item), design.WeaponOff)
+            : (design.WeaponMain, design.SetMainhand(_items, itemId, item), design.WeaponMain);
+        if (!change)
+            return;
+
+        Glamourer.Log.Debug(
+            $"Set {offhand} weapon in design {design.Identifier} from {old.Name} ({old.ItemId}) to {n.Name} ({n.ItemId}).");
+        _saveService.QueueSave(design);
+        DesignChange?.Invoke(DesignChangeType.Weapon, design, offhand);
+    }
+
+    public void ChangeApplyEquip(Design design, EquipSlot slot, bool value)
+    {
+        if (!design.SetApplyEquip(slot, value))
+            return;
+
+        Glamourer.Log.Debug($"Set applying of {slot} equipment piece to {value}.");
+        _saveService.QueueSave(design);
+        DesignChange?.Invoke(DesignChangeType.ApplyEquip, design, slot);
+    }
+
+    public void ChangeStain(Design design, EquipSlot slot, StainId stain)
+    {
+        if (!design.SetStain(slot, stain))
+            return;
+
+        Glamourer.Log.Debug($"Set stain of {slot} equipment piece to {stain.Value}.");
+        _saveService.QueueSave(design);
+        DesignChange?.Invoke(DesignChangeType.Stain, design, slot);
+    }
+
+    public void ChangeApplyStain(Design design, EquipSlot slot, bool value)
+    {
+        if (!design.SetApplyStain(slot, value))
+            return;
+
+        Glamourer.Log.Debug($"Set applying of stain of {slot} equipment piece to {value}.");
+        _saveService.QueueSave(design);
+        DesignChange?.Invoke(DesignChangeType.Stain, design, slot);
+    }
+
+    private Guid CreateNewGuid()
+    {
+        while (true)
+        {
+            var guid = Guid.NewGuid();
+            if (_designs.All(d => d.Identifier != guid))
+                return guid;
+        }
+    }
+
+    private bool Add(Design design, string? message)
+    {
+        if (_designs.Any(d => d == design || d.Identifier == design.Identifier))
+            return false;
+
+        design.Index = _designs.Count;
+        _designs.Add(design);
+        if (!message.IsNullOrEmpty())
+            Glamourer.Log.Debug(message);
+        _saveService.ImmediateSave(design);
+        DesignChange?.Invoke(DesignChangeType.Created, design);
+        return true;
+    }
+
+    private void MigrateOldDesigns(DalamudPluginInterface pi, string filePath)
+    {
+        if (!File.Exists(filePath))
+            return;
+
+        var errors    = 0;
+        var successes = 0;
+        try
+        {
+            var text                    = File.ReadAllText(filePath);
+            var dict                    = JsonConvert.DeserializeObject<Dictionary<string, string>>(text) ?? new Dictionary<string, string>();
+            var migratedFileSystemPaths = new Dictionary<string, string>(dict.Count);
+            foreach (var (name, base64) in dict)
+            {
+                try
+                {
+                    var actualName = Path.GetFileName(name);
+                    var design = new Design(_items)
+                    {
+                        CreationDate = DateTimeOffset.UtcNow,
+                        Identifier   = CreateNewGuid(),
+                        Name         = actualName,
+                    };
+                    design.MigrateBase64(_items, base64);
+                    Add(design, $"Migrated old design to {design.Identifier}.");
+                    migratedFileSystemPaths.Add(design.Identifier.ToString(), name);
+                    ++successes;
+                }
+                catch (Exception ex)
+                {
+                    Glamourer.Log.Error($"Could not migrate design {name}:\n{ex}");
+                    ++errors;
+                }
+            }
+
+            DesignFileSystem.MigrateOldPaths(pi, migratedFileSystemPaths);
+            Glamourer.Log.Information($"Successfully migrated {successes} old designs. Failed to migrate {errors} designs.");
+        }
+        catch (Exception e)
+        {
+            Glamourer.Log.Error($"Could not migrate old design file {filePath}:\n{e}");
+        }
+
+        try
+        {
+            File.Move(filePath, Path.ChangeExtension(filePath, ".json.bak"));
+            Glamourer.Log.Information($"Moved migrated design file {filePath} to backup file.");
+        }
+        catch (Exception ex)
+        {
+            Glamourer.Log.Error($"Could not move migrated design file {filePath} to backup file:\n{ex}");
+        }
+    }
+}