// 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.SphereColliderCast"/> in edit mode.
    /// </summary>
    [TestFixture]
    public class SphereColliderCastTests : TestBase
    {
        private SphereColliderCast sphereCast;

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

        [Test]
        public void Validate_LoadCapsuleColliderSettings()
        {
            sphereCast = CreateGameObject().AddComponent<SphereColliderCast>();
            SphereCollider collider = sphereCast.gameObject.AddComponent<SphereCollider>();
            collider.center = Vector3.zero;
            collider.radius = 0.5f;

            sphereCast.ResetConfigDebug();
            sphereCast.Start();

            Assert.AreEqual(Vector3.zero, sphereCast.center);
            Assert.AreEqual(0.5f, sphereCast.radius);

            collider.center = Vector3.forward;
            collider.radius = 1;

            sphereCast.ResetConfigDebug();
            sphereCast.Start();

            Assert.AreEqual(Vector3.forward, sphereCast.center);
            Assert.AreEqual(1, sphereCast.radius);

            GameObject.DestroyImmediate(collider);
            sphereCast.Start();
            SphereCollider generated = sphereCast.GetComponent<SphereCollider>();
            Assert.IsNotNull(generated);

            Assert.AreEqual(generated.center, sphereCast.center);
            Assert.AreEqual(generated.radius, sphereCast.radius);
        }

        [Test]
        public void Validate_GetBottom()
        {
            SphereCollider sphere = sphereCast.GetComponent<SphereCollider>();
            TestUtils.AssertInBounds(
                sphereCast.GetBottom(Vector3.zero, Quaternion.identity),
                new Vector3(0, -sphere.radius, 0),
                0.05f);
            TestUtils.AssertInBounds(
                sphereCast.GetBottom(Vector3.zero, Quaternion.Euler(0, 0, 180)),
                new Vector3(0, sphere.radius, 0),
                0.05f);
            TestUtils.AssertInBounds(
                sphereCast.GetBottom(Vector3.zero, Quaternion.Euler(0, 0, 90)),
                new Vector3(sphere.radius, 0, 0),
                0.05f);
        }

        [Test]
        public void Validate_GetHits_NoHits()
        {
            bool didHit = sphereCast.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(sphereCast.CastSelf(Vector3.zero, Quaternion.identity, Vector3.forward, dist + 1, out IRaycastHit hit));
            Assert.IsTrue(hit.rigidbody.gameObject == target);
        }

        [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 = sphereCast.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(
                sphereCast.PushOutOverlapping(Vector3.zero, Quaternion.identity, 10.0f),
                Vector3.zero,
                0.01f);
        }

        [Test]
        public void Validate_PushOutOverlapping_OneOverlap()
        {
            MakeCube();
            TestUtils.AssertInBounds(
                sphereCast.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(
                sphereCast.PushOutOverlapping(Vector3.zero, Quaternion.identity, 10.0f).magnitude,
                1.0f,
                0.01f);
        }

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

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

        [Test]
        public void Validate_NoBounceWithZeroDistance()
        {
            MakeCube();
            var config = new KCCConfig
            {
                ColliderCast = sphereCast,
                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(sphereCast.GetOverlapping(Vector3.forward, Quaternion.identity, skinWidth: 0));
            Assert.IsEmpty(sphereCast.GetOverlapping(Vector3.forward, Quaternion.identity, skinWidth: 0.1f));
            Assert.IsEmpty(sphereCast.GetOverlapping(Vector3.forward, Quaternion.identity, skinWidth: 0.2f));
            Assert.IsEmpty(sphereCast.GetOverlapping(Vector3.forward, Quaternion.identity, skinWidth: 0.3f));
            Assert.IsEmpty(sphereCast.GetOverlapping(Vector3.forward, Quaternion.identity, skinWidth: 0.4f));
            Assert.IsEmpty(sphereCast.GetOverlapping(Vector3.forward, Quaternion.identity, skinWidth: 0.5f));
            Assert.IsNotEmpty(sphereCast.GetOverlapping(Vector3.forward * 0.5f, Quaternion.identity, skinWidth: 0.1f));
            Assert.IsEmpty(sphereCast.GetOverlapping(Vector3.forward * 1.5f, Quaternion.identity, skinWidth: 0.1f));
        }

        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;
        }
    }
}