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