/*
 * 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 <QColor>

#include <GraphCanvas/Components/Slots/Data/DataSlotBus.h>
#include <Components/Connections/ConnectionVisualComponent.h>

namespace GraphCanvas
{
    class DataConnectionGraphicsItem;

    class DataPinStyleMonitor
        : public StyleNotificationBus::MultiHandler
    {
    public:
        DataPinStyleMonitor(DataConnectionGraphicsItem& graphicsItem);

        // StyleNotificationBus
        void OnStyleChanged() override;
        ////

        void SetSourceId(const AZ::EntityId& sourceId);
        void SetTargetId(const AZ::EntityId& targetId);

    private:
        DataConnectionGraphicsItem& m_graphicsItem;

        AZ::EntityId m_sourceId;
        AZ::EntityId m_targetId;
    };

    class DataConnectionGraphicsItem
        : public ConnectionGraphicsItem
        , public RootGraphicsItemNotificationBus::Handler
        , public DataSlotNotificationBus::MultiHandler
    {
    public:
        AZ_CLASS_ALLOCATOR(DataConnectionGraphicsItem, AZ::SystemAllocator);
        
        DataConnectionGraphicsItem(const AZ::EntityId& connectionEntityId);
        ~DataConnectionGraphicsItem() override = default;

        // ConnectionGraphicsItem
        void OnStyleChanged() override;
        ////
        
        void OnSourceSlotIdChanged(const AZ::EntityId& oldSlotId, const AZ::EntityId& newSlotId) override;
        void OnTargetSlotIdChanged(const AZ::EntityId& oldSlotId, const AZ::EntityId& newSlotId) override;

        // DataSlotNotificationBus
        void OnDisplayTypeChanged(const AZ::Uuid& dataTypeId, const AZStd::vector<AZ::Uuid>& typeIds) override;
        ////

        void UpdateDataColors();

        // RootGraphicsItemNotifications
        void OnDisplayStateChanged(RootGraphicsItemDisplayState oldState, RootGraphicsItemDisplayState newState) override;
        ////

    protected:
        Styling::ConnectionCurveType GetCurveStyle() const override;
        void UpdatePen() override;
        void OnPathChanged() override;
        
    private:
    
        void PopulateDataColor(QColor& targetColor, const AZ::EntityId& slotId);

        DataPinStyleMonitor m_dataPinStyleMonitor;
    
        QPen    m_pen;
        
        QColor m_sourceDataColor;
        QColor m_targetDataColor;
    };
}