use cynic::SubscriptionBuilder;
use futures::{Stream, TryStreamExt};
use super::{BlokliClient, GraphQlQueries};
use crate::api::{
AccountSelector, BlokliSubscriptionClient, ChannelSelector, Result, TicketSelector, TxId,
internal::{
AccountVariables, ChannelsVariables, SubscribeAccounts, SubscribeChannels, SubscribeGraph, SubscribeHealth,
SubscribeSafeDeployment, SubscribeTicketParams, SubscribeTicketRedeemed, TicketRedeemedVariables,
},
types::{
Account, Channel, OpenedChannelsGraphEntry, ReadinessState, RedeemTicketDetails, Safe, TicketParameters,
Transaction,
},
};
impl GraphQlQueries {
pub fn subscribe_channels(
selector: ChannelSelector,
) -> cynic::StreamingOperation<SubscribeChannels, ChannelsVariables> {
SubscribeChannels::build(ChannelsVariables::from(selector))
}
pub fn subscribe_accounts(
selector: AccountSelector,
) -> cynic::StreamingOperation<SubscribeAccounts, AccountVariables> {
SubscribeAccounts::build(AccountVariables::from(selector))
}
pub fn subscribe_graph() -> cynic::StreamingOperation<SubscribeGraph, ()> {
SubscribeGraph::build(())
}
pub fn subscribe_ticket_params() -> cynic::StreamingOperation<SubscribeTicketParams, ()> {
SubscribeTicketParams::build(())
}
pub fn subscribe_health() -> cynic::StreamingOperation<SubscribeHealth, ()> {
SubscribeHealth::build(())
}
pub fn subscribe_safe_deployments() -> cynic::StreamingOperation<SubscribeSafeDeployment, ()> {
SubscribeSafeDeployment::build(())
}
pub fn subscribe_ticket_redeemed(
selector: TicketSelector,
) -> cynic::StreamingOperation<SubscribeTicketRedeemed, TicketRedeemedVariables> {
SubscribeTicketRedeemed::build(TicketRedeemedVariables::from(selector))
}
}
impl BlokliSubscriptionClient for BlokliClient {
#[tracing::instrument(level = "debug", skip(self), fields(?selector))]
fn subscribe_channels(&self, selector: ChannelSelector) -> Result<impl Stream<Item = Result<Channel>> + Send> {
Ok(self
.build_subscription_stream(GraphQlQueries::subscribe_channels(selector))?
.try_filter_map(|item| futures::future::ok(Some(item.channel_updated))))
}
#[tracing::instrument(level = "debug", skip(self), fields(?selector))]
fn subscribe_accounts(&self, selector: AccountSelector) -> Result<impl Stream<Item = Result<Account>> + Send> {
Ok(self
.build_subscription_stream(GraphQlQueries::subscribe_accounts(selector))?
.try_filter_map(|item| futures::future::ok(Some(item.account_updated))))
}
#[tracing::instrument(level = "debug", skip(self))]
fn subscribe_graph(&self) -> Result<impl Stream<Item = Result<OpenedChannelsGraphEntry>> + Send> {
Ok(self
.build_subscription_stream(GraphQlQueries::subscribe_graph())?
.try_filter_map(|item| futures::future::ok(Some(item.opened_channel_graph_updated))))
}
#[tracing::instrument(level = "debug", skip(self))]
fn subscribe_ticket_params(&self) -> Result<impl Stream<Item = Result<TicketParameters>> + Send> {
Ok(self
.build_subscription_stream(GraphQlQueries::subscribe_ticket_params())?
.try_filter_map(|item| futures::future::ok(Some(item.ticket_parameters_updated))))
}
#[tracing::instrument(level = "debug", skip(self))]
fn subscribe_health(&self) -> Result<impl Stream<Item = Result<ReadinessState>> + Send> {
Ok(self
.build_subscription_stream(GraphQlQueries::subscribe_health())?
.try_filter_map(|item| futures::future::ok(Some(item.health))))
}
#[tracing::instrument(level = "debug", skip(self))]
fn subscribe_safe_deployments(&self) -> Result<impl Stream<Item = Result<Safe>> + Send> {
Ok(self
.build_subscription_stream(GraphQlQueries::subscribe_safe_deployments())?
.try_filter_map(|item| futures::future::ok(Some(item.safe_deployed))))
}
#[tracing::instrument(level = "debug", skip(self))]
fn subscribe_track_transaction(&self, tx_id: TxId) -> Result<impl Stream<Item = Result<Transaction>> + Send> {
Ok(self
.build_subscription_stream(GraphQlQueries::subscribe_track_transaction(tx_id))?
.try_filter_map(|item| futures::future::ok(Some(item.transaction_updated))))
}
#[tracing::instrument(level = "debug", skip(self), fields(?selector))]
fn subscribe_ticket_redeemed(
&self,
selector: TicketSelector,
) -> Result<impl futures::Stream<Item = Result<RedeemTicketDetails>> + Send> {
Ok(self
.build_subscription_stream(GraphQlQueries::subscribe_ticket_redeemed(selector))?
.try_filter_map(|item| futures::future::ok(Some(item.ticket_redeemed))))
}
}