Add support for DefKey and Keys to GraphCollection

Author: Lyeeedar

StructuredXmlEditor.sln

@@ -43,4 +43,7 @@ Global
 	GlobalSection(Performance) = preSolution
 		HasPerformanceSessions = true
 	EndGlobalSection
+	GlobalSection(Performance) = preSolution
+		HasPerformanceSessions = true
+	EndGlobalSection
 EndGlobal

StructuredXmlEditor/Core.xmldef

@@ -327,6 +327,8 @@
 			<Boolean Name="AllowCircularLinks" Default="false" ToolTip="Whether circular links are allowed." VisibleIf="AllowReferenceLinks==true" />
 			<Boolean Name="FlattenData" Default="false" ToolTip="If true each graph node will be placed in a flat list within this element." VisibleIf="AllowReferenceLinks==true" />
 			<String Name="NodeStoreName" Default="Nodes" ToolTip="The name of the node added to this to store all the graph nodes." VisibleIf="FlattenData==true" />
+			<String Name="Keys" ToolTip="The names of the objects to use in this collection." />
+			<String Name="DefKey" ToolTip="The name of the ReferenceDef to use in this collection." />
 			<Number Name="MinCount" Min="0" ToolTip="The minimum number of child elements this can have." />
 			<Number Name="MaxCount" Min="1" Default="99999999" ToolTip="The maximum number of child elements this can have." />
 			<Colour Name="Background" Default="0,0,0,0" />
@@ -346,6 +348,8 @@
 		<Attributes>
 			<String Name="Name" Default="GraphCollection" ToolTip="The name to be written for this element. Place a path to an element inside a '{}' to use that elements value. This will be used as the key when referencing it." />
 			<Boolean Name="ChildrenAreUnique" ToolTip="If set then each child can only be added once" />
+			<String Name="Keys" ToolTip="The names of the objects to use in this collection." />
+			<String Name="DefKey" ToolTip="The name of the ReferenceDef to use in this collection." />
 			<Number Name="MinCount" Min="0" ToolTip="The minimum number of child elements this can have." />
 			<Number Name="MaxCount" Min="1" Default="99999999" ToolTip="The maximum number of child elements this can have." />
 			<Colour Name="Background" Default="0,0,0,0" />

StructuredXmlEditor/Definition/GraphCollectionDefinition.cs

@@ -1,4 +1,5 @@
 using StructuredXmlEditor.Data;
+using StructuredXmlEditor.View;
 using System;
 using System.Collections.Generic;
 using System.Linq;
@@ -17,6 +18,8 @@ public class GraphCollectionDefinition : GraphNodeDefinition
 		public List<CollectionChildDefinition> ChildDefinitions { get; } = new List<CollectionChildDefinition>();
 		public int MinCount { get; set; } = 0;
 		public int MaxCount { get; set; } = int.MaxValue;
+		public List<Tuple<string, string>> DefKeys { get; set; } = new List<Tuple<string, string>>();
+		public string DefKey { get; set; }
 
 		public GraphCollectionDefinition()
 		{
@@ -129,6 +132,27 @@ public override void Parse(XElement definition)
 
 			var currentGroup = "Items";
 
+			DefKey = definition.Attribute("DefKey")?.Value?.ToString();
+			var keyString = definition.Attribute("Keys")?.Value?.ToString();
+			if (!string.IsNullOrWhiteSpace(keyString))
+			{
+				if (!keyString.Contains('('))
+				{
+					DefKeys.AddRange(keyString.Split(new char[] { ',' }, StringSplitOptions.RemoveEmptyEntries).Select(e => new Tuple<string, string>(e.Trim(), "Type")));
+				}
+				else
+				{
+					var categories = keyString.Split(new char[] { ')' }, StringSplitOptions.RemoveEmptyEntries);
+					foreach (var categoryString in categories)
+					{
+						var split = categoryString.Split('(');
+						var category = split[0].Trim();
+						if (category.StartsWith(",")) category = category.Substring(1);
+						DefKeys.AddRange(split[1].Split(new char[] { ',' }, StringSplitOptions.RemoveEmptyEntries).Select(e => new Tuple<string, string>(e.Trim(), category)));
+					}
+				}
+			}
+
 			var childDefs = definition.Nodes();
 			foreach (var childDef in childDefs)
 			{
@@ -152,7 +176,7 @@ public override void Parse(XElement definition)
 				}
 			}
 
-			if (ChildDefinitions.Count == 0)
+			if (ChildDefinitions.Count == 0 && DefKey == null && DefKeys.Count == 0)
 			{
 				throw new Exception("No child definitions in collection '" + Name + "'!");
 			}
@@ -224,6 +248,55 @@ public override void DoSaveData(XElement parent, DataItem item)
 
 		public override void RecursivelyResolve(Dictionary<string, DataDefinition> local, Dictionary<string, DataDefinition> global, Dictionary<string, Dictionary<string, DataDefinition>> referenceableDefinitions)
 		{
+			if (DefKey != null)
+			{
+				var key = DefKey.ToLower();
+
+				Dictionary<string, DataDefinition> defs = null;
+				if (local.ContainsKey(key)) defs = local;
+				else if (global.ContainsKey(key)) defs = global;
+
+				if (defs != null)
+				{
+					var def = defs[key] as ReferenceDefinition;
+
+					foreach (var keydef in def.Keys)
+					{
+						var childDef = def.Definitions[keydef.Item1];
+						var childWrapperDef = new CollectionChildDefinition();
+						childWrapperDef.WrappedDefinition = childDef;
+
+						ChildDefinitions.Add(childWrapperDef);
+						Keys.Add(new Tuple<CollectionChildDefinition, string>(childWrapperDef, keydef.Item2));
+					}
+				}
+				else
+				{
+					Message.Show("Failed to find key " + DefKey + "!", "Collection Resolve Failed", "Ok");
+				}
+			}
+
+			foreach (var key in DefKeys)
+			{
+				Dictionary<string, DataDefinition> defs = null;
+				if (local.ContainsKey(key.Item1.ToLower())) defs = local;
+				else if (global.ContainsKey(key.Item1.ToLower())) defs = global;
+
+				if (defs != null)
+				{
+					var childDef = defs[key.Item1.ToLower()];
+					var childWrapperDef = new CollectionChildDefinition();
+					childWrapperDef.WrappedDefinition = childDef;
+
+					ChildDefinitions.Add(childWrapperDef);
+					Keys.Add(new Tuple<CollectionChildDefinition, string>(childWrapperDef, key.Item2));
+				}
+				else if (key.Item1 != "---")
+				{
+					Message.Show("Failed to find key " + key.Item1 + "!", "Collection Resolve Failed", "Ok");
+				}
+			}
+
 			foreach (var def in ChildDefinitions) def.WrappedDefinition.RecursivelyResolve(local, global, referenceableDefinitions);
 		}
 	}