Implement confirmed balances in GUI

CHANGELOG.md

@@ -22,6 +22,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Dialog to display transaction details and copy TX ID
 - Dialog to send a new transaction
 - Dialog to display Tx ID after successful broadcast
+- Unconfirmed balance display on TUI and GUI
 
 ### Fixed
 

src/Zenith/CLI.hs

@@ -180,6 +180,7 @@ data State = State
   , _timer :: !Int
   , _txForm :: !(Form SendInput () Name)
   , _sentTx :: !(Maybe HexString)
+  , _unconfBalance :: !Integer
   }
 
 makeLenses ''State
@@ -215,6 +216,12 @@ drawUI s = [splashDialog s, helpDialog s, displayDialog s, inputDialog s, ui s]
                if st ^. network == MainNet
                  then displayZec (st ^. balance)
                  else displayTaz (st ^. balance))) <=>
+         C.hCenter
+           (str
+              ("Unconf: " ++
+               if st ^. network == MainNet
+                 then displayZec (st ^. unconfBalance)
+                 else displayTaz (st ^. unconfBalance))) <=>
          listAddressBox "Addresses" (st ^. addresses) <+>
          B.vBorder <+>
          (C.hCenter (str ("Last block seen: " ++ show (st ^. syncBlock))) <=>
@@ -1046,6 +1053,10 @@ runZenithCLI config = do
             if not (null accList)
               then getBalance pool $ entityKey $ head accList
               else return 0
+          uBal <-
+            if not (null accList)
+              then getUnconfirmedBalance pool $ entityKey $ head accList
+              else return 0
           eventChan <- BC.newBChan 10
           _ <-
             forkIO $
@@ -1083,6 +1094,7 @@ runZenithCLI config = do
               0
               (mkSendForm 0 $ SendInput "" 0.0 "")
               Nothing
+              uBal
     Left e -> do
       print $
         "No Zebra node available on port " <>
@@ -1111,6 +1123,10 @@ refreshWallet s = do
     if not (null aL)
       then getBalance pool $ entityKey $ head aL
       else return 0
+  uBal <-
+    if not (null aL)
+      then getUnconfirmedBalance pool $ entityKey $ head aL
+      else return 0
   txL <-
     if not (null addrL)
       then getUserTx pool $ entityKey $ head addrL
@@ -1121,6 +1137,8 @@ refreshWallet s = do
   let txL' = L.listReplace (Vec.fromList txL) (Just 0) (s ^. transactions)
   return $
     s & wallets .~ wL & accounts .~ aL' & syncBlock .~ bl & balance .~ bal &
+    unconfBalance .~
+    uBal &
     addresses .~
     addrL' &
     transactions .~
@@ -1191,6 +1209,7 @@ refreshAccount s = do
          Just (_k, w) -> return w
   aL <- runNoLoggingT $ getAddresses pool $ entityKey selAccount
   bal <- getBalance pool $ entityKey selAccount
+  uBal <- getUnconfirmedBalance pool $ entityKey selAccount
   let aL' = L.listReplace (Vec.fromList aL) (Just 0) (s ^. addresses)
   selAddress <-
     do case L.listSelectedElement aL' of
@@ -1201,13 +1220,17 @@ refreshAccount s = do
   case selAddress of
     Nothing ->
       return $
-      s & balance .~ bal & addresses .~ aL' & msg .~ "Switched to account: " ++
+      s & balance .~ bal & unconfBalance .~ uBal & addresses .~ aL' & msg .~
+      "Switched to account: " ++
       T.unpack (zcashAccountName $ entityVal selAccount)
     Just (_i, a) -> do
       tList <- getUserTx pool $ entityKey a
       let tL' = L.listReplace (Vec.fromList tList) (Just 0) (s ^. transactions)
       return $
-        s & balance .~ bal & addresses .~ aL' & transactions .~ tL' & msg .~
+        s & balance .~ bal & unconfBalance .~ uBal & addresses .~ aL' &
+        transactions .~
+        tL' &
+        msg .~
         "Switched to account: " ++
         T.unpack (zcashAccountName $ entityVal selAccount)
 

src/Zenith/DB.hs

@@ -1392,6 +1392,19 @@ getBalance pool za = do
   let oBal = sum oAmts
   return . fromIntegral $ tBal + sBal + oBal
 
+getUnconfirmedBalance :: ConnectionPool -> ZcashAccountId -> IO Integer
+getUnconfirmedBalance pool za = do
+  trNotes <- getWalletUnspentUnconfirmedTrNotes pool za
+  let tAmts = map (walletTrNoteValue . entityVal) trNotes
+  let tBal = sum tAmts
+  sapNotes <- getWalletUnspentUnconfirmedSapNotes pool za
+  let sAmts = map (walletSapNoteValue . entityVal) sapNotes
+  let sBal = sum sAmts
+  orchNotes <- getWalletUnspentUnconfirmedOrchNotes pool za
+  let oAmts = map (walletOrchNoteValue . entityVal) orchNotes
+  let oBal = sum oAmts
+  return . fromIntegral $ tBal + sBal + oBal
+
 clearWalletTransactions :: ConnectionPool -> IO ()
 clearWalletTransactions pool = do
   runNoLoggingT $
@@ -1429,10 +1442,42 @@ getWalletUnspentTrNotes pool za = do
     PS.retryOnBusy $
     flip PS.runSqlPool pool $ do
       select $ do
-        n <- from $ table @WalletTrNote
+        (txs :& tNotes) <-
-        where_ (n ^. WalletTrNoteAccId ==. val za)
+          from $ table @WalletTransaction `innerJoin` table @WalletTrNote `on`
-        where_ (n ^. WalletTrNoteSpent ==. val False)
+          (\(txs :& tNotes) ->
-        pure n
+             txs ^. WalletTransactionId ==. tNotes ^. WalletTrNoteTx)
+        where_ (tNotes ^. WalletTrNoteAccId ==. val za)
+        where_ (tNotes ^. WalletTrNoteSpent ==. val False)
+        where_
+          ((tNotes ^. WalletTrNoteChange ==. val True &&. txs ^.
+            WalletTransactionConf >=.
+            val 3) ||.
+           (tNotes ^. WalletTrNoteChange ==. val False &&. txs ^.
+            WalletTransactionConf >=.
+            val 10))
+        pure tNotes
+
+getWalletUnspentUnconfirmedTrNotes ::
+     ConnectionPool -> ZcashAccountId -> IO [Entity WalletTrNote]
+getWalletUnspentUnconfirmedTrNotes pool za = do
+  runNoLoggingT $
+    PS.retryOnBusy $
+    flip PS.runSqlPool pool $ do
+      select $ do
+        (txs :& tNotes) <-
+          from $ table @WalletTransaction `innerJoin` table @WalletTrNote `on`
+          (\(txs :& tNotes) ->
+             txs ^. WalletTransactionId ==. tNotes ^. WalletTrNoteTx)
+        where_ (tNotes ^. WalletTrNoteAccId ==. val za)
+        where_ (tNotes ^. WalletTrNoteSpent ==. val False)
+        where_
+          ((tNotes ^. WalletTrNoteChange ==. val True &&. txs ^.
+            WalletTransactionConf <.
+            val 3) ||.
+           (tNotes ^. WalletTrNoteChange ==. val False &&. txs ^.
+            WalletTransactionConf <.
+            val 10))
+        pure tNotes
 
 getWalletUnspentSapNotes ::
      ConnectionPool -> ZcashAccountId -> IO [Entity WalletSapNote]
@@ -1441,10 +1486,42 @@ getWalletUnspentSapNotes pool za = do
     PS.retryOnBusy $
     flip PS.runSqlPool pool $ do
       select $ do
-        n1 <- from $ table @WalletSapNote
+        (txs :& sNotes) <-
-        where_ (n1 ^. WalletSapNoteAccId ==. val za)
+          from $ table @WalletTransaction `innerJoin` table @WalletSapNote `on`
-        where_ (n1 ^. WalletSapNoteSpent ==. val False)
+          (\(txs :& sNotes) ->
-        pure n1
+             txs ^. WalletTransactionId ==. sNotes ^. WalletSapNoteTx)
+        where_ (sNotes ^. WalletSapNoteAccId ==. val za)
+        where_ (sNotes ^. WalletSapNoteSpent ==. val False)
+        where_
+          ((sNotes ^. WalletSapNoteChange ==. val True &&. txs ^.
+            WalletTransactionConf >=.
+            val 3) ||.
+           (sNotes ^. WalletSapNoteChange ==. val False &&. txs ^.
+            WalletTransactionConf >=.
+            val 10))
+        pure sNotes
+
+getWalletUnspentUnconfirmedSapNotes ::
+     ConnectionPool -> ZcashAccountId -> IO [Entity WalletSapNote]
+getWalletUnspentUnconfirmedSapNotes pool za = do
+  runNoLoggingT $
+    PS.retryOnBusy $
+    flip PS.runSqlPool pool $ do
+      select $ do
+        (txs :& sNotes) <-
+          from $ table @WalletTransaction `innerJoin` table @WalletSapNote `on`
+          (\(txs :& sNotes) ->
+             txs ^. WalletTransactionId ==. sNotes ^. WalletSapNoteTx)
+        where_ (sNotes ^. WalletSapNoteAccId ==. val za)
+        where_ (sNotes ^. WalletSapNoteSpent ==. val False)
+        where_
+          ((sNotes ^. WalletSapNoteChange ==. val True &&. txs ^.
+            WalletTransactionConf <.
+            val 3) ||.
+           (sNotes ^. WalletSapNoteChange ==. val False &&. txs ^.
+            WalletTransactionConf <.
+            val 10))
+        pure sNotes
 
 getWalletUnspentOrchNotes ::
      ConnectionPool -> ZcashAccountId -> IO [Entity WalletOrchNote]
@@ -1453,10 +1530,42 @@ getWalletUnspentOrchNotes pool za = do
     PS.retryOnBusy $
     flip PS.runSqlPool pool $ do
       select $ do
-        n2 <- from $ table @WalletOrchNote
+        (txs :& oNotes) <-
-        where_ (n2 ^. WalletOrchNoteAccId ==. val za)
+          from $ table @WalletTransaction `innerJoin` table @WalletOrchNote `on`
-        where_ (n2 ^. WalletOrchNoteSpent ==. val False)
+          (\(txs :& oNotes) ->
-        pure n2
+             txs ^. WalletTransactionId ==. oNotes ^. WalletOrchNoteTx)
+        where_ (oNotes ^. WalletOrchNoteAccId ==. val za)
+        where_ (oNotes ^. WalletOrchNoteSpent ==. val False)
+        where_
+          ((oNotes ^. WalletOrchNoteChange ==. val True &&. txs ^.
+            WalletTransactionConf >=.
+            val 3) ||.
+           (oNotes ^. WalletOrchNoteChange ==. val False &&. txs ^.
+            WalletTransactionConf >=.
+            val 10))
+        pure oNotes
+
+getWalletUnspentUnconfirmedOrchNotes ::
+     ConnectionPool -> ZcashAccountId -> IO [Entity WalletOrchNote]
+getWalletUnspentUnconfirmedOrchNotes pool za = do
+  runNoLoggingT $
+    PS.retryOnBusy $
+    flip PS.runSqlPool pool $ do
+      select $ do
+        (txs :& oNotes) <-
+          from $ table @WalletTransaction `innerJoin` table @WalletOrchNote `on`
+          (\(txs :& oNotes) ->
+             txs ^. WalletTransactionId ==. oNotes ^. WalletOrchNoteTx)
+        where_ (oNotes ^. WalletOrchNoteAccId ==. val za)
+        where_ (oNotes ^. WalletOrchNoteSpent ==. val False)
+        where_
+          ((oNotes ^. WalletOrchNoteChange ==. val True &&. txs ^.
+            WalletTransactionConf <.
+            val 3) ||.
+           (oNotes ^. WalletOrchNoteChange ==. val False &&. txs ^.
+            WalletTransactionConf <.
+            val 10))
+        pure oNotes
 
 selectUnspentNotes ::
      ConnectionPool

src/Zenith/GUI.hs

@@ -75,6 +75,7 @@ data AppEvent
   | SwitchAddr !Int
   | SwitchAcc !Int
   | SwitchWal !Int
+  | UpdateBalance !(Integer, Integer)
   | CopyAddr !(Maybe (Entity WalletAddress))
   | LoadTxs ![Entity UserTx]
   | LoadAddrs ![Entity WalletAddress]
@@ -311,7 +312,7 @@ buildUI wenv model = widgetTree
       hstack
         [ addressBox
         , vstack
-            [ mainButton "Send" ShowSend
+            [ mainButton "Send" ShowSend `styleBasic` [textFont "Bold"]
             , txBox `nodeVisible` not (null $ model ^. transactions)
             ]
         ]
@@ -322,19 +323,24 @@ buildUI wenv model = widgetTree
           box_
             [alignMiddle]
             (vstack
-               [ animFadeIn
+               [ hstack
-                   (label (displayAmount (model ^. network) $ model ^. balance) `styleBasic`
+                   [ filler
+                   , animFadeIn
+                       (label
+                          (displayAmount (model ^. network) $ model ^. balance) `styleBasic`
                         [textSize 20])
+                   , filler
+                   ]
                , hstack
                    [ filler
                    , remixIcon remixHourglassFill `styleBasic` [textSize 8]
                    , label
                        (maybe "0" (displayAmount (model ^. network)) $
                         model ^. unconfBalance) `styleBasic`
-                     [textSize 8] `nodeVisible`
+                     [textSize 8]
-                     isJust (model ^. unconfBalance)
                    , filler
-                   ]
+                   ] `nodeVisible`
+                 isJust (model ^. unconfBalance)
                ]) `styleBasic`
           [bgColor white, radius 5, border 1 btnColor]
         , filler
@@ -923,6 +929,15 @@ handleEvent wenv node model evt =
           case selectAccount i of
             Nothing -> return []
             Just acc -> runNoLoggingT $ getAddresses dbPool $ entityKey acc
+      , Task $
+        UpdateBalance <$> do
+          dbPool <- runNoLoggingT $ initPool $ c_dbPath $ model ^. configuration
+          case selectAccount i of
+            Nothing -> return (0, 0)
+            Just acc -> do
+              b <- getBalance dbPool $ entityKey acc
+              u <- getUnconfirmedBalance dbPool $ entityKey acc
+              return (b, u)
       , Event $ SetPool Orchard
       ]
     SwitchWal i ->
@@ -934,6 +949,13 @@ handleEvent wenv node model evt =
             Nothing -> return []
             Just wal -> runNoLoggingT $ getAccounts dbPool $ entityKey wal
       ]
+    UpdateBalance (b, u) ->
+      [ Model $
+        model & balance .~ b & unconfBalance .~
+        (if u == 0
+           then Nothing
+           else Just u)
+      ]
     CopyAddr a ->
       [ setClipboardData ClipboardEmpty
       , setClipboardData $
@@ -1246,6 +1268,14 @@ runZenithGUI config = do
             if not (null addrList)
               then getQrCode pool Orchard $ entityKey $ head addrList
               else return Nothing
+          bal <-
+            if not (null accList)
+              then getBalance pool $ entityKey $ head accList
+              else return 0
+          unconfBal <-
+            if not (null accList)
+              then getUnconfirmedBalance pool $ entityKey $ head accList
+              else return 0
           let model =
                 AppModel
                   config
@@ -1260,8 +1290,10 @@ runZenithGUI config = do
                   0
                   Nothing
                   True
-                  314259000
+                  bal
-                  (Just 300000)
+                  (if unconfBal == 0
+                     then Nothing
+                     else Just unconfBal)
                   Orchard
                   qr
                   False