Laurens-Packages/Packages/com.unity.visualeffectgraph/Editor/TemplateWindow/CreateFromTemplateDropDownButton.cs

using UnityEditor.Experimental;
using UnityEditor.Experimental.GraphView;
using UnityEditor.VFX.UI;

using UnityEngine;
using UnityEngine.UIElements;

namespace UnityEditor.VFX
{
    class CreateFromTemplateDropDownButton : DropDownButtonBase
    {
        private const string k_MainButtonName = "create-button";

        private readonly Button m_InsertButton;

        public CreateFromTemplateDropDownButton(VFXView vfxView)
            : base(
            nameof(CreateFromTemplateDropDownButton),
            vfxView,
        "VFXCreateFromTemplateDropDownPanel",
        "Insert a template into current graph\nHold CTRL key and click to create a new VFX",
            k_MainButtonName,
        EditorResources.iconsPath + "CreateAddNew.png",
        true,
        false)
        {
            var createNew = m_PopupContent.Q<Button>("createNew");
            createNew.clicked += OnCreateNew;

            var mainButton = this.Q<Button>(k_MainButtonName);
            mainButton.RegisterCallback<MouseUpEvent>(OnMainButtonMouseUp);

            m_InsertButton = m_PopupContent.Q<Button>("insert");
            m_InsertButton.clicked += OnInsert;
        }

        private void OnMainButtonMouseUp(MouseUpEvent evt)
        {
            if (evt.button != (int)MouseButton.LeftMouse || evt.clickCount > 1)
                return;

            // If the current asset is a subgraph do not call "OnInsert" callback and rather open the dropdown
            if (m_VFXView.controller.model.isSubgraph)
            {
                OnTogglePopup();
            }
            else if (evt.modifiers.HasFlag(EventModifiers.Control))
            {
                OnCreateNew();
            }
            else
            {
                OnInsert();
            }
        }

        protected override Vector2 GetPopupSize() => new(200, 54);

        protected override void OnOpenPopup()
        {
            m_InsertButton.SetEnabled(!m_VFXView.controller.model.isSubgraph);
            base.OnOpenPopup();
        }

        private void OnCreateNew()
        {
            GraphViewTemplateWindow.ShowCreateFromTemplate(new VFXTemplateHelperInternal(), CreateNewFromTemplate);
        }

        private void OnInsert()
        {
            GraphViewTemplateWindow.ShowInsertTemplate(new VFXTemplateHelperInternal(), InsertFromTemplate);
        }

        private void InsertFromTemplate(string templatePath, string assetPath)
        {
            var window = VFXViewWindow.GetWindow(m_VFXView);
            window.graphView.CreateTemplateSystem(templatePath, Vector2.zero, null, true);
        }

        private void CreateNewFromTemplate(string templatePath, string assetPath)
        {
            m_VFXView.CreateNewFromTemplate(templatePath, assetPath);
        }
    }
}