feat(starknet_os): compiled classes in OS input

starkware-libs · Feb 5, 2025 · 3623bce · 3623bce
1 parent 26bbea3
commit 3623bce
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/starknet_os/Cargo.toml b/crates/starknet_os/Cargo.toml
@@ -12,6 +12,7 @@ testing = ["dep:strum", "dep:strum_macros"]
 
 [dependencies]
 blockifier.workspace = true
+cairo-lang-starknet-classes.workspace = true
 cairo-vm.workspace = true
 indoc.workspace = true
 serde.workspace = true

diff --git a/crates/starknet_os/src/io/os_input.rs b/crates/starknet_os/src/io/os_input.rs
@@ -1,8 +1,10 @@
 use std::collections::HashMap;
 
+use cairo_lang_starknet_classes::casm_contract_class::CasmContractClass;
+use starknet_api::deprecated_contract_class::ContractClass;
 use starknet_committer::block_committer::input::StarknetStorageValue;
 use starknet_committer::patricia_merkle_tree::leaf::leaf_impl::ContractState;
-use starknet_committer::patricia_merkle_tree::types::CompiledClassHash;
+use starknet_committer::patricia_merkle_tree::types::{ClassHash, CompiledClassHash};
 use starknet_patricia::hash::hash_trait::HashOutput;
 use starknet_patricia::patricia_merkle_tree::node_data::inner_node::NodeData;
 use starknet_patricia::patricia_merkle_tree::node_data::leaf::Leaf;
@@ -24,4 +26,6 @@ pub struct StarknetOsInput {
     _contract_commitments: CommitmentInfo<ContractState>,
     _storage_commitments: CommitmentInfo<StarknetStorageValue>,
     _class_commitments: CommitmentInfo<CompiledClassHash>,
+    _deprecated_compiled_classes: HashMap<ClassHash, ContractClass>,
+    _compiled_classes: HashMap<ClassHash, CasmContractClass>,
 }