Add function fromNestedMap to TokenBundle and TokenMap.

cardano-foundation · Jan 26, 2021 · fcf253e · fcf253e
1 parent 0e87431
commit fcf253e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/lib/core/src/Cardano/Wallet/Primitive/Types/TokenBundle.hs b/lib/core/src/Cardano/Wallet/Primitive/Types/TokenBundle.hs
@@ -24,6 +24,7 @@ module Cardano.Wallet.Primitive.Types.TokenBundle
     , empty
     , fromFlatList
     , fromNestedList
+    , fromNestedMap
     , fromTokenMap
     , fromTokenQuantity
 
@@ -201,6 +202,11 @@ fromNestedList
     -> TokenBundle
 fromNestedList c = TokenBundle c . TokenMap.fromNestedList
 
+fromNestedMap
+    :: (Coin, Map TokenPolicyId (NonEmptyMap TokenName TokenQuantity))
+    -> TokenBundle
+fromNestedMap (c, m) = TokenBundle c (TokenMap.fromNestedMap m)
+
 fromTokenMap :: TokenMap -> TokenBundle
 fromTokenMap = TokenBundle (Coin 0)
 

diff --git a/lib/core/src/Cardano/Wallet/Primitive/Types/TokenMap.hs b/lib/core/src/Cardano/Wallet/Primitive/Types/TokenMap.hs
@@ -44,6 +44,7 @@ module Cardano.Wallet.Primitive.Types.TokenMap
     , singleton
     , fromFlatList
     , fromNestedList
+    , fromNestedMap
 
     -- * Deconstruction
     , toFlatList
@@ -446,6 +447,13 @@ fromNestedList entries = fromFlatList
     , (token, quantity) <- NE.toList tokenQuantities
     ]
 
+-- | Creates a token map from a nested map.
+--
+fromNestedMap
+    :: Map TokenPolicyId (NonEmptyMap TokenName TokenQuantity)
+    -> TokenMap
+fromNestedMap = fromNestedList . Map.toList . fmap NEMap.toList
+
 --------------------------------------------------------------------------------
 -- Deconstruction
 --------------------------------------------------------------------------------