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

use chargrid_render::Style;
use std::marker::PhantomData;

pub struct MenuEntryToRender<'a, E> {
    pub entry: &'a E,
    pub selected: bool,
    pub index: usize,
}

pub trait MenuEntryString {
    type Entry;
    fn render_string(&self, entry: MenuEntryToRender<Self::Entry>, buf: &mut String);
}

#[derive(Clone, Copy)]
pub struct MenuEntryStringIntoStr<E>(PhantomData<E>)
where
    for<'a> &'a E: Into<&'a str>;
impl<E> MenuEntryStringIntoStr<E>
where
    for<'a> &'a E: Into<&'a str>,
{
    pub fn new() -> Self {
        Self(PhantomData)
    }
}
impl<E> MenuEntryString for MenuEntryStringIntoStr<E>
where
    for<'a> &'a E: Into<&'a str>,
{
    type Entry = E;
    fn render_string(&self, entry: MenuEntryToRender<Self::Entry>, buf: &mut String) {
        use std::fmt::Write;
        write!(buf, "{}", entry.entry.into()).unwrap()
    }
}

pub struct MenuEntryStringFn<F, E> {
    f: F,
    e: PhantomData<E>,
}
impl<F, E> MenuEntryStringFn<F, E> {
    pub fn new(f: F) -> Self {
        Self { f, e: PhantomData }
    }
}
impl<F, E> MenuEntryString for MenuEntryStringFn<F, E>
where
    F: Fn(MenuEntryToRender<E>, &mut String),
{
    type Entry = E;
    fn render_string(&self, entry: MenuEntryToRender<Self::Entry>, buf: &mut String) {
        (self.f)(entry, buf);
    }
}

pub trait MenuEntryRichString {
    type Entry;
    fn render_rich_string(&self, entry: MenuEntryToRender<Self::Entry>, buf: &mut String) -> Style;
}

pub struct MenuEntryRichStringFn<F, E> {
    f: F,
    e: PhantomData<E>,
}
impl<F, E> MenuEntryRichStringFn<F, E> {
    pub fn new(f: F) -> Self {
        Self { f, e: PhantomData }
    }
}
impl<F, E> MenuEntryRichString for MenuEntryRichStringFn<F, E>
where
    F: Fn(MenuEntryToRender<E>, &mut String) -> Style,
{
    type Entry = E;
    fn render_rich_string(&self, entry: MenuEntryToRender<Self::Entry>, buf: &mut String) -> Style {
        (self.f)(entry, buf)
    }
}