feat(starknet_committer_and_os_cli): add command to dump the OS compiled program to file

Author: dorimedini-starkware

Cargo.lock

@@ -7,12 +7,10 @@ use crate::program_hash::{ProgramHash, PROGRAM_HASH_PATH};
 
 pub mod program_hash;
 
+pub const OS_PROGRAM_BYTES: &[u8] = include_bytes!(concat!(env!("OUT_DIR"), "/starknet_os_bytes"));
+
 pub const OS_PROGRAM: LazyLock<Program> = LazyLock::new(|| {
-    Program::from_bytes(
-        include_bytes!(concat!(env!("OUT_DIR"), "/starknet_os_bytes")),
-        Some("main"),
-    )
-    .expect("Failed to load the OS bytes.")
+    Program::from_bytes(OS_PROGRAM_BYTES, Some("main")).expect("Failed to load the OS bytes.")
 });
 
 pub const PROGRAM_HASH: LazyLock<ProgramHash> = LazyLock::new(|| {

crates/apollo_starknet_os_program/src/lib.rs

@@ -18,6 +18,7 @@ tempfile.workspace = true
 
 # TODO(Amos): Add `testing` feature and move Python test dependencies under it.
 [dependencies]
+apollo_starknet_os_program.workspace = true
 cairo-lang-starknet-classes.workspace = true
 cairo-vm.workspace = true                                   # Should be moved under `testing` feature, when it exists.
 clap = { workspace = true, features = ["cargo", "derive"] }

crates/starknet_committer_and_os_cli/Cargo.toml

@@ -1,6 +1,7 @@
 use std::fs;
 use std::path::Path;
 
+use apollo_starknet_os_program::OS_PROGRAM_BYTES;
 use cairo_lang_starknet_classes::casm_contract_class::CasmContractClass;
 use cairo_vm::types::layout_name::LayoutName;
 use rand_distr::num_traits::Zero;
@@ -73,3 +74,9 @@ pub fn parse_and_run_os(input_path: String, output_path: String) {
     write_to_file(&output_path, &output);
     info!("OS program ran successfully.");
 }
+
+pub(crate) fn dump_os_program(output_path: String) {
+    let os_program_json = serde_json::from_slice::<serde_json::Value>(OS_PROGRAM_BYTES)
+        .expect("OS bytes are JSON-serializable.");
+    write_to_file(&output_path, &os_program_json);
+}

crates/starknet_committer_and_os_cli/src/os_cli/commands.rs

@@ -4,7 +4,7 @@ use tracing::level_filters::LevelFilter;
 use tracing_subscriber::reload::Handle;
 use tracing_subscriber::Registry;
 
-use crate::os_cli::commands::parse_and_run_os;
+use crate::os_cli::commands::{dump_os_program, parse_and_run_os};
 use crate::os_cli::tests::python_tests::OsPythonTestRunner;
 use crate::shared_utils::types::{run_python_test, IoArgs, PythonTestArg};
 
@@ -16,6 +16,11 @@ pub struct OsCliCommand {
 
 #[derive(Debug, Subcommand)]
 enum Command {
+    DumpOsProgram {
+        /// File path to output.
+        #[clap(long, short = 'o', default_value = "stdout")]
+        output_path: String,
+    },
     PythonTest(PythonTestArg),
     RunOsStateless {
         #[clap(flatten)]
@@ -29,6 +34,7 @@ pub async fn run_os_cli(
 ) {
     info!("Starting starknet-os-cli with command: \n{:?}", os_command);
     match os_command.command {
+        Command::DumpOsProgram { output_path } => dump_os_program(output_path),
         Command::PythonTest(python_test_arg) => {
             run_python_test::<OsPythonTestRunner>(python_test_arg).await;
         }