server: snapshot

crates/brk_query/src/impl/block/status.rs

@@ -0,0 +1,43 @@
+use brk_error::Result;
+use brk_types::{BlockStatus, Height};
+use vecdb::{AnyVec, GenericStoredVec};
+
+use crate::Query;
+
+impl Query {
+    pub fn block_status(&self, hash: &str) -> Result<BlockStatus> {
+        let height = self.height_by_hash(hash)?;
+        self.block_status_by_height(height)
+    }
+
+    fn block_status_by_height(&self, height: Height) -> Result<BlockStatus> {
+        let indexer = self.indexer();
+
+        let max_height = Height::from(
+            indexer
+                .vecs
+                .block
+                .height_to_blockhash
+                .len()
+                .saturating_sub(1),
+        );
+
+        if height > max_height {
+            return Ok(BlockStatus::not_in_best_chain());
+        }
+
+        let next_best = if height < max_height {
+            Some(
+                indexer
+                    .vecs
+                    .block
+                    .height_to_blockhash
+                    .read_once(height.incremented())?,
+            )
+        } else {
+            None
+        };
+
+        Ok(BlockStatus::in_best_chain(height, next_best))
+    }
+}