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
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use std::fmt;
use std::sync::{Arc, RwLock, RwLockReadGuard, RwLockWriteGuard};

use fnv::{FnvHashMap, FnvHashSet};

use super::{traverse_path, Function, Props};

#[derive(Clone)]
pub struct EventManager(Arc<RwLock<EventManagerInner>>);

unsafe impl Send for EventManager {}
unsafe impl Sync for EventManager {}

impl EventManager {
    #[inline]
    pub fn new() -> Self {
        EventManager(Arc::new(RwLock::new(EventManagerInner::new())))
    }

    #[inline]
    pub(crate) fn read(&self) -> RwLockReadGuard<EventManagerInner> {
        self.0
            .read()
            .expect("failed to acquire EventManager read lock")
    }

    #[inline]
    pub(crate) fn write(&self) -> RwLockWriteGuard<EventManagerInner> {
        self.0
            .write()
            .expect("failed to acquire EventManager write lock")
    }

    #[inline]
    pub fn dispatch(&self, id: &str, event: &mut Props) {
        let event_funcs = self.read().event_funcs(id, event);

        for (id, func) in event_funcs {
            event.set("component_id", id);

            (&*func)(event);

            if event.get("propagation").is_false() {
                break;
            }
        }
    }
}

impl fmt::Debug for EventManager {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(
            f,
            "{:?}",
            self.read()
                .0
                .iter()
                .map(|(k, v)| (
                    k.clone(),
                    v.iter()
                        .map(|(i, _)| i.clone())
                        .collect::<FnvHashSet<String>>()
                ))
                .collect::<FnvHashMap<String, FnvHashSet<String>>>()
        )
    }
}

pub(crate) struct EventManagerInner(FnvHashMap<String, FnvHashMap<String, Arc<Function>>>);

impl EventManagerInner {
    #[inline]
    fn new() -> Self {
        EventManagerInner(FnvHashMap::default())
    }
    #[inline]
    pub(crate) fn add(&mut self, id: &str, name: &str, func: Arc<Function>) {
        self.0
            .entry(name.into())
            .or_insert_with(FnvHashMap::default)
            .insert(id.into(), func);
    }

    #[inline]
    pub(crate) fn remove(&mut self, id: &str, name: &str) {
        let remove = if let Some(funcs) = self.0.get_mut(name) {
            funcs.remove(id);
            funcs.len() == 0
        } else {
            false
        };

        if remove {
            self.0.remove(name);
        }
    }

    #[inline]
    fn event_funcs(&self, id: &str, event: &mut Props) -> Vec<(String, Arc<Function>)> {
        let mut funcs = Vec::new();

        if let Some(name) = event.get("name").string() {
            if let Some(events) = self.0.get(name) {
                traverse_path(id, "", false, true, |id, _| {
                    if let Some(func) = events.get(id) {
                        funcs.push((id.to_owned(), func.clone()));
                    }
                    true
                });
            }
        }

        funcs
    }
}