IGP-Fan-Reference/Model/Entity/EntityModel.cs

#nullable enable
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using Model.Entity.Data;
using Model.Entity.Parts;
using Model.Types;
using YamlDotNet.Serialization;

namespace Model.Entity;

public class EntityModel
{
    private static Dictionary<string, EntityModel> _database = null!;

    private static List<EntityModel> _entityModels = null!;

    private static List<EntityModel> _entityModelsOnlyHotkey = null!;

    private static Dictionary<string, List<EntityModel>>? _entityModelsByHotkey;


    public EntityModel(string data, string entity, bool isSpeculative = false)
    {
        DataType = data;
        EntityType = entity;
        IsSpeculative = isSpeculative;
    }

    public string DataType { get; set; }

    // TODO Serilization currently being used for build orders
    public string EntityType { get; set; }

    public List<IEntityPartInterface> EntityParts { get; set; } = new();

    public bool IsSpeculative { get; set; }

    public string Descriptive { get; set; } = DescriptiveType.None;

    public string GetName()
    {
        var entityInfo = EntityParts.FirstOrDefault(a => a.GetType() == typeof(EntityInfoModel)) as EntityInfoModel;
        return entityInfo == null ? string.Empty : entityInfo.Name;
    }

    public string GetFaction()
    {
        var entityInfo =
            EntityParts.FirstOrDefault(a => a.GetType() == typeof(EntityFactionModel)) as EntityFactionModel;
        return entityInfo == null ? string.Empty : EntityData.Get()[entityInfo.Faction].GetName();
    }

    public string GetImmortal()
    {
        var entityInfo =
            EntityParts.FirstOrDefault(a =>
                a.GetType() == typeof(EntityVanguardAddedModel)) as EntityVanguardAddedModel;
        return entityInfo == null ? string.Empty : EntityData.Get()[entityInfo.ImmortalId].GetName();
    }

    public string AsYaml()
    {
        var stringBuilder = new StringBuilder();
        var serializer = new Serializer();
        stringBuilder.AppendLine(serializer.Serialize(this));
        return stringBuilder.ToString();
    }

    public EntityModel Clone()
    {
        return (EntityModel)MemberwiseClone();
    }


    public void Copy(EntityModel entity)
    {
        DataType = entity.DataType;
        EntityType = entity.EntityType;
        EntityParts = entity.EntityParts.ToList();
    }


    public EntityModel AddPart(IEntityPartInterface unitPart)
    {
        unitPart.Parent = this;

        EntityParts.Add(unitPart);
        return this;
    }


    public static Dictionary<string, EntityModel> GetDictionary()
    {
        if (_database == null) _database = EntityData.Get();

        return _database;
    }


    public static EntityModel Get(string entity)
    {
        if (_database == null) _database = EntityData.Get();

        return _database[entity];
    }


    public static List<EntityModel> GetList()
    {
        if (_entityModels == null) _entityModels = EntityData.Get().Values.ToList();

        return _entityModels;
    }


    public static List<EntityModel> GetListOnlyHotkey()
    {
        if (_entityModelsOnlyHotkey == null)
        {
            _entityModelsOnlyHotkey = new List<EntityModel>();

            foreach (var entity in EntityData.Get().Values)
                if (entity.Hotkey() != null)
                    _entityModelsOnlyHotkey.Add(entity);
        }

        return _entityModelsOnlyHotkey;
    }


    public static Dictionary<string, List<EntityModel>> GetEntitiesByHotkey()
    {
        if (_entityModelsByHotkey == null)
        {
            _entityModelsByHotkey = new Dictionary<string, List<EntityModel>>();

            foreach (var entity in GetList())
            {
                var entityHotkey = entity.Hotkey();
                if (entityHotkey != null)
                {
                    if (!_entityModelsByHotkey.ContainsKey(entityHotkey.Hotkey))
                        _entityModelsByHotkey[entityHotkey.Hotkey] = new List<EntityModel>();

                    _entityModelsByHotkey[entityHotkey.Hotkey].Add(entity);
                }
            }
        }


        return _entityModelsByHotkey;
    }


    public static EntityModel? GetFrom(string hotkey, string hotkeyGroup, bool holdSpace, string faction,
        string immortal)
    {
        if (string.IsNullOrEmpty(hotkey)) return null;

        if (!GetEntitiesByHotkey().ContainsKey(hotkey)) return null;


        var foundList = from entity in GetEntitiesByHotkey()[hotkey]
            where entity.Hotkey()?.HotkeyGroup == hotkeyGroup
                  && entity.Hotkey()?.HoldSpace == holdSpace
                  && entity.Faction()?.Faction == faction
                  && (entity.VanguardAdded()?.ImmortalId == immortal || entity.VanguardAdded() == null)
                  && (entity.Replaceds().Count == 0 || (from replace in entity.Replaceds()
                      where replace.ImmortalId == immortal
                      select replace).ToList().Count == 0)
            select entity;

        if (!foundList.Any()) return null;

        var found = (foundList ?? Array.Empty<EntityModel>()).First();

        return found;
    }


    public EntityInfoModel Info()
    {
        return ((EntityInfoModel)EntityParts.Find(x => x.GetType() == typeof(EntityInfoModel))!)!;
    }


    public EntitySupplyModel? Supply()
    {
        return ((EntitySupplyModel)EntityParts.Find(x => x.GetType() == typeof(EntitySupplyModel))!)!;
    }


    public EntityTierModel Tier()
    {
        return ((EntityTierModel)EntityParts.Find(x => x.GetType() == typeof(EntityTierModel))!)!;
    }


    public EntityProductionModel? Production()
    {
        return ((EntityProductionModel)EntityParts.Find(x => x.GetType() == typeof(EntityProductionModel))!)!;
    }


    public EntityMovementModel Movement()
    {
        return ((EntityMovementModel)EntityParts.Find(x => x.GetType() == typeof(EntityMovementModel))!)!;
    }


    public EntityVitalityModel Vitality()
    {
        return ((EntityVitalityModel)EntityParts.Find(x => x.GetType() == typeof(EntityVitalityModel))!)!;
    }


    public List<EntityRequirementModel> Requirements()
    {
        return EntityParts.FindAll(x => x.GetType() == typeof(EntityRequirementModel))
            .Cast<EntityRequirementModel>().ToList();
    }


    public List<EntityWeaponModel> Weapons()
    {
        return EntityParts.FindAll(x => x.GetType() == typeof(EntityWeaponModel))
            .Cast<EntityWeaponModel>().ToList();
    }


    public List<EntityVanguardReplacedModel> Replaceds()
    {
        return EntityParts.FindAll(x => x.GetType() == typeof(EntityVanguardReplacedModel))
            .Cast<EntityVanguardReplacedModel>().ToList();
    }


    public EntityVanguardAddedModel? VanguardAdded()
    {
        return ((EntityVanguardAddedModel)EntityParts.Find(x => x.GetType() == typeof(EntityVanguardAddedModel))!)!;
    }


    public EntityHotkeyModel Hotkey()
    {
        return (EntityHotkeyModel)EntityParts.Find(x => x.GetType() == typeof(EntityHotkeyModel))!;
    }


    public EntityFactionModel Faction()
    {
        return (EntityFactionModel)EntityParts.Find(x => x.GetType() == typeof(EntityFactionModel))!;
    }


    public EntityHarvestModel Harvest()
    {
        return (EntityHarvestModel)EntityParts.Find(x => x.GetType() == typeof(EntityHarvestModel))!;
    }


    public List<EntityIdAbilityModel> IdAbilities()
    {
        return EntityParts.FindAll(x => x.GetType() == typeof(EntityIdAbilityModel))
            .Cast<EntityIdAbilityModel>().ToList();
    }


    public List<EntityIdArmyModel> IdArmies()
    {
        return EntityParts.FindAll(x => x.GetType() == typeof(EntityIdArmyModel))
            .Cast<EntityIdArmyModel>().ToList();
    }


    public List<EntityIdPassiveModel> IdPassives()
    {
        return EntityParts.FindAll(x => x.GetType() == typeof(EntityIdPassiveModel))
            .Cast<EntityIdPassiveModel>().ToList();
    }


    public List<EntityIdUpgradeModel> IdUpgrades()
    {
        return EntityParts.FindAll(x => x.GetType() == typeof(EntityIdUpgradeModel))
            .Cast<EntityIdUpgradeModel>().ToList();
    }


    public List<EntityIdVanguardModel> IdVanguards()
    {
        return EntityParts.FindAll(x => x.GetType() == typeof(EntityIdVanguardModel))
            .Cast<EntityIdVanguardModel>().ToList();
    }


    public List<EntityIdPyreSpellModel> IdPyreSpells()
    {
        return EntityParts.FindAll(x => x.GetType() == typeof(EntityIdPyreSpellModel))
            .Cast<EntityIdPyreSpellModel>().ToList();
    }

    public List<EntityMechanicModel> Mechanics()
    {
        return EntityParts.FindAll(x => x.GetType() == typeof(EntityMechanicModel))
            .Cast<EntityMechanicModel>().ToList();
    }

    public List<EntityPassiveModel> Passives()
    {
        return EntityParts.FindAll(x => x.GetType() == typeof(EntityPassiveModel))
            .Cast<EntityPassiveModel>().ToList();
    }

    public List<EntityStrategyModel> Strategies()
    {
        return EntityParts.FindAll(x => x.GetType() == typeof(EntityStrategyModel))
            .Cast<EntityStrategyModel>().ToList();
    }
}