From 939d3d6a93b670f5ca9e0abf8b6f1850f36fb79c Mon Sep 17 00:00:00 2001
From: Kentaro Hayashi <hayashi@clear-code.com>
Date: Thu, 21 Dec 2023 15:43:27 +0900
Subject: [PATCH] msi: keep fluentdopt when previous options is set

Before:

* always reset fluentdopt

After:

* If fluentdopt is empty, set default value
* If fluentdopt configuration is changed from default one, keep it.

Signed-off-by: Kentaro Hayashi <hayashi@clear-code.com>
---
 fluent-package/Rakefile                       |  1 +
 .../fluent-package-post-fluentdwinsvc.rb      | 37 +++++++++++++++++++
 fluent-package/msi/source.wxs                 | 18 ++++++++-
 3 files changed, 54 insertions(+), 2 deletions(-)
 create mode 100644 fluent-package/msi/assets/fluent-package-post-fluentdwinsvc.rb

diff --git a/fluent-package/Rakefile b/fluent-package/Rakefile
index 759830c8e..536ec994d 100755
--- a/fluent-package/Rakefile
+++ b/fluent-package/Rakefile
@@ -539,6 +539,7 @@ class BuildTask
         cp("msi/assets/#{PACKAGE_NAME}-prompt.bat", fluent_package_staging_dir)
         cp("msi/assets/#{PACKAGE_NAME}-post-install.bat", staging_bindir)
         cp("msi/assets/#{PACKAGE_NAME}-post-migration.bat", staging_bindir)
+        cp("msi/assets/#{PACKAGE_NAME}-post-fluentdwinsvc.rb", staging_bindir)
         cp("msi/assets/#{SERVICE_NAME}.bat", fluent_package_staging_dir)
         cp("msi/assets/fluent-gem.bat", fluent_package_staging_dir)
         cp("msi/assets/#{PACKAGE_NAME}-version.rb", staging_bindir)
diff --git a/fluent-package/msi/assets/fluent-package-post-fluentdwinsvc.rb b/fluent-package/msi/assets/fluent-package-post-fluentdwinsvc.rb
new file mode 100644
index 000000000..196e16a91
--- /dev/null
+++ b/fluent-package/msi/assets/fluent-package-post-fluentdwinsvc.rb
@@ -0,0 +1,37 @@
+require 'fileutils'
+require "win32/service"
+require "win32/registry"
+require "optparse"
+include Win32
+
+install_dir = "#{ARGV[0]}"
+
+default_fluentdwinsvc = "-c '#{install_dir}etc\\fluent\\fluentd.conf' -o '#{install_dir}fluentd.log'"
+registry_key = "SYSTEM\\CurrentControlSet\\Services\\fluentdwinsvc"
+
+puts("fluentdwinsvc default: #{default_fluentdwinsvc}")
+begin
+  Win32::Registry::HKEY_LOCAL_MACHINE.open(registry_key, Win32::Registry::KEY_ALL_ACCESS) do |reg|
+    # Check whether RegValue exists or not
+    puts("fluentdwinsvc registry key was opened: #{registry_key}")
+    begin
+      previous_fluentdopt = reg['fluentdopt', Win32::Registry::REG_SZ]
+      puts("fluentdwinsvc current value: #{previous_fluentdopt}")
+      if previous_fluentdopt != default_fluentdwinsvc
+        puts("fluentdwinsvc: fluentdopt configuration was kept because option was changed from default")
+      else
+        puts("fluentdwinsvc: fluentdopt configuration was same as default one: #{default_fluentdwinsvc}")
+      end
+    rescue Win32::Registry::Error
+      # As fluentdopt value does not exist, then set default configuration
+      puts("fluentdwinsvc: reset to default fluentdopt configuration")
+      reg['fluentdopt', Win32::Registry::REG_SZ] = default_fluentdwinsvc
+    end
+  end
+rescue Win32::Registry::Error
+  # No fluentdwinsvc key yet.
+  puts("fluentdwinsvc: create #{registory_key} and set default fluentdopt configuration")
+  Win32::Registry::HKEY_LOCAL_MACHINE.create(registory_key, Win32::Registry::KEY_ALL_ACCESS) do |reg|
+    reg['fluentdopt', Win32::Registry::REG_SZ] = default_fluentdwinsvc
+  end
+end
diff --git a/fluent-package/msi/source.wxs b/fluent-package/msi/source.wxs
index faeb7f1b7..4f3a1a372 100644
--- a/fluent-package/msi/source.wxs
+++ b/fluent-package/msi/source.wxs
@@ -185,7 +185,7 @@
     <Property Id="InstallFluentdWinSvc" Value=" "/>
     <CustomAction Id="SetInstallFluentdWinSvcCommand"
                   Property="InstallFluentdWinSvc"
-                  Value="&quot;[FLUENTPROJECTLOCATION]fluentd.bat&quot; --reg-winsvc i --reg-winsvc-delay-start --reg-winsvc-fluentdopt &quot;-c '[FLUENTPROJECTLOCATION]etc\fluent\fluentd.conf' -o '[FLUENTPROJECTLOCATION]fluentd.log'&quot;"/>
+                  Value="&quot;[FLUENTPROJECTLOCATION]fluentd.bat&quot; --reg-winsvc i --reg-winsvc-delay-start"/>
     <CustomAction Id="InstallFluentdWinSvc"
                   BinaryKey="WixCA"
                   DllEntry="WixQuietExec64"
@@ -193,6 +193,17 @@
                   Return="check"
                   Impersonate="no" />
 
+    <Property Id="InstallFluentdWinSvcFluentdopt" Value=" "/>
+    <CustomAction Id="SetInstallFluentdWinSvcFluentdoptCommand"
+                  Property="InstallFluentdWinSvcFluentdopt"
+                  Value="&quot;[FLUENTPROJECTLOCATION]bin\ruby.exe&quot; &quot;[FLUENTPROJECTLOCATION]bin\fluent-package-post-fluentdwinsvc.rb&quot; &quot;[FLUENTPROJECTLOCATION]&quot;"/>
+    <CustomAction Id="InstallFluentdWinSvcFluentdopt"
+                  BinaryKey="WixCA"
+                  DllEntry="WixQuietExec64"
+                  Execute="deferred"
+                  Return="check"
+                  Impersonate="no" />
+
     <Property Id="CreateCompatTdAgentBat" Value=" "/>
     <CustomAction Id="SetCreateCompatTdAgentBat"
                   Property="CreateCompatTdAgentBat"
@@ -242,9 +253,12 @@
       <Custom Action="SetPostMigrationCommand" Before="PostMigration">WIX_UPGRADE_DETECTED</Custom>
       <Custom Action="PostMigration" Before="StartServices">WIX_UPGRADE_DETECTED</Custom>
 
-      <Custom Action="SetInstallFluentdWinSvcCommand" After="InstallFiles">NOT Installed</Custom>
+      <Custom Action="SetInstallFluentdWinSvcCommand" After="PostInstall">NOT Installed</Custom>
       <Custom Action="InstallFluentdWinSvc" After="SetInstallFluentdWinSvcCommand">NOT Installed</Custom>
 
+      <Custom Action="SetInstallFluentdWinSvcFluentdoptCommand" After="InstallFluentdWinSvc">NOT Installed OR WIX_UPGRADE_DETECTED</Custom>
+      <Custom Action="InstallFluentdWinSvcFluentdopt" After="SetInstallFluentdWinSvcFluentdoptCommand">NOT Installed OR WIX_UPGRADE_DETECTED</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>