From feeb1c56cfac74d00110875624f4a0bcf55706b7 Mon Sep 17 00:00:00 2001
From: Shannon <sdeminick@gmail.com>
Date: Thu, 20 Sep 2018 14:58:15 +1000
Subject: [PATCH] wip for IDocumentEntitySlim changes

---
 .../Models/Entities/DocumentEntitySlim.cs     | 21 +++++++++++++
 .../Models/Entities/IContentEntitySlim.cs     |  2 +-
 .../Models/Entities/IDocumentEntitySlim.cs    | 31 ++++++++++++-------
 .../Implement/EntityRepository.cs             |  3 +-
 4 files changed, 44 insertions(+), 13 deletions(-)

diff --git a/src/Umbraco.Core/Models/Entities/DocumentEntitySlim.cs b/src/Umbraco.Core/Models/Entities/DocumentEntitySlim.cs
index 5b63ad81c5..39ece5fa10 100644
--- a/src/Umbraco.Core/Models/Entities/DocumentEntitySlim.cs
+++ b/src/Umbraco.Core/Models/Entities/DocumentEntitySlim.cs
@@ -1,4 +1,5 @@
 using System.Collections.Generic;
+using System.Linq;
 
 namespace Umbraco.Core.Models.Entities
 {
@@ -8,13 +9,32 @@ namespace Umbraco.Core.Models.Entities
     public class DocumentEntitySlim : ContentEntitySlim, IDocumentEntitySlim
     {
         private static readonly IReadOnlyDictionary<string, string> Empty = new Dictionary<string, string>();
+
         private IReadOnlyDictionary<string, string> _cultureNames;
+        private IEnumerable<string> _publishedCultures;
+        private IEnumerable<string> _editedCultures;
+
+        /// <inheritdoc />
         public IReadOnlyDictionary<string, string> CultureNames
         {
             get => _cultureNames ?? Empty;
             set => _cultureNames = value;
         }
 
+        /// <inheritdoc />
+        public IEnumerable<string> PublishedCultures
+        {
+            get => _publishedCultures ?? Enumerable.Empty<string>();
+            set => _publishedCultures = value;
+        }
+
+        /// <inheritdoc />
+        public IEnumerable<string> EditedCultures
+        {
+            get => _editedCultures ?? Enumerable.Empty<string>();
+            set => _editedCultures = value;
+        }
+
         public ContentVariation Variations { get; set; }
 
         /// <inheritdoc />
@@ -22,5 +42,6 @@ namespace Umbraco.Core.Models.Entities
 
         /// <inheritdoc />
         public bool Edited { get; set; }
+        
     }
 }
diff --git a/src/Umbraco.Core/Models/Entities/IContentEntitySlim.cs b/src/Umbraco.Core/Models/Entities/IContentEntitySlim.cs
index dc986a4cd9..9ab557b02c 100644
--- a/src/Umbraco.Core/Models/Entities/IContentEntitySlim.cs
+++ b/src/Umbraco.Core/Models/Entities/IContentEntitySlim.cs
@@ -20,4 +20,4 @@
         /// </summary>
         string ContentTypeThumbnail { get; }
     }
-}
\ No newline at end of file
+}
diff --git a/src/Umbraco.Core/Models/Entities/IDocumentEntitySlim.cs b/src/Umbraco.Core/Models/Entities/IDocumentEntitySlim.cs
index 6b72fd4a2b..38fd9a02f1 100644
--- a/src/Umbraco.Core/Models/Entities/IDocumentEntitySlim.cs
+++ b/src/Umbraco.Core/Models/Entities/IDocumentEntitySlim.cs
@@ -7,26 +7,35 @@ namespace Umbraco.Core.Models.Entities
     /// </summary>
     public interface IDocumentEntitySlim : IContentEntitySlim
     {
-        //fixme we need to supply more information than this and change this property name. This will need to include Published/Editor per variation since we need this information for the tree
+        /// <summary>
+        /// Gets the variant name for each culture
+        /// </summary>
         IReadOnlyDictionary<string, string> CultureNames { get; }
 
+        /// <summary>
+        /// Gets the published cultures.
+        /// </summary>
+        IEnumerable<string> PublishedCultures { get; }
+
+        /// <summary>
+        /// Gets the edited cultures.
+        /// </summary>
+        IEnumerable<string> EditedCultures { get; }
+
+        /// <summary>
+        /// Gets the content variation of the content type.
+        /// </summary>
         ContentVariation Variations { get; }
 
         /// <summary>
-        /// At least one variation is published
+        /// Gets a value indicating whether the content is published.
         /// </summary>
-        /// <remarks>
-        /// If the document is invariant, this simply means there is a published version
-        /// </remarks>
-        bool Published { get; set; }
+        bool Published { get; }
 
         /// <summary>
-        /// At least one variation has pending changes
+        /// Gets a value indicating whether the content has been edited.
         /// </summary>
-        /// <remarks>
-        /// If the document is invariant, this simply means there is pending changes
-        /// </remarks>
-        bool Edited { get; set; }
+        bool Edited { get; }
 
     }
 }
diff --git a/src/Umbraco.Core/Persistence/Repositories/Implement/EntityRepository.cs b/src/Umbraco.Core/Persistence/Repositories/Implement/EntityRepository.cs
index 340eecb538..7dd2b7ff27 100644
--- a/src/Umbraco.Core/Persistence/Repositories/Implement/EntityRepository.cs
+++ b/src/Umbraco.Core/Persistence/Repositories/Implement/EntityRepository.cs
@@ -901,10 +901,11 @@ namespace Umbraco.Core.Persistence.Repositories.Implement
             var entity = new DocumentEntitySlim();
             BuildContentEntity(entity, dto);
             
-            //fixme we need to set these statuses for each variant, see notes in IDocumentEntitySlim
+            //fill in the invariant info
             entity.Edited = dto.Edited;
             entity.Published = dto.Published;
 
+            //fill in the variant info
             if (dto.Variations.VariesByCulture() && dto.VariationInfo != null && dto.VariationInfo.Count > 0)
             {
                 var variantInfo = new Dictionary<string, string>(StringComparer.InvariantCultureIgnoreCase);