Update to WiX Intermediate Representation

1 files changed, 222 insertions, 0 deletions

diff --git a/src/WixToolset.Data.WindowsInstaller/Rows/WixActionRowCollection.cs b/src/WixToolset.Data.WindowsInstaller/Rows/WixActionRowCollection.cs
new file mode 100644
index 00000000..513a104f
--- /dev/null
+++ b/src/WixToolset.Data.WindowsInstaller/Rows/WixActionRowCollection.cs
@@ -0,0 +1,222 @@
+// 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.Rows
+{
+    using System;
+    using System.Collections;
+    using System.Diagnostics;
+    using System.Xml;
+
+    /// <summary>
+    /// A collection of action rows sorted by their sequence table and action name.
+    /// </summary>
+    public sealed class WixActionRowCollection : ICollection
+    {
+        private SortedList collection;
+
+        /// <summary>
+        /// Creates a new action table object.
+        /// </summary>
+        public WixActionRowCollection()
+        {
+            this.collection = new SortedList();
+        }
+
+        /// <summary>
+        /// Gets the number of items in the collection.
+        /// </summary>
+        /// <value>Number of items in collection.</value>
+        public int Count
+        {
+            get { return this.collection.Count; }
+        }
+
+        /// <summary>
+        /// Gets if the collection has been synchronized.
+        /// </summary>
+        /// <value>True if the collection has been synchronized.</value>
+        public bool IsSynchronized
+        {
+            get { return this.collection.IsSynchronized; }
+        }
+
+        /// <summary>
+        /// Gets the object used to synchronize the collection.
+        /// </summary>
+        /// <value>Oject used the synchronize the collection.</value>
+        public object SyncRoot
+        {
+            get { return this; }
+        }
+
+        /// <summary>
+        /// Get an ActionRow by its sequence table and action name.
+        /// </summary>
+        /// <param name="sequenceTable">The sequence table of the ActionRow.</param>
+        /// <param name="action">The action name of the ActionRow.</param>
+        public WixActionRow this[SequenceTable sequenceTable, string action]
+        {
+            get { return (WixActionRow)this.collection[GetKey(sequenceTable, action)]; }
+        }
+
+        /// <summary>
+        /// Add an ActionRow to the collection.
+        /// </summary>
+        /// <param name="actionRow">The ActionRow to add.</param>
+        /// <param name="overwrite">true to overwrite an existing ActionRow; false otherwise.</param>
+        public void Add(WixActionRow actionRow, bool overwrite)
+        {
+            string key = GetKey(actionRow.SequenceTable, actionRow.Action);
+
+            if (overwrite)
+            {
+                this.collection[key] = actionRow;
+            }
+            else
+            {
+                this.collection.Add(key, actionRow);
+            }
+        }
+
+        /// <summary>
+        /// Add an ActionRow to the collection.
+        /// </summary>
+        /// <param name="actionRow">The ActionRow to add.</param>
+        public void Add(WixActionRow actionRow)
+        {
+            this.Add(actionRow, false);
+        }
+
+        /// <summary>
+        /// Determines if the collection contains an ActionRow with a specific sequence table and name.
+        /// </summary>
+        /// <param name="sequenceTable">The sequence table of the ActionRow.</param>
+        /// <param name="action">The action name of the ActionRow.</param>
+        /// <returns>true if the ActionRow was found; false otherwise.</returns>
+        public bool Contains(SequenceTable sequenceTable, string action)
+        {
+            return this.collection.Contains(GetKey(sequenceTable, action));
+        }
+
+        /// <summary>
+        /// Copies the collection into an array.
+        /// </summary>
+        /// <param name="array">Array to copy the collection into.</param>
+        /// <param name="index">Index to start copying from.</param>
+        public void CopyTo(System.Array array, int index)
+        {
+            this.collection.Values.CopyTo(array, index);
+        }
+
+        /// <summary>
+        /// Gets the enumerator for the collection.
+        /// </summary>
+        /// <returns>The enumerator for the collection.</returns>
+        public IEnumerator GetEnumerator()
+        {
+            return this.collection.Values.GetEnumerator();
+        }
+
+        /// <summary>
+        /// Remove an ActionRow from the collection.
+        /// </summary>
+        /// <param name="sequenceTable">The sequence table of the ActionRow.</param>
+        /// <param name="action">The action name of the ActionRow.</param>
+        public void Remove(SequenceTable sequenceTable, string action)
+        {
+            this.collection.Remove(GetKey(sequenceTable, action));
+        }
+
+        /// <summary>
+        /// Load an action table from an XmlReader.
+        /// </summary>
+        /// <param name="reader">Reader to get data from.</param>
+        /// <returns>The ActionRowCollection represented by the xml.</returns>
+        internal static WixActionRowCollection Load(XmlReader reader)
+        {
+            reader.MoveToContent();
+
+            return Parse(reader);
+        }
+
+        /// <summary>
+        /// Creates a new action table object and populates it from an Xml reader.
+        /// </summary>
+        /// <param name="reader">Reader to get data from.</param>
+        /// <returns>The parsed ActionTable.</returns>
+        private static WixActionRowCollection Parse(XmlReader reader)
+        {
+            if (!reader.LocalName.Equals("actions"))
+            {
+                throw new XmlException();
+            }
+
+            WixActionRowCollection actionRows = new WixActionRowCollection();
+            bool empty = reader.IsEmptyElement;
+
+            while (reader.MoveToNextAttribute())
+            {
+            }
+
+            if (!empty)
+            {
+                bool done = false;
+
+                // loop through all the fields in a row
+                while (!done && reader.Read())
+                {
+                    switch (reader.NodeType)
+                    {
+                        case XmlNodeType.Element:
+                        switch (reader.LocalName)
+                        {
+                            case "action":
+                                WixActionRow[] parsedActionRows = WixActionRow.Parse(reader);
+
+                                foreach (WixActionRow actionRow in parsedActionRows)
+                                {
+                                    actionRows.Add(actionRow);
+                                }
+                                break;
+                            default:
+                                throw new XmlException();
+                        }
+                            break;
+                        case XmlNodeType.EndElement:
+                            done = true;
+                            break;
+                    }
+                }
+
+                if (!done)
+                {
+                    throw new XmlException();
+                }
+            }
+
+            return actionRows;
+        }
+
+        /// <summary>
+        /// Get the key for storing an ActionRow.
+        /// </summary>
+        /// <param name="sequenceTable">The sequence table of the ActionRow.</param>
+        /// <param name="action">The action name of the ActionRow.</param>
+        /// <returns>The string key.</returns>
+        private static string GetKey(SequenceTable sequenceTable, string action)
+        {
+            return GetKey(sequenceTable.ToString(), action);
+        }
+
+        /// <summary>
+        /// Get the key for storing an ActionRow.
+        /// </summary>
+        /// <param name="sequenceTable">The sequence table of the ActionRow.</param>
+        /// <param name="action">The action name of the ActionRow.</param>
+        /// <returns>The string key.</returns>
+        private static string GetKey(string sequenceTable, string action)
+        {
+            return String.Concat(sequenceTable, '/', action);
+        }
+    }
+}