1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

use crate::widget::WidgetId;
use std::{
    any::Any,
    collections::HashMap,
    sync::mpsc::{Receiver, Sender},
};

pub type Message = Box<dyn Any>;
pub type Messages = Vec<Message>;

pub struct MessageReceiver(Receiver<(WidgetId, Message)>);

impl MessageReceiver {
    pub fn new(receiver: Receiver<(WidgetId, Message)>) -> Self {
        Self(receiver)
    }

    pub fn process(&mut self) -> HashMap<WidgetId, Messages> {
        let mut result = HashMap::<WidgetId, Messages>::new();
        while let Ok((id, message)) = self.0.try_recv() {
            if let Some(list) = result.get_mut(&id) {
                list.push(message);
            } else {
                let mut list = Messages::with_capacity(1);
                list.push(message);
                result.insert(id, list);
            }
        }
        result
    }
}

#[derive(Clone)]
pub struct MessageSender(Sender<(WidgetId, Message)>);

impl MessageSender {
    pub fn new(sender: Sender<(WidgetId, Message)>) -> Self {
        Self(sender)
    }

    pub fn write<T>(&self, id: WidgetId, message: T) -> bool
    where
        T: 'static + Any,
    {
        self.0.send((id, Box::new(message))).is_ok()
    }

    pub fn write_raw(&self, id: WidgetId, message: Message) -> bool {
        self.0.send((id, message)).is_ok()
    }

    pub fn write_raw_all<I>(&self, messages: I)
    where
        I: IntoIterator<Item = (WidgetId, Message)>,
    {
        for data in messages {
            drop(self.0.send(data));
        }
    }
}

pub struct Messenger<'a> {
    sender: MessageSender,
    pub messages: &'a [Message],
}

impl<'a> Messenger<'a> {
    pub fn new(sender: MessageSender, messages: &'a [Message]) -> Self {
        Self { sender, messages }
    }

    pub fn write<T>(&self, id: WidgetId, message: T) -> bool
    where
        T: 'static,
    {
        self.sender.write(id, message)
    }

    pub fn write_raw(&self, id: WidgetId, message: Message) -> bool {
        self.sender.write_raw(id, message)
    }

    pub fn write_raw_all<I>(&self, messages: I)
    where
        I: IntoIterator<Item = (WidgetId, Message)>,
    {
        self.sender.write_raw_all(messages);
    }
}