/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/

#pragma once

#include <AzToolsFramework/ComponentMode/ComponentModeDelegate.h>
#include <AzToolsFramework/ToolsComponents/EditorComponentBase.h>
#include <PhysX/EditorColliderComponentRequestBus.h>
#include <Editor/ColliderComponentMode.h>

namespace UnitTest
{
    /// Mock EditorColliderComponent for testing component mode.
    class TestColliderComponentMode
        : public AzToolsFramework::Components::EditorComponentBase
        , public PhysX::EditorColliderComponentRequestBus::Handler
    {
    public:
        AZ_EDITOR_COMPONENT(TestColliderComponentMode, "{D4EEA05C-4620-4A63-8816-2D0380158DF9}");

        static void Reflect(AZ::ReflectContext* context)
        {
            if (auto serializeContext = azrtti_cast<AZ::SerializeContext*>(context))
            {
                serializeContext->Class<TestColliderComponentMode>()
                    ->Version(0)
                    ->Field("ComponentMode", &TestColliderComponentMode::m_componentModeDelegate);
            }
        }

        void Activate() override
        {
            EditorComponentBase::Activate();
            PhysX::EditorColliderComponentRequestBus::Handler::BusConnect(AZ::EntityComponentIdPair(GetEntityId(), GetId()));
            m_componentModeDelegate.ConnectWithSingleComponentMode<
                TestColliderComponentMode, PhysX::ColliderComponentMode>(
                    AZ::EntityComponentIdPair(GetEntityId(), GetId()), nullptr);
        }

        void Deactivate() override
        {
            PhysX::EditorColliderComponentRequestBus::Handler::BusDisconnect();
            EditorComponentBase::Deactivate();
            m_componentModeDelegate.Disconnect();
        }

        void SetColliderOffset(const AZ::Vector3& offset) { m_offset = offset; }
        AZ::Vector3 GetColliderOffset() override { return m_offset; }
        void SetColliderRotation(const AZ::Quaternion& rotation) override { m_rotation = rotation; }
        AZ::Quaternion GetColliderRotation() override { return m_rotation; }
        AZ::Transform GetColliderWorldTransform() override { return m_transform; }
        void SetShapeType(Physics::ShapeType shapeType) override { m_shapeType = shapeType; }
        Physics::ShapeType GetShapeType() override { return m_shapeType; }
        void SetSphereRadius(float radius) override { m_sphereRadius = radius; }
        float GetSphereRadius() override { return m_sphereRadius; }
        void SetCapsuleRadius(float radius) override { m_capsuleRadius = radius; }
        float GetCapsuleRadius() override { return m_capsuleRadius; }
        void SetCapsuleHeight(float height) override { m_capsuleHeight = height; }
        float GetCapsuleHeight() override { return m_capsuleHeight; }
        void SetAssetScale(const AZ::Vector3& scale) override { m_assetScale = scale; }
        AZ::Vector3 GetAssetScale() override { return m_assetScale; }

    private:
        AzToolsFramework::ComponentModeFramework::ComponentModeDelegate m_componentModeDelegate;
        AZ::Vector3 m_offset;
        AZ::Quaternion m_rotation;
        AZ::Transform m_transform;
        Physics::ShapeType m_shapeType;
        float m_sphereRadius;
        float m_capsuleHeight;
        float m_capsuleRadius;
        AZ::Vector3 m_assetScale;
    };
} // namespace UnitTest