diff --git a/src/Zenith/DB.hs b/src/Zenith/DB.hs index 6a7909a..384f3bb 100644 --- a/src/Zenith/DB.hs +++ b/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