Merge pull request #5 from apollodao/pacman/fee-msgs-from-coin

Implement `fee_msgs_from_coin`

packages/fee-config/CHANGELOG.md

@@ -0,0 +1,7 @@
+# Changelog
+
+## [0.1.2] - 2024-05-09
+
+### Added
+
+- Implemented public function `fee_msgs_from_coin` on `FeeConfig<Addr>`

packages/fee-config/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "cw-fee-config"
-version = "0.1.1"
+version = "0.1.2"
 edition = "2021"
 description = "Fee configuration for CosmWasm contracts"
 license = { workspace = true }

packages/fee-config/src/lib.rs

@@ -173,6 +173,25 @@ impl FeeConfig<Addr> {
 
         Ok((fee_msgs, coins_after_fees))
     }
+
+    /// Calculates the fee from the input coin and returns messages to send it
+    /// to the fee recipients.
+    ///
+    /// # Arguments
+    /// * `coin` - The coin to take the fee from.
+    ///
+    /// # Returns
+    /// * `Vec<CosmosMsg>` - The messages to send the fees to the fee
+    ///   recipients.
+    /// * `Coin` - The asset after the fee has been taken.
+    pub fn fee_msgs_from_coin(&self, coin: Coin, env: &Env) -> StdResult<(Vec<CosmosMsg>, Coin)> {
+        let (msgs, coins_after_fee) =
+            self.fee_msgs_from_coins(&Coins::try_from(vec![coin.clone()])?, env)?;
+        Ok((
+            msgs,
+            cosmwasm_std::coin(coins_after_fee.amount_of(&coin.denom).u128(), coin.denom),
+        ))
+    }
 }
 
 impl From<FeeConfig<Addr>> for FeeConfig<String> {
@@ -266,6 +285,27 @@ pub mod tests {
         assert_eq!(asset_after_fee.amount, Uint128::new(99));
     }
 
+    #[test]
+    fn fee_msgs_from_coin_works() {
+        let env = mock_env();
+
+        let fee_config = super::FeeConfig {
+            fee_rate: Decimal::percent(1),
+            fee_recipients: vec![(Addr::unchecked("addr1"), Decimal::percent(100))],
+        };
+        let coin = coin(100u128, "uusdc");
+        let (msgs, coin_after_fee) = fee_config.fee_msgs_from_coin(coin.clone(), &env).unwrap();
+        assert_eq!(msgs.len(), 1);
+        assert_eq!(
+            msgs[0],
+            CosmosMsg::Bank(BankMsg::Send {
+                to_address: "addr1".to_string(),
+                amount: vec![cosmwasm_std::coin(1u128, "uusdc".to_string())]
+            })
+        );
+        assert_eq!(coin_after_fee, cosmwasm_std::coin(99u128, "uusdc"));
+    }
+
     #[test]
     fn fee_msg_from_coins_works() {
         let env = mock_env();