feat: pass through burn_from_address and mint_to_address

neutron-org · Aug 1, 2024 · c931dd4 · c931dd4
1 parent 0406eae
commit c931dd4
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 5 deletions.
diff --git a/contracts/tokenfactory/schema/execute_msg.json b/contracts/tokenfactory/schema/execute_msg.json
@@ -64,6 +64,12 @@
             },
             "denom": {
               "type": "string"
+            },
+            "mint_to_address": {
+              "type": [
+                "string",
+                "null"
+              ]
             }
           }
         }
@@ -86,6 +92,12 @@
             "amount": {
               "$ref": "#/definitions/Uint128"
             },
+            "burn_from_address": {
+              "type": [
+                "string",
+                "null"
+              ]
+            },
             "denom": {
               "type": "string"
             }

diff --git a/contracts/tokenfactory/src/contract.rs b/contracts/tokenfactory/src/contract.rs
@@ -33,11 +33,28 @@ pub fn execute(
             denom,
             new_admin_address,
         } => NeutronMsg::submit_change_admin(denom, new_admin_address).into(),
-        ExecuteMsg::MintTokens { denom, amount } => {
-            NeutronMsg::submit_mint_tokens(denom, amount, env.contract.address).into()
-        }
-        ExecuteMsg::BurnTokens { denom, amount } => {
-            NeutronMsg::submit_burn_tokens(denom, amount).into()
+        ExecuteMsg::MintTokens {
+            denom,
+            amount,
+            mint_to_address,
+        } => NeutronMsg::submit_mint_tokens(
+            denom,
+            amount,
+            mint_to_address.unwrap_or(env.contract.address.into()),
+        )
+        .into(),
+        ExecuteMsg::BurnTokens {
+            denom,
+            amount,
+            burn_from_address,
+        } => {
+            // TODO: update neutron-sdk `submit_burn_tokens`
+            NeutronMsg::BurnTokens {
+                denom,
+                amount,
+                burn_from_address: burn_from_address.unwrap_or_default(),
+            }
+            .into()
         }
         ExecuteMsg::SetBeforeSendHook {
             denom,

diff --git a/contracts/tokenfactory/src/msg.rs b/contracts/tokenfactory/src/msg.rs
@@ -18,10 +18,12 @@ pub enum ExecuteMsg {
     MintTokens {
         denom: String,
         amount: Uint128,
+        mint_to_address: Option<String>,
     },
     BurnTokens {
         denom: String,
         amount: Uint128,
+        burn_from_address: Option<String>,
     },
     SendTokens {
         recipient: String,