From 451641e466466785f4a0738934be40ce6222038a Mon Sep 17 00:00:00 2001
From: Paul Cioanca <paul.cioanca@supabase.io>
Date: Wed, 17 Jan 2024 14:02:28 +0200
Subject: [PATCH] chore: patch pg_cron perms post-pg_upgrade

---
 .../pg_upgrade_scripts/complete.sh            | 22 +++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/ansible/files/admin_api_scripts/pg_upgrade_scripts/complete.sh b/ansible/files/admin_api_scripts/pg_upgrade_scripts/complete.sh
index 7711f8403..b139091ed 100755
--- a/ansible/files/admin_api_scripts/pg_upgrade_scripts/complete.sh
+++ b/ansible/files/admin_api_scripts/pg_upgrade_scripts/complete.sh
@@ -48,6 +48,28 @@ EOF
 
         run_sql -c "$PG_NET_GRANT_QUERY"
     fi
+
+    # Patching pg_cron ownership as it resets during upgrade
+    HAS_PG_CRON_OWNED_BY_POSTGRES=$(run_sql -A -t -c "select count(*) > 0 from pg_extension where extname = 'pg_cron' and extowner::regrole::text = 'postgres';")
+
+    if [ "$HAS_PG_CRON_OWNED_BY_POSTGRES" = "t" ]; then
+        RECREATE_PG_CRON_QUERY=$(cat <<EOF
+        begin;
+        create temporary table cron_job as select * from cron.job;
+        create temporary table cron_job_run_details as select * from cron.job_run_details;
+        drop extension pg_cron;
+        create extension pg_cron schema pg_catalog;
+        insert into cron.job select * from cron_job;
+        insert into cron.job_run_details select * from cron_job_run_details;
+        select setval('cron.jobid_seq', coalesce(max(jobid), 0) + 1, false) from cron.job;
+        select setval('cron.runid_seq', coalesce(max(runid), 0) + 1, false) from cron.job_run_details;
+        update cron.job set username = 'postgres' where username = 'supabase_admin';
+        commit;
+EOF
+        )
+
+        run_sql -c "$RECREATE_PG_CRON_QUERY"
+    fi
 }
 
 function complete_pg_upgrade {