Add custom toolbar actions

Add custom buttons to the Asset Management toolbar in the Graph window.

You can add custom buttons to the Asset Management toolbar by creating toolbar element classes and applying the [GraphToolbarElement] attribute.

Prerequisites

You must have created and implemented a graph tool. - Implement a graph tool

Create a toolbar element class

To create a toolbar element class, do the following:

Create a class that inherits from any VisualElement (EditorToolbarButton in the example below).
Apply the [GraphToolbarElement] attribute to bind it to your Graph type.
Implement IAccessContainerWindow to get a reference to the editor window. This lets you access the current graph from within the button’s click handler.

[GraphToolbarElement(id, typeof(MySimpleGraph), order: 200)]
class MyCustomToolbarAction : EditorToolbarButton, IAccessContainerWindow
{
    public const string id = "MyGraphTool/CustomAction";
    public EditorWindow containerWindow { get; set; }

    public MyCustomToolbarAction()
    {
        text = "My Action";
        tooltip = "Performs a custom action on the graph";
        icon = EditorGUIUtility.IconContent("console.infoicon").image as Texture2D;
        clicked += OnClick;
    }

    void OnClick()
    {
        if (containerWindow is IGraphWindow graphWindow)
            UnityEngine.Debug.Log(graphWindow.Graph.Name);
    }
}

When you open your graph asset, your custom buttons appear in the Asset Management toolbar at the position determined by their order value.

Additional resources

Implement node options

Add subgraph support