From eee2ce52b5f0a8283ae45cbd425276ea830ba399 Mon Sep 17 00:00:00 2001
From: Hongrui Fang <chfanghr@gmail.com>
Date: Thu, 13 Oct 2022 18:56:56 +0800
Subject: [PATCH] filter stake UTxO by `ScriptCredential`

---
 agora/Agora/Stake/Scripts.hs | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/agora/Agora/Stake/Scripts.hs b/agora/Agora/Stake/Scripts.hs
index 67aa70c..cb5676b 100644
--- a/agora/Agora/Stake/Scripts.hs
+++ b/agora/Agora/Stake/Scripts.hs
@@ -284,7 +284,9 @@ mkStakeValidator
                 # (pfield @"_0" # stakeInputRef)
                 # txInfoF.inputs
 
-          stakeValidatorAddress = pfield @"address" # validatedInput
+          stakeValidatorCredential =
+            pfield @"credential"
+              #$ pfield @"address" # validatedInput
 
       --------------------------------------------------------------------------
 
@@ -303,8 +305,10 @@ mkStakeValidator
                   PGT -> ptraceError "More than one SST in one UTxO"
                   -- 1
                   PEQ ->
-                    let ownedByStakeValidator =
-                          txOutF.address #== stakeValidatorAddress
+                    let ownerCredential = pfield @"credential" # txOutF.address
+
+                        isOwnedByStakeValidator =
+                          ownerCredential #== stakeValidatorCredential
 
                         datum =
                           ptrace "Resolve stake datum" $
@@ -314,7 +318,7 @@ mkStakeValidator
                                 # txInfoF.datums
                      in passert
                           "Should owned by stake validator"
-                          ownedByStakeValidator
+                          isOwnedByStakeValidator
                           (pjust # datum)
                   -- 0
                   PLT -> pnothing