freya-devtools-app 0.4.0-rc.15

Devtools App for Freya app
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

use std::{
    collections::{
        HashMap,
        HashSet,
    },
    sync::Arc,
};

use async_lock::Mutex;
use async_tungstenite::WebSocketSender;
use freya_core::{
    integration::NodeId,
    prelude::spawn,
};
use freya_devtools::{
    IncomingMessage,
    IncomingMessageAction,
    NodeInfo,
};
use freya_radio::hooks::RadioChannel;
use smol::net::TcpStream;
use tungstenite::Message;

pub struct DevtoolsState {
    pub(crate) nodes: HashMap<u64, Vec<NodeInfo>>,
    pub(crate) expanded_nodes: HashSet<(u64, NodeId)>,
    pub(crate) client: Arc<Mutex<Option<WebSocketSender<TcpStream>>>>,
    pub(crate) animation_speed: f32,
}

impl DevtoolsState {
    pub fn send_action(&self, action: IncomingMessageAction) {
        let message = Message::Text(
            serde_json::to_string(&IncomingMessage { action })
                .unwrap()
                .into(),
        );
        let client = self.client.clone();
        spawn(async move {
            if let Some(client) = client.lock().await.as_mut() {
                client.send(message).await.ok();
            }
        });
    }
}

#[derive(PartialEq, Eq, Clone, Debug, Hash)]
pub enum DevtoolsChannel {
    Global,
    UpdatedTree,
    Misc,
}

impl RadioChannel<DevtoolsState> for DevtoolsChannel {}