2 files changed, 48 insertions, 1 deletions

diff --git a/src/test/dtf/EmbeddedUI/SetupWizard.xaml b/src/test/dtf/EmbeddedUI/SetupWizard.xaml
index 9fd493a7..97e406c2 100644
--- a/src/test/dtf/EmbeddedUI/SetupWizard.xaml
+++ b/src/test/dtf/EmbeddedUI/SetupWizard.xaml
@@ -8,6 +8,8 @@
         Title="Sample Embedded UI" Height="400" Width="540" Visibility="Visible">
         <Grid>
                 <TextBox Margin="8,8,8,63" Name="messagesTextBox" IsReadOnly="True" VerticalScrollBarVisibility="Visible" HorizontalScrollBarVisibility="Auto" FontFamily="Lucida Console" FontSize="10" />
+                <Button Height="23" HorizontalAlignment="Left" Name="repairButton" VerticalAlignment="Bottom" Width="75" Visibility="Hidden" Click="repairButton_Click" Margin="8,0,0,8">Repair</Button>
+                <Button Height="23" HorizontalAlignment="Left" Name="uninstallButton" VerticalAlignment="Bottom" Width="75" Visibility="Hidden" Click="uninstallButton_Click" Margin="91,0,0,8">Uninstall</Button>
                 <Button Height="23" HorizontalAlignment="Right" Name="installButton" VerticalAlignment="Bottom" Width="75" Click="installButton_Click" Margin="0,0,91,8">Install</Button>
                 <Button Height="23" HorizontalAlignment="Right" Name="exitButton" VerticalAlignment="Bottom" Width="75" Visibility="Hidden" Click="exitButton_Click" Margin="0,0,8,8">Exit</Button>
                 <Button Height="23" Margin="0,0,8,8" Name="cancelButton" VerticalAlignment="Bottom" Width="75" HorizontalAlignment="Right" Click="cancelButton_Click">Cancel</Button>
diff --git a/src/test/dtf/EmbeddedUI/SetupWizard.xaml.cs b/src/test/dtf/EmbeddedUI/SetupWizard.xaml.cs
index b846d61f..a4345481 100644
--- a/src/test/dtf/EmbeddedUI/SetupWizard.xaml.cs
+++ b/src/test/dtf/EmbeddedUI/SetupWizard.xaml.cs
@@ -16,19 +16,44 @@ namespace WixToolset.Samples.EmbeddedUI
     using System.Windows.Shapes;
     using WixToolset.Dtf.WindowsInstaller;
 
+    public enum SetupOperationType
+    {
+        Install,
+        Repair,
+        Uninstall
+    }
+
     /// <summary>
     /// Interaction logic for SetupWizard.xaml
     /// </summary>
     public partial class SetupWizard : Window
     {
+        private bool isMaintenance;
         private ManualResetEvent installStartEvent;
         private InstallProgressCounter progressCounter;
         private bool canceled;
 
-        public SetupWizard(ManualResetEvent installStartEvent)
+        public SetupOperationType Operation { get; private set; }
+
+        public SetupWizard(ManualResetEvent installStartEvent, bool isMaintenance)
         {
             this.installStartEvent = installStartEvent;
             this.progressCounter = new InstallProgressCounter(0.5);
+            this.isMaintenance = isMaintenance;
+
+            this.Loaded += this.SetupWizard_Loaded;
+        }
+
+        private void SetupWizard_Loaded(object sender, RoutedEventArgs e)
+        {
+            this.Loaded -= this.SetupWizard_Loaded;
+
+            if (this.isMaintenance)
+            {
+                this.installButton.Visibility = Visibility.Hidden;
+                this.repairButton.Visibility = Visibility.Visible;
+                this.uninstallButton.Visibility = Visibility.Visible;
+            }
         }
 
         public MessageResult ProcessMessage(InstallMessage messageType, Record messageRecord,
@@ -82,7 +107,27 @@ namespace WixToolset.Samples.EmbeddedUI
 
         private void installButton_Click(object sender, RoutedEventArgs e)
         {
+            this.Operation = SetupOperationType.Install;
+            this.StartInstall();
+        }
+
+        private void repairButton_Click(object sender, RoutedEventArgs e)
+        {
+            this.Operation = SetupOperationType.Repair;
+            this.StartInstall();
+        }
+
+        private void uninstallButton_Click(object sender, RoutedEventArgs e)
+        {
+            this.Operation = SetupOperationType.Uninstall;
+            this.StartInstall();
+        }
+
+        private void StartInstall()
+        {
             this.installButton.Visibility = Visibility.Hidden;
+            this.repairButton.Visibility = Visibility.Hidden;
+            this.uninstallButton.Visibility = Visibility.Hidden;
             this.progressBar.Visibility = Visibility.Visible;
             this.progressLabel.Visibility = Visibility.Visible;
             this.installStartEvent.Set();