Update ReferenceDrawer.cs

marc-antoine-girard · marc-antoine-girard · commit 64c42a4e1294 · 2022-07-12T18:21:54.000-04:00
diff --git a/Editor/Drawers/ReferenceDrawer.cs b/Editor/Drawers/ReferenceDrawer.cs
@@ -226,14 +226,21 @@ private void HandleDragPerform()
                     break;
                 case DragAndDropMode.Unity:
                     ModeValue = ReferenceMode.Unity;
-                    PropertyValue = DragAndDrop.objectReferences[0];
+                    PropertyValue = GetUnityObject(DragAndDrop.objectReferences[0]);
                     break;
                 case DragAndDropMode.None:
                 default:
                     throw new ArgumentOutOfRangeException();
             }
         }
 
+        private Object GetUnityObject(Object objectReference)
+        {
+            if(objectReference is GameObject gameObject)
+                return gameObject.GetComponent(GenericType);
+            return objectReference;
+        }
+
         protected abstract void PingObject();
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -226,14 +226,21 @@ private void HandleDragPerform()`
`226`	`226`	`break;`
`227`	`227`	`case DragAndDropMode.Unity:`
`228`	`228`	`ModeValue = ReferenceMode.Unity;`
`229`		`- PropertyValue = DragAndDrop.objectReferences[0];`
	`229`	`+ PropertyValue = GetUnityObject(DragAndDrop.objectReferences[0]);`
`230`	`230`	`break;`
`231`	`231`	`case DragAndDropMode.None:`
`232`	`232`	`default:`
`233`	`233`	`throw new ArgumentOutOfRangeException();`
`234`	`234`	`}`
`235`	`235`	`}`
`236`	`236`
	`237`	`+ private Object GetUnityObject(Object objectReference)`
	`238`	`+ {`
	`239`	`+ if(objectReference is GameObject gameObject)`
	`240`	`+ return gameObject.GetComponent(GenericType);`
	`241`	`+ return objectReference;`
	`242`	`+ }`
	`243`	`+`
`237`	`244`	`protected abstract void PingObject();`
`238`	`245`	`}`
`239`	`246`	`}`