Update to WiX Intermediate Representation

1 files changed, 23 insertions, 0 deletions

diff --git a/src/WixToolset.Data.WindowsInstaller/TableExtensions.cs b/src/WixToolset.Data.WindowsInstaller/TableExtensions.cs
new file mode 100644
index 00000000..1be64ffe
--- /dev/null
+++ b/src/WixToolset.Data.WindowsInstaller/TableExtensions.cs
@@ -0,0 +1,23 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+namespace WixToolset.Data
+{
+    using System.Collections.Generic;
+    using System.Linq;
+
+    /// <summary>
+    /// Methods that extend <see cref="Table"/>.
+    /// </summary>
+    public static class TableExtensions
+    {
+        /// <summary>
+        /// Gets the rows contained in the table as a particular row type.
+        /// </summary>
+        /// <param name="table">Table to get rows from.</param>
+        /// <remarks>If the <paramref name="table"/> is null, an empty enumerable will be returned.</remarks>
+        public static IEnumerable<T> RowsAs<T>(this Table table) where T : Row
+        {
+            return (null == table) ? Enumerable.Empty<T>() : table.Rows.Cast<T>();
+        }
+    }
+}