using System;
using System.Linq;

using ACE.Entity;
using ACE.Entity.Enum.Properties;

namespace ACE.Database.Models.Shard
{
    public static class BiotaExtensions
    {
        // =====================================
        // Get
        // Bool, DID, Float, IID, Int, Int64, String, Position
        // =====================================

        public static bool? GetProperty(this Biota biota, PropertyBool property)
        {
            return biota.BiotaPropertiesBool.FirstOrDefault(x => x.Type == (uint)property)?.Value;
        }

        public static uint? GetProperty(this Biota biota, PropertyDataId property)
        {
            return biota.BiotaPropertiesDID.FirstOrDefault(x => x.Type == (uint)property)?.Value;
        }

        public static double? GetProperty(this Biota biota, PropertyFloat property)
        {
            return biota.BiotaPropertiesFloat.FirstOrDefault(x => x.Type == (ushort)property)?.Value;
        }

        public static uint? GetProperty(this Biota biota, PropertyInstanceId property)
        {
            return biota.BiotaPropertiesIID.FirstOrDefault(x => x.Type == (uint)property)?.Value;
        }

        public static int? GetProperty(this Biota biota, PropertyInt property)
        {
            return biota.BiotaPropertiesInt.FirstOrDefault(x => x.Type == (uint)property)?.Value;
        }

        public static long? GetProperty(this Biota biota, PropertyInt64 property)
        {
            return biota.BiotaPropertiesInt64.FirstOrDefault(x => x.Type == (uint)property)?.Value;
        }

        public static string GetProperty(this Biota biota, PropertyString property)
        {
            return biota.BiotaPropertiesString.FirstOrDefault(x => x.Type == (uint)property)?.Value;
        }

        public static BiotaPropertiesPosition GetProperty(this Biota weenie, PositionType positionType)
        {
            return weenie.BiotaPropertiesPosition.FirstOrDefault(x => x.PositionType == (uint)positionType);
        }

        public static Position GetPosition(this Biota biota, PositionType positionType)
        {
            var result = biota.BiotaPropertiesPosition.FirstOrDefault(x => x.PositionType == (uint)positionType);

            if (result == null)
                return null;

            return new Position(result.ObjCellId, result.OriginX, result.OriginY, result.OriginZ, result.AnglesX, result.AnglesY, result.AnglesZ, result.AnglesW);
        }


        // =====================================
        // Set
        // Bool, DID, Float, IID, Int, Int64, String, Position
        // =====================================

        public static void SetProperty(this Biota biota, PropertyBool property, bool value)
        {
            var result = biota.BiotaPropertiesBool.FirstOrDefault(x => x.Type == (uint)property);

            if (result != null)
                result.Value = value;
            else
            {
                var entity = new BiotaPropertiesBool { ObjectId = biota.Id, Type = (ushort)property, Value = value, Object = biota };

                biota.BiotaPropertiesBool.Add(entity);
            }
        }

        public static void SetProperty(this Biota biota, PropertyDataId property, uint value)
        {
            var result = biota.BiotaPropertiesDID.FirstOrDefault(x => x.Type == (uint)property);

            if (result != null)
                result.Value = value;
            else
            {
                var entity = new BiotaPropertiesDID { ObjectId = biota.Id, Type = (ushort)property, Value = value, Object = biota };

                biota.BiotaPropertiesDID.Add(entity);
            }
        }

        public static void SetProperty(this Biota biota, PropertyFloat property, double value)
        {
            var result = biota.BiotaPropertiesFloat.FirstOrDefault(x => x.Type == (ushort)property);

            if (result != null)
                result.Value = value;
            else
            {
                var entity = new BiotaPropertiesFloat { ObjectId = biota.Id, Type = (ushort)property, Value = value, Object = biota };

                biota.BiotaPropertiesFloat.Add(entity);
            }
        }

        public static void SetProperty(this Biota biota, PropertyInstanceId property, uint value)
        {
            var result = biota.BiotaPropertiesIID.FirstOrDefault(x => x.Type == (uint)property);

            if (result != null)
                result.Value = value;
            else
            {
                var entity = new BiotaPropertiesIID { ObjectId = biota.Id, Type = (ushort)property, Value = value, Object = biota };

                biota.BiotaPropertiesIID.Add(entity);
            }
        }

        public static void SetProperty(this Biota biota, PropertyInt property, int value)
        {
            var result = biota.BiotaPropertiesInt.FirstOrDefault(x => x.Type == (uint)property);

            if (result != null)
                result.Value = value;
            else
            {
                var entity = new BiotaPropertiesInt { ObjectId = biota.Id, Type = (ushort)property, Value = value, Object = biota };

                biota.BiotaPropertiesInt.Add(entity);
            }
        }

        public static void SetProperty(this Biota biota, PropertyInt64 property, long value)
        {
            var result = biota.BiotaPropertiesInt64.FirstOrDefault(x => x.Type == (uint)property);

            if (result != null)
                result.Value = value;
            else
            {
                var entity = new BiotaPropertiesInt64 { ObjectId = biota.Id, Type = (ushort)property, Value = value, Object = biota };

                biota.BiotaPropertiesInt64.Add(entity);
            }
        }

        public static void SetProperty(this Biota biota, PropertyString property, string value)
        {
            var result = biota.BiotaPropertiesString.FirstOrDefault(x => x.Type == (uint)property);

            if (result != null)
                result.Value = value;
            else
            {
                var entity = new BiotaPropertiesString { ObjectId = biota.Id, Type = (ushort)property, Value = value, Object = biota };

                biota.BiotaPropertiesString.Add(entity);
            }
        }

        public static void SetPosition(this Biota biota, PositionType positionType, Position position)
        {
            var result = biota.BiotaPropertiesPosition.FirstOrDefault(x => x.PositionType == (uint)positionType);

            if (result != null)
            {
                result.ObjCellId = position.Cell;
                result.OriginX = position.PositionX;
                result.OriginY = position.PositionY;
                result.OriginZ = position.PositionZ;
                result.AnglesW = position.RotationW;
                result.AnglesX = position.RotationX;
                result.AnglesY = position.RotationY;
                result.AnglesZ = position.RotationZ;
            }
            else
            {
                var entity = new BiotaPropertiesPosition { ObjectId = biota.Id, PositionType = (ushort)positionType, ObjCellId = position.Cell, OriginX = position.PositionX, OriginY = position.PositionY, OriginZ = position.PositionZ, AnglesW = position.RotationW, AnglesX = position.RotationX, AnglesY = position.RotationY, AnglesZ = position.RotationZ, Object = biota };

                biota.BiotaPropertiesPosition.Add(entity);
            }
        }


        // =====================================
        // Remove
        // Bool, DID, Float, IID, Int, Int64, String, Position
        // =====================================

        public static bool TryRemoveProperty(this Biota biota, PropertyBool property, out BiotaPropertiesBool entity)
        {
            entity = biota.BiotaPropertiesBool.FirstOrDefault(x => x.Type == (uint)property);

            if (entity != null)
            {
                biota.BiotaPropertiesBool.Remove(entity);
                entity.Object = null;
                return true;
            }

            return false;
        }

        public static bool TryRemoveProperty(this Biota biota, PropertyDataId property, out BiotaPropertiesDID entity)
        {
            entity = biota.BiotaPropertiesDID.FirstOrDefault(x => x.Type == (uint)property);

            if (entity != null)
            {
                biota.BiotaPropertiesDID.Remove(entity);
                entity.Object = null;
                return true;
            }

            return false;
        }

        public static bool TryRemoveProperty(this Biota biota, PropertyFloat property, out BiotaPropertiesFloat entity)
        {
            entity = biota.BiotaPropertiesFloat.FirstOrDefault(x => x.Type == (ushort)property);

            if (entity != null)
            {
                biota.BiotaPropertiesFloat.Remove(entity);
                entity.Object = null;
                return true;
            }

            return false;
        }

        public static bool TryRemoveProperty(this Biota biota, PropertyInstanceId property, out BiotaPropertiesIID entity)
        {
            entity = biota.BiotaPropertiesIID.FirstOrDefault(x => x.Type == (uint)property);

            if (entity != null)
            {
                biota.BiotaPropertiesIID.Remove(entity);
                entity.Object = null;
                return true;
            }

            return false;
        }

        public static bool TryRemoveProperty(this Biota biota, PropertyInt property, out BiotaPropertiesInt entity)
        {
            entity = biota.BiotaPropertiesInt.FirstOrDefault(x => x.Type == (uint)property);

            if (entity != null)
            {
                biota.BiotaPropertiesInt.Remove(entity);
                entity.Object = null;
                return true;
            }

            return false;
        }

        public static bool TryRemoveProperty(this Biota biota, PropertyInt64 property, out BiotaPropertiesInt64 entity)
        {
            entity = biota.BiotaPropertiesInt64.FirstOrDefault(x => x.Type == (uint)property);

            if (entity != null)
            {
                biota.BiotaPropertiesInt64.Remove(entity);
                entity.Object = null;
                return true;
            }

            return false;
        }

        public static bool TryRemoveProperty(this Biota biota, PropertyString property, out BiotaPropertiesString entity)
        {
            entity = biota.BiotaPropertiesString.FirstOrDefault(x => x.Type == (uint)property);

            if (entity != null)
            {
                biota.BiotaPropertiesString.Remove(entity);
                entity.Object = null;
                return true;
            }

            return false;
        }

        public static bool TryRemovePosition(this Biota biota, PositionType positionType, out BiotaPropertiesPosition entity)
        {
            entity = biota.BiotaPropertiesPosition.FirstOrDefault(x => x.PositionType == (uint)positionType);

            if (entity != null)
            {
                biota.BiotaPropertiesPosition.Remove(entity);
                entity.Object = null;
                return true;
            }

            return false;
        }


        // =====================================
        // BiotaPropertiesSkill
        // =====================================
        public static BiotaPropertiesSkill GetSkill(this Biota biota, ushort type)
        {
            return biota.BiotaPropertiesSkill.FirstOrDefault(x => x.Type == type);
        }

        public static BiotaPropertiesSkill GetOrAddSkill(this Biota biota, ushort type, out bool skillAdded)
        {
            var entity = biota.BiotaPropertiesSkill.FirstOrDefault(x => x.Type == type);

            if (entity != null)
            {
                skillAdded = false;
                return entity;
            }

            entity = new BiotaPropertiesSkill { ObjectId = biota.Id, Type = type, Object = biota };

            biota.BiotaPropertiesSkill.Add(entity);
            skillAdded = true;
            return entity;
        }
    }
}