🔨 Add a view function on the multi vesting contracts

frak-id · Oct 11, 2023 · db1c792 · db1c792
1 parent f4854d3
commit db1c792
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 26 deletions.
diff --git a/.changeset/README.md b/.changeset/README.md
diff --git a/.changeset/config.json b/.changeset/config.json
diff --git a/.changeset/few-jokes-ring.md b/.changeset/few-jokes-ring.md
diff --git a/contracts/wallets/MultiVestingWallets.sol b/contracts/wallets/MultiVestingWallets.sol
@@ -128,7 +128,7 @@ contract MultiVestingWallets is FrakAccessControlUpgradeable {
     }
 
     /* -------------------------------------------------------------------------- */
-    /*                              Create vesting's                              */
+    /*                              Create vestings                               */
     /* -------------------------------------------------------------------------- */
 
     /**
@@ -328,7 +328,7 @@ contract MultiVestingWallets is FrakAccessControlUpgradeable {
     }
 
     /* -------------------------------------------------------------------------- */
-    /*                                View method's                               */
+    /*                                View methods                                */
     /* -------------------------------------------------------------------------- */
 
     /**
@@ -358,6 +358,15 @@ contract MultiVestingWallets is FrakAccessControlUpgradeable {
         return owned[beneficiary].length();
     }
 
+    /**
+     * @notice Get all the vesting id owned by an address.
+     * @param beneficiary Address to check.
+     * @return All the vesting ids of the address.
+     */
+    function ownedVestingIds(address beneficiary) public view returns (uint256[] memory) {
+        return owned[beneficiary].values();
+    }
+
     /**
      * @notice Get the remaining amount of token of a beneficiary.
      * @param beneficiary Address to check.