Merge pull request #234 from paltalabs/feat/collect-fees-method

collect fees method
paltalabs · Nov 25, 2024 · f1af417 · f1af417
2 parents 25f5523 + a82293c
commit f1af417
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/apps/contracts/vault/src/interface.rs b/apps/contracts/vault/src/interface.rs
@@ -200,6 +200,9 @@ pub trait VaultTrait {
     fn get_asset_amounts_for_dftokens(e: Env, df_token: i128) -> Map<Address, i128>;
 
     fn get_fees(e: Env) -> (u32, u32);
+
+    /// Collects the fees from the vault and transfers them to the fee receiver addresses. 
+    fn collect_fees(e: Env) -> Result<(), ContractError>;
 }
 
 pub trait AdminInterfaceTrait {

diff --git a/apps/contracts/vault/src/lib.rs b/apps/contracts/vault/src/lib.rs
@@ -574,6 +574,11 @@ impl VaultTrait for DeFindexVault {
         let vault_fee = get_vault_fee(&e);
         (defindex_protocol_fee, vault_fee)
     }
+
+    fn collect_fees(e: Env) -> Result<(), ContractError> {
+        extend_instance_ttl(&e);
+        collect_fees(&e)
+    }
 }
 
 #[contractimpl]