Make all template field info fields available so custom fields are al…

…so accessible.
blipson89 · Aug 27, 2019 · 0cc50bc · 0cc50bc
1 parent 745c255
commit 0cc50bc
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 1 deletion.
diff --git a/src/Leprechaun/Model/TemplateFieldCodeGenerationMetadata.cs b/src/Leprechaun/Model/TemplateFieldCodeGenerationMetadata.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using System.Diagnostics;
 
 namespace Leprechaun.Model
@@ -32,6 +33,8 @@ public TemplateFieldCodeGenerationMetadata(TemplateFieldInfo field, string codeN
 
 		public virtual int SortOrder => Field.SortOrder;
 
+        public virtual Dictionary<Guid, string> AllFields => Field.AllFields;
+
 		/// <summary>
 		/// A unique name for this template field, usable as a C# identifier (e.g. property name)
 		/// </summary>

diff --git a/src/Leprechaun/Model/TemplateFieldInfo.cs b/src/Leprechaun/Model/TemplateFieldInfo.cs
@@ -1,11 +1,17 @@
 using System;
+using System.Collections.Generic;
 using System.Diagnostics;
 
 namespace Leprechaun.Model
 {
 	[DebuggerDisplay("{Path} ({Id})")]
 	public class TemplateFieldInfo
 	{
+        public TemplateFieldInfo()
+        {
+            AllFields = new Dictionary<Guid, string>();
+		}
+
 		public Guid Id { get; set; }
 
 		public string Name { get; set; }
@@ -23,5 +29,7 @@ public class TemplateFieldInfo
 		public string Section { get; set; }
 
 		public int SortOrder { get; set; }
+
+		public Dictionary<Guid, string> AllFields { get; set; }
 	}
 }
diff --git a/src/Leprechaun/TemplateReaders/DataStoreTemplateReader.cs b/src/Leprechaun/TemplateReaders/DataStoreTemplateReader.cs
@@ -111,7 +111,8 @@ protected virtual TemplateFieldInfo[] ParseTemplateFields(IItemData templateItem
 						Section = section.Name,
 						SortOrder = GetFieldValueAsInt(field, SortOrderFieldId, 100),
 						Source = GetFieldValue(field, SourceFieldId, string.Empty),
-						Type = GetFieldValue(field, FieldTypeFieldId, string.Empty)
+						Type = GetFieldValue(field, FieldTypeFieldId, string.Empty),
+						AllFields = GetAllFields(field)
 					});
 				}
 			}
@@ -180,5 +181,42 @@ protected virtual Guid[] ParseMultilistValue(string value)
 		}
 
 		protected virtual ICollection<Guid> IgnoredBaseTemplateIds => new HashSet<Guid> { TemplateIDs.StandardTemplate.Guid, TemplateIDs.Folder.Guid };
+
+        protected virtual Dictionary<Guid, string> GetAllFields(IItemData item)
+        {
+			var allFields = new Dictionary<Guid, string>();
+
+            foreach (var field in item.SharedFields)
+            {
+                if (!allFields.ContainsKey(field.FieldId))
+                {
+                    allFields.Add(field.FieldId, field.Value);
+                }
+            }
+
+            foreach (var language in item.UnversionedFields)
+            {
+                foreach (var field in language.Fields)
+                {
+					if (!allFields.ContainsKey(field.FieldId))
+                    {
+                        allFields.Add(field.FieldId, field.Value);
+                    }
+				}
+            }
+
+            foreach (var version in item.Versions)
+            {
+                foreach (var field in version.Fields)
+                {
+					if (!allFields.ContainsKey(field.FieldId))
+                    {
+                        allFields.Add(field.FieldId, field.Value);
+                    }
+				}
+            }
+
+            return allFields;
+        }
 	}
 }