feat(torii-indexer): add option for strict model reader block

crates/torii/cli/src/options.rs

@@ -90,6 +90,16 @@
 #[derive(Debug, clap::Args, Clone, Serialize, Deserialize, PartialEq)]
 #[command(next_help_heading = "Indexing options")]
 pub struct IndexingOptions {
+    /// Whether or not to read models from the block number they were registered in.
+    /// If false, models will be read from the latest block.
+    #[arg(
+        long = "indexing.strict_model_reader",
+        default_value_t = false,
+        help = "Whether or not to read models from the block number they were registered in."
+    )]
+    #[serde(default)]
+    pub strict_model_reader: bool,
+
     /// Chunk size of the events page when indexing using events
     #[arg(long = "indexing.events_chunk_size", default_value_t = DEFAULT_EVENTS_CHUNK_SIZE, help = "Chunk size of the events page to fetch from the sequencer.")]
     #[serde(default = "default_events_chunk_size")]
@@ -175,6 +185,7 @@
 impl Default for IndexingOptions {
     fn default() -> Self {
         Self {
+            strict_model_reader: false,
             events_chunk_size: DEFAULT_EVENTS_CHUNK_SIZE,
             blocks_chunk_size: DEFAULT_BLOCKS_CHUNK_SIZE,
             pending: true,

crates/torii/indexer/src/processors/mod.rs

@@ -30,6 +30,7 @@ pub mod upgrade_model;
 pub struct EventProcessorConfig {
     pub historical_events: HashSet<String>,
     pub namespaces: HashSet<String>,
+    pub strict_model_reader: bool,
 }
 
 impl EventProcessorConfig {

crates/torii/indexer/src/processors/register_event.rs

@@ -5,7 +5,7 @@
 use dojo_world::contracts::abigen::world::Event as WorldEvent;
 use dojo_world::contracts::model::ModelReader;
 use dojo_world::contracts::world::WorldContractReader;
-use starknet::core::types::Event;
+use starknet::core::types::{BlockId, Event};
 use starknet::providers::Provider;
 use torii_sqlite::Sql;
 use tracing::{debug, info};
@@ -47,7 +47,7 @@
         &self,
         world: &WorldContractReader<P>,
         db: &mut Sql,
-        _block_number: u64,
+        block_number: u64,
         block_timestamp: u64,
         _event_id: &str,
         event: &Event,
@@ -79,7 +79,11 @@
 
         // Called model here by language, but it's an event. Torii rework will make clear
         // distinction.
-        let model = world.model_reader(&namespace, &name).await?;
+        let model = if config.strict_model_reader {
+            world.model_reader_with_block(&namespace, &name, BlockId::Number(block_number)).await?
+        } else {
+            world.model_reader(&namespace, &name).await?
+        };
         let schema = model.schema().await?;
         let layout = model.layout().await?;
 

crates/torii/indexer/src/processors/register_model.rs

@@ -5,7 +5,7 @@
 use dojo_world::contracts::abigen::world::Event as WorldEvent;
 use dojo_world::contracts::model::ModelReader;
 use dojo_world::contracts::world::WorldContractReader;
-use starknet::core::types::Event;
+use starknet::core::types::{BlockId, Event};
 use starknet::providers::Provider;
 use torii_sqlite::Sql;
 use tracing::{debug, info};
@@ -47,7 +47,7 @@
         &self,
         world: &WorldContractReader<P>,
         db: &mut Sql,
-        _block_number: u64,
+        block_number: u64,
         block_timestamp: u64,
         _event_id: &str,
         event: &Event,
@@ -77,7 +77,11 @@
             return Ok(());
         }
 
-        let model = world.model_reader(&namespace, &name).await?;
+        let model = if config.strict_model_reader {
+            world.model_reader_with_block(&namespace, &name, BlockId::Number(block_number)).await?
+        } else {
+            world.model_reader(&namespace, &name).await?
+        };
         let schema = model.schema().await?;
         let layout = model.layout().await?;
 

crates/torii/indexer/src/processors/upgrade_event.rs

@@ -5,7 +5,7 @@
 use dojo_world::contracts::abigen::world::Event as WorldEvent;
 use dojo_world::contracts::model::ModelReader;
 use dojo_world::contracts::world::WorldContractReader;
-use starknet::core::types::Event;
+use starknet::core::types::{BlockId, Event};
 use starknet::providers::Provider;
 use torii_sqlite::Sql;
 use tracing::{debug, info};
@@ -47,11 +47,11 @@
         &self,
         world: &WorldContractReader<P>,
         db: &mut Sql,
-        _block_number: u64,
+        block_number: u64,
         block_timestamp: u64,
         _event_id: &str,
         event: &Event,
-        _config: &EventProcessorConfig,
+        config: &EventProcessorConfig,
     ) -> Result<(), Error> {
         // Torii version is coupled to the world version, so we can expect the event to be well
         // formed.
@@ -88,7 +88,11 @@
         let namespace = model.namespace;
         let prev_schema = model.schema;
 
-        let model = world.model_reader(&namespace, &name).await?;
+        let model = if config.strict_model_reader {
+            world.model_reader_with_block(&namespace, &name, BlockId::Number(block_number)).await?
+        } else {
+            world.model_reader(&namespace, &name).await?
+        };
         let new_schema = model.schema().await?;
         let schema_diff = new_schema.diff(&prev_schema);
         // No changes to the schema. This can happen if torii is re-run with a fresh database.

crates/torii/indexer/src/processors/upgrade_model.rs

@@ -5,7 +5,7 @@
 use dojo_world::contracts::abigen::world::Event as WorldEvent;
 use dojo_world::contracts::model::ModelReader;
 use dojo_world::contracts::world::WorldContractReader;
-use starknet::core::types::Event;
+use starknet::core::types::{BlockId, Event};
 use starknet::providers::Provider;
 use torii_sqlite::Sql;
 use tracing::{debug, info};
@@ -47,11 +47,11 @@
         &self,
         world: &WorldContractReader<P>,
         db: &mut Sql,
-        _block_number: u64,
+        block_number: u64,
         block_timestamp: u64,
         _event_id: &str,
         event: &Event,
-        _config: &EventProcessorConfig,
+        config: &EventProcessorConfig,
     ) -> Result<(), Error> {
         // Torii version is coupled to the world version, so we can expect the event to be well
         // formed.
@@ -86,7 +86,11 @@
         let namespace = model.namespace;
         let prev_schema = model.schema;
 
-        let model = world.model_reader(&namespace, &name).await?;
+        let model = if config.strict_model_reader {
+            world.model_reader_with_block(&namespace, &name, BlockId::Number(block_number)).await?
+        } else {
+            world.model_reader(&namespace, &name).await?
+        };
         let new_schema = model.schema().await?;
         let schema_diff = new_schema.diff(&prev_schema);
         // No changes to the schema. This can happen if torii is re-run with a fresh database.

crates/torii/runner/src/lib.rs

@@ -177,6 +177,7 @@
                 polling_interval: Duration::from_millis(self.args.indexing.polling_interval),
                 flags,
                 event_processor_config: EventProcessorConfig {
+                    strict_model_reader: self.args.indexing.strict_model_reader,
                     historical_events: self.args.events.historical.into_iter().collect(),
                     namespaces: self.args.indexing.namespaces.into_iter().collect(),
                 },