T-Node/TNodeGraphViewImpl/Editor/Inspector/NodeInspectorInNode.cs

using System.Reflection;
using TNode.TNodeCore.Editor.Serialization;
using TNodeCore.Editor.NodeGraphView;
using TNodeCore.Runtime.Attributes;
using TNodeCore.Runtime.Models;
using UnityEditor;
using UnityEditor.UIElements;
using UnityEngine.UIElements;

namespace TNodeGraphViewImpl.Editor.Inspector{
    public class NodeInspectorInNode:VisualElement{
        private NodeData _data;
        public NodeData Data{
            get => _data;
            set{
                _data = value;
                UpdateData();
                
            }
            
        }

        public NodeInspectorInNode():base(){
        }
        private void UpdateData(){
            if (_data != null){
                RefreshInspector();
            }
        }

        private void RefreshInspector(){
            //Set size
           
            Clear();
            //RefreshItems();
            RefreshPropertyDrawer();
        }

  
        private void RefreshPropertyDrawer(){
            //Check if the model's type is a generic type of  BlackboardDragNode<>
            if (_data.GetType().IsSubclassOf(typeof(BlackboardDragNode))){
                return;
            }
            var serializedObject = new SerializedObject((NodeDataWrapper)_data);
            foreach (var field in _data.GetType().GetFields(BindingFlags.Instance | BindingFlags.Public|BindingFlags.NonPublic)){
                //Create corresponding property field
                //check if the field has ShowInNodeView attribute
                var showInNodeViewAttribute = field.GetCustomAttribute<ShowInNode>() != null;
                if (!showInNodeViewAttribute)
                    continue;
                var drawer = new PropertyField(serializedObject.FindProperty("model").FindPropertyRelative(field.Name));
                drawer.RegisterValueChangeCallback((evt) => {
                    serializedObject.Update();
                    serializedObject.ApplyModifiedProperties();
                    ((NodeDataWrapper)_data).ForceNotify();
                });
                if (_data is SceneNode && _data is  BlackboardDragNode==false){
                    
                }
                else{
                    if (drawer.Q<ObjectField>() != null){
                        drawer.Q<ObjectField>().allowSceneObjects = false;
                    }
                }
                drawer.Bind(serializedObject);
                Add(drawer);
            }

            var globalTest = GetFirstAncestorOfType<IBaseDataGraphView>()?.AutoUpdate;
            if(globalTest??false){
                CreateTestButton();
            }
            else if (_data.isTest){
                //Add a test button for the node
                CreateTestButton();
            }
        }

        private void CreateTestButton(){
            var testButton = new Button(() => {
                var test = GetFirstAncestorOfType<IBaseDataGraphView>();
                if (test != null){
                    if(!test.IsRuntimeGraph) return;
                    var runtimeGraph = test.GetRuntimeGraph();
                    if (runtimeGraph != null){
                       var res = runtimeGraph.RunOnDependency(_data);
                    }
                    _data.OnTest();
                }
            }){
                text = "Test"
            };

            Add(testButton);
     
        }
    }
}