add replace code

tvm_debugger/src/main.rs

@@ -8,7 +8,10 @@ use std::path::PathBuf;
 
 use clap::ArgAction;
 use clap::Parser;
-
+use tvm_block::StateInit;
+use tvm_types::{base64_decode, read_single_root_boc};
+use tvm_block::Deserializable;
+use tvm_block::Serializable;
 use crate::execute::execute;
 use crate::result::ExecutionResult;
 
@@ -82,16 +85,43 @@ struct Args {
     /// Trace VM execution
     #[arg(long, action=ArgAction::SetTrue, default_value = "false")]
     trace: bool,
+
+    /// Update code in tvc without executing anything
+    #[arg(long)]
+    replace_code: Option<String>,
 }
 
 fn main() -> anyhow::Result<()> {
     let args: Args = Args::parse();
+    if let Some(   new_code) = args.replace_code {
+        replace_code(args.input_file, new_code)?;
+        return Ok(());
+    }
     let mut res: ExecutionResult = ExecutionResult::new(args.json);
     execute(&args, &mut res)?;
     println!("{}", res.output());
     Ok(())
 }
 
+fn replace_code(input_file: PathBuf, code: String) -> anyhow::Result<()> {
+    let mut contract_state_init =
+        StateInit::construct_from_file(&input_file).map_err(|e| {
+            anyhow::format_err!(
+                "Failed to load state init from input file {:?}: {e}",
+                input_file
+            )
+        })?;
+    let bytes = base64_decode(&code).map_err(|e| anyhow::format_err!("Failed to decode code as base64: {e}"))?;
+    let code_cell = read_single_root_boc(bytes).map_err(|e| anyhow::format_err!(
+                "Failed to construct code cell from base64 decoded code cell: {e}",
+            ))?;
+    contract_state_init.set_code(code_cell);
+    contract_state_init
+        .write_to_file(&input_file)
+        .map_err(|e| anyhow::format_err!("Failed to save state init after execution: {e}"))?;
+    Ok(())
+}
+
 #[cfg(test)]
 mod tests {
     use std::fs;
@@ -127,6 +157,7 @@ mod tests {
             decode_out_messages: false,
             json: true,
             trace: false,
+            replace_code: None,
         }
     }