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

//! This module provides a mirror of the VirtualDOM Element Slab using a Vector.

use std::ops::{Index, IndexMut};
use web_sys::Node;

pub(crate) struct NodeSlab {
    nodes: Vec<Option<Node>>,
}

impl NodeSlab {
    pub fn new(capacity: usize) -> NodeSlab {
        let nodes = Vec::with_capacity(capacity);
        NodeSlab { nodes }
    }
}
impl Index<usize> for NodeSlab {
    type Output = Option<Node>;
    fn index(&self, index: usize) -> &Self::Output {
        &self.nodes[index]
    }
}

impl IndexMut<usize> for NodeSlab {
    fn index_mut(&mut self, index: usize) -> &mut Self::Output {
        if index >= self.nodes.capacity() * 3 {
            panic!("Trying to mutate an element way too far out of bounds");
        }

        if index + 1 > self.nodes.len() {
            self.nodes.resize_with(index + 1, || None);
        }
        &mut self.nodes[index]
    }
}