1 files changed, 86 insertions, 0 deletions

diff --git a/src/api/burn/WixToolset.Mba.Core/BundleInfo.cs b/src/api/burn/WixToolset.Mba.Core/BundleInfo.cs
new file mode 100644
index 00000000..3d5d535d
--- /dev/null
+++ b/src/api/burn/WixToolset.Mba.Core/BundleInfo.cs
@@ -0,0 +1,86 @@
+// 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.Mba.Core
+{
+    using System;
+    using System.Collections.Generic;
+    using System.IO;
+    using System.Xml;
+    using System.Xml.XPath;
+
+    /// <summary>
+    /// Default implementation of <see cref="IBundleInfo"/>.
+    /// </summary>
+    public class BundleInfo : IBundleInfo
+    {
+        /// <inheritdoc/>
+        public bool PerMachine { get; internal set; }
+
+        /// <inheritdoc/>
+        public string Name { get; internal set; }
+
+        /// <inheritdoc/>
+        public string LogVariable { get; internal set; }
+
+        /// <inheritdoc/>
+        public IDictionary<string, IPackageInfo> Packages { get; internal set; }
+
+        internal BundleInfo()
+        {
+            this.Packages = new Dictionary<string, IPackageInfo>();
+        }
+
+        /// <inheritdoc/>
+        public IPackageInfo AddRelatedBundleAsPackage(DetectRelatedBundleEventArgs e)
+        {
+            var package = PackageInfo.GetRelatedBundleAsPackage(e.ProductCode, e.RelationType, e.PerMachine, e.Version);
+            this.Packages.Add(package.Id, package);
+            return package;
+        }
+
+        /// <summary>
+        /// Parses BA manifest from the given stream.
+        /// </summary>
+        /// <param name="stream"></param>
+        /// <returns></returns>
+        public static IBundleInfo ParseBundleFromStream(Stream stream)
+        {
+            XPathDocument manifest = new XPathDocument(stream);
+            XPathNavigator root = manifest.CreateNavigator();
+            return ParseBundleFromXml(root);
+        }
+
+        /// <summary>
+        /// Parses BA manifest from the given <see cref="XPathNavigator"/>.
+        /// </summary>
+        /// <param name="root">The root of the BA manifest.</param>
+        /// <returns></returns>
+        public static IBundleInfo ParseBundleFromXml(XPathNavigator root)
+        {
+            BundleInfo bundle = new BundleInfo();
+
+            XmlNamespaceManager namespaceManager = new XmlNamespaceManager(root.NameTable);
+            namespaceManager.AddNamespace("p", BootstrapperApplicationData.XMLNamespace);
+            XPathNavigator bundleNode = root.SelectSingleNode("/p:BootstrapperApplicationData/p:WixBundleProperties", namespaceManager);
+
+            if (bundleNode == null)
+            {
+                throw new Exception("Failed to select bundle information.");
+            }
+
+            bool? perMachine = BootstrapperApplicationData.GetYesNoAttribute(bundleNode, "PerMachine");
+            if (perMachine.HasValue)
+            {
+                bundle.PerMachine = perMachine.Value;
+            }
+
+            bundle.Name = BootstrapperApplicationData.GetAttribute(bundleNode, "DisplayName");
+
+            bundle.LogVariable = BootstrapperApplicationData.GetAttribute(bundleNode, "LogPathVariable");
+
+            bundle.Packages = PackageInfo.ParsePackagesFromXml(root);
+
+            return bundle;
+        }
+    }
+}