aboutsummaryrefslogtreecommitdiffstats
path: root/components/devtools/resource.rs
blob: 7cef8188cc85a624b2b6bc5e94323aa16b62e612 (plain) (blame)
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
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */

use std::cell::RefCell;
use std::collections::HashMap;
use std::net::TcpStream;

use serde::Serialize;

use crate::StreamId;
use crate::protocol::JsonPacketStream;

#[derive(Serialize)]
pub(crate) struct ResourceAvailableReply<T: Serialize> {
    pub from: String,
    #[serde(rename = "type")]
    pub type_: String,
    pub array: Vec<(String, Vec<T>)>,
}

pub(crate) trait ResourceAvailable {
    fn actor_name(&self) -> String;

    fn get_streams(&self) -> &RefCell<HashMap<StreamId, TcpStream>>;

    fn resource_available<T: Serialize>(&self, resource: T, resource_type: String) {
        self.resources_available(vec![resource], resource_type);
    }

    fn resources_available<T: Serialize>(&self, resources: Vec<T>, resource_type: String) {
        let msg = ResourceAvailableReply::<T> {
            from: self.actor_name(),
            type_: "resources-available-array".into(),
            array: vec![(resource_type, resources)],
        };

        for stream in self.get_streams().borrow_mut().values_mut() {
            let _ = stream.write_json_packet(&msg);
        }
    }
}