fix: update ext custom scripts to follow changes in Salt

Author: soedirgo

ansible/files/postgresql_extension_custom_scripts/pgmq/after-create.sql

@@ -1,6 +1,7 @@
 do $$
 declare
   extoid oid := (select oid from pg_extension where extname = 'pgmq');
+  extversion text := (select extversion from pg_extension where extname = 'pgmq');
   r record;
   cls pg_class%rowtype;
 begin
@@ -18,8 +19,13 @@ begin
     physical backups everywhere
 */
 -- Detach and delete the official function
-alter extension pgmq drop function pgmq.drop_queue(TEXT);
-drop function pgmq.drop_queue(TEXT);
+if extversion = '1.4.4' then
+  alter extension pgmq drop function pgmq.drop_queue;
+  drop function pgmq.drop_queue;
+else -- 1.5.1+
+  alter extension pgmq drop function pgmq.drop_queue(TEXT);
+  drop function pgmq.drop_queue(TEXT);
+end if;
 
 -- Create and reattach the patched function
 CREATE FUNCTION pgmq.drop_queue(queue_name TEXT)
@@ -134,7 +140,11 @@ BEGIN
 END;
 $func$ LANGUAGE plpgsql;
 
-alter extension pgmq add function pgmq.drop_queue(TEXT);
+if extversion = '1.4.4' then
+  alter extension pgmq add function pgmq.drop_queue;
+else -- 1.5.1+
+  alter extension pgmq add function pgmq.drop_queue(TEXT);
+end if;
 
 
   update pg_extension set extowner = 'postgres'::regrole where extname = 'pgmq';

ansible/files/postgresql_extension_custom_scripts/supabase_vault/after-create.sql

@@ -1,8 +1,17 @@
-grant usage on schema vault to postgres with grant option;
-grant select, delete, truncate, references on vault.secrets, vault.decrypted_secrets to postgres with grant option;
-grant execute on function vault.create_secret, vault.update_secret, vault._crypto_aead_det_decrypt to postgres with grant option;
+do $$
+declare
+  extversion text := (select extversion from pg_extension where extname = 'supabase_vault');
+begin
+  set local search_path = '';
 
--- service_role used to be able to manage secrets in Vault <=0.2.8 because it had privileges to pgsodium functions
-grant usage on schema vault to service_role;
-grant select, delete on vault.secrets, vault.decrypted_secrets to service_role;
-grant execute on function vault.create_secret, vault.update_secret, vault._crypto_aead_det_decrypt to service_role;
+  if extversion != '0.2.8' then
+    grant usage on schema vault to postgres with grant option;
+    grant select, delete, truncate, references on vault.secrets, vault.decrypted_secrets to postgres with grant option;
+    grant execute on function vault.create_secret, vault.update_secret, vault._crypto_aead_det_decrypt to postgres with grant option;
+
+    -- service_role used to be able to manage secrets in Vault <=0.2.8 because it had privileges to pgsodium functions
+    grant usage on schema vault to service_role;
+    grant select, delete on vault.secrets, vault.decrypted_secrets to service_role;
+    grant execute on function vault.create_secret, vault.update_secret, vault._crypto_aead_det_decrypt to service_role;
+  end if;
+end $$;