Merge pull request #57 from lordubbe/ClassesOnly-attribute

Thundernerd · Nov 14, 2023 · 16d5b04 · 16d5b04
2 parents 809922d + 08cf75a
commit 16d5b04
Show file tree

Hide file tree

Showing 5 changed files with 42 additions and 6 deletions.
diff --git a/Editor/Attributes.meta b/Editor/Attributes.meta
diff --git a/Editor/Attributes/ClassesOnlyAttribute.cs b/Editor/Attributes/ClassesOnlyAttribute.cs
@@ -0,0 +1,9 @@
+using System.Collections;
+using System.Collections.Generic;
+using UnityEngine;
+
+namespace TNRD
+{
+    /// <summary>An attribute to use when you want to exclude non-raw references in the reference picker.</summary>
+    public class ClassesOnlyAttribute : PropertyAttribute { }
+}
diff --git a/Editor/Attributes/ClassesOnlyAttribute.cs.meta b/Editor/Attributes/ClassesOnlyAttribute.cs.meta
diff --git a/Editor/Drawers/ReferenceDrawer.cs b/Editor/Drawers/ReferenceDrawer.cs
@@ -71,7 +71,7 @@ private void OnButtonClicked(Rect position, SerializedProperty property)
         {
             AdvancedDropdownState state = new AdvancedDropdownState();
             SerializableInterfaceAdvancedDropdown dropdown =
-                new SerializableInterfaceAdvancedDropdown(state, GenericType, GetRelevantScene(property), property);
+                new SerializableInterfaceAdvancedDropdown(state, GenericType, GetRelevantScene(property), property, FieldInfo.GetCustomAttribute<ClassesOnlyAttribute>() != null);
             dropdown.ItemSelectedEvent += OnItemSelected;
             dropdown.Show(position);
         }

diff --git a/Editor/Utilities/SerializableInterfaceAdvancedDropdown.cs b/Editor/Utilities/SerializableInterfaceAdvancedDropdown.cs
@@ -16,6 +16,7 @@ internal sealed class SerializableInterfaceAdvancedDropdown : AdvancedDropdown
         private readonly Type interfaceType;
         private readonly MethodInfo sortChildrenMethod;
         private readonly bool canSort;
+        private readonly bool classesOnly;
         private readonly Scene? relevantScene;
         private readonly SerializedProperty property;
 
@@ -28,7 +29,8 @@ internal sealed class SerializableInterfaceAdvancedDropdown : AdvancedDropdown
             AdvancedDropdownState state,
             Type interfaceType,
             Scene? relevantScene,
-            SerializedProperty property
+            SerializedProperty property,
+            bool classesOnly
         )
             : base(state)
         {
@@ -42,15 +44,21 @@ SerializedProperty property
             this.interfaceType = interfaceType;
             this.relevantScene = relevantScene;
             this.property = property;
+            this.classesOnly = classesOnly;
         }
 
         /// <inheritdoc />
         protected override AdvancedDropdownItem BuildRoot()
         {
-            AdvancedDropdownItemWrapper item = new AdvancedDropdownItemWrapper(interfaceType.Name)
-                .AddChild(new AssetsItemBuilder(interfaceType).Build())
-                .AddChild(new ClassesItemBuilder(interfaceType).Build())
-                .AddChild(new SceneItemBuilder(interfaceType, relevantScene).Build());
+            AdvancedDropdownItemWrapper item = new AdvancedDropdownItemWrapper(interfaceType.Name);
+
+            item.AddChild(new ClassesItemBuilder(interfaceType).Build());
+
+            if (!classesOnly)
+            {
+                item.AddChild(new AssetsItemBuilder(interfaceType).Build());
+                item.AddChild(new SceneItemBuilder(interfaceType, relevantScene).Build());
+            }
 
             foreach (AdvancedDropdownItem dropdownItem in item.children)
             {