From deacd569ef03faee2d1e731d5b99b9f4dfa9d13a Mon Sep 17 00:00:00 2001
From: hustcer <hustcer@outlook.com>
Date: Wed, 7 May 2025 08:11:51 +0800
Subject: [PATCH] Add WindowsTerminalProfile feature for installation

---
 wix/main.wixproj |   1 +
 wix/main.wxs     | 121 +++++++++++++++++++++++++++++++++--------------
 2 files changed, 87 insertions(+), 35 deletions(-)

diff --git a/wix/main.wixproj b/wix/main.wixproj
index cab83fb431..77e48b12e7 100644
--- a/wix/main.wixproj
+++ b/wix/main.wixproj
@@ -29,6 +29,7 @@
     <BindPath Include="." />
     <Content Include="nu.ico" />
     <Content Include="License.rtf" />
+    <Content Include="windows-terminal-profile.json" />
   </ItemGroup>
 
 </Project>
diff --git a/wix/main.wxs b/wix/main.wxs
index df2504d6f3..88657b00f1 100644
--- a/wix/main.wxs
+++ b/wix/main.wxs
@@ -23,7 +23,7 @@
     <MajorUpgrade
       MigrateFeatures="yes"
       Schedule="afterInstallInitialize"
-      DowngradeErrorMessage="A newer version of [ProductName] is already installed. Setup will now exit."/>
+      DowngradeErrorMessage="A newer version of [ProductName] is already installed. Setup will now exit." />
 
     <!-- Embed cab media to MSI file -->
     <Media Id="1" Cabinet="cab1.cab" EmbedCab="yes" />
@@ -54,6 +54,31 @@
           <Directory Id="BINDIR_USER" Name="bin" />
         </Directory>
       </Directory>
+
+      <!-- Windows Terminal Profile Directories -->
+      <Directory Id="AppDataMicrosoftFolder" Name="Microsoft">
+        <Directory Id="AppDataWindowsTerminalFolder" Name="Windows Terminal">
+          <Directory Id="WindowsTerminalProfileFolder" Name="Fragments">
+            <Directory Id="WindowsTerminalProfileAppFolder" Name="nu">
+              <Component Id="WindowsTerminalProfile" Guid="957239F4-7B87-4399-9F91-7DF2ABE5ED8B">
+                <File Id="WindowsTerminalProfileFile"
+                      Name="nu.json"
+                      Source="$(var.ProjectDir)\windows-terminal-profile.json" />
+                <RegistryValue Root="HKCU"
+                               Key="Software\nu"
+                               Name="WindowsTerminalProfile"
+                               Value="1"
+                               Type="integer"
+                               KeyPath="yes" />
+                <RemoveFolder Id="RemoveWindowsTerminalProfileFolderA" Directory="WindowsTerminalProfileAppFolder" On="uninstall" />
+                <RemoveFolder Id="RemoveWindowsTerminalProfileFolderB" Directory="WindowsTerminalProfileFolder" On="uninstall" />
+                <RemoveFolder Id="RemoveWindowsTerminalProfileFolderC" Directory="AppDataWindowsTerminalFolder" On="uninstall" />
+                <RemoveFolder Id="RemoveWindowsTerminalProfileFolderD" Directory="AppDataMicrosoftFolder" On="uninstall" />
+              </Component>
+            </Directory>
+          </Directory>
+        </Directory>
+      </Directory>
     </StandardDirectory>
 
     <ComponentGroup Id="NushellBinaries" Directory="BINDIR">
@@ -92,35 +117,35 @@
       <!-- Per-user PATH component -->
       <Component Id="EnvironmentPathUser" Guid="{D6A3A7B2-1F3A-4B6A-8A3B-3A7B2D6A3A7B}" Condition="MSIINSTALLPERUSER=1">
         <Environment Id="PATHUser"
-                    Name="PATH"
-                    Value="[BINDIR]"
-                    Permanent="no"
-                    Part="last"
-                    Action="set"
-                    System="no"/>
+                     Name="PATH"
+                     Value="[BINDIR]"
+                     Permanent="no"
+                     Part="last"
+                     Action="set"
+                     System="no" />
         <RegistryValue Root="HKCU"
-                      Key="Software\nu"
-                      Name="installed"
-                      Type="integer"
-                      Value="1"
-                      KeyPath="yes"/>
+                       Key="Software\nu"
+                       Name="installed"
+                       Type="integer"
+                       Value="1"
+                       KeyPath="yes" />
       </Component>
 
       <!-- Per-machine PATH component -->
       <Component Id="EnvironmentPathMachine" Guid="{47781325-8A2B-4F8D-A058-9A2C4136E25F}" Condition="ALLUSERS=1">
         <Environment Id="PATHMachine"
-                    Name="PATH"
-                    Value="[BINDIR]"
-                    Permanent="no"
-                    Part="last"
-                    Action="set"
-                    System="yes"/>
+                     Name="PATH"
+                     Value="[BINDIR]"
+                     Permanent="no"
+                     Part="last"
+                     Action="set"
+                     System="yes" />
         <RegistryValue Root="HKLM"
-                      Key="Software\nu"
-                      Name="installed"
-                      Type="integer"
-                      Value="1"
-                      KeyPath="yes"/>
+                       Key="Software\nu"
+                       Name="installed"
+                       Type="integer"
+                       Value="1"
+                       KeyPath="yes" />
       </Component>
     </DirectoryRef>
 
@@ -132,6 +157,14 @@
       <ComponentRef Id="EnvironmentPathMachine" />
     </Feature>
 
+    <!-- Windows Terminal Profile Feature -->
+    <Feature Id="WindowsTerminalProfile"
+             Title="Windows Terminal Profile"
+             Description="Add $(var.ProductName) profile to Windows Terminal."
+             Level="1">
+      <ComponentRef Id="WindowsTerminalProfile" />
+    </Feature>
+
     <!-- Load Advanced UI -->
     <WixVariable Id="WixUILicenseRtf" Value="$(var.ProjectDir)\License.rtf" />
     <ui:WixUI Id="WixUI_Advanced" />
@@ -153,26 +186,44 @@
                  Value="[INSTALLDIR_USER]"
                  After="AppSearch"
                  Condition="MSIINSTALLPERUSER=1"
-                 Sequence="both"/>
+                 Sequence="both" />
     <SetProperty Id="BINDIR"
                  Action="SetBINDIR_User"
                  Value="[BINDIR_USER]"
                  After="AppSearch"
                  Condition="MSIINSTALLPERUSER=1"
-                 Sequence="both"/>
+                 Sequence="both" />
 
     <!-- If Installed with ALLUSERS=1 set the default Install dir to ProgramFiles64Folder -->
     <SetProperty Id="INSTALLDIR"
-             Action="SetINSTALLDIR_Machine"
-             Value="[ProgramFiles64Folder][ApplicationFolderName]"
-             After="AppSearch"
-             Condition="ALLUSERS=1"
-             Sequence="both"/>
+                 Action="SetINSTALLDIR_Machine"
+                 Value="[ProgramFiles64Folder][ApplicationFolderName]"
+                 After="AppSearch"
+                 Condition="ALLUSERS=1"
+                 Sequence="both" />
     <SetProperty Id="BINDIR"
-             Action="SetBINDIR_Machine"
-             Value="[ProgramFiles64Folder][ApplicationFolderName]\bin"
-             After="AppSearch"
-             Condition="ALLUSERS=1"
-             Sequence="both"/>
+                 Action="SetBINDIR_Machine"
+                 Value="[ProgramFiles64Folder][ApplicationFolderName]\bin"
+                 After="AppSearch"
+                 Condition="ALLUSERS=1"
+                 Sequence="both" />
+
+    <!-- Custom Action for Windows Terminal Profile -->
+    <SetProperty Id="ReplacePathsInWindowsTerminalProfile"
+                 Sequence="execute"
+                 Value="&quot;[#nu.exe]&quot; -c &quot;let doc = (open `[#WindowsTerminalProfileFile]` | update profiles.commandline `[#nu.exe]` | update profiles.icon `[#nu.ico]`); $doc | save -f `[#WindowsTerminalProfileFile]`&quot;"
+                 After="CostFinalize" />
+
+    <CustomAction Id="ReplacePathsInWindowsTerminalProfile"
+                  Return="check"
+                  Impersonate="yes"
+                  Execute="deferred"
+                  DllEntry="WixQuietExec"
+                  BinaryRef="Wix4UtilCA_X64" />
+
+    <InstallExecuteSequence>
+      <Custom Action="ReplacePathsInWindowsTerminalProfile" Before="InstallFinalize"
+              Condition="(&amp;WindowsTerminalProfile=3) OR ((!WindowsTerminalProfile=3) AND (REINSTALL&lt;&gt;&quot;&quot;))"/>
+    </InstallExecuteSequence>
   </Package>
 </Wix>