From c52adf76f7fe5daedd1a51daf9c994b235c838ef Mon Sep 17 00:00:00 2001
From: Bjarke Berg <mail@bergmania.dk>
Date: Mon, 28 Oct 2019 12:56:54 +0100
Subject: [PATCH] AB3226 - Added MediaPicker references

---
 .../MediaPickerPropertyEditor.cs              | 24 +++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/src/Umbraco.Web/PropertyEditors/MediaPickerPropertyEditor.cs b/src/Umbraco.Web/PropertyEditors/MediaPickerPropertyEditor.cs
index dd755ee0ba..0d0f1a8498 100644
--- a/src/Umbraco.Web/PropertyEditors/MediaPickerPropertyEditor.cs
+++ b/src/Umbraco.Web/PropertyEditors/MediaPickerPropertyEditor.cs
@@ -1,5 +1,7 @@
-using Umbraco.Core;
+using System.Collections.Generic;
+using Umbraco.Core;
 using Umbraco.Core.Logging;
+using Umbraco.Core.Models.Editors;
 using Umbraco.Core.PropertyEditors;
 
 namespace Umbraco.Web.PropertyEditors
@@ -17,14 +19,32 @@ namespace Umbraco.Web.PropertyEditors
         Icon = Constants.Icons.MediaImage)]
     public class MediaPickerPropertyEditor : DataEditor
     {
+
         /// <summary>
         /// Initializes a new instance of the <see cref="MediaPickerPropertyEditor"/> class.
         /// </summary>
         public MediaPickerPropertyEditor(ILogger logger)
             : base(logger)
-        { }
+        {
+        }
 
         /// <inheritdoc />
         protected override IConfigurationEditor CreateConfigurationEditor() => new MediaPickerConfigurationEditor();
+
+        protected override IDataValueEditor CreateValueEditor() => new MediaPickerPropertyValueEditor(Attribute);
+    }
+
+    public class MediaPickerPropertyValueEditor : DataValueEditor, IDataValueReference
+    {
+        public MediaPickerPropertyValueEditor(DataEditorAttribute attribute) : base(attribute)
+        {
+        }
+
+        public IEnumerable<UmbracoEntityReference> GetReferences(object value)
+        {
+            var asString = value == null ? string.Empty : value is string str ? str : value.ToString();
+
+            yield return new UmbracoEntityReference(Udi.Parse(asString));
+        }
     }
 }