Unify triggers to check for method support

Author: ekzyis

prisma/migrations/20250502042253_wallet_v2/migration.sql

@@ -356,105 +356,94 @@ ALTER TABLE "UserWallet" ADD CONSTRAINT "UserWallet_userId_fkey" FOREIGN KEY ("u
 -- AddForeignKey
 ALTER TABLE "UserWallet" ADD CONSTRAINT "UserWallet_walletId_fkey" FOREIGN KEY ("walletId") REFERENCES "WalletV2"("id") ON DELETE CASCADE ON UPDATE CASCADE;
 
-CREATE OR REPLACE FUNCTION wallet_check_send_method()
+CREATE OR REPLACE FUNCTION wallet_check_support()
 RETURNS TRIGGER AS $$
 DECLARE
     wallet "WalletV2";
+    method_type TEXT;
 BEGIN
-    SELECT w.* INTO wallet
-    FROM "UserWallet" uw
-    JOIN "WalletV2" w ON uw."walletId" = w.id
-    WHERE uw.id = NEW."walletId";
-
-    IF NOT NEW."type"::"WalletSendMethod" = ANY(wallet."sendMethods") THEN
-        RAISE EXCEPTION 'Wallet % does not support send method %', wallet.name, NEW."type";
-    END IF;
-
-    RETURN NEW;
-END;
-$$ LANGUAGE plpgsql;
+    method_type := TG_ARGV[0];
 
-CREATE OR REPLACE FUNCTION wallet_check_recv_method()
-RETURNS TRIGGER AS $$
-DECLARE
-    wallet "WalletV2";
-BEGIN
     SELECT w.* INTO wallet
     FROM "UserWallet" uw
     JOIN "WalletV2" w ON uw."walletId" = w.id
     WHERE uw.id = NEW."walletId";
 
-    IF NOT NEW."type"::"WalletRecvMethod" = ANY(wallet."recvMethods") THEN
-        RAISE EXCEPTION 'Wallet % does not support receive method %', wallet.name, NEW."type";
+    IF method_type = 'SEND' THEN
+        IF NOT NEW."type"::"WalletSendMethod" = ANY(wallet."sendMethods") THEN
+            RAISE EXCEPTION 'Wallet % does not support send method %', wallet.name, NEW."type";
+        END IF;
+    ELSIF method_type = 'RECEIVE' THEN
+        IF NOT NEW."type"::"WalletRecvMethod" = ANY(wallet."recvMethods") THEN
+            RAISE EXCEPTION 'Wallet % does not support receive method %', wallet.name, NEW."type";
+        END IF;
     END IF;
 
     RETURN NEW;
 END;
 $$ LANGUAGE plpgsql;
 
--- Create triggers for each WalletSend table
-CREATE TRIGGER wallet_check_send_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletSendNWC"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_send_method();
+    EXECUTE FUNCTION wallet_check_support('SEND');
 
-CREATE TRIGGER wallet_check_send_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletSendLNbits"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_send_method();
+    EXECUTE FUNCTION wallet_check_support('SEND');
 
-CREATE TRIGGER wallet_check_send_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletSendPhoenixd"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_send_method();
+    EXECUTE FUNCTION wallet_check_support('SEND');
 
-CREATE TRIGGER wallet_check_send_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletSendBlink"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_send_method();
+    EXECUTE FUNCTION wallet_check_support('SEND');
 
-CREATE TRIGGER wallet_check_send_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletSendWebLN"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_send_method();
+    EXECUTE FUNCTION wallet_check_support('SEND');
 
-CREATE TRIGGER wallet_check_send_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletSendLNC"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_send_method();
+    EXECUTE FUNCTION wallet_check_support('SEND');
 
--- Create triggers for each WalletRecv table
-CREATE TRIGGER wallet_check_recv_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletRecvNWC"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_recv_method();
+    EXECUTE FUNCTION wallet_check_support('RECEIVE');
 
-CREATE TRIGGER wallet_check_recv_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletRecvLNbits"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_recv_method();
+    EXECUTE FUNCTION wallet_check_support('RECEIVE');
 
-CREATE TRIGGER wallet_check_recv_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletRecvPhoenixd"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_recv_method();
+    EXECUTE FUNCTION wallet_check_support('RECEIVE');
 
-CREATE TRIGGER wallet_check_recv_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletRecvBlink"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_recv_method();
+    EXECUTE FUNCTION wallet_check_support('RECEIVE');
 
-CREATE TRIGGER wallet_check_recv_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletRecvLightningAddress"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_recv_method();
+    EXECUTE FUNCTION wallet_check_support('RECEIVE');
 
-CREATE TRIGGER wallet_check_recv_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletRecvCLNRest"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_recv_method();
+    EXECUTE FUNCTION wallet_check_support('RECEIVE');
 
-CREATE TRIGGER wallet_check_recv_method
+CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletRecvLNDGRPC"
     FOR EACH ROW
-    EXECUTE FUNCTION wallet_check_recv_method();
+    EXECUTE FUNCTION wallet_check_support('RECEIVE');