void-focus 0.3.0-alpha.5

A feature-rich terminal focus timer with task tracking
Documentation
use ratatui::style::Color;

use crate::canvas_timer::SceneStyle;

mod builtin;
mod catalog;
mod color;
mod file;
mod tokens;

pub use catalog::{themes_dir, ThemeCatalog, ThemeEntry};
pub use tokens::{ThemeTokens, NAMES as TOKEN_NAMES};

use anyhow::{Context, Result};

use self::builtin::builtin_tokens;
use self::catalog::ThemeSource;
use self::file::ThemeFile;

#[derive(Clone)]
pub struct Theme {
    pub bg: Color,
    pub text: Color,
    pub dim: Color,
    pub accent: Color,
    pub on_accent: Color,
    pub success: Color,
    pub warning: Color,
    pub error: Color,
    pub info: Color,
    pub progress_dim: Color,
    pub task_track: Color,
    pub panel: Color,
    pub panel_border: Color,
    pub select_bg: Color,
    pub select_fg: Color,
    pub active_bg: Color,
    pub active_fg: Color,
}

impl Theme {
    pub fn from_tokens(tokens: ThemeTokens) -> Self {
        Self {
            bg: tokens.bg,
            text: tokens.text,
            dim: tokens.dim,
            accent: tokens.accent,
            on_accent: tokens.on_accent,
            success: tokens.success,
            warning: tokens.warning,
            error: tokens.error,
            info: tokens.info,
            progress_dim: tokens.progress_dim,
            task_track: tokens.task_track,
            panel: tokens.panel,
            panel_border: tokens.panel_border,
            select_bg: tokens.select_bg,
            select_fg: tokens.select_fg,
            active_bg: tokens.active_bg,
            active_fg: tokens.active_fg,
        }
    }

    pub fn into_tokens(self) -> ThemeTokens {
        ThemeTokens {
            bg: self.bg,
            text: self.text,
            dim: self.dim,
            accent: self.accent,
            on_accent: self.on_accent,
            success: self.success,
            warning: self.warning,
            error: self.error,
            info: self.info,
            progress_dim: self.progress_dim,
            task_track: self.task_track,
            panel: self.panel,
            panel_border: self.panel_border,
            select_bg: self.select_bg,
            select_fg: self.select_fg,
            active_bg: self.active_bg,
            active_fg: self.active_fg,
        }
    }

    pub fn dark() -> Self {
        Self {
            bg: Color::Rgb(15, 15, 20),
            text: Color::Rgb(225, 225, 230),
            dim: Color::Rgb(90, 90, 100),
            accent: Color::Rgb(100, 180, 255),
            on_accent: Color::Rgb(10, 10, 15),
            success: Color::Rgb(80, 210, 130),
            warning: Color::Rgb(245, 185, 70),
            error: Color::Rgb(245, 85, 85),
            info: Color::Rgb(170, 140, 250),
            progress_dim: Color::Rgb(45, 45, 55),
            task_track: Color::Rgb(35, 35, 42),
            panel: Color::Rgb(20, 22, 30),
            panel_border: Color::Rgb(55, 60, 75),
            select_bg: Color::Rgb(38, 52, 78),
            select_fg: Color::Rgb(230, 235, 245),
            active_bg: Color::Rgb(32, 48, 72),
            active_fg: Color::Rgb(170, 210, 255),
        }
    }

    pub fn light() -> Self {
        Self {
            bg: Color::Rgb(250, 250, 252),
            text: Color::Rgb(30, 30, 35),
            dim: Color::Rgb(140, 140, 150),
            accent: Color::Rgb(25, 110, 200),
            on_accent: Color::Rgb(255, 255, 255),
            success: Color::Rgb(30, 150, 80),
            warning: Color::Rgb(200, 130, 20),
            error: Color::Rgb(200, 50, 50),
            info: Color::Rgb(110, 70, 190),
            progress_dim: Color::Rgb(200, 205, 215),
            task_track: Color::Rgb(220, 225, 232),
            panel: Color::Rgb(242, 245, 250),
            panel_border: Color::Rgb(190, 198, 210),
            select_bg: Color::Rgb(210, 225, 245),
            select_fg: Color::Rgb(20, 40, 70),
            active_bg: Color::Rgb(195, 218, 245),
            active_fg: Color::Rgb(15, 60, 120),
        }
    }

    pub fn polaris() -> Self {
        Self {
            bg: Color::Rgb(10, 14, 30),
            text: Color::Rgb(215, 225, 250),
            dim: Color::Rgb(100, 120, 160),
            accent: Color::Rgb(90, 200, 255),
            on_accent: Color::Rgb(10, 14, 30),
            success: Color::Rgb(80, 235, 180),
            warning: Color::Rgb(255, 160, 75),
            error: Color::Rgb(255, 95, 120),
            info: Color::Rgb(180, 135, 255),
            progress_dim: Color::Rgb(40, 50, 80),
            task_track: Color::Rgb(25, 35, 60),
            panel: Color::Rgb(14, 20, 38),
            panel_border: Color::Rgb(55, 70, 110),
            select_bg: Color::Rgb(28, 42, 78),
            select_fg: Color::Rgb(210, 225, 255),
            active_bg: Color::Rgb(22, 38, 68),
            active_fg: Color::Rgb(140, 210, 255),
        }
    }

    pub fn matrix() -> Self {
        Self {
            bg: Color::Rgb(3, 12, 5),
            text: Color::Rgb(150, 240, 140),
            dim: Color::Rgb(60, 110, 65),
            accent: Color::Rgb(80, 230, 90),
            on_accent: Color::Rgb(0, 0, 0),
            success: Color::Rgb(70, 220, 110),
            warning: Color::Rgb(220, 200, 60),
            error: Color::Rgb(255, 80, 90),
            info: Color::Rgb(100, 190, 240),
            progress_dim: Color::Rgb(15, 40, 18),
            task_track: Color::Rgb(8, 28, 10),
            panel: Color::Rgb(5, 16, 7),
            panel_border: Color::Rgb(35, 85, 40),
            select_bg: Color::Rgb(10, 32, 14),
            select_fg: Color::Rgb(160, 255, 165),
            active_bg: Color::Rgb(14, 42, 18),
            active_fg: Color::Rgb(110, 255, 120),
        }
    }

    pub fn scene_style(&self, mode: Color) -> SceneStyle {
        SceneStyle {
            mode,
            track: self.progress_dim,
            task: self.success,
            task_dim: self.task_track,
            bg: self.bg,
            bg_mid: mix(self.bg, self.panel, 160),
            bg_light: self.panel,
            wave: self.accent,
            core: mode,
            glow: self.accent,
            particle: self.dim,
            text: self.text,
            session_on: self.accent,
            session_off: self.dim,
        }
    }
}

pub fn resolve(id: &str, catalog: &ThemeCatalog) -> Result<Theme> {
    if let Some(tokens) = builtin_tokens(id) {
        return Ok(Theme::from_tokens(tokens));
    }

    let entry = catalog.resolve_entry(id)?;
    let tokens = match &entry.source {
        ThemeSource::Builtin => builtin_tokens(id).context("builtin theme missing tokens")?,
        ThemeSource::Embedded(source) => ThemeFile::from_str(source)?.into_tokens()?,
        ThemeSource::File(path) => ThemeFile::from_path(path)?.into_tokens()?,
    };
    Ok(Theme::from_tokens(tokens))
}

pub fn normalize_theme_id(raw: &str) -> String {
    raw.trim().to_ascii_lowercase()
}

fn mix(a: Color, b: Color, t: u8) -> Color {
    let (ar, ag, ab) = rgb(a);
    let (br, bg, bb) = rgb(b);
    let t = t as u16;
    let inv = 255 - t;
    Color::Rgb(
        ((ar as u16 * inv + br as u16 * t) / 255) as u8,
        ((ag as u16 * inv + bg as u16 * t) / 255) as u8,
        ((ab as u16 * inv + bb as u16 * t) / 255) as u8,
    )
}

fn rgb(c: Color) -> (u8, u8, u8) {
    match c {
        Color::Rgb(r, g, b) => (r, g, b),
        Color::Black => (0, 0, 0),
        Color::White => (255, 255, 255),
        _ => (128, 128, 128),
    }
}