From f2e02ea05eff34f533bb60e6da049eb9f54c8c62 Mon Sep 17 00:00:00 2001
From: Bjarke Berg <mail@bergmania.dk>
Date: Mon, 28 Oct 2019 13:57:55 +0100
Subject: [PATCH] AB3329 - Added Content picker references

---
 .../ContentPickerPropertyEditor.cs            | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/src/Umbraco.Web/PropertyEditors/ContentPickerPropertyEditor.cs b/src/Umbraco.Web/PropertyEditors/ContentPickerPropertyEditor.cs
index c6de91f560..0e6eaa9254 100644
--- a/src/Umbraco.Web/PropertyEditors/ContentPickerPropertyEditor.cs
+++ b/src/Umbraco.Web/PropertyEditors/ContentPickerPropertyEditor.cs
@@ -1,6 +1,7 @@
 using System.Collections.Generic;
 using Umbraco.Core;
 using Umbraco.Core.Logging;
+using Umbraco.Core.Models.Editors;
 using Umbraco.Core.PropertyEditors;
 
 namespace Umbraco.Web.PropertyEditors
@@ -25,5 +26,23 @@ namespace Umbraco.Web.PropertyEditors
         {
             return new ContentPickerConfigurationEditor();
         }
+
+        protected override IDataValueEditor CreateValueEditor() => new ContentPickerPropertyValueEditor(Attribute);
+
+        internal class ContentPickerPropertyValueEditor  : DataValueEditor, IDataValueReference
+        {
+            public ContentPickerPropertyValueEditor(DataEditorAttribute attribute) : base(attribute)
+            {
+            }
+
+            public IEnumerable<UmbracoEntityReference> GetReferences(object value)
+            {
+                if (value == null) yield break;
+
+                var asString = value is string str ? str : value.ToString();
+
+                yield return new UmbracoEntityReference(Udi.Parse(asString));
+            }
+        }
     }
 }