Use multi-way if statement to clarify applyOurTxToUTxO.

cardano-foundation · Nov 30, 2021 · 09d571a · 09d571a
1 parent edb39cc
commit 09d571a
Showing 1 changed file with 23 additions and 20 deletions.
diff --git a/lib/core/src/Cardano/Wallet/Primitive/Model.hs b/lib/core/src/Cardano/Wallet/Primitive/Model.hs
@@ -1,11 +1,13 @@
 {-# LANGUAGE BangPatterns #-}
+{-# LANGUAGE BlockArguments #-}
 {-# LANGUAGE DataKinds #-}
 {-# LANGUAGE DeriveGeneric #-}
 {-# LANGUAGE DuplicateRecordFields #-}
 {-# LANGUAGE FlexibleContexts #-}
 {-# LANGUAGE FlexibleInstances #-}
 {-# LANGUAGE GADTs #-}
 {-# LANGUAGE LambdaCase #-}
+{-# LANGUAGE MultiWayIf #-}
 {-# LANGUAGE NamedFieldPuns #-}
 {-# LANGUAGE OverloadedLabels #-}
 {-# LANGUAGE RankNTypes #-}
@@ -554,26 +556,27 @@ applyOurTxToUTxO !slotNo !blockHeight !tx !prevUTxO = do
                 Just $ distance totalIn totalOut
             (_, Incoming) ->
                 Nothing
-    return $ if hasKnownOutput && not hasKnownInput then
-        let dir = Incoming in
-        Just
-            ( ( tx { fee = actualFee dir }
-              , mkTxMeta (TB.getCoin received) dir
-              )
-            , ourNextUTxO
-            )
-    else if hasKnownInput || hasKnownWithdrawal then
-        let adaSpent = TB.getCoin spent
-            adaReceived = TB.getCoin received
-            dir = if adaSpent > adaReceived then Outgoing else Incoming
-            amount = distance adaSpent adaReceived
-        in
-        Just
-            ( (tx { fee = actualFee dir }, mkTxMeta amount dir)
-            , ourNextUTxO
-            )
-    else
-        Nothing
+    return if
+        | hasKnownOutput && not hasKnownInput ->
+            let dir = Incoming in
+            Just
+                ( ( tx { fee = actualFee dir }
+                  , mkTxMeta (TB.getCoin received) dir
+                  )
+                , ourNextUTxO
+                )
+        | hasKnownInput || hasKnownWithdrawal ->
+            let adaSpent = TB.getCoin spent
+                adaReceived = TB.getCoin received
+                dir = if adaSpent > adaReceived then Outgoing else Incoming
+                amount = distance adaSpent adaReceived
+            in
+            Just
+                ( (tx { fee = actualFee dir }, mkTxMeta amount dir)
+                , ourNextUTxO
+                )
+        | otherwise ->
+            Nothing
   where
     mkTxMeta :: Coin -> Direction -> TxMeta
     mkTxMeta amount dir = TxMeta