/*
 * 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 <AzCore/Component/Component.h>
#include <AzCore/std/containers/unordered_map.h>

#include <GraphCanvas/Components/Nodes/NodeBus.h>

#include <Editor/Include/ScriptCanvas/GraphCanvas/MappingBus.h>
#include <ScriptCanvas/Core/NodeBus.h>

namespace ScriptCanvasEditor
{
    class SceneMemberMappingComponent
        : public AZ::Component
        , public SceneMemberMappingRequestBus::Handler
        , public SceneMemberMappingConfigurationRequestBus::Handler
        , public GraphCanvas::NodeNotificationBus::Handler
    {
    public:
        AZ_COMPONENT(SceneMemberMappingComponent, "{145667DE-EBD6-4EC5-B630-7C9B1A5ACFF0}");
        static void Reflect(AZ::ReflectContext* context);

        SceneMemberMappingComponent() = default;
        SceneMemberMappingComponent(const AZ::EntityId& sourceId);
        ~SceneMemberMappingComponent() = default;

        void Activate() override;
        void Deactivate() override;

        // SceneMemberMappingConfigurationRequestBus
        void ConfigureMapping(const AZ::EntityId& scriptCanvasMemberId) override;
        ////

        // SceneMemberMappingRequestBus
        AZ::EntityId GetGraphCanvasEntityId() const override;
        ////

        // NodeNotificationBus
        void OnBatchedConnectionManipulationBegin() override;
        void OnBatchedConnectionManipulationEnd() override;
        ////

    private:

        AZ::EntityId m_sourceId;
    };

    class SlotMappingComponent
        : public AZ::Component
        , public GraphCanvas::NodeNotificationBus::Handler
        , public SlotMappingRequestBus::MultiHandler
        , public ScriptCanvas::NodeNotificationsBus::Handler
        , public SceneMemberMappingConfigurationRequestBus::Handler
    {
    public:
        AZ_COMPONENT(SlotMappingComponent, "{94DBC04C-964D-46A0-AD66-6A779FE4DC61}");
        
        static void Reflect(AZ::ReflectContext* context);
        
        SlotMappingComponent() = default;
        SlotMappingComponent(const AZ::EntityId& sourceId);
        ~SlotMappingComponent() = default;
        
        void Activate() override;
        void Deactivate() override;
        
        // GraphCanvas::NodeNotificationBus
        void OnAddedToScene(const AZ::EntityId&) override;
        
        void OnSlotAddedToNode(const AZ::EntityId& slotId) override;
        void OnSlotRemovedFromNode(const AZ::EntityId& slotId) override;
        ////
        
        // ScriptCanvasSlotRemappingBus
        AZ::EntityId MapToGraphCanvasId(const ScriptCanvas::SlotId& slotId) override;
        ////

        // ScriptCanvas::NodeNotificationBus::Handler
        void OnSlotRenamed(const ScriptCanvas::SlotId& slotId, AZStd::string_view newName) override;
        void OnSlotDisplayTypeChanged(const ScriptCanvas::SlotId& slotId, const ScriptCanvas::Data::Type& slotType) override;
        ////

        // SceneMemberMappingConfigurationRequestBus
        void ConfigureMapping(const AZ::EntityId& scriptCanvasMemberId) override;
        ////
        
    private:

        AZ::EntityId m_sourceId;
        AZStd::unordered_map< ScriptCanvas::SlotId, AZ::EntityId > m_slotMapping;

        AZStd::unordered_set< ScriptCanvas::SlotId > m_ignoreRenameSlots;
    };    
}