From 1287b5e430ae991f3c22228b8a2add30c0005983 Mon Sep 17 00:00:00 2001
From: Raul Metsma <raul@metsma.ee>
Date: Sun, 31 Mar 2024 14:43:40 +0300
Subject: [PATCH] Merge shell extension installer to main installer

IB-7980

Signed-off-by: Raul Metsma <raul@metsma.ee>
---
 .github/workflows/build.yml                 |  1 -
 client/CMakeLists.txt                       |  3 +-
 extensions/windows/CMakeLists.txt           | 35 +---------------
 extensions/windows/EsteidShellExtension.wxs | 45 ---------------------
 qdigidoc4.wxs                               | 24 +++++++++++
 5 files changed, 27 insertions(+), 81 deletions(-)
 delete mode 100644 extensions/windows/EsteidShellExtension.wxs

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 1abf12ef1..023f077f3 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -199,7 +199,6 @@ jobs:
           -DCMAKE_TOOLCHAIN_FILE=${{ env.RUNVCPKG_VCPKG_ROOT }}/scripts/buildsystems/vcpkg.cmake `
           "-DLibDigiDocpp_ROOT=libs/PFiles64/libdigidocpp"
         cmake --build build --target msi
-        cmake --build build --target msishellext
         cmake --build build --target appx
     - name: Archive artifacts
       uses: actions/upload-artifact@v4
diff --git a/client/CMakeLists.txt b/client/CMakeLists.txt
index 9d9a4bdcd..7ced4bdfd 100644
--- a/client/CMakeLists.txt
+++ b/client/CMakeLists.txt
@@ -230,11 +230,12 @@ elseif(WIN32)
 		-d libs_path=${LIBS_PATH}
 		-d client_path=$<TARGET_FILE:${PROJECT_NAME}>
 		-d qt_suffix=$<$<CONFIG:Debug>:d>
+		-d ShellExt=$<TARGET_FILE:EsteidShellExtension>
 		${CMAKE_SOURCE_DIR}/${PROJECT_NAME}.wxs
 		${CMAKE_SOURCE_DIR}/cmake/modules/WelcomeDlg.wxs
 		${CMAKE_SOURCE_DIR}/cmake/modules/WixUI_Minimal.wxs
 	)
-	add_custom_target(msi DEPENDS ${PROJECT_NAME}
+	add_custom_target(msi DEPENDS ${PROJECT_NAME} EsteidShellExtension
 		COMMAND ${WIX_CMD} -o "${MSI_FILE}.msi"
 		#Build MSI with QT
 		COMMAND ${WIX_CMD} -d qt_path=${qtCore_install_prefix} -o "${MSI_FILE}.qt.msi"
diff --git a/extensions/windows/CMakeLists.txt b/extensions/windows/CMakeLists.txt
index fa67950b5..e69030a35 100644
--- a/extensions/windows/CMakeLists.txt
+++ b/extensions/windows/CMakeLists.txt
@@ -1,17 +1,10 @@
-cmake_minimum_required(VERSION 3.16)
 project(EsteidShellExtension VERSION 3.13.9)
-
-set(CMAKE_MODULE_PATH ${CMAKE_SOURCE_DIR}/cmake/modules)
-set(CMAKE_INCLUDE_CURRENT_DIR ON)
-
 include(VersionInfo)
 
 if(CMAKE_SIZEOF_VOID_P EQUAL 8)
     set(MIDL_TARGET "x64")
-    set(PLATFORM "x64")
 else()
     set(MIDL_TARGET "win32")
-    set(PLATFORM "x86")
 endif()
 
 add_custom_command(
@@ -37,41 +30,15 @@ set_target_properties(${PROJECT_NAME} PROPERTIES
     MSVC_RUNTIME_LIBRARY "MultiThreaded$<$<CONFIG:Debug>:Debug>"
     COMPILE_DEFINITIONS "_UNICODE;UNICODE;_WINDLL"
     INCLUDE_DIRECTORIES ${CMAKE_CURRENT_BINARY_DIR}
-    INTERPROCEDURAL_OPTIMIZATION YES
     COMPILE_OPTIONS "/guard:cf"
     LINK_OPTIONS "/guard:cf"
     LINK_LIBRARIES "uxtheme.lib"
     SKIP_AUTOMOC ON
 )
 
-add_custom_target(msishellext DEPENDS ${PROJECT_NAME}
-    COMMAND wix.exe build -nologo
-        -arch ${PLATFORM}
-        -d MSI_VERSION=${VERSION}
-        -d ShellExt=$<TARGET_FILE:EsteidShellExtension>
-        ${CMAKE_CURRENT_SOURCE_DIR}/EsteidShellExtension.wxs
-        ${CMAKE_MODULE_PATH}/WelcomeDlg.wxs
-        ${CMAKE_MODULE_PATH}/WixUI_Minimal.wxs
-        -ext WixToolset.UI.wixext
-        -bv WixUIDialogBmp=${CMAKE_MODULE_PATH}/dlgbmp.bmp
-        -bv WixUIBannerBmp=${CMAKE_MODULE_PATH}/banner.bmp
-        -o Digidoc_ShellExt-${VERSION}$ENV{VER_SUFFIX}.${PLATFORM}.msi
-    WORKING_DIRECTORY ${CMAKE_BINARY_DIR}
-)
-
 if(SIGNCERT)
-    if(CROSSSIGNCERT)
-        target_link_options(${PROJECT_NAME} PRIVATE "/INTEGRITYCHECK")
-    endif()
     add_custom_command(TARGET ${PROJECT_NAME} POST_BUILD
         COMMAND signtool.exe sign /a /v /s MY /n "${SIGNCERT}" /fd SHA256 /du http://installer.id.ee
-            "$<$<BOOL:${CROSSSIGNCERT}>:/ph;/ac;${CROSSSIGNCERT}>"
-            /tr http://sha256timestamp.ws.symantec.com/sha256/timestamp /td SHA256 $<TARGET_FILE:EsteidShellExtension>
-        COMMAND_EXPAND_LISTS
-    )
-    add_custom_command(TARGET msishellext POST_BUILD
-        COMMAND signtool.exe sign /a /v /s MY /n "${SIGNCERT}" /fd SHA256 /du http://installer.id.ee
-            /tr http://sha256timestamp.ws.symantec.com/sha256/timestamp /td SHA256
-            "${CMAKE_BINARY_DIR}/Digidoc_ShellExt-${VERSION}$ENV{VER_SUFFIX}.${PLATFORM}.msi"
+            /tr http://sha256timestamp.ws.symantec.com/sha256/timestamp /td SHA256 $<TARGET_FILE:${PROJECT_NAME}>
     )
 endif()
diff --git a/extensions/windows/EsteidShellExtension.wxs b/extensions/windows/EsteidShellExtension.wxs
deleted file mode 100644
index 4a3d80e1c..000000000
--- a/extensions/windows/EsteidShellExtension.wxs
+++ /dev/null
@@ -1,45 +0,0 @@
-<?if $(sys.BUILDARCH) = x64 ?>
-  <!-- Separate upgrade code / component ID created for x64 installer because both 32-
-       and 64-bit shell extensions are installed on x64 machine and x64 installer
-       removes x86 component installed directly before it if same guids are used.
-  -->
-  <?define ShellExtId = "{5606A547-759D-43DA-AEEB-D3BF1D1E816D}" ?>
-  <?define ShellUpgradeCode = "dc5723f9-386f-4581-a872-b9c823dfe40f" ?>
-  <?define ShellComponentId = "7adaa2cc-c3f5-4e2e-a38d-5d9c9730444e" ?>
-<?else?>
-  <?define ShellExtId = "{310AAB39-76FE-401B-8A7F-0F578C5F6AB5}" ?>
-  <?define ShellUpgradeCode = "8c4b5dae-dba9-4445-8584-3d156a48da27" ?>
-  <?define ShellComponentId = "00ad9482-2770-4dc7-b564-f68268d3f457" ?>
-<?endif?>
-
-<Wix xmlns="http://wixtoolset.org/schemas/v4/wxs"
-     xmlns:ui="http://wixtoolset.org/schemas/v4/wxs/ui">
-  <Package Name="EstEID Shell Extension" UpgradeCode="$(var.ShellUpgradeCode)"
-      Language="1033" Version="$(var.MSI_VERSION)" Codepage="1251" Manufacturer="RIA" InstallerVersion="500">
-    <MediaTemplate EmbedCab="yes" CompressionLevel="high" />
-    <MajorUpgrade AllowSameVersionUpgrades="yes" Schedule="afterInstallInitialize" DowngradeErrorMessage=
-       "A newer version of [ProductName] is already installed. If you are trying to downgrade, please uninstall the newer version first." />
-    <Launch Condition="Installed OR (VersionNT &gt;= 601)" Message="[ProductName] requires Windows 7 or higher." />
-    <ui:WixUI Id="WixUI_Minimal2" />
-
-    <StandardDirectory Id="ProgramFiles6432Folder">
-      <Directory Id="INSTALLFOLDER" Name="DigiDoc4 Client">
-        <Component Id="Application" Guid="$(var.ShellComponentId)">
-          <!--Does not work with multiple files -->
-          <!--RegistryValue Root='HKCR' Key='*\shell\[ProductName]' Type='string' Value='Sign with [ProductName]' />
-          <RegistryValue Root='HKCR' Key='*\shell\[ProductName]' Name="Icon" Type='string' Value='"[INSTALLFOLDER]qdigidoc4.exe",0' />
-          <RegistryValue Root='HKCR' Key='*\shell\[ProductName]\command' Type='string' Value='"[INSTALLFOLDER]qdigidoc4.exe" "%1"' /-->
-          <File Source="$(var.ShellExt)" />
-          <RegistryKey Root="HKCR" Key="CLSID\$(var.ShellExtId)\InprocServer32">
-            <RegistryValue Type="string" Value="[INSTALLFOLDER]EsteidShellExtension.dll" />
-            <RegistryValue Type="string" Name="ThreadingModel" Value="Apartment" />
-          </RegistryKey>
-          <RegistryValue Root="HKCR" Key="*\shellex\ContextMenuHandlers\DigiDoc4ShellExtension$(sys.BUILDARCH)"
-                         Value="$(var.ShellExtId)" Type="string" />
-          <RegistryValue Root="HKLM" Key="Software\Microsoft\Windows\CurrentVersion\Shell Extensions\Approved"
-                         Name="$(var.ShellExtId)" Value="DigiDoc4 Shell Extension" Type="string" />
-        </Component>
-      </Directory>
-    </StandardDirectory>
-  </Package>
-</Wix>
diff --git a/qdigidoc4.wxs b/qdigidoc4.wxs
index d7b257d67..01f564681 100644
--- a/qdigidoc4.wxs
+++ b/qdigidoc4.wxs
@@ -1,7 +1,13 @@
 <?if $(sys.BUILDARCH) = x64 ?>
   <?define OpenSSLSuffix = "-x64" ?>
+  <!-- Separate upgrade code / component ID created for x64 installer because both 32-
+       and 64-bit shell extensions are installed on x64 machine and x64 installer
+       removes x86 component installed directly before it if same guids are used.
+  -->
+  <?define ShellExtId = "{5606A547-759D-43DA-AEEB-D3BF1D1E816D}" ?>
 <?else?>
   <?define OpenSSLSuffix = "" ?>
+  <?define ShellExtId = "{310AAB39-76FE-401B-8A7F-0F578C5F6AB5}" ?>
 <?endif?>
 
 <?if $(env.VisualStudioVersion) = "17.0" ?>
@@ -24,6 +30,7 @@
     <Icon Id="qdigidoc4.ico" SourceFile="$(var.ico_path)" />
     <Property Id="ARPPRODUCTICON" Value="qdigidoc4.ico" />
     <Property Id="DESKTOP_SHORTCUT" Value="0" />
+    <Property Id="SHELL_EXTENSION" Value="1" />
     <MajorUpgrade AllowSameVersionUpgrades="yes" Schedule="afterInstallInitialize" DowngradeErrorMessage=
         "A newer version of [ProductName] is already installed. If you are trying to downgrade, please uninstall the newer version first." />
     <Launch Condition="Installed OR (VersionNT &gt;= 601)" Message="[ProductName] requires Windows 7 or higher." />
@@ -140,6 +147,23 @@
         <Directory Name="imageformats">
           <File Source="$(var.qt_path)\..\plugins\imageformats\qsvg$(var.qt_suffix).dll" />
         </Directory>
+<?endif?>
+<?ifdef var.ShellExt ?>
+        <Component Id="ShellExt" Guid="7adaa2cc-c3f5-4e2e-a38d-5d9c9730444e" Transitive="yes" Condition="SHELL_EXTENSION = 1">
+          <!--Does not work with multiple files -->
+          <!--RegistryValue Root="HKCR" Key="*\shell\[ProductName]" Type="string" Value="Sign with [ProductName]" />
+          <RegistryValue Root="HKCR" Key="*\shell\[ProductName]" Name="Icon" Type="string" Value=""[INSTALLFOLDER]qdigidoc4.exe",0" />
+          <RegistryValue Root="HKCR" Key="*\shell\[ProductName]\command" Type="string" Value=""[INSTALLFOLDER]qdigidoc4.exe" "%1"" /-->
+          <File Source="$(var.ShellExt)"/>
+          <RegistryKey Root="HKCR" Key="CLSID\$(var.ShellExtId)\InprocServer32">
+            <RegistryValue Type="string" Value="[INSTALLFOLDER]EsteidShellExtension.dll"/>
+            <RegistryValue Type="string" Name="ThreadingModel" Value="Apartment" />
+          </RegistryKey>
+          <RegistryValue Root="HKCR" Key="*\shellex\ContextMenuHandlers\DigiDoc4ShellExtension$(sys.BUILDARCH)"
+                         Value="$(var.ShellExtId)" Type="string" />
+          <RegistryValue Root="HKLM" Key="Software\Microsoft\Windows\CurrentVersion\Shell Extensions\Approved"
+                         Name="$(var.ShellExtId)" Value="DigiDoc4 Shell Extension" Type="string" />
+        </Component>
 <?endif?>
       </Directory>
     </StandardDirectory>