WIXFEAT:6164 Add OnPlanMsiPackage.

3 files changed, 214 insertions, 3 deletions

diff --git a/src/WixToolset.Mba.Core/BootstrapperApplication.cs b/src/WixToolset.Mba.Core/BootstrapperApplication.cs
index 249c17c9..9ee909dc 100644
--- a/src/WixToolset.Mba.Core/BootstrapperApplication.cs
+++ b/src/WixToolset.Mba.Core/BootstrapperApplication.cs
@@ -162,6 +162,11 @@ namespace WixToolset.Mba.Core
         public event EventHandler<PlanMsiFeatureEventArgs> PlanMsiFeature;
 
         /// <summary>
+        /// Fired when the engine is planning an MSI or MSP package.
+        /// </summary>
+        public event EventHandler<PlanMsiPackageEventArgs> PlanMsiPackage;
+
+        /// <summary>
         /// Fired when the engine has completed planning the installation of a specific package.
         /// </summary>
         public event EventHandler<PlanPackageCompleteEventArgs> PlanPackageComplete;
@@ -659,6 +664,19 @@ namespace WixToolset.Mba.Core
         }
 
         /// <summary>
+        /// Called when the engine is planning an MSI or MSP package.
+        /// </summary>
+        /// <param name="args">Additional arguments for this event.</param>
+        protected virtual void OnPlanMsiPackage(PlanMsiPackageEventArgs args)
+        {
+            EventHandler<PlanMsiPackageEventArgs> handler = this.PlanMsiPackage;
+            if (null != handler)
+            {
+                handler(this, args);
+            }
+        }
+
+        /// <summary>
         /// Called when then engine has completed planning the installation of a specific package.
         /// </summary>
         /// <param name="args">Additional arguments for this event.</param>
@@ -1288,6 +1306,18 @@ namespace WixToolset.Mba.Core
             return args.HResult;
         }
 
+        int IBootstrapperApplication.OnPlanMsiPackage(string wzPackageId, bool fExecute, ActionState action, ref bool fCancel, ref BURN_MSI_PROPERTY actionMsiProperty, ref INSTALLUILEVEL uiLevel, ref bool fDisableExternalUiHandler)
+        {
+            PlanMsiPackageEventArgs args = new PlanMsiPackageEventArgs(wzPackageId, fExecute, action, fCancel, actionMsiProperty, uiLevel, fDisableExternalUiHandler);
+            this.OnPlanMsiPackage(args);
+
+            fCancel = args.Cancel;
+            actionMsiProperty = args.ActionMsiProperty;
+            uiLevel = args.UiLevel;
+            fDisableExternalUiHandler = args.DisableExternalUiHandler;
+            return args.HResult;
+        }
+
         int IBootstrapperApplication.OnPlanPackageComplete(string wzPackageId, int hrStatus, PackageState state, RequestState requested, ActionState execute, ActionState rollback)
         {
             var args = new PlanPackageCompleteEventArgs(wzPackageId, hrStatus, state, requested, execute, rollback);
@@ -1466,9 +1496,9 @@ namespace WixToolset.Mba.Core
             return args.HResult;
         }
 
-        int IBootstrapperApplication.OnExecutePackageBegin(string wzPackageId, bool fExecute, ActionState action, ref bool fCancel)
+        int IBootstrapperApplication.OnExecutePackageBegin(string wzPackageId, bool fExecute, ActionState action, INSTALLUILEVEL uiLevel, bool fDisableExternalUiHandler, ref bool fCancel)
         {
-            ExecutePackageBeginEventArgs args = new ExecutePackageBeginEventArgs(wzPackageId, fExecute, action, fCancel);
+            ExecutePackageBeginEventArgs args = new ExecutePackageBeginEventArgs(wzPackageId, fExecute, action, uiLevel, fDisableExternalUiHandler, fCancel);
             this.OnExecutePackageBegin(args);
 
             fCancel = args.Cancel;
diff --git a/src/WixToolset.Mba.Core/EventArgs.cs b/src/WixToolset.Mba.Core/EventArgs.cs
index ca0fa173..e739a853 100644
--- a/src/WixToolset.Mba.Core/EventArgs.cs
+++ b/src/WixToolset.Mba.Core/EventArgs.cs
@@ -996,6 +996,64 @@ namespace WixToolset.Mba.Core
     }
 
     /// <summary>
+    /// Additional arguments used when the engine is planning an MSI or MSP package.
+    /// </summary>
+    [Serializable]
+    public class PlanMsiPackageEventArgs : CancellableHResultEventArgs
+    {
+        /// <summary>
+        /// Creates a new instance of the <see cref="PlanMsiPackageEventArgs"/> class.
+        /// </summary>
+        /// <param name="packageId">The identity of the package planned for.</param>
+        /// <param name="shouldExecute">Whether the package is planned to execute or roll back.</param>
+        /// <param name="action">The action planned for the package.</param>
+        /// <param name="cancelRecommendation">The recommendation from the engine.</param>
+        /// <param name="actionMsiProperty">The requested MSI property to add.</param>
+        /// <param name="uiLevel">The requested internal UI level.</param>
+        /// <param name="disableExternalUiHandler">Whether Burn is requested to set up an external UI handler.</param>
+        public PlanMsiPackageEventArgs(string packageId, bool shouldExecute, ActionState action, bool cancelRecommendation, BURN_MSI_PROPERTY actionMsiProperty, INSTALLUILEVEL uiLevel, bool disableExternalUiHandler)
+            : base(cancelRecommendation)
+        {
+            this.PackageId = packageId;
+            this.ShouldExecute = shouldExecute;
+            this.Action = action;
+            this.ActionMsiProperty = actionMsiProperty;
+            this.UiLevel = uiLevel;
+            this.DisableExternalUiHandler = disableExternalUiHandler;
+        }
+
+        /// <summary>
+        /// Gets identity of the package planned for.
+        /// </summary>
+        public string PackageId { get; private set; }
+
+        /// <summary>
+        /// Gets whether the package is planned to execute or roll back.
+        /// </summary>
+        public bool ShouldExecute { get; private set; }
+
+        /// <summary>
+        /// Gets the action planned for the package.
+        /// </summary>
+        public ActionState Action { get; private set; }
+
+        /// <summary>
+        /// Gets or sets the requested MSI property to add.
+        /// </summary>
+        public BURN_MSI_PROPERTY ActionMsiProperty { get; set; }
+
+        /// <summary>
+        /// Gets or sets the requested internal UI level.
+        /// </summary>
+        public INSTALLUILEVEL UiLevel { get; set; }
+
+        /// <summary>
+        /// Gets or sets whether Burn is requested to set up an external UI handler.
+        /// </summary>
+        public bool DisableExternalUiHandler { get; set; }
+    }
+
+    /// <summary>
     /// Additional arguments used when then engine has completed planning the installation of a specific package.
     /// </summary>
     [Serializable]
@@ -1503,13 +1561,17 @@ namespace WixToolset.Mba.Core
         /// <param name="packageId">The identity of the package to act on.</param>
         /// <param name="shouldExecute">Whether the package is being executed or rolled back.</param>
         /// <param name="action">The action about to be executed.</param>
+        /// <param name="uiLevel">The internal UI level (if this is an MSI or MSP package).</param>
+        /// <param name="disableExternalUiHandler">Whether Burn will set up an external UI handler (if this is an MSI or MSP package).</param>
         /// <param name="cancelRecommendation">The recommendation from the engine.</param>
-        public ExecutePackageBeginEventArgs(string packageId, bool shouldExecute, ActionState action, bool cancelRecommendation)
+        public ExecutePackageBeginEventArgs(string packageId, bool shouldExecute, ActionState action, INSTALLUILEVEL uiLevel, bool disableExternalUiHandler, bool cancelRecommendation)
             : base(cancelRecommendation)
         {
             this.PackageId = packageId;
             this.ShouldExecute = shouldExecute;
             this.Action = action;
+            this.UiLevel = uiLevel;
+            this.DisableExternalUiHandler = disableExternalUiHandler;
         }
 
         /// <summary>
@@ -1526,6 +1588,16 @@ namespace WixToolset.Mba.Core
         /// Gets the action about to be executed.
         /// </summary>
         public ActionState Action { get; private set; }
+
+        /// <summary>
+        /// Gets the internal UI level (if this is an MSI or MSP package).
+        /// </summary>
+        public INSTALLUILEVEL UiLevel { get; private set; }
+
+        /// <summary>
+        /// Gets whether Burn will set up an external UI handler (if this is an MSI or MSP package).
+        /// </summary>
+        public bool DisableExternalUiHandler { get; private set; }
     }
 
     /// <summary>
diff --git a/src/WixToolset.Mba.Core/IBootstrapperApplication.cs b/src/WixToolset.Mba.Core/IBootstrapperApplication.cs
index c4daaf32..fa655282 100644
--- a/src/WixToolset.Mba.Core/IBootstrapperApplication.cs
+++ b/src/WixToolset.Mba.Core/IBootstrapperApplication.cs
@@ -221,6 +221,18 @@ namespace WixToolset.Mba.Core
 
         [PreserveSig]
         [return: MarshalAs(UnmanagedType.I4)]
+        int OnPlanMsiPackage(
+            [MarshalAs(UnmanagedType.LPWStr)] string wzPackageId,
+            [MarshalAs(UnmanagedType.Bool)] bool fExecute,
+            [MarshalAs(UnmanagedType.U4)] ActionState action,
+            [MarshalAs(UnmanagedType.Bool)] ref bool fCancel,
+            [MarshalAs(UnmanagedType.U4)] ref BURN_MSI_PROPERTY actionMsiProperty,
+            [MarshalAs(UnmanagedType.U4)] ref INSTALLUILEVEL uiLevel,
+            [MarshalAs(UnmanagedType.Bool)] ref bool fDisableExternalUiHandler
+            );
+
+        [PreserveSig]
+        [return: MarshalAs(UnmanagedType.I4)]
         int OnPlanPackageComplete(
             [MarshalAs(UnmanagedType.LPWStr)] string wzPackageId,
             int hrStatus,
@@ -393,6 +405,8 @@ namespace WixToolset.Mba.Core
             [MarshalAs(UnmanagedType.LPWStr)] string wzPackageId,
             [MarshalAs(UnmanagedType.Bool)] bool fExecute,
             [MarshalAs(UnmanagedType.U4)] ActionState action,
+            [MarshalAs(UnmanagedType.U4)] INSTALLUILEVEL uiLevel,
+            [MarshalAs(UnmanagedType.Bool)] bool fDisableExternalUiHandler,
             [MarshalAs(UnmanagedType.Bool)] ref bool fCancel
             );
 
@@ -832,4 +846,99 @@ namespace WixToolset.Mba.Core
         Restart,
         ReloadBootstrapper,
     }
+
+    /// <summary>
+    /// The property Burn will add so the MSI can know the planned action for the package.
+    /// </summary>
+    public enum BURN_MSI_PROPERTY
+    {
+        /// <summary>
+        /// No property will be added.
+        /// </summary>
+        None,
+
+        /// <summary>
+        /// Add BURNMSIINSTALL=1
+        /// </summary>
+        Install,
+
+        /// <summary>
+        /// Add BURNMSIMODFIY=1
+        /// </summary>
+        Modify,
+
+        /// <summary>
+        /// Add BURNMSIREPAIR=1
+        /// </summary>
+        Repair,
+
+        /// <summary>
+        /// Add BURNMSIUNINSTALL=1
+        /// </summary>
+        Uninstall,
+    }
+
+    /// <summary>
+    /// From msi.h
+    /// https://docs.microsoft.com/en-us/windows/win32/api/msi/nf-msi-msisetinternalui
+    /// </summary>
+    [Flags]
+    public enum INSTALLUILEVEL
+    {
+        /// <summary>
+        /// UI level is unchanged
+        /// </summary>
+        NoChange = 0,
+
+        /// <summary>
+        /// default UI is used
+        /// </summary>
+        Default = 1,
+
+        /// <summary>
+        /// completely silent installation
+        /// </summary>
+        None = 2,
+
+        /// <summary>
+        /// simple progress and error handling
+        /// </summary>
+        Basic = 3,
+
+        /// <summary>
+        /// authored UI, wizard dialogs suppressed
+        /// </summary>
+        Reduced = 4,
+
+        /// <summary>
+        /// authored UI with wizards, progress, errors
+        /// </summary>
+        Full = 5,
+
+        /// <summary>
+        /// display success/failure dialog at end of install
+        /// </summary>
+        EndDialog = 0x80,
+
+        /// <summary>
+        /// display only progress dialog
+        /// </summary>
+        ProgressOnly = 0x40,
+
+        /// <summary>
+        /// do not display the cancel button in basic UI
+        /// </summary>
+        HideCancel = 0x20,
+
+        /// <summary>
+        /// force display of source resolution even if quiet
+        /// </summary>
+        SourceResOnly = 0x100,
+
+        /// <summary>
+        /// show UAC prompt even if quiet
+        /// Can only be used if on Windows Installer 5.0 or later
+        /// </summary>
+        UacOnly = 0x200,
+    }
 }