using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using RimWorld;
using Verse;
using AlienRace;
using UnityEngine;

namespace Rimworld_Animations_Patch
{
    public class CompPawnSexData : ThingComp
    {
        public HandAnimationDef handAnimationDef = null;
        public Graphic handGraphic = null;

        public List<BodyPartRecord> hands = new List<BodyPartRecord>();
        public float sizeOfPenis = 0f;
        public float sizeOfBreasts = 0f;

        public Dictionary<AlienPartGenerator.BodyAddon, BodyAddonData> bodyAddonData = new Dictionary<AlienPartGenerator.BodyAddon, BodyAddonData>();
        public Dictionary<AlienPartGenerator.BodyAddon, BodyAddonData> bodyAddonDataPortraits = new Dictionary<AlienPartGenerator.BodyAddon, BodyAddonData>();

        private Pawn pawn;

        public BodyAddonData GetBodyAddonData(AlienPartGenerator.BodyAddon bodyAddon, bool isPortrait)
        {
            if (pawn == null)
            { pawn = parent as Pawn; }

            if (pawn == null || (pawn.Map != Find.CurrentMap && pawn.holdingOwner == null) || bodyAddon == null) return null;
            
            if (isPortrait)
            {
                if (bodyAddonDataPortraits.TryGetValue(bodyAddon, out BodyAddonData bodyAddonDatum) == false)
                {
                    bodyAddonDatum = new BodyAddonData(pawn, bodyAddon, true);
                    bodyAddonDataPortraits.Add(bodyAddon, bodyAddonDatum);
                }

                return bodyAddonDatum;
            }

            else
            {
                if (bodyAddonData.TryGetValue(bodyAddon, out BodyAddonData bodyAddonDatum) == false)
                {
                    bodyAddonDatum = new BodyAddonData(pawn, bodyAddon);
                    bodyAddonData.Add(bodyAddon, bodyAddonDatum);
                }

                return bodyAddonDatum;
            }
        }

        public void UpdateBodyAddonVisibility()
        {
            foreach (KeyValuePair<AlienPartGenerator.BodyAddon, BodyAddonData> kvp in bodyAddonData)
            { kvp.Value?.UpdateVisibility(); }

            foreach (KeyValuePair<AlienPartGenerator.BodyAddon, BodyAddonData> kvp in bodyAddonDataPortraits)
            { kvp.Value?.UpdateVisibility(); }
        }

        public void UpdateBodyPartCountAndSize()
        {
            hands = pawn?.health?.hediffSet?.GetNotMissingParts()?.Where(x => x.def.tags.Contains(BodyPartTagDefOf.ManipulationLimbCore))?.ToList();

            Hediff hediffPenis = pawn?.health?.hediffSet?.hediffs?.FirstOrDefault(x => x.def.defName.Contains("penis") == true || x.def.defName.Contains("Penis"));
            sizeOfPenis = hediffPenis != null ? hediffPenis.Severity : 0f;

            Hediff hediffBreasts = pawn?.health?.hediffSet?.hediffs?.FirstOrDefault(x => x.def.defName.Contains("breasts") == true || x.def.defName.Contains("Breasts") == true);
            sizeOfBreasts = hediffBreasts != null ? hediffBreasts.Severity : 0f;
        }

        public int GetNumberOfHands()
        {
            if (hands.Any() == false) return 0;

            return hands.Count;
        }
    }
}