Add confirmation box support for the ReferencePickerAttribute

Author: arimger

Assets/Editor Toolbox/Editor/Drawers/Toolbox/PropertySelf/ReferencePickerAttributeDrawer.cs

@@ -75,26 +75,41 @@ private static Type GetCurrentManagedReferenceType(SerializedProperty property,
 
         private void CreateTypeProperty(SerializedProperty property, Type parentType, ReferencePickerAttribute attribute, Rect position)
         {
+            var searchFieldRequired = attribute.AddTextSearchField;
+            var confirmationRequired = attribute.AddConfirmationBox;
+
             var currentType = GetCurrentManagedReferenceType(property, out var hasMixedValues);
             var hadMixedValues = EditorGUI.showMixedValue;
             EditorGUI.showMixedValue = hasMixedValues;
-            typeField.OnGui(position, attribute.AddTextSearchField, (type) =>
+            typeField.OnGui(position, searchFieldRequired, (type) =>
             {
                 try
                 {
-                    if (!property.serializedObject.isEditingMultipleObjects)
+                    var isOperationAllowed = true;
+                    if (confirmationRequired)
                     {
-                        UpdateTypeProperty(property, type, attribute);
+                        isOperationAllowed = EditorUtility.DisplayDialog("Confirm Assignment",
+                            "Are you sure you want to assign a new reference? This action will replace the current one.",
+                            "Assign",
+                            "Cancel");
                     }
-                    else
+
+                    if (isOperationAllowed)
                     {
-                        var targets = property.serializedObject.targetObjects;
-                        foreach (var target in targets)
+                        if (!property.serializedObject.isEditingMultipleObjects)
+                        {
+                            UpdateTypeProperty(property, type, attribute);
+                        }
+                        else
                         {
-                            using (var so = new SerializedObject(target))
+                            var targets = property.serializedObject.targetObjects;
+                            foreach (var target in targets)
                             {
-                                var sp = so.FindProperty(property.propertyPath);
-                                UpdateTypeProperty(sp, type, attribute);
+                                using (var so = new SerializedObject(target))
+                                {
+                                    var sp = so.FindProperty(property.propertyPath);
+                                    UpdateTypeProperty(sp, type, attribute);
+                                }
                             }
                         }
                     }

Assets/Editor Toolbox/Runtime/Attributes/Property/Toolbox/PropertySelfAttributes/ReferencePickerAttribute.cs

@@ -42,6 +42,11 @@ public ReferencePickerAttribute(Type parentType, TypeGrouping typeGrouping)
         /// Indicates if created popup menu should have an additional search field.
         /// </summary>
         public bool AddTextSearchField { get; set; } = true;
+
+        /// <summary>
+        /// Indicates if confirmation box should appear after picking a new reference. 
+        /// </summary>
+        public bool AddConfirmationBox { get; set; }
     }
 }
 #endif