cosmic-terminal/src/terminal_theme.rs

use alacritty_terminal::{
    term::color::Colors,
    vte::ansi::{NamedColor, Rgb},
};
use hex_color::HexColor;
use std::{collections::HashMap, fs};

use crate::config::{
    COSMIC_THEME_DARK, COSMIC_THEME_LIGHT, ColorScheme, ColorSchemeAnsi, ColorSchemeKind,
};

fn auto_colors() -> Colors {
    let mut colors = Colors::default();

    // These colors come from `ransid`: https://gitlab.redox-os.org/redox-os/ransid/-/blob/master/src/color.rs
    /* Indexed colors */
    for value in 16..=231 {
        let convert = |value: u8| -> u8 {
            match value {
                0 => 0,
                _ => value * 0x28 + 0x28,
            }
        };

        let r = convert((value - 16) / 36 % 6);
        let g = convert((value - 16) / 6 % 6);
        let b = convert((value - 16) % 6);
        colors[value as usize] = Some(Rgb { r, g, b });
    }

    /* Grays */
    for value in 232..=255 {
        let gray = (value - 232) * 10 + 8;
        colors[value as usize] = Some(Rgb {
            r: gray,
            g: gray,
            b: gray,
        });
    }

    colors
}

impl From<&ColorScheme> for Colors {
    fn from(color_scheme: &ColorScheme) -> Self {
        let mut colors = auto_colors();

        let encode_rgb = |rgb_opt: Option<HexColor>| -> Option<Rgb> {
            let rgb = rgb_opt?;
            Some(Rgb {
                r: rgb.r,
                g: rgb.g,
                b: rgb.b,
            })
        };

        // Set normal colors
        colors[NamedColor::Black] = encode_rgb(color_scheme.normal.black);
        colors[NamedColor::Red] = encode_rgb(color_scheme.normal.red);
        colors[NamedColor::Green] = encode_rgb(color_scheme.normal.green);
        colors[NamedColor::Yellow] = encode_rgb(color_scheme.normal.yellow);
        colors[NamedColor::Blue] = encode_rgb(color_scheme.normal.blue);
        colors[NamedColor::Magenta] = encode_rgb(color_scheme.normal.magenta);
        colors[NamedColor::Cyan] = encode_rgb(color_scheme.normal.cyan);
        colors[NamedColor::White] = encode_rgb(color_scheme.normal.white);

        // Set bright colors
        colors[NamedColor::BrightBlack] = encode_rgb(color_scheme.bright.black);
        colors[NamedColor::BrightRed] = encode_rgb(color_scheme.bright.red);
        colors[NamedColor::BrightGreen] = encode_rgb(color_scheme.bright.green);
        colors[NamedColor::BrightYellow] = encode_rgb(color_scheme.bright.yellow);
        colors[NamedColor::BrightBlue] = encode_rgb(color_scheme.bright.blue);
        colors[NamedColor::BrightMagenta] = encode_rgb(color_scheme.bright.magenta);
        colors[NamedColor::BrightCyan] = encode_rgb(color_scheme.bright.cyan);
        colors[NamedColor::BrightWhite] = encode_rgb(color_scheme.bright.white);

        // Set dim colors
        colors[NamedColor::DimBlack] = encode_rgb(color_scheme.dim.black);
        colors[NamedColor::DimRed] = encode_rgb(color_scheme.dim.red);
        colors[NamedColor::DimGreen] = encode_rgb(color_scheme.dim.green);
        colors[NamedColor::DimYellow] = encode_rgb(color_scheme.dim.yellow);
        colors[NamedColor::DimBlue] = encode_rgb(color_scheme.dim.blue);
        colors[NamedColor::DimMagenta] = encode_rgb(color_scheme.dim.magenta);
        colors[NamedColor::DimCyan] = encode_rgb(color_scheme.dim.cyan);
        colors[NamedColor::DimWhite] = encode_rgb(color_scheme.dim.white);

        // Set special colors
        colors[NamedColor::Foreground] = encode_rgb(color_scheme.foreground);
        colors[NamedColor::Background] = encode_rgb(color_scheme.background);
        colors[NamedColor::Cursor] = encode_rgb(color_scheme.cursor);
        colors[NamedColor::BrightForeground] = encode_rgb(color_scheme.bright_foreground);
        colors[NamedColor::DimForeground] = encode_rgb(color_scheme.dim_foreground);

        colors
    }
}

impl From<(&str, &Colors)> for ColorScheme {
    fn from(tuple: (&str, &Colors)) -> Self {
        let (name, colors) = tuple;

        let encode_rgb = |rgb_opt: Option<Rgb>| -> Option<HexColor> {
            let rgb = rgb_opt?;
            Some(HexColor::rgb(rgb.r, rgb.g, rgb.b))
        };

        Self {
            name: name.to_string(),
            foreground: encode_rgb(colors[NamedColor::Foreground]),
            background: encode_rgb(colors[NamedColor::Background]),
            cursor: encode_rgb(colors[NamedColor::Cursor]),
            bright_foreground: encode_rgb(colors[NamedColor::BrightForeground]),
            dim_foreground: encode_rgb(colors[NamedColor::DimForeground]),
            normal: ColorSchemeAnsi {
                black: encode_rgb(colors[NamedColor::Black]),
                red: encode_rgb(colors[NamedColor::Red]),
                green: encode_rgb(colors[NamedColor::Green]),
                yellow: encode_rgb(colors[NamedColor::Yellow]),
                blue: encode_rgb(colors[NamedColor::Blue]),
                magenta: encode_rgb(colors[NamedColor::Magenta]),
                cyan: encode_rgb(colors[NamedColor::Cyan]),
                white: encode_rgb(colors[NamedColor::White]),
            },
            bright: ColorSchemeAnsi {
                black: encode_rgb(colors[NamedColor::BrightBlack]),
                red: encode_rgb(colors[NamedColor::BrightRed]),
                green: encode_rgb(colors[NamedColor::BrightGreen]),
                yellow: encode_rgb(colors[NamedColor::BrightYellow]),
                blue: encode_rgb(colors[NamedColor::BrightBlue]),
                magenta: encode_rgb(colors[NamedColor::BrightMagenta]),
                cyan: encode_rgb(colors[NamedColor::BrightCyan]),
                white: encode_rgb(colors[NamedColor::BrightWhite]),
            },
            dim: ColorSchemeAnsi {
                black: encode_rgb(colors[NamedColor::DimBlack]),
                red: encode_rgb(colors[NamedColor::DimRed]),
                green: encode_rgb(colors[NamedColor::DimGreen]),
                yellow: encode_rgb(colors[NamedColor::DimYellow]),
                blue: encode_rgb(colors[NamedColor::DimBlue]),
                magenta: encode_rgb(colors[NamedColor::DimMagenta]),
                cyan: encode_rgb(colors[NamedColor::DimCyan]),
                white: encode_rgb(colors[NamedColor::DimWhite]),
            },
        }
    }
}

pub fn cosmic_dark() -> Colors {
    let mut colors = auto_colors();

    let encode_rgb = |data: u32| -> Rgb {
        Rgb {
            r: (data >> 16) as u8,
            g: (data >> 8) as u8,
            b: data as u8,
        }
    };

    colors[NamedColor::Black] = Some(encode_rgb(0x1B1B1B));
    colors[NamedColor::Red] = Some(encode_rgb(0xF16161));
    colors[NamedColor::Green] = Some(encode_rgb(0x7CB987));
    colors[NamedColor::Yellow] = Some(encode_rgb(0xDDC74C));
    colors[NamedColor::Blue] = Some(encode_rgb(0x6296BE));
    colors[NamedColor::Magenta] = Some(encode_rgb(0xBE6DEE));
    colors[NamedColor::Cyan] = Some(encode_rgb(0x49BAC8));
    colors[NamedColor::White] = Some(encode_rgb(0xBEBEBE));

    colors[NamedColor::BrightBlack] = Some(encode_rgb(0x808080));
    colors[NamedColor::BrightRed] = Some(encode_rgb(0xFF8985));
    colors[NamedColor::BrightGreen] = Some(encode_rgb(0x97D5A0));
    colors[NamedColor::BrightYellow] = Some(encode_rgb(0xFAE365));
    colors[NamedColor::BrightBlue] = Some(encode_rgb(0x7DB1DA));
    colors[NamedColor::BrightMagenta] = Some(encode_rgb(0xD68EFF));
    colors[NamedColor::BrightCyan] = Some(encode_rgb(0x49BAC8));
    colors[NamedColor::BrightWhite] = Some(encode_rgb(0xC4C4C4));

    colors[NamedColor::DimBlack] = Some(encode_rgb(0x000000));
    colors[NamedColor::DimRed] = Some(encode_rgb(0xA04040));
    colors[NamedColor::DimGreen] = Some(encode_rgb(0x5D7D62));
    colors[NamedColor::DimYellow] = Some(encode_rgb(0x9E914A));
    colors[NamedColor::DimBlue] = Some(encode_rgb(0x486073));
    colors[NamedColor::DimMagenta] = Some(encode_rgb(0x7F46A1));
    colors[NamedColor::DimCyan] = Some(encode_rgb(0x3F7F87));
    colors[NamedColor::DimWhite] = Some(encode_rgb(0x898989));

    // Set special colors
    colors[NamedColor::Foreground] = colors[NamedColor::BrightWhite];
    // Background comes from theme settings: colors[NamedColor::Background] = colors[NamedColor::Black];
    colors[NamedColor::Cursor] = colors[NamedColor::BrightWhite];
    colors[NamedColor::BrightForeground] = colors[NamedColor::BrightWhite];
    colors[NamedColor::DimForeground] = colors[NamedColor::DimWhite];

    colors
}

pub fn cosmic_light() -> Colors {
    let mut colors = auto_colors();

    let encode_rgb = |data: u32| -> Rgb {
        Rgb {
            r: (data >> 16) as u8,
            g: (data >> 8) as u8,
            b: data as u8,
        }
    };

    colors[NamedColor::Black] = Some(encode_rgb(0x1B1B1B));
    colors[NamedColor::Red] = Some(encode_rgb(0xA24F50));
    colors[NamedColor::Green] = Some(encode_rgb(0x437E4F));
    colors[NamedColor::Yellow] = Some(encode_rgb(0x7D6E1E));
    colors[NamedColor::Blue] = Some(encode_rgb(0x516D94));
    colors[NamedColor::Magenta] = Some(encode_rgb(0x9A30CA));
    colors[NamedColor::Cyan] = Some(encode_rgb(0x227A85));
    colors[NamedColor::White] = Some(encode_rgb(0xD7D7D7));

    colors[NamedColor::BrightBlack] = Some(encode_rgb(0x000000));
    colors[NamedColor::BrightRed] = Some(encode_rgb(0x890418));
    colors[NamedColor::BrightGreen] = Some(encode_rgb(0x185529));
    colors[NamedColor::BrightYellow] = Some(encode_rgb(0x534800));
    colors[NamedColor::BrightBlue] = Some(encode_rgb(0x2E496D));
    colors[NamedColor::BrightMagenta] = Some(encode_rgb(0x6B0091));
    colors[NamedColor::BrightCyan] = Some(encode_rgb(0x00525A));
    colors[NamedColor::BrightWhite] = Some(encode_rgb(0xE8E8E8));

    colors[NamedColor::DimBlack] = Some(encode_rgb(0x262626));
    colors[NamedColor::DimRed] = Some(encode_rgb(0xB25D5E));
    colors[NamedColor::DimGreen] = Some(encode_rgb(0x528D5E));
    colors[NamedColor::DimYellow] = Some(encode_rgb(0x8C7D30));
    colors[NamedColor::DimBlue] = Some(encode_rgb(0x5F7CA4));
    colors[NamedColor::DimMagenta] = Some(encode_rgb(0xAA43DB));
    colors[NamedColor::DimCyan] = Some(encode_rgb(0x358994));
    colors[NamedColor::DimWhite] = Some(encode_rgb(0xC4C4C4));

    // Set special colors
    colors[NamedColor::Foreground] = colors[NamedColor::Black];
    // Background comes from theme settings: colors[NamedColor::Background] = colors[NamedColor::BrightWhite];
    colors[NamedColor::Cursor] = colors[NamedColor::Black];
    colors[NamedColor::BrightForeground] = colors[NamedColor::BrightBlack];
    colors[NamedColor::DimForeground] = colors[NamedColor::DimBlack];

    colors
}

// Get builtin themes
pub fn terminal_themes() -> HashMap<(String, ColorSchemeKind), Colors> {
    let mut themes = HashMap::new();
    themes.insert(
        (COSMIC_THEME_DARK.to_string(), ColorSchemeKind::Dark),
        cosmic_dark(),
    );
    themes.insert(
        (COSMIC_THEME_LIGHT.to_string(), ColorSchemeKind::Light),
        cosmic_light(),
    );
    themes
}

// Helper function to export builtin themes to theme files
#[allow(dead_code)]
pub fn export() {
    for ((name, _color_scheme_kind), theme) in terminal_themes() {
        let color_scheme = ColorScheme::from((name.as_str(), &theme));

        // Ensure conversion to and from ColorScheme matches original theme
        {
            let theme_conv = Colors::from(&color_scheme);
            for i in 0..alacritty_terminal::term::color::COUNT {
                assert_eq!(theme[i], theme_conv[i]);
            }
        }

        let ron = match ron::ser::to_string_pretty(&color_scheme, ron::ser::PrettyConfig::new()) {
            Ok(ok) => ok,
            Err(err) => {
                log::error!("failed to export {name:?}: {err}");
                continue;
            }
        };

        let path = format!("color-schemes/{name}.ron");
        match fs::write(&path, ron) {
            Ok(()) => {
                log::info!("exported {path:?}");
            }
            Err(err) => {
                log::error!("failed to esport {path:?}: {err}");
            }
        }
    }
}