add optional post_init_data to set in result in InstantiateMsg

a hack to workaround scrtlabs/SecretNetwork#1323

Useful when you want to instantiate this contract and return data back to the calling contract that is available in the Reply

src/contract.rs

@@ -4,8 +4,8 @@ use std::collections::HashSet;
 
 use base64::{engine::general_purpose, Engine as _};
 use cosmwasm_std::{
-    Addr, Api, attr, Binary, BlockInfo, CanonicalAddr, CosmosMsg, Deps, DepsMut, Env,
-    MessageInfo, Response, StdError, StdResult, Storage, to_binary, WasmMsg,
+    attr, to_binary, Addr, Api, Binary, BlockInfo, CanonicalAddr, CosmosMsg, Deps, DepsMut, Env,
+    MessageInfo, Response, StdError, StdResult, Storage, WasmMsg,
 };
 use cosmwasm_storage::{PrefixedStorage, ReadonlyPrefixedStorage};
 use primitive_types::U256;
@@ -121,7 +121,12 @@ pub fn instantiate(
     } else {
         Vec::new()
     };
-    Ok(Response::new().add_messages(messages))
+    let res = Response::new().add_messages(messages);
+    if let Some(some_post_init_data) = msg.post_init_data {
+        Ok(res.set_data(some_post_init_data))
+    } else {
+        Ok(res)
+    }
 }
 
 ///////////////////////////////////// Handle //////////////////////////////////////
@@ -133,12 +138,7 @@ pub fn instantiate(
 /// * `env` - Env of contract's environment
 /// * `info` - contract execution info for authorization - identity of the call, and payment.
 /// * `msg` - HandleMsg passed in with the execute message
-pub fn execute(
-    deps: DepsMut,
-    env: Env,
-    info: MessageInfo,
-    msg: ExecuteMsg,
-) -> StdResult<Response> {
+pub fn execute(deps: DepsMut, env: Env, info: MessageInfo, msg: ExecuteMsg) -> StdResult<Response> {
     let mut config: Config = load(deps.storage, CONFIG_KEY)?;
     let mut deps = deps;
     let mut _deps = &mut deps;

src/msg.rs

@@ -31,6 +31,9 @@ pub struct InstantiateMsg {
     /// contract that instantiated it, but it could be used to execute any
     /// contract
     pub post_init_callback: Option<PostInstantiateCallback>,
+    /// optional data to set in result
+    /// a hack to workaround https://github.com/scrtlabs/SecretNetwork/issues/1323
+    pub post_init_data: Option<Binary>,
 }
 
 /// This type represents optional configuration values.

src/unittest_handles.rs

@@ -47,6 +47,7 @@ mod tests {
             royalty_info: None,
             config: None,
             post_init_callback: None,
+            post_init_data: None,
         };
 
         (instantiate(&mut deps.as_mut(), &env, info, init_msg), deps)
@@ -96,6 +97,7 @@ mod tests {
             royalty_info: None,
             config: Some(init_config),
             post_init_callback: None,
+            post_init_data: None,
         };
 
         (instantiate(&mut deps.as_mut(), &env, info, init_msg), deps)
@@ -183,6 +185,7 @@ mod tests {
             royalty_info: None,
             config: None,
             post_init_callback,
+            post_init_data: None,
         };
 
         let init_response = instantiate(&mut deps.as_mut(), &env, info, init_msg).unwrap();

src/unittest_mint_run.rs

@@ -26,6 +26,7 @@ mod tests {
             royalty_info: None,
             config: None,
             post_init_callback: None,
+            post_init_data: None,
         };
 
         (instantiate(&mut deps.as_mut(), &env, info, init_msg), deps)

src/unittest_non_transferable.rs

@@ -40,6 +40,7 @@ mod tests {
             royalty_info: None,
             config: None,
             post_init_callback: None,
+            post_init_data: None,
         };
 
         (instantiate(&mut deps.as_mut(), &env, info, init_msg), deps)
@@ -89,6 +90,7 @@ mod tests {
             royalty_info: None,
             config: Some(init_config),
             post_init_callback: None,
+            post_init_data: None,
         };
 
         (instantiate(&mut deps.as_mut(), &env, info, init_msg), deps)
@@ -140,6 +142,7 @@ mod tests {
             royalty_info,
             config: Some(init_config),
             post_init_callback: None,
+            post_init_data: None,
         };
 
         (instantiate(&mut deps.as_mut(), &env, info, init_msg), deps)
@@ -220,6 +223,7 @@ mod tests {
             royalty_info: None,
             config: None,
             post_init_callback,
+            post_init_data: None,
         };
 
         let init_response = instantiate(&mut deps.as_mut(), &env, info, init_msg).unwrap();

src/unittest_queries.rs

@@ -34,6 +34,7 @@ mod tests {
             royalty_info: None,
             config: None,
             post_init_callback: None,
+            post_init_data: None,
         };
 
         (instantiate(&mut deps.as_mut(), &env, info, init_msg), deps)
@@ -83,6 +84,7 @@ mod tests {
             royalty_info: None,
             config: Some(init_config),
             post_init_callback: None,
+            post_init_data: None,
         };
 
         (instantiate(&mut deps.as_mut(), &env, info, init_msg), deps)

src/unittest_royalties.rs

@@ -35,6 +35,7 @@ mod tests {
             royalty_info,
             config: None,
             post_init_callback: None,
+            post_init_data: None,
         };
 
         (instantiate(&mut deps.as_mut(), &env, info, init_msg), deps)
@@ -86,6 +87,7 @@ mod tests {
             royalty_info,
             config: Some(init_config),
             post_init_callback: None,
+            post_init_data: None,
         };
 
         (instantiate(&mut deps.as_mut(), &env, info, init_msg), deps)
@@ -286,6 +288,7 @@ mod tests {
             royalty_info: None,
             config: None,
             post_init_callback,
+            post_init_data: None,
         };
 
         let init_response = instantiate(&mut deps.as_mut(), &env, info, init_msg).unwrap();