apollo_storage: implement starknet version query in storage reader (#11210)

Author: nadin-Starkware

crates/apollo_storage/src/header.rs

@@ -133,6 +133,12 @@ pub trait HeaderStorageReader {
         block_number: BlockNumber,
     ) -> StorageResult<Option<StarknetVersion>>;
 
+    /// Returns the Starknet version for a given block number using direct key-value lookup.
+    fn get_starknet_version_by_key(
+        &self,
+        block_number: BlockNumber,
+    ) -> StorageResult<Option<StarknetVersion>>;
+
     /// Returns the signature of the block with the given number.
     fn get_block_signature(
         &self,
@@ -251,6 +257,15 @@ impl<Mode: TransactionKind> HeaderStorageReader for StorageTxn<'_, Mode> {
         }
     }
 
+    fn get_starknet_version_by_key(
+        &self,
+        block_number: BlockNumber,
+    ) -> StorageResult<Option<StarknetVersion>> {
+        let starknet_version_table = self.open_table(&self.tables.starknet_version)?;
+        let starknet_version = starknet_version_table.get(&self.txn, &block_number)?;
+        Ok(starknet_version)
+    }
+
     fn get_block_signature(
         &self,
         block_number: BlockNumber,

crates/apollo_storage/src/storage_reader_types.rs

@@ -314,8 +314,14 @@ impl StorageReaderServerHandler<StorageReaderRequest, StorageReaderResponse>
             StorageReaderRequest::FileOffsets(_offset_kind) => {
                 unimplemented!()
             }
-            StorageReaderRequest::StarknetVersion(_block_number) => {
-                unimplemented!()
+            StorageReaderRequest::StarknetVersion(block_number) => {
+                let starknet_version = txn.get_starknet_version_by_key(block_number)?.ok_or(
+                    StorageError::NotFound {
+                        resource_type: "Starknet version".to_string(),
+                        resource_id: format!("block: {}", block_number),
+                    },
+                )?;
+                Ok(StorageReaderResponse::StarknetVersion(starknet_version))
             }
             StorageReaderRequest::StateStorageVersion => {
                 let version = txn.get_state_version()?.ok_or(StorageError::NotFound {