T-Node/TNodeGraphViewImpl/Editor/Search/NodeSearchWindowProvider.cs

using System;
using System.Collections.Generic;
using TNode.Cache;
using TNode.Editor.NodeGraphView;
using TNode.Editor.Tools.NodeCreator;
using TNode.Models;
using UnityEditor;
using UnityEditor.Experimental.GraphView;
using UnityEngine;
using UnityEngine.UIElements;

namespace TNode.Editor.Search{
    public class NodeSearchWindowProvider:ScriptableObject,ISearchWindowProvider{
        private Type _graphType;
        private GraphView _graphView;
        private EditorWindow _editor;
        public void Setup(Type graph,GraphView graphView,EditorWindow editor){
            _graphType = graph;
            _graphView = graphView;
            _editor = editor;
        }
        public List<SearchTreeEntry> CreateSearchTree(SearchWindowContext context){
            var nodeDataTypes = NodeEditorExtensions.GetGraphDataUsage(_graphType);
            
            var list = new List<SearchTreeEntry>{
                new SearchTreeGroupEntry(new GUIContent("Add New Node"), 0),
            };
            //TODO a node icon shall be given by some way
            Texture2D icon = new Texture2D(2,2);
            foreach (var nodeDataType in nodeDataTypes){
                Debug.Log(nodeDataType.Name);

                
                list.Add(new SearchTreeEntry(new GUIContent($" {nodeDataType.Name} ",icon)){
                    level =  1,
                    userData = nodeDataType,
                });
            }
            return list;
        }
        public bool OnSelectEntry(SearchTreeEntry SearchTreeEntry, SearchWindowContext context){
            var userData = SearchTreeEntry.userData;
            var relativePos = context.screenMousePosition - _editor.position.position;
            var localPos = _graphView.WorldToLocal(relativePos);
            if (userData is Type type){
                //Check if type is derived from NodeData
                if (typeof(NodeData).IsAssignableFrom(type)){
                    //Make an instance of the type
                    if (NodeCreator.InstantiateNodeData(type) is { } nodeData){
                        nodeData.nodeName = $"New {type.Name}";
                        ((IBaseDataGraphView) _graphView).AddTNode(nodeData, new Rect(localPos.x, localPos.y, 100, 100));
                    }
                }
                return true;
            }
            return false;
        }
        
        public NodeSearchWindowProvider(){
        }
    }
}