use std::sync::Arc;
use async_std::sync::Mutex;
use ng_repo::errors::*;
use ng_repo::log::*;
use ng_repo::types::*;
#[cfg(not(target_arch = "wasm32"))]
use crate::broker::BROKER;
use crate::connection::NoiseFSM;
use crate::types::*;
use crate::{actor::*, types::ProtocolMessage};
impl PublishEvent {
pub fn get_actor(&self, id: i64) -> Box<dyn EActor> {
Actor::<PublishEvent, ()>::new_responder(id)
}
pub fn new(event: Event, overlay: OverlayId) -> PublishEvent {
PublishEvent(event, Some(overlay))
}
pub fn set_overlay(&mut self, overlay: OverlayId) {
self.1 = Some(overlay);
}
pub fn overlay(&self) -> &OverlayId {
self.1.as_ref().unwrap()
}
pub fn event(&self) -> &Event {
&self.0
}
pub fn take_event(self) -> Event {
self.0
}
}
impl TryFrom<ProtocolMessage> for PublishEvent {
type Error = ProtocolError;
fn try_from(msg: ProtocolMessage) -> Result<Self, Self::Error> {
let req: ClientRequestContentV0 = msg.try_into()?;
if let ClientRequestContentV0::PublishEvent(a) = req {
Ok(a)
} else {
log_debug!("INVALID {:?}", req);
Err(ProtocolError::InvalidValue)
}
}
}
impl From<PublishEvent> for ProtocolMessage {
fn from(msg: PublishEvent) -> ProtocolMessage {
let overlay = msg.1.unwrap();
ProtocolMessage::from_client_request_v0(ClientRequestContentV0::PublishEvent(msg), overlay)
}
}
impl Actor<'_, PublishEvent, ()> {}
#[async_trait::async_trait]
impl EActor for Actor<'_, PublishEvent, ()> {
async fn respond(
&mut self,
_msg: ProtocolMessage,
_fsm: Arc<Mutex<NoiseFSM>>,
) -> Result<(), ProtocolError> {
#[cfg(not(target_arch = "wasm32"))]
{
let req = PublishEvent::try_from(_msg)?;
req.event().verify()?;
let overlay = req.overlay().clone();
let (user_id, remote_peer) = {
let fsm = _fsm.lock().await;
(
fsm.user_id()?,
fsm.remote_peer().ok_or(ProtocolError::ActorError)?,
)
};
let res = {
let broker = BROKER.read().await;
broker
.dispatch_event(&overlay, req.take_event(), &user_id, &remote_peer)
.await
};
if res.is_err() {
let res: Result<(), ServerError> = Err(res.unwrap_err());
_fsm.lock()
.await
.send_in_reply_to(res.into(), self.id())
.await?;
} else {
let broker = { BROKER.read().await.get_server_broker()? };
for client in res.unwrap() {
broker
.read()
.await
.remove_all_subscriptions_of_client(&client)
.await;
}
let finalres: Result<(), ServerError> = Ok(());
_fsm.lock()
.await
.send_in_reply_to(finalres.into(), self.id())
.await?;
}
}
Ok(())
}
}