From 134855b739d638a494f14bacb54b2640b00a1e6e Mon Sep 17 00:00:00 2001
From: Odysseas Gabrielides <odysseas.gabrielides@gmail.com>
Date: Thu, 9 Jan 2025 15:36:17 +0200
Subject: [PATCH] feat: move state sync read operations in consensus app

---
 packages/rs-drive-abci/src/abci/app/consensus.rs | 15 +++++++++++++++
 packages/rs-drive-abci/src/server.rs             |  5 -----
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/packages/rs-drive-abci/src/abci/app/consensus.rs b/packages/rs-drive-abci/src/abci/app/consensus.rs
index 0740c5c449..dd4cd4bb71 100644
--- a/packages/rs-drive-abci/src/abci/app/consensus.rs
+++ b/packages/rs-drive-abci/src/abci/app/consensus.rs
@@ -15,6 +15,7 @@ use drive::grovedb::Transaction;
 use std::fmt::Debug;
 use std::sync::RwLock;
 use tenderdash_abci::proto::abci as proto;
+use dapi_grpc::tonic;
 
 /// AbciApp is an implementation of ABCI Application, as defined by Tenderdash.
 ///
@@ -201,4 +202,18 @@ where
     ) -> Result<proto::ResponseApplySnapshotChunk, proto::ResponseException> {
         handler::apply_snapshot_chunk(self, request).map_err(error_into_exception)
     }
+
+    fn list_snapshots(
+        &self,
+        request: proto::RequestListSnapshots,
+    ) -> Result<proto::ResponseListSnapshots, proto::ResponseException> {
+        handler::list_snapshots(self, request).map_err(error_into_exception)
+    }
+
+    fn load_snapshot_chunk(
+        &self,
+        request: proto::RequestLoadSnapshotChunk,
+    ) -> Result<proto::ResponseLoadSnapshotChunk, proto::ResponseException> {
+        handler::load_snapshot_chunk(self, request).map_err(error_into_exception)
+    }
 }
diff --git a/packages/rs-drive-abci/src/server.rs b/packages/rs-drive-abci/src/server.rs
index 3011af708d..665f482bc6 100644
--- a/packages/rs-drive-abci/src/server.rs
+++ b/packages/rs-drive-abci/src/server.rs
@@ -40,11 +40,6 @@ pub fn start(
             tenderdash_abci::proto::abci::abci_application_server::AbciApplicationServer::new(
                 check_tx_service,
             ),
-        )
-        .add_service(
-            tenderdash_abci::proto::abci::abci_application_server::AbciApplicationServer::new(
-                state_source_service,
-            ),
         );
 
     let grpc_server_cancel = cancel.clone();