using System;
using System.Collections.Generic;
using System.Linq;
using Assets.HeroEditor4D.Common.Scripts.Collections;
using Assets.HeroEditor4D.Common.Scripts.Data;
using Assets.HeroEditor4D.Common.Scripts.EditorScripts;
using Assets.HeroEditor4D.Common.Scripts.Enums;
using UnityEngine;

namespace Assets.HeroEditor4D.Common.Scripts.CharacterScripts
{
    /// <summary>
    /// Character presentation in editor. Contains sprites, renderers, animation and so on.
    /// </summary>
    public partial class Character : MonoBehaviour
    {
        public SpriteCollection SpriteCollection;

        [Header("Body")]
        public List<Sprite> Ears;
        public Sprite Hair;
        public Sprite HairCropped;
        public Sprite Beard;
        public List<Sprite> Body;
        public Sprite Head;

        [Header("Expressions")]
        public string Expression = "Default";
        public List<Expression> Expressions;

        [Header("Equipment")]
        public Sprite Helmet;
        public Sprite PrimaryWeapon;
        public Sprite SecondaryWeapon;
        public Sprite Cape;
        public Sprite Back;
        public List<Sprite> Shield;
        public List<Sprite> Armor;
        public List<Sprite> CompositeWeapon;

        [Header("Accessories")]
        public Sprite Makeup;
        public Sprite Mask;
        public List<Sprite> Earrings;

        [Header("Body renderers")]
        public List<SpriteRenderer> BodyRenderers;
        public SpriteRenderer HeadRenderer;
        public List<SpriteRenderer> EarsRenderers;
        public SpriteRenderer HairRenderer;
        public SpriteRenderer BeardRenderer;
        public SpriteRenderer EyebrowsRenderer;
        public SpriteRenderer EyesRenderer;
        public SpriteRenderer MouthRenderer;

        [Header("Equipment renderers")]
        public SpriteRenderer HelmetRenderer;
        public SpriteRenderer PrimaryWeaponRenderer;
        public SpriteRenderer SecondaryWeaponRenderer;
        public List<SpriteRenderer> ArmorRenderers;
        public List<SpriteRenderer> VestRenderers;
        public List<SpriteRenderer> BracersRenderers;
        public List<SpriteRenderer> LeggingsRenderers;
        public SpriteRenderer CapeRenderer;
        public SpriteRenderer BackRenderer;
        public List<SpriteRenderer> ShieldRenderers;
        public List<SpriteRenderer> BowRenderers;
        public List<SpriteRenderer> FirearmsRenderers;

        [Header("Accessories renderers")]
        public SpriteRenderer MakeupRenderer;
        public SpriteRenderer MaskRenderer;
        public List<SpriteRenderer> EarringsRenderers;

        [Header("Materials")]
        public Material DefaultMaterial;
        public Material EyesPaintMaterial;
        public Material EquipmentPaintMaterial;

        [Header("Animation")]
        public WeaponType WeaponType;

        [Header("Meta")]
        public bool HideEars;
        public bool CropHair;

        [Header("Anchors")]
	    public Transform AnchorBody;
	    public Transform AnchorSword;
	    public Transform AnchorBow;
        public Transform AnchorFireMuzzle;

        [Header("Service")]
		public LayerManager LayerManager;

        [Header("Custom")]
        public List<Sprite> Underwear;
        public Color UnderwearColor;
        public bool ShowHelmet = true;

        /// <summary>
        /// Initializes character renderers with selected sprites.
        /// </summary>
        public void Initialize()
        {
            try // Disable try/catch for debugging.
            {
                TryInitialize();
            }
            catch (Exception e)
            {
                Debug.LogWarningFormat("Unable to initialize character {0}: {1}", name, e.Message);
            }
        }

		/// <summary>
		/// Initializes character renderers with selected sprites.
		/// </summary>
		private void TryInitialize()
        {
            var expressionNames = Expressions.Select(i => i.Name).ToList();

            if (!expressionNames.Contains("Default") || !expressionNames.Contains("Angry") || !expressionNames.Contains("Dead"))
            {
                throw new Exception("Character must have at least 3 basic expressions: Default, Angry and Dead.");
            }

            if (ShowHelmet)
            {
                HairRenderer.sprite = Hair == null ? null : CropHair ? HairCropped : Hair;
                EarsRenderers.ForEach(i => i.enabled = !HideEars);
				HelmetRenderer.sprite = Helmet;
			}
            else
            {
                HairRenderer.sprite = Hair;
                EarsRenderers.ForEach(i => i.enabled = true);
				HelmetRenderer.sprite = null;
			}

            MapSprites(EarsRenderers, Ears);
            SetExpression(Expression);
			if (BeardRenderer != null) BeardRenderer.sprite = Beard;
			MapSprites(BodyRenderers, Body);
            HeadRenderer.sprite = Head;
            MapSprites(ArmorRenderers, Armor);
			//CapeRenderer.sprite = Cape;
			//BackRenderer.sprite = Back;
	        PrimaryWeaponRenderer.sprite = PrimaryWeapon;
            PrimaryWeaponRenderer.enabled = true;
			SecondaryWeaponRenderer.sprite = SecondaryWeapon;
            SecondaryWeaponRenderer.enabled = WeaponType == WeaponType.Paired;
			MapSprites(BowRenderers, CompositeWeapon);
            BowRenderers.ForEach(i => i.enabled = WeaponType == WeaponType.Bow);
			MapSprites(ShieldRenderers, Shield);
            ShieldRenderers.ForEach(i => i.enabled = WeaponType == WeaponType.Melee1H);
			
			if (MakeupRenderer != null) MakeupRenderer.sprite = Makeup;
            if (MaskRenderer != null) MaskRenderer.sprite = Mask;
			MapSprites(EarringsRenderers, Earrings);

			if (WeaponType == WeaponType.Crossbow)
            {
                var quiver = BowRenderers.Single(i => i.name == "Quiver");

                quiver.enabled = true;
				MapSprites(new List<SpriteRenderer> { PrimaryWeaponRenderer, quiver }, CompositeWeapon);
			}

            ApplyMaterials();
		}

		private void MapSprites(List<SpriteRenderer> spriteRenderers, List<Sprite> sprites)
        {
            spriteRenderers.ForEach(i => MapSprite(i, sprites));
        }

        private void MapSprite(SpriteRenderer spriteRenderer, List<Sprite> sprites)
        {
            spriteRenderer.sprite = sprites == null ? null : spriteRenderer.GetComponent<SpriteMapping>().FindSprite(sprites);
        }

        private void ApplyMaterials()
        {
            var renderers = ArmorRenderers.ToList();

            renderers.Add(HairRenderer);
            renderers.Add(PrimaryWeaponRenderer);
            renderers.Add(SecondaryWeaponRenderer);
            renderers.ForEach(i => i.sharedMaterial = i.color == Color.white ? DefaultMaterial : EquipmentPaintMaterial);
        }

        public void CopyFrom(Character character)
        {
            if (character == null) throw new ArgumentNullException(nameof(character), "Can't copy from empty character!");

            Body = character.Body;
            Ears = character.Ears;
            Hair = character.Hair;
            Expression = character.Expression;
            Expressions = character.Expressions;
            Beard = character.Beard;

            Helmet = character.Helmet;
            Armor = character.Armor;
            PrimaryWeapon = character.PrimaryWeapon;
            SecondaryWeapon = character.SecondaryWeapon;
            Cape = character.Cape;
            Back = character.Back;
            Shield = character.Shield;
            CompositeWeapon = character.CompositeWeapon;

            Makeup = character.Makeup;
            Mask = character.Mask;
            Earrings = character.Earrings;

            foreach (var target in GetComponentsInChildren<SpriteRenderer>(true).Where(i => i.sprite != null))
            {
                foreach (var source in character.GetComponentsInChildren<SpriteRenderer>(true))
                {
                    if (target.name == source.name && target.transform.parent.name == source.transform.parent.name)
                    {
                        target.color = source.color;
                        break;
                    }
                }
            }

            WeaponType = character.WeaponType;
            HideEars = character.HideEars;
            CropHair = character.CropHair;
            Initialize();
            SetExpression("Default");
        }
    }
}