Added Get and Set methods to TreeNode

rappen · Aug 6, 2023 · b99b0bd · b99b0bd
1 parent 9030b63
commit b99b0bd
Showing 1 changed file with 24 additions and 1 deletion.
diff --git a/FetchXmlBuilder/Builder/TreeNodeHelper.cs b/FetchXmlBuilder/Builder/TreeNodeHelper.cs
@@ -8,7 +8,7 @@
 
 namespace Rappen.XTB.FetchXmlBuilder.Builder
 {
-    internal class TreeNodeHelper
+    internal static class TreeNodeHelper
     {
         /// <summary>
         /// Reuses an existing node in a tree to represent a new FetchXML item
@@ -590,5 +590,28 @@ private static List<string> GetEntitysChilds(XmlNode ent)
             }
             return result;
         }
+
+        internal static void SetNodeAttributeValue(this TreeNode node, string attribute, string value)
+        {
+            var collec = (Dictionary<string, string>)node.Tag;
+            if (collec.ContainsKey(attribute))
+            {
+                collec[attribute] = value;
+            }
+            else
+            {
+                collec.Add(attribute, value);
+            }
+        }
+
+        internal static string GetNodeAttributeValue(this TreeNode node, string attribute)
+        {
+            var collec = (Dictionary<string, string>)node.Tag;
+            if (collec.ContainsKey(attribute))
+            {
+                return collec[attribute];
+            }
+            return null;
+        }
     }
 }