Add_option_to_save_run_outputs_and_execution_resources_to_file

Author: YairVaknin-starkware

.github/workflows/upload_artifacts_workflow.yml

@@ -4,6 +4,7 @@ on:
   push:
     branches:
       - main
+      - yairv/add_option_to_save_run_outputs_and_execution_resources_to_file
 
 jobs:
   artifacts-push:

crates/cairo-program-runner-lib/src/hints/types.rs

@@ -10,6 +10,7 @@ use cairo_vm::types::layout::CairoLayoutParams;
 use cairo_vm::types::layout_name::LayoutName;
 use cairo_vm::types::program::Program;
 use cairo_vm::vm::runners::cairo_pie::{CairoPie, StrippedProgram};
+use cairo_vm::vm::runners::cairo_runner::ExecutionResources;
 use cairo_vm::Felt252;
 use num_traits::ToPrimitive;
 use serde::de::Error as SerdeError;
@@ -539,3 +540,9 @@ pub struct PedersenMerkleInput {
     pub prev_leaf: Felt252,
     pub new_leaf: Felt252,
 }
+
+#[derive(Serialize)]
+pub struct OutputER {
+    pub output: Vec<Felt252>,
+    pub execution_resources: ExecutionResources,
+}

crates/cairo-program-runner/src/bin/cairo_program_runner/main.rs

@@ -4,12 +4,13 @@ use std::io::{self, Write};
 use std::path::PathBuf;
 
 use bincode::enc::write::Writer;
+use cairo_program_runner_lib::types::OutputER;
 use cairo_program_runner_lib::utils::{get_cairo_run_config, get_program, get_program_input};
 use cairo_vm::types::layout_name::LayoutName;
 
 use cairo_program_runner_lib::cairo_run_program;
-use cairo_vm::cairo_run;
 use cairo_vm::vm::errors::cairo_run_errors::CairoRunError;
+use cairo_vm::{cairo_run, Felt252};
 use clap::Parser;
 use tempfile::NamedTempFile;
 
@@ -87,6 +88,11 @@ struct Args {
         the layout."
     )]
     allow_missing_builtins: bool,
+    #[clap(
+        long = "output_and_er_file",
+        help = "Write program output and execution resources as JSON to this file"
+    )]
+    output_and_er_file: Option<PathBuf>,
 }
 struct FileWriter {
     buf_writer: io::BufWriter<std::fs::File>,
@@ -137,7 +143,26 @@ fn main() -> Result<(), Box<dyn Error>> {
         args.allow_missing_builtins,
     )?;
 
-    let runner = cairo_run_program(&program, program_input_contents, cairo_run_config)?;
+    let mut runner = cairo_run_program(&program, program_input_contents, cairo_run_config)?;
+
+    if let Some(output_and_er_file) = args.output_and_er_file {
+        let mut output_buffer = String::new();
+        runner.vm.write_output(&mut output_buffer)?;
+        let output_lines = output_buffer
+            .lines()
+            .map(|line| {
+                Felt252::from_dec_str(line)
+                    .map_err(|_| format!("Failed to parse output line as Felt decimal: {}", line))
+            })
+            .collect::<Result<Vec<Felt252>, _>>()?;
+        let er = runner.get_execution_resources()?;
+        let to_dump = OutputER {
+            output: output_lines,
+            execution_resources: er,
+        };
+        let json = serde_json::to_string_pretty(&to_dump)?;
+        std::fs::write(&output_and_er_file, json)?;
+    }
 
     // Handle Cairo PIE output if specified
     if let Some(pie_output_path) = args.cairo_pie_output {