From feb3a8f896802ff274333012c3b399beb5c86f41 Mon Sep 17 00:00:00 2001
From: Lucas Fernandes Nogueira <lucasfernandesnog@gmail.com>
Date: Thu, 9 Dec 2021 01:27:11 -0300
Subject: [PATCH] feat(bundler): configure WiX to add launch option, closes
 #3015 (#3043)

---
 .changes/wix-run-app.md                               | 5 +++++
 tooling/bundler/src/bundle/windows/msi/wix.rs         | 2 ++
 tooling/bundler/src/bundle/windows/templates/main.wxs | 8 ++++++++
 3 files changed, 15 insertions(+)
 create mode 100644 .changes/wix-run-app.md

diff --git a/.changes/wix-run-app.md b/.changes/wix-run-app.md
new file mode 100644
index 000000000..30db1d3ff
--- /dev/null
+++ b/.changes/wix-run-app.md
@@ -0,0 +1,5 @@
+---
+"tauri-bundler": patch
+---
+
+Configure WiX to add an option to launch the application after finishing setup.
diff --git a/tooling/bundler/src/bundle/windows/msi/wix.rs b/tooling/bundler/src/bundle/windows/msi/wix.rs
index c72de28c7..5c8f0d685 100644
--- a/tooling/bundler/src/bundle/windows/msi/wix.rs
+++ b/tooling/bundler/src/bundle/windows/msi/wix.rs
@@ -330,6 +330,8 @@ fn run_light(
   let mut args: Vec<String> = vec![
     "-ext".to_string(),
     "WixUIExtension".to_string(),
+    "-ext".to_string(),
+    "WixUtilExtension".to_string(),
     "-o".to_string(),
     output_path.display().to_string(),
   ];
diff --git a/tooling/bundler/src/bundle/windows/templates/main.wxs b/tooling/bundler/src/bundle/windows/templates/main.wxs
index 50c2004ec..132f5150b 100644
--- a/tooling/bundler/src/bundle/windows/templates/main.wxs
+++ b/tooling/bundler/src/bundle/windows/templates/main.wxs
@@ -46,7 +46,15 @@
         <Property Id="ARPNOREPAIR" Value="yes" Secure="yes" />      <!-- Remove repair -->
         <SetProperty Id="ARPNOMODIFY" Value="1" After="InstallValidate" Sequence="execute"/>
 
+        <!-- launch app checkbox -->
+        <Property Id="WIXUI_EXITDIALOGOPTIONALCHECKBOXTEXT" Value="Launch {{{product_name}}}" />
+        <Property Id="WixShellExecTarget" Value="{{{app_exe_source}}}" />
+        <CustomAction Id="LaunchApplication" BinaryKey="WixCA" DllEntry="WixShellExec" Impersonate="yes" />
+
         <UI>
+            <!-- launch app checkbox -->
+            <Publish Dialog="ExitDialog" Control="Finish" Event="DoAction" Value="LaunchApplication">WIXUI_EXITDIALOGOPTIONALCHECKBOX = 1 and NOT Installed</Publish>
+
             <Property Id="WIXUI_INSTALLDIR" Value="INSTALLDIR" />
 
             {{#unless license}}