// Copyright (C) 2023 Nicholas Maltbie
//
// Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
// associated documentation files (the "Software"), to deal in the Software without restriction,
// including without limitation the rights to use, copy, modify, merge, publish, distribute,
// sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all copies or
// substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
// BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
// CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.

using System.Collections.Generic;
using System.Linq;
using nickmaltbie.OpenKCC.Utils;
using nickmaltbie.OpenKCC.Utils.ColliderCast;
using nickmaltbie.TestUtilsUnity.Tests.TestCommon;
using NUnit.Framework;
using UnityEngine;
using static nickmaltbie.OpenKCC.Utils.KCCUtils;

namespace nickmaltbie.OpenKCC.Tests.EditMode.Utils.ColliderCast
{
    /// <summary>
    /// Basic tests for <see cref="nickmaltbie.OpenKCC.Utils.ColliderCast.BoxColliderCast"/> in edit mode.
    /// </summary>
    [TestFixture]
    public class BoxColliderCastTests : TestBase
    {
        private BoxColliderCast boxCast;

        [SetUp]
        public void SetUp()
        {
            GameObject go = CreateGameObject();
            boxCast = go.AddComponent<BoxColliderCast>();
            boxCast.Start();
        }

        /// <summary>
        /// Verify that the awake function for the collider cast will
        /// load properties form the attached collider if one exists.
        /// </summary>
        [Test]
        public void Validate_LoadBoxColliderSettings()
        {
            boxCast = CreateGameObject().AddComponent<BoxColliderCast>();
            BoxCollider collider = boxCast.gameObject.AddComponent<BoxCollider>();
            boxCast.enabled = false;
            collider.center = Vector3.zero;
            collider.size = Vector3.one;

            boxCast.ResetConfigDebug();
            boxCast.SetupCollider();

            Assert.AreEqual(Vector3.zero, boxCast.center);
            Assert.AreEqual(Vector3.one, boxCast.size);

            collider.center = Vector3.forward;
            collider.size = Vector3.one * 5;

            boxCast.ResetConfigDebug();
            boxCast.SetupCollider();

            Assert.AreEqual(Vector3.forward, boxCast.center);
            Assert.AreEqual(Vector3.one * 5, boxCast.size);

            GameObject.DestroyImmediate(collider);
            boxCast.ResetConfigDebug();
            boxCast.SetupCollider();
            BoxCollider generated = boxCast.GetComponent<BoxCollider>();
            Assert.IsNotNull(generated);

            Assert.AreEqual(generated.center, boxCast.center);
            Assert.AreEqual(generated.size, boxCast.size);
        }

        [Test]
        public void Validate_GetBottom()
        {
            BoxCollider box = boxCast.GetComponent<BoxCollider>();
            TestUtils.AssertInBounds(
                boxCast.GetBottom(Vector3.zero, Quaternion.identity),
                new Vector3(0, -(box.size / 2).y, 0),
                0.05f);
            TestUtils.AssertInBounds(
                boxCast.GetBottom(Vector3.zero, Quaternion.Euler(0, 0, 180)),
                new Vector3(0, (box.size / 2).y, 0),
                0.05f);
            TestUtils.AssertInBounds(
                boxCast.GetBottom(Vector3.zero, Quaternion.Euler(0, 0, 90)),
                new Vector3((box.size / 2).x, 0, 0),
                0.05f);
        }

        [Test]
        public void Validate_GetHits_NoHits()
        {
            bool didHit = boxCast.CastSelf(Vector3.zero, Quaternion.identity, Vector3.forward, 1, out _);
            Assert.IsFalse(didHit);
        }

        [Test]
        public void Validate_GetHits_OneHit([NUnit.Framework.Range(1, 10, 2)] float dist)
        {
            GameObject target = MakeCube(Vector3.forward * dist);
            Assert.IsTrue(boxCast.CastSelf(Vector3.zero, Quaternion.identity, Vector3.forward, dist + 1, out IRaycastHit hit));
            Assert.IsTrue(hit.rigidbody.gameObject == target);
        }

        [Test]
        public void Validate_NoBounceWithZeroDistance()
        {
            MakeCube();
            var config = new KCCConfig
            {
                ColliderCast = boxCast,
                SkinWidth = 0.1f,
            };
            KCCBounce bounce = KCCUtils.SingleKCCBounce(Vector3.up, Vector3.forward, Vector3.forward, Quaternion.identity, config);
            Assert.AreEqual(MovementAction.Move, bounce.action);
            Assert.AreEqual(bounce.finalPosition - bounce.initialPosition, Vector3.forward);
        }

        [Test]
        public void Validate_GetOverlapSkinWidth()
        {
            MakeCube();
            Assert.IsNotEmpty(boxCast.GetOverlapping(Vector3.forward, Quaternion.identity, skinWidth: 0));
            Assert.IsEmpty(boxCast.GetOverlapping(Vector3.forward, Quaternion.identity, skinWidth: 0.1f));
            Assert.IsEmpty(boxCast.GetOverlapping(Vector3.forward, Quaternion.identity, skinWidth: 0.2f));
            Assert.IsEmpty(boxCast.GetOverlapping(Vector3.forward, Quaternion.identity, skinWidth: 0.3f));
            Assert.IsEmpty(boxCast.GetOverlapping(Vector3.forward, Quaternion.identity, skinWidth: 0.4f));
            Assert.IsEmpty(boxCast.GetOverlapping(Vector3.forward, Quaternion.identity, skinWidth: 0.5f));
            Assert.IsNotEmpty(boxCast.GetOverlapping(Vector3.forward * 0.5f, Quaternion.identity, skinWidth: 0.1f));
            Assert.IsEmpty(boxCast.GetOverlapping(Vector3.forward * 1.5f, Quaternion.identity, skinWidth: 0.1f));
        }

        [Test]
        public void Validate_GetOverlapping([NUnit.Framework.Range(1, 10, 2)] int numOverlap)
        {
            GameObject[] targets = Enumerable.Range(0, numOverlap).Select(_ => MakeCube()).ToArray();
            IEnumerable<Collider> overlapping = boxCast.GetOverlapping(Vector3.zero, Quaternion.identity);
            Assert.IsTrue(new HashSet<GameObject>(overlapping.Select(o => o.gameObject)).SetEquals(targets));
        }

        [Test]
        public void Validate_PushOutOverlapping_NoOverlap()
        {
            TestUtils.AssertInBounds(
                boxCast.PushOutOverlapping(Vector3.zero, Quaternion.identity, 10.0f),
                Vector3.zero,
                0.01f);
        }

        [Test]
        public void Validate_PushOutOverlapping_OneOverlap()
        {
            MakeCube();
            TestUtils.AssertInBounds(
                boxCast.PushOutOverlapping(Vector3.zero, Quaternion.identity, 10.0f).magnitude,
                1.0f,
                0.01f);
        }

        [Test]
        public void Validate_PushOutOverlapping_MultipleOverlap()
        {
            MakeCube();
            var target1 = GameObject.CreatePrimitive(PrimitiveType.Sphere);
            target1.transform.localScale = Vector3.one * 3;
            RegisterGameObject(target1);
            TestUtils.AssertInBounds(
                boxCast.PushOutOverlapping(Vector3.zero, Quaternion.identity, 10.0f).magnitude,
                1.0f,
                0.01f);
        }

        [Test]
        public void Validate_DoRaycastInDirection_NoHit()
        {
            Assert.IsFalse(boxCast.DoRaycastInDirection(Vector3.zero, Vector3.forward, 1, out IRaycastHit _));
        }

        [Test]
        public void Validate_DoRaycastInDirection_Hit()
        {
            GameObject target = MakeCube(Vector3.forward * 1);
            Assert.IsTrue(boxCast.CastSelf(Vector3.zero, Quaternion.identity, Vector3.forward, 5, out IRaycastHit hit));
            Assert.IsTrue(hit.rigidbody.gameObject == target);
        }

        private GameObject MakeCube(Vector3? position = null)
        {
            var target = GameObject.CreatePrimitive(PrimitiveType.Cube);
            target.AddComponent<BoxCollider>();
            target.AddComponent<Rigidbody>();
            RegisterGameObject(target);
            target.transform.position = position ?? Vector3.zero;
            return target;
        }
    }
}