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

#![allow(unused_imports)]
use std::collections::HashMap;
use std::io::Stdout;
use std::{
    borrow::{Borrow, BorrowMut},
    fmt,
    ops::Deref,
};
use tui::layout::Rect;
use tui::{
    backend::{Backend, CrosstermBackend, TestBackend},
    Frame,
};

type Callback<B> = fn(&mut Frame<B>);

pub trait IRendererStorage<B: Backend> {
    fn has_component(self: &Self, tagname: &str) -> bool;
    fn add_renderer<'b>(self: &'b mut Self, tagname: &'b str, render: Callback<B>) -> &'b mut Self;
    fn render(self: &Self, tagname: &str, frame: &mut Frame<B>);
}

pub struct RendererStorage<B: Backend> {
    storage: HashMap<String, Callback<B>>,
}

impl<B: Backend> RendererStorage<B> {
    pub fn new() -> Self {
        RendererStorage {
            storage: HashMap::new(),
        }
    }
}

impl<B: Backend> IRendererStorage<B> for RendererStorage<B> {
    fn add_renderer<'b>(self: &'b mut Self, tagname: &'b str, render: Callback<B>) -> &'b mut Self {
        self.storage.entry(tagname.to_owned()).or_insert(render);
        self
    }

    fn has_component(self: &Self, tagname: &str) -> bool {
        self.storage.contains_key(tagname)
    }

    fn render(self: &Self, tagname: &str, frame: &mut Frame<B>) {
        let opt = self.storage.get(tagname).clone();
        if opt.is_some() {
            let f = opt.unwrap();
            f(frame);
        }
    }
}

impl<B: Backend> fmt::Debug for RendererStorage<B> {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        let mut r = f.debug_struct("RenderStorage");
        r.field("Components", &self.storage.keys());
        r.finish()
    }
}