use cdk::subscription::{IndexableParams, Params};
use cdk::ws::{WsResponseResult, WsSubscribeResponse};
use super::{WsContext, WsError};
pub(crate) async fn handle(
context: &mut WsContext,
params: Params,
) -> Result<WsResponseResult, WsError> {
let sub_id = params.id.clone();
if context.subscriptions.contains_key(&sub_id) {
return Err(WsError::InvalidParams);
}
let params: IndexableParams = params.into();
let mut subscription = context
.state
.mint
.pubsub_manager()
.try_subscribe(params)
.await
.map_err(|_| WsError::ParseError)?;
let publisher = context.publisher.clone();
context.subscriptions.insert(
sub_id.clone(),
tokio::spawn(async move {
while let Some(response) = subscription.recv().await {
let _ = publisher.send(response).await;
}
}),
);
Ok(WsSubscribeResponse {
status: "OK".to_string(),
sub_id,
}
.into())
}