feat(proof_data_handler): add new endpoints to the TEE prover interface API

Cargo.toml

@@ -116,6 +116,7 @@ google-cloud-storage = "0.15.0"
 governor = "0.4.2"
 hex = "0.4"
 http = "0.2.9"
+hyper = "0.14.29"
 iai = "0.1"
 insta = "1.29.0"
 itertools = "0.10"

checks-config/era.dic

@@ -973,3 +973,11 @@ uncached
 untrimmed
 UNNEST
 semver
+TeeRequestProcessor
+l1_batch_number
+RequestProcessorError
+map_err
+proof_inputs
+submit_proofs
+ready_to_be_proven
+privkey

core/lib/config/src/configs/proof_data_handler.rs

@@ -6,6 +6,7 @@ use serde::Deserialize;
 pub struct ProofDataHandlerConfig {
     pub http_port: u16,
     pub proof_generation_timeout_in_secs: u16,
+    pub tee_support: bool,
 }
 
 impl ProofDataHandlerConfig {

core/lib/config/src/testonly.rs

@@ -626,6 +626,7 @@ impl Distribution<configs::ProofDataHandlerConfig> for EncodeDist {
         configs::ProofDataHandlerConfig {
             http_port: self.sample(rng),
             proof_generation_timeout_in_secs: self.sample(rng),
+            tee_support: self.sample(rng),
         }
     }
 }

core/lib/dal/migrations/20240523085604_add_tee_proof_generation_details_table.down.sql

@@ -0,0 +1,4 @@
+DROP TABLE IF EXISTS tee_attestations;
+DROP TABLE IF EXISTS tee_proof_generation_details;
+
+DROP INDEX IF EXISTS idx_tee_proof_generation_details_status_prover_taken_at;

core/lib/dal/migrations/20240523085604_add_tee_proof_generation_details_table.up.sql

@@ -0,0 +1,22 @@
+CREATE TABLE IF NOT EXISTS tee_attestations
+(
+    pubkey                  BYTEA PRIMARY KEY,
+    attestation             BYTEA
+);
+
+CREATE TABLE IF NOT EXISTS tee_proof_generation_details
+(
+    l1_batch_number         BIGINT PRIMARY KEY REFERENCES tee_verifier_input_producer_jobs (l1_batch_number) ON DELETE CASCADE,
+    status                  TEXT      NOT NULL,
+    signature               BYTEA,
+    pubkey                  BYTEA REFERENCES tee_attestations (pubkey) ON DELETE SET NULL,
+    proof                   BYTEA,
+    tee_type                TEXT,
+    created_at              TIMESTAMP NOT NULL,
+    updated_at              TIMESTAMP NOT NULL,
+    prover_taken_at         TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS idx_tee_proof_generation_details_status_prover_taken_at
+    ON tee_proof_generation_details (prover_taken_at)
+    WHERE status = 'picked_by_prover';

core/lib/dal/src/lib.rs

@@ -20,7 +20,7 @@ use crate::{
     snapshot_recovery_dal::SnapshotRecoveryDal, snapshots_creator_dal::SnapshotsCreatorDal,
     snapshots_dal::SnapshotsDal, storage_logs_dal::StorageLogsDal,
     storage_logs_dedup_dal::StorageLogsDedupDal, storage_web3_dal::StorageWeb3Dal,
-    sync_dal::SyncDal, system_dal::SystemDal,
+    sync_dal::SyncDal, system_dal::SystemDal, tee_proof_generation_dal::TeeProofGenerationDal,
     tee_verifier_input_producer_dal::TeeVerifierInputProducerDal, tokens_dal::TokensDal,
     tokens_web3_dal::TokensWeb3Dal, transactions_dal::TransactionsDal,
     transactions_web3_dal::TransactionsWeb3Dal, vm_runner_dal::VmRunnerDal,
@@ -50,6 +50,7 @@ pub mod storage_logs_dedup_dal;
 pub mod storage_web3_dal;
 pub mod sync_dal;
 pub mod system_dal;
+pub mod tee_proof_generation_dal;
 pub mod tee_verifier_input_producer_dal;
 pub mod tokens_dal;
 pub mod tokens_web3_dal;
@@ -111,6 +112,8 @@ where
 
     fn proof_generation_dal(&mut self) -> ProofGenerationDal<'_, 'a>;
 
+    fn tee_proof_generation_dal(&mut self) -> TeeProofGenerationDal<'_, 'a>;
+
     fn system_dal(&mut self) -> SystemDal<'_, 'a>;
 
     fn snapshots_dal(&mut self) -> SnapshotsDal<'_, 'a>;
@@ -213,6 +216,10 @@ impl<'a> CoreDal<'a> for Connection<'a, Core> {
         ProofGenerationDal { storage: self }
     }
 
+    fn tee_proof_generation_dal(&mut self) -> TeeProofGenerationDal<'_, 'a> {
+        TeeProofGenerationDal { storage: self }
+    }
+
     fn system_dal(&mut self) -> SystemDal<'_, 'a> {
         SystemDal { storage: self }
     }