lunar-lib 0.8.0

Common utilities for lunar applications
Documentation 
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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141

use std::{collections::HashMap, str::FromStr};

use ratatui::style::{Color, Modifier, Style};
use serde::Deserialize;

use crate::{iterator_ext::IteratorExtensions, ratatui_theme::ResolvedPalette};

#[derive(Debug, Deserialize)]
#[serde(untagged)]
enum StyleType {
    Normal(Style),
    Raw(RawStyle),
    FgOnlyNormal(Color),
    FgOnlyPalette(String),
}

#[derive(Debug, Deserialize, Clone)]
struct RawStyle {
    /// The foreground color.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub fg: Option<String>,

    /// The background color.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub bg: Option<String>,

    /// The underline color.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub underline_color: Option<String>,

    /// The modifiers to add.
    #[serde(
        default,
        skip_serializing_if = "Modifier::is_empty",
        deserialize_with = "deserialize_modifier"
    )]
    pub add_modifier: Modifier,

    /// The modifiers to remove.
    #[serde(
        default,
        skip_serializing_if = "Modifier::is_empty",
        deserialize_with = "deserialize_modifier"
    )]
    pub sub_modifier: Modifier,
}

impl RawStyle {
    fn into_style(self, palette: &HashMap<String, Color>) -> Style {
        Style {
            fg: self.fg.and_then(|fg| palette.get(&fg).copied()),
            bg: self.bg.and_then(|bg| palette.get(&bg).copied()),
            underline_color: self
                .underline_color
                .and_then(|uc| palette.get(&uc).copied()),
            add_modifier: self.add_modifier,
            sub_modifier: self.sub_modifier,
        }
    }
}

impl FromStr for RawStyle {
    type Err = toml::de::Error;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        toml::from_str(s)
    }
}

fn deserialize_modifier<'de, D>(deserializer: D) -> Result<Modifier, D::Error>
where
    D: serde::Deserializer<'de>,
{
    use serde::Deserialize;

    Option::<Modifier>::deserialize(deserializer)
        .map(|modifier| modifier.unwrap_or_else(Modifier::empty))
}

#[derive(Deserialize)]
pub(super) struct RawTheme {
    #[serde(default)]
    styles: HashMap<String, StyleType>,

    #[serde(default)]
    palette: HashMap<String, Color>,
}

impl RawTheme {
    pub fn resolve(self) -> ResolvedPalette {
        let mut resolved_styles = HashMap::with_capacity(self.styles.len());

        for key in self.styles.keys() {
            let mut final_style = Style::default();
            let parts = key.split('.').to_vec();

            for i in 1..=parts.len() {
                let parent_key = parts[..i].join(".");

                if let Some(style_type) = self.styles.get(&parent_key) {
                    let style = match style_type {
                        StyleType::Normal(style) => *style,
                        StyleType::Raw(raw_style) => raw_style.clone().into_style(&self.palette),
                        StyleType::FgOnlyNormal(color) => {
                            resolved_styles.insert(key.clone(), Style::default().fg(*color));
                            continue;
                        }
                        StyleType::FgOnlyPalette(key) => {
                            if let Some(copied) = self.palette.get(key) {
                                final_style.fg = Some(*copied)
                            };

                            resolved_styles.insert(key.clone(), final_style);
                            continue;
                        }
                    };

                    if let Some(fg) = style.fg {
                        final_style.fg = Some(fg);
                    }
                    if let Some(bg) = style.bg {
                        final_style.bg = Some(bg);
                    }
                    if let Some(uc) = style.underline_color {
                        final_style.underline_color = Some(uc)
                    }

                    final_style.add_modifier |= style.add_modifier & !Modifier::UNDERLINED;
                    final_style.sub_modifier |= style.sub_modifier & !Modifier::UNDERLINED;
                }
            }

            resolved_styles.insert(key.clone(), final_style);
        }

        ResolvedPalette {
            styles: resolved_styles,
            palette: self.palette,
        }
    }
}