fix: Fix functions return types

Author: whiterabbit1983

memory-store/migrations/000042_add_api_keys.down.sql

@@ -1,13 +1,9 @@
 BEGIN;
 
--- Drop helper functions (updated signatures due to composite type returns)
+-- Drop helper functions (now using RETURNS TABLE syntax)
 DROP FUNCTION IF EXISTS decrypt_api_key(BYTEA, TEXT);
 DROP FUNCTION IF EXISTS encrypt_api_key(TEXT, TEXT);
 
--- Drop composite types
-DROP TYPE IF EXISTS decrypted_api_key_result;
-DROP TYPE IF EXISTS encrypted_api_key_result;
-
 -- Drop trigger and function
 DROP TRIGGER IF EXISTS update_api_keys_timestamp_trigger ON api_keys;
 DROP FUNCTION IF EXISTS update_api_keys_timestamp();

memory-store/migrations/000042_add_api_keys.up.sql

@@ -42,55 +42,32 @@ CREATE INDEX IF NOT EXISTS idx_api_keys_name ON api_keys(name);
 CREATE INDEX IF NOT EXISTS idx_api_keys_metadata ON api_keys USING gin(metadata);
 CREATE INDEX IF NOT EXISTS idx_api_keys_deleted_at ON api_keys(deleted_at) WHERE deleted_at IS NULL;
 
--- Create composite types for function return values (so they can be tracked by Hasura)
-DO $$
-BEGIN
-    IF NOT EXISTS (
-        SELECT 1 FROM pg_type WHERE typname = 'encrypted_api_key_result'
-    ) THEN
-        CREATE TYPE encrypted_api_key_result AS (
-            encrypted_value BYTEA
-        );
-    END IF;
-END $$;
-
-DO $$
-BEGIN
-    IF NOT EXISTS (
-        SELECT 1 FROM pg_type WHERE typname = 'decrypted_api_key_result'
-    ) THEN
-        CREATE TYPE decrypted_api_key_result AS (
-            decrypted_value TEXT
-        );
-    END IF;
-END $$;
-
 -- Helper functions for encryption/decryption (following secrets pattern)
--- Modified to return SETOF composite types so they return tables that Hasura can track
+-- Modified to return TABLE so they can be tracked by Hasura  
 CREATE OR REPLACE FUNCTION encrypt_api_key(
     p_value TEXT,
     p_key TEXT
-) RETURNS SETOF encrypted_api_key_result AS $$
+) RETURNS TABLE(encrypted_value BYTEA) AS $$
 BEGIN
     RETURN QUERY SELECT 
         pgp_sym_encrypt(
             p_value,
             p_key,
             'cipher-algo=aes256'
-        ) AS encrypted_value;
+        );
 END;
 $$ LANGUAGE plpgsql SECURITY DEFINER;
 
 CREATE OR REPLACE FUNCTION decrypt_api_key(
     p_encrypted_value BYTEA,
     p_key TEXT
-) RETURNS SETOF decrypted_api_key_result AS $$
+) RETURNS TABLE(decrypted_value TEXT) AS $$
 BEGIN
     RETURN QUERY SELECT 
         pgp_sym_decrypt(
             p_encrypted_value,
             p_key
-        ) AS decrypted_value;
+        );
 END;
 $$ LANGUAGE plpgsql SECURITY DEFINER;