Reintegrate MSI constructs into WxToolset.Data.WindowsInstaller namespace

1 files changed, 218 insertions, 0 deletions

diff --git a/src/WixToolset.Data/WindowsInstaller/TableDefinition.cs b/src/WixToolset.Data/WindowsInstaller/TableDefinition.cs
new file mode 100644
index 00000000..518f0926
--- /dev/null
+++ b/src/WixToolset.Data/WindowsInstaller/TableDefinition.cs
@@ -0,0 +1,218 @@
+// 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.WindowsInstaller
+{
+    using System;
+    using System.Collections.Generic;
+    using System.Collections.ObjectModel;
+    using System.Xml;
+
+    /// <summary>
+    /// Definition of a table in a database.
+    /// </summary>
+    public sealed class TableDefinition : IComparable<TableDefinition>
+    {
+        /// <summary>
+        /// Tracks the maximum number of columns supported in a real table.
+        /// This is a Windows Installer limitation.
+        /// </summary>
+        public const int MaxColumnsInRealTable = 32;
+
+        /// <summary>
+        /// Creates a table definition.
+        /// </summary>
+        /// <param name="name">Name of table to create.</param>
+        /// <param name="columns">Column definitions for the table.</param>
+        /// <param name="unreal">Flag if table is unreal.</param>
+        /// <param name="bootstrapperApplicationData">Flag if table is part of UX Manifest.</param>
+        public TableDefinition(string name, IList<ColumnDefinition> columns, bool unreal = false, bool bootstrapperApplicationData = false)
+        {
+            this.Name = name;
+            this.Unreal = unreal;
+            this.BootstrapperApplicationData = bootstrapperApplicationData;
+
+            this.Columns = new ReadOnlyCollection<ColumnDefinition>(columns);
+        }
+
+        /// <summary>
+        /// Gets the name of the table.
+        /// </summary>
+        /// <value>Name of the table.</value>
+        public string Name { get; private set; }
+
+        /// <summary>
+        /// Gets if the table is unreal.
+        /// </summary>
+        /// <value>Flag if table is unreal.</value>
+        public bool Unreal { get; private set; }
+
+        /// <summary>
+        /// Gets if the table is a part of the bootstrapper application data manifest.
+        /// </summary>
+        /// <value>Flag if table is a part of the bootstrapper application data manifest.</value>
+        public bool BootstrapperApplicationData { get; private set; }
+
+        /// <summary>
+        /// Gets the collection of column definitions for this table.
+        /// </summary>
+        /// <value>Collection of column definitions for this table.</value>
+        public IList<ColumnDefinition> Columns { get; private set; }
+
+        /// <summary>
+        /// Gets the column definition in the table by index.
+        /// </summary>
+        /// <param name="columnIndex">Index of column to locate.</param>
+        /// <value>Column definition in the table by index.</value>
+        public ColumnDefinition this[int columnIndex]
+        {
+            get { return this.Columns[columnIndex]; }
+        }
+
+        /// <summary>
+        /// Compares this table definition to another table definition.
+        /// </summary>
+        /// <remarks>
+        /// Only Windows Installer traits are compared, allowing for updates to WiX-specific table definitions.
+        /// </remarks>
+        /// <param name="updated">The updated <see cref="TableDefinition"/> to compare with this target definition.</param>
+        /// <returns>0 if the tables' core properties are the same; otherwise, non-0.</returns>
+        public int CompareTo(TableDefinition updated)
+        {
+            // by definition, this object is greater than null
+            if (null == updated)
+            {
+                return 1;
+            }
+
+            // compare the table names
+            int ret = String.Compare(this.Name, updated.Name, StringComparison.Ordinal);
+
+            // compare the column count
+            if (0 == ret)
+            {
+                // transforms can only add columns
+                ret = Math.Min(0, updated.Columns.Count - this.Columns.Count);
+
+                // compare name, type, and length of each column
+                for (int i = 0; 0 == ret && this.Columns.Count > i; i++)
+                {
+                    ColumnDefinition thisColumnDef = this.Columns[i];
+                    ColumnDefinition updatedColumnDef = updated.Columns[i];
+
+                    ret = thisColumnDef.CompareTo(updatedColumnDef);
+                }
+            }
+
+            return ret;
+        }
+
+        /// <summary>
+        /// Parses table definition from xml reader.
+        /// </summary>
+        /// <param name="reader">Reader to get data from.</param>
+        /// <returns>The TableDefintion represented by the Xml.</returns>
+        internal static TableDefinition Read(XmlReader reader)
+        {
+            bool empty = reader.IsEmptyElement;
+            string name = null;
+            bool unreal = false;
+            bool bootstrapperApplicationData = false;
+
+            while (reader.MoveToNextAttribute())
+            {
+                switch (reader.LocalName)
+                {
+                    case "name":
+                        name = reader.Value;
+                        break;
+                    case "unreal":
+                        unreal = reader.Value.Equals("yes");
+                        break;
+                    case "bootstrapperApplicationData":
+                        bootstrapperApplicationData = reader.Value.Equals("yes");
+                        break;
+                }
+            }
+
+            if (null == name)
+            {
+                throw new XmlException();
+            }
+
+            List<ColumnDefinition> columns = new List<ColumnDefinition>();
+            bool hasPrimaryKeyColumn = false;
+
+            // parse the child elements
+            if (!empty)
+            {
+                bool done = false;
+
+                while (!done && reader.Read())
+                {
+                    switch (reader.NodeType)
+                    {
+                        case XmlNodeType.Element:
+                            switch (reader.LocalName)
+                            {
+                                case "columnDefinition":
+                                    ColumnDefinition columnDefinition = ColumnDefinition.Read(reader);
+                                    columns.Add(columnDefinition);
+
+                                    if (columnDefinition.PrimaryKey)
+                                    {
+                                        hasPrimaryKeyColumn = true;
+                                    }
+                                    break;
+                                default:
+                                    throw new XmlException();
+                            }
+                            break;
+                        case XmlNodeType.EndElement:
+                            done = true;
+                            break;
+                    }
+                }
+
+                if (!unreal && !bootstrapperApplicationData && !hasPrimaryKeyColumn)
+                {
+                    throw new WixException(WixDataErrors.RealTableMissingPrimaryKeyColumn(SourceLineNumber.CreateFromUri(reader.BaseURI), name));
+                }
+
+                if (!done)
+                {
+                    throw new XmlException();
+                }
+            }
+
+            return new TableDefinition(name, columns, unreal, bootstrapperApplicationData);
+        }
+
+        /// <summary>
+        /// Persists an output in an XML format.
+        /// </summary>
+        /// <param name="writer">XmlWriter where the Output should persist itself as XML.</param>
+        internal void Write(XmlWriter writer)
+        {
+            writer.WriteStartElement("tableDefinition", TableDefinitionCollection.XmlNamespaceUri);
+
+            writer.WriteAttributeString("name", this.Name);
+
+            if (this.Unreal)
+            {
+                writer.WriteAttributeString("unreal", "yes");
+            }
+
+            if (this.BootstrapperApplicationData)
+            {
+                writer.WriteAttributeString("bootstrapperApplicationData", "yes");
+            }
+
+            foreach (ColumnDefinition columnDefinition in this.Columns)
+            {
+                columnDefinition.Write(writer);
+            }
+
+            writer.WriteEndElement();
+        }
+    }
+}