Easy function to get wallet donated amounts

fluidity-money · Aug 5, 2024 · 74459b2 · 74459b2
1 parent 9ac82ef
commit 74459b2
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 2 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/.gitignore b/.gitignore
@@ -12,3 +12,5 @@ docs/
 
 # Dotenv file
 .env
+
+.DS_Store
diff --git a/pkg/Donation.sol b/pkg/Donation.sol
@@ -30,8 +30,7 @@ contract Donation is IDonation {
     uint128 donationEpoch_;
 
     /**
-     * @notice positions_ held by the user. The latest element in the array is the
-     *         deposited amounts for the current epoch.
+     * @notice positions_ donated to a cat by each user.
      */
     mapping (address => uint256) private positions_;
 
@@ -60,6 +59,11 @@ contract Donation is IDonation {
         return cats_[donationEpoch_][_cat];
     }
 
+    /// @inheritdoc IDonationView
+    function wallet(address _wallet) external view returns (uint256) {
+        return positions_[_wallet];
+    }
+
     /// @inheritdoc IDonationView
     function walletCount(bytes8 _cat) external view returns (uint256) {
         return catWallets_[donationEpoch_][_cat];

diff --git a/pkg/IDonation.sol b/pkg/IDonation.sol
@@ -10,6 +10,13 @@ interface IDonationView {
      */
     function get(bytes8 cat) external view returns (uint256);
 
+    /**
+     * @notice wallet amount donated
+     * @param wallet to get the amount donated for.
+     * @return the amount
+     */
+    function wallet(address wallet) external view returns (uint256);
+
     /**
      * @notice walletCount that interacted with this cat.
      * @param cat to use.