From fa853d3373f02825962f35fcac6ad36dde13bc4d Mon Sep 17 00:00:00 2001
From: dahall <dahall@byu.net>
Date: Wed, 12 Feb 2020 20:06:52 -0700
Subject: [PATCH] Added GetItem and SetItem extension methods to TreeNode

---
 WIndows.Forms/Extensions/TreeViewExtension.cs | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/WIndows.Forms/Extensions/TreeViewExtension.cs b/WIndows.Forms/Extensions/TreeViewExtension.cs
index b7df5bac..40111d5f 100644
--- a/WIndows.Forms/Extensions/TreeViewExtension.cs
+++ b/WIndows.Forms/Extensions/TreeViewExtension.cs
@@ -81,6 +81,28 @@ namespace Vanara.Extensions
 		/// <returns>An <see cref="IEnumerable{T}"/> of all <see cref="TreeNode"/> instances in this <see cref="TreeView"/>.</returns>
 		public static IEnumerable<TreeNode> EnumerateAllNodes(this TreeView treeView) => AsEnumerable(treeView.Nodes, true);
 
+		/// <summary>Sets the <see cref="TVITEM"/> values.</summary>
+		/// <param name="node">The <see cref="TreeNode"/> instance for which to set details.</param>
+		/// <param name="tvItem">The <see cref="TVITEMEX"/> instance.</param>
+		public static bool SetItem(this TreeNode node, ref TVITEMEX tvItem) => SendMessage(node.TreeView.Handle, TreeViewMessage.TVM_SETITEM, default, ref tvItem).ToInt32() != 0;
+
+		/// <summary>Gets the node values.</summary>
+		/// <param name="node">The <see cref="TreeNode"/> instance for which to get details.</param>
+		/// <param name="mask">The mask of items to get.</param>
+		/// <param name="stateMask">The mask of states to get.</param>
+		/// <returns>A <see cref="TVITEMEX"/> structure with the information.</returns>
+		public static TVITEMEX GetItem(this TreeNode node, TreeViewItemMask mask = (TreeViewItemMask)0x13FF, TreeViewItemStates stateMask = (TreeViewItemStates)0xFFFF)
+		{
+			var tvItem = new TVITEMEX
+			{
+				hItem = node.Handle,
+				mask = mask.SetFlags(TreeViewItemMask.TVIF_HANDLE).SetFlags(TreeViewItemMask.TVIF_TEXT, false),
+				stateMask = stateMask
+			};
+			SendMessage(node.TreeView.Handle, TreeViewMessage.TVM_GETITEM, default, ref tvItem);
+			return tvItem;
+		}
+
 		private static IconSize GetIconSizeFromSize(Size sz)
 		{
 			switch (sz.Height)