Add ibc_query_client_connections RPC

Author: majecty

core/src/ibc/connection_03/mod.rs

@@ -27,4 +27,5 @@ pub fn client_connections_path(client_id: &str) -> String {
 
 pub use manager::Manager;
 pub use types::ConnectionEnd;
+pub use types::ConnectionIdentifiersInClient;
 pub use types::ConnectionState;

core/src/ibc/connection_03/types.rs

@@ -66,6 +66,10 @@ impl ConnectionIdentifiersInClient {
     pub fn add(&mut self, identifier: Identifier) {
         self.0.push(identifier);
     }
+
+    pub fn into_vec(self) -> Vec<Identifier> {
+        self.0
+    }
 }
 
 #[cfg(test)]

core/src/ibc/querier/mod.rs

@@ -45,6 +45,12 @@ impl DebugName for ibc::client_02::types::ConsensusState {
     }
 }
 
+impl DebugName for ibc::connection_03::types::ConnectionIdentifiersInClient {
+    fn debug_name() -> &'static str {
+        "ConnectionIdentifiersInClient"
+    }
+}
+
 /// Queries the path and returns the result in decoded struct
 pub fn query<T>(ctx: &dyn ibc::Context, path: &CommitmentPath) -> Option<T>
 where
@@ -89,3 +95,9 @@ pub fn path_connection_end(id: IdentifierSlice) -> CommitmentPath {
         raw: ibc::connection_03::path(id),
     }
 }
+
+pub fn path_connection_identifiers(client_id: IdentifierSlice) -> CommitmentPath {
+    CommitmentPath {
+        raw: ibc::connection_03::client_connections_path(client_id),
+    }
+}

rpc/src/v1/impls/ibc.rs

@@ -16,7 +16,7 @@
 
 use super::super::errors;
 use super::super::traits::IBC;
-use super::super::types::{ClientState, ConnectionEnd, ConsensusState, IBCQuery};
+use super::super::types::{ClientState, ConnectionEnd, ConnectionIdentifiersInClient, ConsensusState, IBCQuery};
 use ccore::ibc;
 use ccore::ibc::querier;
 use ccore::{BlockChainClient, BlockId, StateInfo};
@@ -151,4 +151,29 @@ where
         };
         Ok(Some(response))
     }
+
+    fn query_client_connections(
+        &self,
+        client_identifier: String,
+        block_number: Option<u64>,
+    ) -> Result<Option<IBCQuery<ConnectionIdentifiersInClient>>> {
+        let block_id = block_number.map(BlockId::Number).unwrap_or(BlockId::Latest);
+        let mut state = self.client.state_at(block_id).ok_or_else(errors::state_not_exist)?;
+        let block_number = match self.client.block_number(&block_id) {
+            None => return Ok(None),
+            Some(block_number) => block_number,
+        };
+
+        let context = ibc::context::TopLevelContext::new(&mut state, block_number);
+        let path = querier::path_connection_identifiers(&client_identifier);
+        let connections_in_client: Option<ibc::connection_03::ConnectionIdentifiersInClient> =
+            querier::query(&context, &path);
+
+        let response = IBCQuery {
+            number: block_number,
+            data: connections_in_client.map(|from_core| from_core.into_vec()),
+            proof: querier::make_proof(&context, &path),
+        };
+        Ok(Some(response))
+    }
 }

rpc/src/v1/traits/ibc.rs

@@ -15,7 +15,7 @@
 // along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 use super::super::types::IBCQuery;
-use super::super::types::{ClientState, ConnectionEnd, ConsensusState};
+use super::super::types::{ClientState, ConnectionEnd, ConnectionIdentifiersInClient, ConsensusState};
 use jsonrpc_core::Result;
 use primitives::Bytes;
 
@@ -46,4 +46,11 @@ pub trait IBC {
         identifier: String,
         block_number: Option<u64>,
     ) -> Result<Option<IBCQuery<ConnectionEnd>>>;
+
+    #[rpc(name = "ibc_query_client_connections")]
+    fn query_client_connections(
+        &self,
+        client_identifier: String,
+        block_number: Option<u64>,
+    ) -> Result<Option<IBCQuery<ConnectionIdentifiersInClient>>>;
 }

rpc/src/v1/types/ibc.rs

@@ -113,3 +113,5 @@ impl ConnectionEnd {
         }
     }
 }
+
+pub type ConnectionIdentifiersInClient = Vec<String>;

rpc/src/v1/types/mod.rs

@@ -25,7 +25,7 @@ mod work;
 pub use self::action::{Action, ActionWithTracker};
 pub use self::block::Block;
 pub use self::block::BlockNumberAndHash;
-pub use self::ibc::{ClientState, ConnectionEnd, ConsensusState, IBCQuery};
+pub use self::ibc::{ClientState, ConnectionEnd, ConnectionIdentifiersInClient, ConsensusState, IBCQuery};
 pub use self::mem_pool::MemPoolMinFees;
 pub use self::transaction::{PendingTransactions, Transaction};
 pub use self::unsigned_transaction::UnsignedTransaction;