o3de/Gems/ScriptCanvas/Code/Editor/View/Widgets/VariablePanel/VariableConfigurationWidget.cpp

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

#include <QAction>
#include <QCompleter>
#include <QEvent>
#include <QGraphicsScene>
#include <QGraphicsView>
#include <QAction>
#include <QMenu>
#include <QMessageBox>
#include <QScopedValueRollback>
#include <QLineEdit>
#include <QTimer>
#include <QPushButton>
#include <QHeaderView>

#include <AzCore/Component/ComponentApplication.h>
#include <AzCore/RTTI/BehaviorContext.h>
#include <AzCore/UserSettings/UserSettings.h>
#include <AzCore/Serialization/SerializeContext.h>
#include <AzCore/Serialization/EditContext.h>
#include <AzToolsFramework/Entity/EditorEntityContextPickingBus.h>
#include <AzToolsFramework/ToolsComponents/EditorComponentBase.h>

#include <GraphCanvas/Components/VisualBus.h>
#include <GraphCanvas/Components/SceneBus.h>
#include <GraphCanvas/Components/StyleBus.h>

#include <Editor/View/Widgets/VariablePanel/VariableConfigurationWidget.h>
#include <Editor/View/Widgets/VariablePanel/ui_VariableConfigurationWidget.h>

#include <Data/Data.h>

#include <Editor/GraphCanvas/GraphCanvasEditorNotificationBusId.h>
#include <Editor/QtMetaTypes.h>
#include <Editor/Settings.h>
#include <Editor/Translation/TranslationHelper.h>
#include <Editor/View/Widgets/DataTypePalette/DataTypePaletteModel.h>
#include <Editor/View/Widgets/NodePalette/VariableNodePaletteTreeItemTypes.h>
#include <Editor/View/Widgets/PropertyGridBus.h>
#include <Editor/Include/ScriptCanvas/Bus/EditorScriptCanvasBus.h>
#include <Editor/Include/ScriptCanvas/GraphCanvas/NodeDescriptorBus.h>

#include <ScriptCanvas/Core/NodeBus.h>
#include <ScriptCanvas/Data/DataRegistry.h>
#include <ScriptCanvas/GraphCanvas/NodeDescriptorBus.h>
#include <ScriptCanvas/Asset/RuntimeAsset.h>

namespace ScriptCanvasEditor
{
    ///////////////////////
    // VariableConfigurationWidget
    ///////////////////////

    VariableConfigurationWidget::VariableConfigurationWidget
        ( const ScriptCanvas::ScriptCanvasId& scriptCanvasId
        , const VariablePaletteRequests::VariableConfigurationInput& input
        , QWidget* parent)
        : AzQtComponents::StyledDialog(parent)
        , ui(new Ui::VariableConfigurationWidget())
        , m_scriptCanvasId(scriptCanvasId)
        , m_input(input)
    {
        ui->setupUi(this);
        {
            const auto windowTitle = AZStd::string::format("Pick %s name/type", input.m_configurationVariableTitle.c_str());
            setWindowTitle(QCoreApplication::translate("VariableConfigurationWidget", windowTitle.c_str(), nullptr));
            const auto labelText = AZStd::string::format("%s Name", input.m_configurationVariableTitle.c_str());
            ui->label->setText(QCoreApplication::translate("VariableConfigurationWidget", labelText.c_str(), nullptr));
            const auto slotPlaceHolderText = AZStd::string::format("Type the name for your %s here...", input.m_configurationVariableTitle.c_str());
            ui->slotName->setPlaceholderText(QCoreApplication::translate("VariableConfigurationWidget", slotPlaceHolderText.c_str(), nullptr));
            const auto label_2Text = AZStd::string::format("%s Type", input.m_configurationVariableTitle.c_str());
            ui->label_2->setText(QCoreApplication::translate("VariableConfigurationWidget", label_2Text.c_str(), nullptr));
        }

        ui->variablePalette->SetActiveScene(scriptCanvasId);

        ui->searchFilter->setClearButtonEnabled(true);
        QObject::connect(ui->searchFilter, &QLineEdit::textChanged, this, &VariableConfigurationWidget::OnQuickFilterChanged);
        QObject::connect(ui->slotName, &QLineEdit::returnPressed, this, &VariableConfigurationWidget::OnReturnPressed);
        QObject::connect(ui->slotName, &QLineEdit::textChanged, this, &VariableConfigurationWidget::OnNameChanged);
        QObject::connect(ui->variablePalette, &QTableView::clicked, this, [this]() { ui->buttonBox->button(QDialogButtonBox::Ok)->setEnabled(true); });
        QObject::connect(ui->variablePalette, &VariablePaletteTableView::CreateNamedVariable, this, [this](const AZStd::string& variableName, const ScriptCanvas::Data::Type& variableType)
            {
                // only emitted on container types
                OnCreateVariable(variableType);
                OnNameChanged(variableName.c_str());
            });

        // Tell the widget to auto create our context menu, for now
        setContextMenuPolicy(Qt::ActionsContextMenu);

        ui->searchFilter->setEnabled(true);

        QObject::connect(ui->variablePalette, &VariablePaletteTableView::CreateVariable, this, &VariableConfigurationWidget::OnCreateVariable);

        m_filterTimer.setInterval(250);
        m_filterTimer.setSingleShot(true);
        m_filterTimer.stop();

        QObject::connect(&m_filterTimer, &QTimer::timeout, this, &VariableConfigurationWidget::UpdateFilter);

        ui->buttonBox->button(QDialogButtonBox::Ok)->setEnabled(false);

    }

    VariableConfigurationWidget::~VariableConfigurationWidget()
    {
    }

    void VariableConfigurationWidget::PopulateVariablePalette(const AZStd::unordered_set< AZ::Uuid >& objectTypes)
    {
        ui->variablePalette->PopulateVariablePalette(objectTypes);
    }

    void VariableConfigurationWidget::OnEscape()
    {
        reject();
    }

    void VariableConfigurationWidget::focusOutEvent(QFocusEvent* )
    {
        reject();
    }

    const ScriptCanvas::ScriptCanvasId& VariableConfigurationWidget::GetActiveScriptCanvasId() const
    {
        return m_scriptCanvasId;
    }

    void VariableConfigurationWidget::ShowVariablePalette()
    {
        ClearFilter();

        ui->searchFilter->setPlaceholderText("Variable Type...");
        FocusOnSearchFilter();

        ui->searchFilter->setCompleter(ui->variablePalette->GetVariableCompleter());

        AzToolsFramework::EditorEvents::Bus::Handler::BusConnect();
    }

    void VariableConfigurationWidget::FocusOnSearchFilter()
    {
        ui->searchFilter->setFocus(Qt::FocusReason::MouseFocusReason);
    }

    void VariableConfigurationWidget::ClearFilter()
    {
        {
            QSignalBlocker blocker(ui->searchFilter);
            ui->searchFilter->setText("");
        }

        UpdateFilter();
    }

    void VariableConfigurationWidget::UpdateFilter()
    {
        ui->variablePalette->SetFilter(ui->searchFilter->userInputText());
    }

    void VariableConfigurationWidget::OnReturnPressed()
    {
        // Set the type to the slot
        if (!m_selectedType.IsNull())
        {
            m_slotName = ui->slotName->text().toUtf8().data();
            accept();
        }
    }

    AZ::Uuid VariableConfigurationWidget::GetSelectedType() const
    {
        return m_selectedType;
    }

    AZStd::string VariableConfigurationWidget::GetSlotName() const
    {
        return m_slotName;
    }

    void VariableConfigurationWidget::SetSlotName(AZStd::string name)
    {
        m_slotName = name;
        ui->slotName->setText(QString(name.c_str()));
    }

    void VariableConfigurationWidget::OnQuickFilterChanged(const QString& text)
    {
        if (text.isEmpty())
        {
            //If field was cleared, update immediately
            UpdateFilter();
            return;
        }
        m_filterTimer.stop();
        m_filterTimer.start();
    }

    void VariableConfigurationWidget::OnNameChanged(const QString& text)
    {
        bool nameInUse = false;

        const AZStd::unordered_map<ScriptCanvas::VariableId, ScriptCanvas::GraphVariable>* properties = nullptr;
        ScriptCanvas::GraphVariableManagerRequestBus::EventResult(properties, m_scriptCanvasId, &ScriptCanvas::GraphVariableManagerRequests::GetVariables);

        int numInUse = 0;
        if (properties)
        {
            for (const auto& variablePair : (*properties))
            {
                if (&variablePair.second != m_input.m_graphVariable)
                {
                    AZStd::string testName = text.toUtf8().data();
                    if (testName.compare(variablePair.second.GetVariableName()) == 0)
                    {
                        nameInUse = true;
                        ++numInUse;
                    }
                }
                else
                {
                    // the variable being renamed is the one being modified, so it can use its own name
                    break;
                }
            }
        }

        m_slotName = text.toUtf8().data();

        if (nameInUse)
        {
            m_slotName.append(AZStd::string::format(" (%d)", numInUse));
            ui->slotName->setText(m_slotName.c_str());
        }
    }

    void VariableConfigurationWidget::OnContextMenuRequested(const QPoint&)
    {
    }

    void VariableConfigurationWidget::OnCreateVariable(ScriptCanvas::Data::Type varType)
    {
        m_selectedType = varType.GetAZType();
    }

#include <Editor/View/Widgets/VariablePanel/moc_VariableConfigurationWidget.cpp>
}