From 5f2813aa84d6d5d2938937b81bf0783dfe91a974 Mon Sep 17 00:00:00 2001
From: Samir Jha <samirjha1525@gmail.com>
Date: Tue, 6 Aug 2024 13:01:43 -0400
Subject: [PATCH] Fixes #37717 - Update evr extension ownership to foreman

---
 .../pre/35-change-evr-extension-ownership.rb  | 26 +++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 hooks/pre/35-change-evr-extension-ownership.rb

diff --git a/hooks/pre/35-change-evr-extension-ownership.rb b/hooks/pre/35-change-evr-extension-ownership.rb
new file mode 100644
index 00000000..81bdb39a
--- /dev/null
+++ b/hooks/pre/35-change-evr-extension-ownership.rb
@@ -0,0 +1,26 @@
+# In Katello 4.16, the 'evr' extension is removed from PostgreSQL and integrated into the Katello database via a migration.
+# This hook ensures the 'evr' extension's ownership is transferred to the 'foreman' user so migrations can act on it.
+
+if local_postgresql? && execute("rpm -q postgresql-evr", false, false)
+  is_postgresql_active = execute_command("systemctl is-active postgresql", false, true)&.last&.exitstatus == 0
+
+  # Ensure the PostgreSQL service is running
+  unless is_postgresql_active
+    logger.debug("Starting postgresql service")
+    start_services(['postgresql']) unless app_value(:noop)
+  end
+
+  # Update the ownership of the evr extension
+  logger.debug("Updating ownership of the evr extension if it is enabled")
+  database = param_value('foreman', 'db_database') || 'foreman'
+  username = param_value('foreman', 'db_username') || 'foreman'
+  sql = "psql -d '#{database}' -c \\\"UPDATE pg_extension SET extowner = (SELECT oid FROM pg_authid WHERE rolname='#{username}') WHERE extname='evr';\\\""
+  logger.debug("Executing: #{sql}")
+  execute_as!('postgres', sql, false, true) unless app_value(:noop)
+
+  # Stop the PostgreSQL service if it was not running
+  unless is_postgresql_active
+    logger.debug("Stopping postgresql service")
+    stop_services(['postgresql']) unless app_value(:noop)
+  end
+end