/*
 * Copyright (c) Contributors to the Open 3D Engine Project. For complete copyright and license terms please see the LICENSE at the root of this distribution.
 *
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */

#pragma once

#include <AzFramework/Physics/RigidBodyBus.h>
#include <Source/AutoGen/GemComponent.AutoComponent.h>

#if AZ_TRAIT_SERVER
#include <AzFramework/Spawnable/SpawnableEntitiesInterface.h>
#include <Source/Components/Multiplayer/GemSpawnerComponent.h>
#endif


namespace MultiplayerSample
{
    //! @brief Gems have a physical static body on the server with a trigger volume
    //! On clients, gems do not have physics and have a local bouncing effect.
    class GemComponent
        : public GemComponentBase
        , public Physics::RigidBodyNotificationBus::Handler
    {
    public:
        AZ_MULTIPLAYER_COMPONENT(MultiplayerSample::GemComponent, s_gemComponentConcreteUuid, MultiplayerSample::GemComponentBase);

        static void Reflect(AZ::ReflectContext* context);

        void OnActivate(Multiplayer::EntityIsMigrating entityIsMigrating) override;
        void OnDeactivate(Multiplayer::EntityIsMigrating entityIsMigrating) override;

        //! RigidBodyNotificationBus overrides ...
        //! @{
        void OnPhysicsEnabled(const AZ::EntityId& entityId) override;
        //! }@

    private:
        // Animate the gem on clients without spending network traffic. (The gem will not spin on the authority server.)
        void ClientAnimationTick();
        AZ::ScheduledEvent m_clientAnimationEvent{ [this]()
        {
            ClientAnimationTick();
        }, AZ::Name("GemComponent") };

        void OnNetworkLocationChanged(const AZ::Vector3& location);
        AZ::Event<AZ::Vector3>::Handler m_networkLocationHandler{ [this](const AZ::Vector3& location)
        {
            OnNetworkLocationChanged(location);
        } };

        AZ::Vector3 m_rootLocation = AZ::Vector3::CreateZero();
        AZ::TimeMs m_lifetime = AZ::Time::ZeroTimeMs;
    };

    class GemComponentController
        : public GemComponentControllerBase
    {
    public:
        explicit GemComponentController(GemComponent& parent);

        void OnActivate(Multiplayer::EntityIsMigrating entityIsMigrating) override;
        void OnDeactivate(Multiplayer::EntityIsMigrating entityIsMigrating) override;

#if AZ_TRAIT_SERVER
        void SetGemSpawnerController(GemSpawnerComponentController* controller);

        void HandleRPC_CollectedByPlayer(AzNetworking::IConnection* invokingConnection) override;

    private:
        GemSpawnerComponentController* m_controller = nullptr;
#endif
    };
}