blockifier: get proof_facts_length from tx

Author: avivg-starkware

crates/blockifier/src/fee/receipt.rs

@@ -184,8 +184,7 @@ impl TransactionReceipt {
             tx_type: account_tx.tx_type(),
             reverted_steps,
             reverted_sierra_gas,
-            // TODO(AvivG): Get proof facts length from account_tx when it supports retrieving it.
-            proof_facts_length: 0,
+            proof_facts_length: account_tx.proof_facts_length(),
         })
     }
 }

crates/blockifier/src/transaction/account_transaction.rs

@@ -220,6 +220,10 @@ impl AccountTransaction {
         self.signature().0.len()
     }
 
+    pub fn proof_facts_length(&self) -> usize {
+        if let Transaction::Invoke(tx) = &self.tx { tx.proof_facts_length() } else { 0 }
+    }
+
     pub fn enforce_fee(&self) -> bool {
         self.create_tx_info().enforce_fee()
     }

crates/blockifier/src/transaction/transactions_test.rs

@@ -623,6 +623,7 @@ fn test_invoke_tx(
     // transaction.
     let calldata_length = invoke_tx.calldata_length();
     let signature_length = invoke_tx.signature_length();
+    let proof_facts_length = invoke_tx.proof_facts_length();
     let state_changes_for_fee = StateChangesCount {
         n_storage_updates: 1,
         n_modified_contracts: 1,
@@ -635,8 +636,7 @@ fn test_invoke_tx(
         StateResources::new_for_testing(state_changes_for_fee, 0),
         None,
         ExecutionSummary::default(),
-        // TODO(AvivG): Test non-zero proof facts length after invoke_tx supports retrieving it.
-        0,
+        proof_facts_length,
     );
     let sender_address = invoke_tx.sender_address();
 

crates/starknet_api/src/executable_transaction.rs

@@ -345,6 +345,16 @@ impl InvokeTransaction {
         let tx_hash = invoke_tx.calculate_transaction_hash(chain_id, &invoke_tx.version())?;
         Ok(Self { tx: invoke_tx, tx_hash })
     }
+
+    /// Returns the length of proof facts for client-side proving.
+    pub fn proof_facts_length(&self) -> usize {
+        match &self.tx {
+            crate::transaction::InvokeTransaction::V3(tx) => tx.proof_facts.0.len(),
+            // Client-side proving is only supported starting from V3.
+            crate::transaction::InvokeTransaction::V0(_)
+            | crate::transaction::InvokeTransaction::V1(_) => 0,
+        }
+    }
 }
 
 #[derive(Clone, Debug, Default, Deserialize, Eq, PartialEq, Serialize, Hash)]