Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement CanQueryUnreceivedAcksSequences for StarknetChain #260

Merged
merged 8 commits into from
Feb 10, 2025
Original file line number Diff line number Diff line change
Expand Up @@ -89,6 +89,7 @@ use crate::impls::queries::packet_receipt::QueryStarknetPacketReceipt;
use crate::impls::queries::packet_received::QueryPacketIsReceivedOnStarknet;
use crate::impls::queries::status::QueryStarknetChainStatus;
use crate::impls::queries::token_balance::QueryErc20TokenBalance;
use crate::impls::queries::unreceived_acks::QueryStarknetUnreceivedAckSequences;
use crate::impls::queries::unreceived_packets::QueryStarknetUnreceivedPacketSequences;
use crate::impls::send_message::SendCallMessages;
use crate::impls::submit_tx::SubmitCallTransaction;
Expand Down Expand Up @@ -387,6 +388,8 @@ cgp_preset! {
QueryStarknetPacketCommitment,
UnreceivedPacketSequencesQuerierComponent:
QueryStarknetUnreceivedPacketSequences,
UnreceivedAcksSequencesQuerierComponent:
QueryStarknetUnreceivedAckSequences,
PacketAcknowledgementQuerierComponent:
QueryStarknetAckCommitment,
PacketReceiptQuerierComponent:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,4 +12,5 @@ pub mod packet_receipt;
pub mod packet_received;
pub mod status;
pub mod token_balance;
pub mod unreceived_acks;
pub mod unreceived_packets;
Original file line number Diff line number Diff line change
@@ -0,0 +1,75 @@
use core::marker::PhantomData;

use cgp::prelude::*;
use hermes_cairo_encoding_components::strategy::ViaCairo;
use hermes_cairo_encoding_components::types::as_felt::AsFelt;
use hermes_chain_components::traits::queries::unreceived_acks_sequences::UnreceivedAcksSequencesQuerier;
use hermes_chain_components::traits::types::ibc::{
HasChannelIdType, HasIbcChainTypes, HasPortIdType, HasSequenceType,
};
use hermes_encoding_components::traits::decode::CanDecode;
use hermes_encoding_components::traits::encode::CanEncode;
use hermes_encoding_components::traits::has_encoding::HasEncoding;
use hermes_encoding_components::traits::types::encoded::HasEncodedType;
use ibc::core::host::types::identifiers::{PortId as IbcPortId, Sequence as IbcSequence};
use starknet::core::types::Felt;
use starknet::macros::selector;

use crate::traits::contract::call::CanCallContract;
use crate::traits::queries::address::CanQueryContractAddress;
use crate::traits::types::blob::HasBlobType;
use crate::traits::types::method::HasSelectorType;
use crate::types::channel_id::ChannelId;
use crate::types::messages::ibc::channel::PortId as CairoPortId;
use crate::types::messages::ibc::packet::Sequence;

pub struct QueryStarknetUnreceivedAckSequences;

impl<Chain, Counterparty, Encoding> UnreceivedAcksSequencesQuerier<Chain, Counterparty>
for QueryStarknetUnreceivedAckSequences
where
Chain: HasChannelIdType<Counterparty, ChannelId = ChannelId>
+ HasPortIdType<Counterparty, PortId = IbcPortId>
+ HasSequenceType<Counterparty, Sequence = IbcSequence>
+ HasBlobType<Blob = Vec<Felt>>
+ HasSelectorType<Selector = Felt>
+ CanQueryContractAddress<symbol!("ibc_core_contract_address")>
+ HasEncoding<AsFelt, Encoding = Encoding>
+ CanCallContract
+ CanRaiseAsyncError<Encoding::Error>,
Counterparty: HasIbcChainTypes<Chain, Sequence = IbcSequence>,
Encoding: CanEncode<ViaCairo, Product![CairoPortId, ChannelId, Vec<Sequence>]>
+ CanDecode<ViaCairo, Product![Vec<Sequence>]>
+ HasEncodedType<Encoded = Vec<Felt>>,
{
async fn query_unreceived_acknowledgments_sequences(
chain: &Chain,
channel_id: &ChannelId,
port_id: &IbcPortId,
packet_ack_sequences: &[IbcSequence],
) -> Result<Vec<IbcSequence>, Chain::Error> {
let encoding = chain.encoding();

let contract_address = chain.query_contract_address(PhantomData).await?;

let calldata = encoding
.encode(&product![
port_id.clone(),
channel_id.clone(),
packet_ack_sequences.to_owned()
])
.map_err(Chain::raise_error)?;

let output = chain
.call_contract(
&contract_address,
&selector!("unreceived_ack_sequences"),
&calldata,
)
.await?;

let product![ack_sequences,] = encoding.decode(&output).map_err(Chain::raise_error)?;

Ok(ack_sequences)
}
}
Loading
Loading