Modify SQLite function checkpointFromEntity to process tokens.

cardano-foundation · Dec 2, 2020 · 203df6e · 203df6e
1 parent 2af6f8c
commit 203df6e
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/lib/core/src/Cardano/Wallet/DB/Sqlite.hs b/lib/core/src/Cardano/Wallet/DB/Sqlite.hs
@@ -1173,9 +1173,15 @@ checkpointFromEntity cp utxo s =
         ) = cp
     header = (W.BlockHeader slot (Quantity bh) headerHash parentHeaderHash)
     utxo' = W.UTxO . Map.fromList $
-        [ (W.TxIn input ix, W.TxOut addr (TB.fromCoin coin))
-        | UTxO _ _ (TxId input) ix addr coin <- fst <$> utxo
+        [ (W.TxIn input ix, W.TxOut addr (mkTokenBundle coin tokens))
+        | (UTxO _ _ (TxId input) ix addr coin, tokens) <- utxo
         ]
+    mkTokenBundle coin tokens =
+        TB.fromFlatList coin (mkTokenEntry <$> tokens)
+    mkTokenEntry token =
+        ( TB.AssetId (utxoTokenPolicyId token) (utxoTokenName token)
+        , utxoTokenQuantity token
+        )
     gp = W.GenesisParameters
         { getGenesisBlockHash = coerce genesisHash
         , getGenesisBlockDate = W.StartTime genesisStart