use super::{
stream::AsyncComponentStream, AsyncComponent, AsyncComponentController, AsyncController,
};
use crate::{Receiver, Sender, ShutdownOnDrop};
use std::fmt::{self, Debug};
pub struct AsyncConnector<C: AsyncComponent> {
pub(super) widget: C::Root,
pub(super) sender: Sender<C::Input>,
pub(super) receiver: Receiver<C::Output>,
pub(super) shutdown_on_drop: ShutdownOnDrop,
}
impl<C: AsyncComponent> AsyncConnector<C> {
pub fn forward<X: 'static, F: (Fn(C::Output) -> X) + 'static>(
self,
sender_: &Sender<X>,
transform: F,
) -> AsyncController<C> {
let Self {
widget,
sender,
receiver,
shutdown_on_drop,
} = self;
crate::spawn_local(receiver.forward(sender_.clone(), transform));
AsyncController {
widget,
sender,
shutdown_on_drop,
}
}
pub fn connect_receiver<F: FnMut(&mut Sender<C::Input>, C::Output) + 'static>(
self,
mut func: F,
) -> AsyncController<C> {
let Self {
widget,
sender,
receiver,
shutdown_on_drop,
} = self;
let mut sender_ = sender.clone();
crate::spawn_local(async move {
while let Some(event) = receiver.recv().await {
func(&mut sender_, event);
}
});
AsyncController {
widget,
sender,
shutdown_on_drop,
}
}
pub fn detach(self) -> AsyncController<C> {
let Self {
widget,
sender,
shutdown_on_drop,
..
} = self;
AsyncController {
widget,
sender,
shutdown_on_drop,
}
}
pub fn into_stream(self) -> AsyncComponentStream<C> {
let Self {
receiver,
shutdown_on_drop,
..
} = self;
AsyncComponentStream {
stream: receiver.into_stream(),
shutdown_on_drop,
}
}
}
impl<C: AsyncComponent> AsyncComponentController<C> for AsyncConnector<C> {
fn sender(&self) -> &Sender<C::Input> {
&self.sender
}
fn widget(&self) -> &C::Root {
&self.widget
}
fn detach_runtime(&mut self) {
self.shutdown_on_drop.deactivate();
}
}
impl<C> Debug for AsyncConnector<C>
where
C: AsyncComponent + Debug,
C::Widgets: Debug,
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("Connector")
.field("widget", &self.widget)
.field("sender", &self.sender)
.field("receiver", &self.receiver)
.field("shutdown_on_drop", &self.shutdown_on_drop)
.finish()
}
}