MinhHai
82 lines
3.2 KiB
C#
82 lines
3.2 KiB
C#
using System.Linq;
|
|
using UnityEditor;
|
|
using UnityEngine.UIElements;
|
|
using EditorAttributes.Editor.Utility;
|
|
|
|
namespace EditorAttributes.Editor
|
|
{
|
|
[CustomPropertyDrawer(typeof(InlineButtonAttribute))]
|
|
public class InlineButtonDrawer : PropertyDrawerBase
|
|
{
|
|
public override VisualElement CreatePropertyGUI(SerializedProperty property)
|
|
{
|
|
var root = new VisualElement() { style = { flexDirection = FlexDirection.Row } };
|
|
|
|
var propertyField = CreatePropertyField(property);
|
|
|
|
propertyField.name = "CustomPropertyField"; // This is used to identify the our property field from the one automatically created by unity for the drawer
|
|
propertyField.style.flexGrow = 1f;
|
|
|
|
root.Add(propertyField);
|
|
|
|
root.RegisterCallback<GeometryChangedEvent>(OnGeometryChanged);
|
|
|
|
void OnGeometryChanged(GeometryChangedEvent changeEvent)
|
|
{
|
|
root.UnregisterCallback<GeometryChangedEvent>(OnGeometryChanged);
|
|
|
|
// Adding multiple InlineButton attributes on the same property causes a root and propertyField to be continously added to the hierarchy with each property drawer call
|
|
// So we move children recursively from the bottom of the hierarchy to the root and remove their original parents so there is only one root and property field no matter how many times this attribute is used
|
|
MoveChildren(root, propertyField.name);
|
|
MoveChildren(propertyField, propertyField.name);
|
|
|
|
var propertyInfo = ReflectionUtility.GetValidMemberInfo(property.name, property);
|
|
var inlineButtonAttributes = propertyInfo.GetCustomAttributes(typeof(InlineButtonAttribute), true) as InlineButtonAttribute[];
|
|
|
|
foreach (var inlineButtonAttribute in inlineButtonAttributes)
|
|
root.Add(CreateInlineButton(inlineButtonAttribute, property));
|
|
}
|
|
|
|
return root;
|
|
}
|
|
|
|
private VisualElement CreateInlineButton(InlineButtonAttribute inlineButtonAttribute, SerializedProperty property)
|
|
{
|
|
var methodInfo = ReflectionUtility.FindFunction(inlineButtonAttribute.FunctionName, property.serializedObject.targetObject);
|
|
string buttonLabel = inlineButtonAttribute.ButtonLabel == string.Empty ? inlineButtonAttribute.FunctionName : inlineButtonAttribute.ButtonLabel;
|
|
|
|
if (methodInfo.GetParameters().Length > 0)
|
|
return new HelpBox("The function cannot have parameters", HelpBoxMessageType.Error);
|
|
|
|
if (inlineButtonAttribute.IsRepetable)
|
|
{
|
|
var repeatButton = new RepeatButton(() => InvokeFunctionOnAllTargets(property.serializedObject.targetObjects, methodInfo.Name), inlineButtonAttribute.PressDelay, inlineButtonAttribute.RepetitionInterval) { text = buttonLabel };
|
|
|
|
repeatButton.style.width = inlineButtonAttribute.ButtonWidth;
|
|
repeatButton.AddToClassList(Button.ussClassName);
|
|
|
|
return repeatButton;
|
|
}
|
|
else
|
|
{
|
|
var button = new Button(() => InvokeFunctionOnAllTargets(property.serializedObject.targetObjects, methodInfo.Name)) { text = buttonLabel };
|
|
|
|
button.style.width = inlineButtonAttribute.ButtonWidth;
|
|
|
|
return button;
|
|
}
|
|
}
|
|
|
|
private void MoveChildren(VisualElement element, string parentName)
|
|
{
|
|
if (element.parent.name == parentName)
|
|
{
|
|
foreach (var child in element.Children().ToList())
|
|
element.parent.Add(child);
|
|
|
|
element.RemoveFromHierarchy();
|
|
}
|
|
}
|
|
}
|
|
}
|