1 files changed, 129 insertions, 0 deletions

diff --git a/src/WixToolset.WixBA/Model.cs b/src/WixToolset.WixBA/Model.cs
new file mode 100644
index 00000000..9f03e95b
--- /dev/null
+++ b/src/WixToolset.WixBA/Model.cs
@@ -0,0 +1,129 @@
+// 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.UX
+{
+    using System;
+    using System.Collections.Generic;
+    using System.Net;
+    using WixToolset.Bootstrapper;
+
+    /// <summary>
+    /// The model.
+    /// </summary>
+    public class Model
+    {
+        private const string BurnBundleInstallDirectoryVariable = "InstallFolder";
+        private const string BurnBundleLayoutDirectoryVariable = "WixBundleLayoutDirectory";
+        private const string BurnBundleVersionVariable = "WixBundleVersion";
+
+        /// <summary>
+        /// Creates a new model for the UX.
+        /// </summary>
+        /// <param name="bootstrapper">Bootstrapper hosting the UX.</param>
+        public Model(BootstrapperApplication bootstrapper)
+        {
+            this.Bootstrapper = bootstrapper;
+            this.Telemetry = new List<KeyValuePair<string, string>>();
+            this.Version = this.Engine.VersionVariables[BurnBundleVersionVariable];
+        }
+
+        /// <summary>
+        /// Gets the bootstrapper.
+        /// </summary>
+        public BootstrapperApplication Bootstrapper { get; private set; }
+
+        /// <summary>
+        /// Gets the bootstrapper command-line.
+        /// </summary>
+        public Command Command { get { return this.Bootstrapper.Command; } }
+
+        /// <summary>
+        /// Gets the bootstrapper engine.
+        /// </summary>
+        public Engine Engine { get { return this.Bootstrapper.Engine; } }
+
+        /// <summary>
+        /// Gets the key/value pairs used in telemetry.
+        /// </summary>
+        public List<KeyValuePair<string, string>> Telemetry { get; private set; }
+
+        /// <summary>
+        /// Get or set the final result of the installation.
+        /// </summary>
+        public int Result { get; set; }
+
+        /// <summary>
+        /// Get the version of the install.
+        /// </summary>
+        public Version Version { get; private set; }
+
+        /// <summary>
+        /// Get or set the path where the bundle is installed.
+        /// </summary>
+        public string InstallDirectory
+        {
+            get
+            {
+                if (!this.Engine.StringVariables.Contains(BurnBundleInstallDirectoryVariable))
+                {
+                    return null;
+                }
+
+                return this.Engine.StringVariables[BurnBundleInstallDirectoryVariable];
+            }
+
+            set
+            {
+                this.Engine.StringVariables[BurnBundleInstallDirectoryVariable] = value;
+            }
+        }
+
+        /// <summary>
+        /// Get or set the path for the layout to be created.
+        /// </summary>
+        public string LayoutDirectory
+        {
+            get
+            {
+                if (!this.Engine.StringVariables.Contains(BurnBundleLayoutDirectoryVariable))
+                {
+                    return null;
+                }
+
+                return this.Engine.StringVariables[BurnBundleLayoutDirectoryVariable];
+            }
+
+            set
+            {
+                this.Engine.StringVariables[BurnBundleLayoutDirectoryVariable] = value;
+            }
+        }
+
+        public LaunchAction PlannedAction { get; set; }
+
+        /// <summary>
+        /// Creates a correctly configured HTTP web request.
+        /// </summary>
+        /// <param name="uri">URI to connect to.</param>
+        /// <returns>Correctly configured HTTP web request.</returns>
+        public HttpWebRequest CreateWebRequest(string uri)
+        {
+            HttpWebRequest request = (HttpWebRequest)WebRequest.Create(uri);
+            request.UserAgent = String.Concat("WixInstall", this.Version.ToString());
+
+            return request;
+        }
+
+        /// <summary>
+        /// Gets the display name for a package if possible.
+        /// </summary>
+        /// <param name="packageId">Identity of the package to find the display name.</param>
+        /// <returns>Display name of the package if found or the package id if not.</returns>
+        public string GetPackageName(string packageId)
+        {
+            PackageInfo package;
+
+            return this.Bootstrapper.BAManifest.Bundle.Packages.TryGetValue(packageId, out package) ? package.DisplayName : packageId;
+        }
+    }
+}