Add walletPubKey and update the definition of walletPubKeyHash (fix I…

…ntersectMBO#104)
jeffhappily · Nov 12, 2021 · 7f1adb1 · 7f1adb1
1 parent eabb8c0
commit 7f1adb1
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/plutus-contract/src/Plutus/Contract/Trace.hs b/plutus-contract/src/Plutus/Contract/Trace.hs
@@ -41,6 +41,7 @@ module Plutus.Contract.Trace
     , defaultDistFor
     -- * Wallets
     , EM.Wallet(..)
+    , EM.walletPubKey
     , EM.walletPubKeyHash
     , EM.knownWallets
     , EM.knownWallet

diff --git a/plutus-contract/src/Wallet/Emulator/Types.hs b/plutus-contract/src/Wallet/Emulator/Types.hs
@@ -17,6 +17,7 @@ module Wallet.Emulator.Types(
     WalletId(..),
     Crypto.XPrv,
     Crypto.XPub,
+    walletPubKey,
     walletPubKeyHash,
     addSignature,
     knownWallets,

diff --git a/plutus-contract/src/Wallet/Emulator/Wallet.hs b/plutus-contract/src/Wallet/Emulator/Wallet.hs
@@ -128,14 +128,18 @@ fromBase16 s = bimap show WalletId (fromText s)
 walletMockWallet :: Wallet -> Maybe MockWallet
 walletMockWallet (Wallet wid) = find ((==) wid . WalletId . CW.mwWalletId) CW.knownWallets
 
--- | The public key hash of a mock wallet.  (Fails if the wallet is not a mock wallet).
-walletPubKeyHash :: Wallet -> PubKeyHash
-walletPubKeyHash w = CW.pubKeyHash
-                 $ fromMaybe (error $ "Wallet.Emulator.Wallet.walletPubKeyHash: Wallet "
+-- | The public key of a mock wallet.  (Fails if the wallet is not a mock wallet).
+walletPubKey :: Wallet -> PubKey
+walletPubKey w = CW.pubKey
+                 $ fromMaybe (error $ "Wallet.Emulator.Wallet.walletPubKey: Wallet "
                                    <> show w
                                    <> " is not a mock wallet")
                  $ walletMockWallet w
 
+-- | The public key hash of a mock wallet.  (Fails if the wallet is not a mock wallet).
+walletPubKeyHash :: Wallet -> PubKeyHash
+walletPubKeyHash = pubKeyHash . walletPubKey
+
 -- | Get the address of a mock wallet. (Fails if the wallet is not a mock wallet).
 walletAddress :: Wallet -> Address
 walletAddress = pubKeyHashAddress . walletPubKeyHash