From 75fb5596d76a0c054b3066f702dce47af12b2b68 Mon Sep 17 00:00:00 2001
From: Shizuo Fujita <fujita@clear-code.com>
Date: Mon, 27 Jan 2025 19:18:38 +0900
Subject: [PATCH] windows: backup and restore registry with update install

Signed-off-by: Shizuo Fujita <fujita@clear-code.com>
---
 fluent-package/msi/source.wxs | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/fluent-package/msi/source.wxs b/fluent-package/msi/source.wxs
index faeb7f1b7..73c6b72a5 100644
--- a/fluent-package/msi/source.wxs
+++ b/fluent-package/msi/source.wxs
@@ -235,7 +235,31 @@
                   Return="check"
                   Impersonate="no" />
 
+    <Property Id="BackupRegistryKey" Value=" "/>
+    <CustomAction Id="SetBackupRegistryKeyCommand"
+                  Property="BackupRegistryKey"
+                  Value="&quot;cmd&quot; /c &quot;chcp 437 &amp;&amp; reg export HKLM\SYSTEM\CurrentControlSet\Services\fluentdwinsvc C:/fluentdwinsvc.reg&quot;"/>
+    <CustomAction Id="BackupRegistryKey"
+                  BinaryKey="WixCA"
+                  DllEntry="WixQuietExec64"
+                  Execute="deferred"
+                  Return="ignore"
+                  Impersonate="no" />
+    <Property Id="RestoreRegistryKey" Value=" "/>
+    <CustomAction Id="SetRestoreRegistryKeyCommand"
+                  Property="RestoreRegistryKey"
+                  Value="&quot;cmd&quot; /c &quot;reg import C:/fluentdwinsvc.reg&quot;"/>
+    <CustomAction Id="RestoreRegistryKey"
+                  BinaryKey="WixCA"
+                  DllEntry="WixQuietExec64"
+                  Execute="deferred"
+                  Return="ignore"
+                  Impersonate="no" />
+
     <InstallExecuteSequence>
+      <Custom Action="SetBackupRegistryKeyCommand" Before="BackupRegistryKey">REMOVE="ALL"</Custom>
+      <Custom Action="BackupRegistryKey" Before="StopServices">REMOVE="ALL"</Custom>
+
       <Custom Action="SetPostInstallCommand" After="InstallFiles">NOT Installed</Custom>
       <Custom Action="PostInstall" After="SetPostInstallCommand">NOT Installed</Custom>
 
@@ -245,6 +269,9 @@
       <Custom Action="SetInstallFluentdWinSvcCommand" After="InstallFiles">NOT Installed</Custom>
       <Custom Action="InstallFluentdWinSvc" After="SetInstallFluentdWinSvcCommand">NOT Installed</Custom>
 
+      <Custom Action="SetRestoreRegistryKeyCommand" After="InstallFluentdWinSvc">NOT Installed</Custom>
+      <Custom Action="RestoreRegistryKey" After="SetRestoreRegistryKeyCommand">NOT Installed</Custom>
+
       <Custom Action="SetCreateCompatTdAgentBat" Before="InstallFinalize">NOT Installed</Custom>
       <Custom Action="CreateCompatTdAgentBat" After="SetCreateCompatTdAgentBat">NOT Installed</Custom>
       <Custom Action="SetCreateCompatTdAgentGemBat" Before="InstallFinalize">NOT Installed</Custom>