src/Zenith/DB.hs

@@ -1390,6 +1390,22 @@ getBalance pool za = do
   let oBal = sum oAmts
   return . fromIntegral $ tBal + sBal + oBal
 
+getTransparentBalance :: ConnectionPool -> ZcashAccountId -> IO Integer
+getTransparentBalance pool za = do
+  trNotes <- getWalletUnspentTrNotes pool za
+  let tAmts = map (walletTrNoteValue . entityVal) trNotes
+  return . fromIntegral $ sum tAmts
+
+getShieldedBalance :: ConnectionPool -> ZcashAccountId -> IO Integer
+getShieldedBalance pool za = do
+  sapNotes <- getWalletUnspentSapNotes pool za
+  let sAmts = map (walletSapNoteValue . entityVal) sapNotes
+  let sBal = sum sAmts
+  orchNotes <- getWalletUnspentOrchNotes pool za
+  let oAmts = map (walletOrchNoteValue . entityVal) orchNotes
+  let oBal = sum oAmts
+  return . fromIntegral $ sBal + oBal
+
 getUnconfirmedBalance :: ConnectionPool -> ZcashAccountId -> IO Integer
 getUnconfirmedBalance pool za = do
   trNotes <- getWalletUnspentUnconfirmedTrNotes pool za