rusty-rich 0.3.0

//! Theme system — equivalent to Rich's `theme.py`.
//!
//! A Theme maps style names (like "repr.number", "repr.str") to Style values,
//! allowing customizable color schemes for different renderable types.

use std::collections::HashMap;

use crate::style::Style;

// ---------------------------------------------------------------------------
// Theme
// ---------------------------------------------------------------------------

/// A set of named styles that can be applied to themed output.
#[derive(Debug, Clone)]
pub struct Theme {
    /// Mapping from style name → Style.
    pub styles: HashMap<String, Style>,
    /// Optional inherited base theme.
    pub inherit: Option<Box<Theme>>,
}

impl Theme {
    /// Create a new empty theme.
    pub fn new() -> Self {
        Self {
            styles: HashMap::new(),
            inherit: None,
        }
    }

    /// Create a theme that inherits from another.
    pub fn with_inherit(inherit: Theme) -> Self {
        Self {
            styles: HashMap::new(),
            inherit: Some(Box::new(inherit)),
        }
    }

    /// Look up a style by name. Falls back to the inherited theme.
    pub fn get(&self, name: &str) -> Option<&Style> {
        self.styles
            .get(name)
            .or_else(|| self.inherit.as_ref().and_then(|i| i.get(name)))
    }

    /// Set a named style.
    pub fn set(&mut self, name: impl Into<String>, style: Style) {
        self.styles.insert(name.into(), style);
    }

    /// Merge another theme's styles into this one.
    pub fn merge(&mut self, other: &Theme) {
        for (name, style) in &other.styles {
            self.styles.entry(name.clone()).or_insert_with(|| style.clone());
        }
    }
}

impl Default for Theme {
    fn default() -> Self {
        Self::new()
    }
}

// ---------------------------------------------------------------------------
// ThemeStack
// ---------------------------------------------------------------------------

/// A stack of themes — when looking up a style, themes are checked from
/// top to bottom until a match is found.
#[derive(Debug, Clone)]
pub struct ThemeStack {
    themes: Vec<Theme>,
}

impl ThemeStack {
    pub fn new() -> Self {
        Self { themes: Vec::new() }
    }

    pub fn push(&mut self, theme: Theme) {
        self.themes.push(theme);
    }

    pub fn pop(&mut self) -> Option<Theme> {
        self.themes.pop()
    }

    /// Look up a style name across the stack (top-down).
    pub fn get(&self, name: &str) -> Option<Style> {
        for theme in self.themes.iter().rev() {
            if let Some(s) = theme.get(name) {
                return Some(s.clone());
            }
        }
        None
    }
}

// ---------------------------------------------------------------------------
// Default themes
// ---------------------------------------------------------------------------

/// Built-in theme style names used by various Rich renderables.
pub mod names {
    // repr / pretty
    pub const REPR_NUMBER: &str = "repr.number";
    pub const REPR_STR: &str = "repr.str";
    pub const REPR_BOOL_TRUE: &str = "repr.bool_true";
    pub const REPR_BOOL_FALSE: &str = "repr.bool_false";
    pub const REPR_NONE: &str = "repr.none";
    pub const REPR_URL: &str = "repr.url";
    pub const REPR_PATH: &str = "repr.path";
    pub const REPR_IPV4: &str = "repr.ipv4";
    pub const REPR_IPV6: &str = "repr.ipv6";
    pub const REPR_ELLIPSIS: &str = "repr.ellipsis";
    pub const REPR_ATTRIB_NAME: &str = "repr.attrib_name";
    pub const REPR_ATTRIB_VALUE: &str = "repr.attrib_value";
    pub const REPR_TAG_NAME: &str = "repr.tag_name";
    pub const REPR_TAG_CONTENTS: &str = "repr.tag_contents";
    pub const REPR_TAG_PUNCTUATION: &str = "repr.tag_punctuation";
    pub const REPR_BRACE: &str = "repr.brace";
    pub const REPR_CALL: &str = "repr.call";
    pub const REPR_COMMA: &str = "repr.comma";
    pub const REPR_BOOL: &str = "repr.bool";
    pub const REPR_ERROR: &str = "repr.error";

    // table
    pub const TABLE_HEADER: &str = "table.header";
    pub const TABLE_CELL: &str = "table.cell";
    pub const TABLE_ROW: &str = "table.row";
    pub const TABLE_FOOTER: &str = "table.footer";
    pub const TABLE_TITLE: &str = "table.title";
    pub const TABLE_CAPTION: &str = "table.caption";
    pub const TABLE_BORDER: &str = "table.border";

    // logging
    pub const LOGGING_KEYWORD: &str = "logging.keyword";
    pub const LOGGING_LEVEL_DEBUG: &str = "logging.level.debug";
    pub const LOGGING_LEVEL_INFO: &str = "logging.level.info";
    pub const LOGGING_LEVEL_WARNING: &str = "logging.level.warning";
    pub const LOGGING_LEVEL_ERROR: &str = "logging.level.error";
    pub const LOGGING_LEVEL_CRITICAL: &str = "logging.level.critical";
    pub const LOGGING_LEVEL_NOTSET: &str = "logging.level.notset";

    // traceback
    pub const TRACEBACK_BORDER: &str = "traceback.border";
    pub const TRACEBACK_TITLE: &str = "traceback.title";
    pub const TRACEBACK_ERROR: &str = "traceback.error";
    pub const TRACEBACK_ERROR_MARK: &str = "traceback.error_mark";
    pub const TRACEBACK_FILENAME: &str = "traceback.filename";
    pub const TRACEBACK_LINE_NO: &str = "traceback.line_no";
    pub const TRACEBACK_LOCALS_HEADER: &str = "traceback.locals_header";
    pub const TRACEBACK_LOCALS_KEY: &str = "traceback.locals_key";
    pub const TRACEBACK_LOCALS_VALUE: &str = "traceback.locals_value";
    pub const TRACEBACK_EXC_TYPE: &str = "traceback.exc_type";
    pub const TRACEBACK_EXC_VALUE: &str = "traceback.exc_value";

    // general
    pub const RULE_LINE: &str = "rule.line";
    pub const RULE_TEXT: &str = "rule.text";
    pub const BAR_COMPLETE: &str = "bar.complete";
    pub const BAR_FINISHED: &str = "bar.finished";
    pub const BAR_PULSE: &str = "bar.pulse";
    pub const BAR_REMAINING: &str = "bar.remaining";
    pub const PROGRESS_DESCRIPTION: &str = "progress.description";
    pub const PROGRESS_PERCENTAGE: &str = "progress.percentage";
    pub const PROGRESS_REMAINING: &str = "progress.remaining";
    pub const PROGRESS_ELAPSED: &str = "progress.elapsed";
    pub const PROGRESS_DATA: &str = "progress.data";
    pub const PROGRESS_DOWNLOAD: &str = "progress.download";
    pub const PROGRESS_TRANSFER: &str = "progress.transfer";
    pub const PROGRESS_FILESIZE: &str = "progress.filesize";
    pub const PROGRESS_TOTAL: &str = "progress.total";
    pub const TREE: &str = "tree";
    pub const TREE_LINE: &str = "tree.line";
    pub const MARKDOWN_H1: &str = "markdown.h1";
    pub const MARKDOWN_H2: &str = "markdown.h2";
    pub const MARKDOWN_CODE: &str = "markdown.code";
    pub const MARKDOWN_LINK: &str = "markdown.link";
    pub const MARKDOWN_ITEM: &str = "markdown.item";
    pub const MARKDOWN_H3: &str = "markdown.h3";
    pub const MARKDOWN_H4: &str = "markdown.h4";
    pub const MARKDOWN_H5: &str = "markdown.h5";
    pub const MARKDOWN_H6: &str = "markdown.h6";
    pub const MARKDOWN_H7: &str = "markdown.h7";
    pub const MARKDOWN_PARAGRAPH: &str = "markdown.paragraph";
    pub const MARKDOWN_BLOCKQUOTE: &str = "markdown.blockquote";
    pub const MARKDOWN_LIST: &str = "markdown.list";
    pub const MARKDOWN_ITEM_BULLET: &str = "markdown.item.bullet";
    pub const MARKDOWN_ITEM_NUMBER: &str = "markdown.item.number";
    pub const MARKDOWN_TABLE: &str = "markdown.table";
    pub const MARKDOWN_TABLE_HEADER: &str = "markdown.table.header";
    pub const MARKDOWN_CODE_BLOCK: &str = "markdown.code.block";
    pub const MARKDOWN_CODE_INLINE: &str = "markdown.code.inline";
    pub const MARKDOWN_HR: &str = "markdown.hr";
    pub const JSON_KEY: &str = "json.key";
    pub const JSON_STR: &str = "json.str";
    pub const JSON_NUMBER: &str = "json.number";
    pub const JSON_BOOL: &str = "json.bool";
    pub const JSON_NULL: &str = "json.null";
    pub const JSON_BOOL_TRUE: &str = "json.bool_true";
    pub const JSON_BOOL_FALSE: &str = "json.bool_false";
    pub const JSON_BRACE: &str = "json.brace";

    // status
    pub const STATUS_SPINNER: &str = "status.spinner";
    pub const STATUS_MESSAGE: &str = "status.message";

    // prompt
    pub const PROMPT: &str = "prompt";
    pub const PROMPT_CHOICES: &str = "prompt.choices";
    pub const PROMPT_DEFAULT: &str = "prompt.default";

    // syntax highlighting (via syntect)
    pub const SYNTAX_COMMENT: &str = "syntax.comment";
    pub const SYNTAX_KEYWORD: &str = "syntax.keyword";
    pub const SYNTAX_STRING: &str = "syntax.string";
    pub const SYNTAX_NUMBER: &str = "syntax.number";
    pub const SYNTAX_FUNCTION: &str = "syntax.function";
    pub const SYNTAX_TYPE: &str = "syntax.type";
}

/// Create the default Rich-like theme.
pub fn default_theme() -> Theme {
    let mut t = Theme::new();
    use crate::color::Color;
    use crate::style::Style;

    // repr styles
    t.set(names::REPR_NUMBER, Style::new().color(Color::parse("cyan").unwrap()).bold(true));
    t.set(names::REPR_STR, Style::new().color(Color::parse("green").unwrap()));
    t.set(names::REPR_BOOL_TRUE, Style::new().color(Color::parse("bright_green").unwrap()).bold(true));
    t.set(names::REPR_BOOL_FALSE, Style::new().color(Color::parse("bright_red").unwrap()).bold(true));
    t.set(names::REPR_NONE, Style::new().color(Color::parse("bright_yellow").unwrap()).dim(true));
    t.set(names::REPR_URL, Style::new().color(Color::parse("bright_blue").unwrap()).underline(true));
    t.set(names::REPR_PATH, Style::new().color(Color::parse("magenta").unwrap()));
    t.set(names::REPR_ATTRIB_NAME, Style::new().color(Color::parse("bright_cyan").unwrap()));
    t.set(names::REPR_ATTRIB_VALUE, Style::new().color(Color::parse("white").unwrap()));
    t.set(names::REPR_ELLIPSIS, Style::new().dim(true).color(Color::parse("white").unwrap()));
    t.set(names::REPR_IPV4, Style::new().color(Color::parse("bright_blue").unwrap()).bold(true));
    t.set(names::REPR_IPV6, Style::new().color(Color::parse("bright_magenta").unwrap()).bold(true));
    t.set(names::REPR_TAG_NAME, Style::new().color(Color::parse("bright_blue").unwrap()).bold(true));
    t.set(names::REPR_TAG_CONTENTS, Style::new().color(Color::parse("white").unwrap()));
    t.set(names::REPR_TAG_PUNCTUATION, Style::new().color(Color::parse("bright_black").unwrap()));
    t.set(names::REPR_BRACE, Style::new().color(Color::parse("bright_black").unwrap()));
    t.set(names::REPR_CALL, Style::new().color(Color::parse("bright_cyan").unwrap()));
    t.set(names::REPR_COMMA, Style::new().color(Color::parse("bright_black").unwrap()));
    t.set(names::REPR_BOOL, Style::new().color(Color::parse("bright_yellow").unwrap()).bold(true));
    t.set(names::REPR_ERROR, Style::new().color(Color::parse("bright_red").unwrap()).bold(true));

    // table styles
    t.set(names::TABLE_HEADER, Style::new().bold(true).color(Color::parse("white").unwrap()));
    t.set(names::TABLE_FOOTER, Style::new().bold(true));
    t.set(names::TABLE_TITLE, Style::new().bold(true));
    t.set(names::TABLE_CAPTION, Style::new().dim(true));
    t.set(names::TABLE_BORDER, Style::new().color(Color::parse("bright_black").unwrap()));
    t.set(names::TABLE_CELL, Style::new().color(Color::parse("white").unwrap()));
    t.set(names::TABLE_ROW, Style::new().color(Color::parse("white").unwrap()));

    // rule
    t.set(names::RULE_LINE, Style::new().color(Color::parse("bright_black").unwrap()));
    t.set(names::RULE_TEXT, Style::new().bold(true));

    // tree
    t.set(names::TREE, Style::new().color(Color::parse("white").unwrap()));
    t.set(names::TREE_LINE, Style::new().color(Color::parse("bright_black").unwrap()));

    // progress
    t.set(names::BAR_COMPLETE, Style::new().color(Color::parse("green").unwrap()));
    t.set(names::BAR_FINISHED, Style::new().color(Color::parse("bright_green").unwrap()));
    t.set(names::BAR_PULSE, Style::new().color(Color::parse("bright_cyan").unwrap()));
    t.set(names::PROGRESS_DESCRIPTION, Style::new().color(Color::parse("white").unwrap()));
    t.set(names::PROGRESS_PERCENTAGE, Style::new().color(Color::parse("cyan").unwrap()));
    t.set(names::PROGRESS_REMAINING, Style::new().color(Color::parse("bright_black").unwrap()));
    t.set(names::PROGRESS_ELAPSED, Style::new().color(Color::parse("cyan").unwrap()));
    t.set(names::PROGRESS_DATA, Style::new().color(Color::parse("bright_blue").unwrap()));
    t.set(names::PROGRESS_DOWNLOAD, Style::new().color(Color::parse("bright_cyan").unwrap()));
    t.set(names::PROGRESS_TRANSFER, Style::new().color(Color::parse("bright_magenta").unwrap()));
    t.set(names::PROGRESS_FILESIZE, Style::new().color(Color::parse("bright_green").unwrap()));
    t.set(names::PROGRESS_TOTAL, Style::new().color(Color::parse("bright_blue").unwrap()));
    t.set(names::BAR_REMAINING, Style::new().color(Color::parse("bright_black").unwrap()));

    // markdown
    t.set(names::MARKDOWN_H1, Style::new().bold(true).color(Color::parse("bright_cyan").unwrap()));
    t.set(names::MARKDOWN_H2, Style::new().bold(true).color(Color::parse("cyan").unwrap()));
    t.set(names::MARKDOWN_CODE, Style::new().color(Color::parse("bright_yellow").unwrap()).bgcolor(Color::parse("black").unwrap()));
    t.set(names::MARKDOWN_LINK, Style::new().color(Color::parse("bright_blue").unwrap()).underline(true));
    t.set(names::MARKDOWN_H3, Style::new().bold(true).color(Color::parse("yellow").unwrap()));
    t.set(names::MARKDOWN_H4, Style::new().bold(true).color(Color::parse("green").unwrap()));
    t.set(names::MARKDOWN_H5, Style::new().bold(true).color(Color::parse("blue").unwrap()));
    t.set(names::MARKDOWN_H6, Style::new().bold(true).color(Color::parse("bright_black").unwrap()));
    t.set(names::MARKDOWN_H7, Style::new().dim(true).color(Color::parse("bright_black").unwrap()));
    t.set(names::MARKDOWN_PARAGRAPH, Style::new().color(Color::parse("white").unwrap()));
    t.set(names::MARKDOWN_BLOCKQUOTE, Style::new().color(Color::parse("bright_black").unwrap()));
    t.set(names::MARKDOWN_LIST, Style::new().color(Color::parse("white").unwrap()));
    t.set(names::MARKDOWN_ITEM, Style::new().color(Color::parse("cyan").unwrap()));
    t.set(names::MARKDOWN_ITEM_BULLET, Style::new().color(Color::parse("cyan").unwrap()));
    t.set(names::MARKDOWN_ITEM_NUMBER, Style::new().color(Color::parse("cyan").unwrap()));
    t.set(names::MARKDOWN_TABLE, Style::new().color(Color::parse("white").unwrap()));
    t.set(names::MARKDOWN_TABLE_HEADER, Style::new().bold(true).color(Color::parse("white").unwrap()));
    t.set(names::MARKDOWN_CODE_BLOCK, Style::new().color(Color::parse("bright_yellow").unwrap()).bgcolor(Color::parse("black").unwrap()));
    t.set(names::MARKDOWN_CODE_INLINE, Style::new().color(Color::parse("bright_yellow").unwrap()));
    t.set(names::MARKDOWN_HR, Style::new().color(Color::parse("bright_black").unwrap()));

    // json
    t.set(names::JSON_KEY, Style::new().color(Color::parse("cyan").unwrap()));
    t.set(names::JSON_STR, Style::new().color(Color::parse("green").unwrap()));
    t.set(names::JSON_NUMBER, Style::new().color(Color::parse("bright_blue").unwrap()).bold(true));
    t.set(names::JSON_BOOL, Style::new().color(Color::parse("bright_yellow").unwrap()).bold(true));
    t.set(names::JSON_NULL, Style::new().color(Color::parse("bright_red").unwrap()).dim(true));
    t.set(names::JSON_BOOL_TRUE, Style::new().color(Color::parse("bright_green").unwrap()).bold(true));
    t.set(names::JSON_BOOL_FALSE, Style::new().color(Color::parse("bright_red").unwrap()).bold(true));
    t.set(names::JSON_BRACE, Style::new().color(Color::parse("bright_black").unwrap()));

    // traceback
    t.set(names::TRACEBACK_BORDER, Style::new().color(Color::parse("red").unwrap()));
    t.set(names::TRACEBACK_TITLE, Style::new().bold(true));
    t.set(names::TRACEBACK_ERROR, Style::new().color(Color::parse("bright_red").unwrap()).bold(true));
    t.set(names::TRACEBACK_ERROR_MARK, Style::new().color(Color::parse("bright_red").unwrap()).bold(true));
    t.set(names::TRACEBACK_FILENAME, Style::new().color(Color::parse("cyan").unwrap()));
    t.set(names::TRACEBACK_LINE_NO, Style::new().color(Color::parse("bright_black").unwrap()));
    t.set(names::TRACEBACK_LOCALS_HEADER, Style::new().bold(true));
    t.set(names::TRACEBACK_LOCALS_KEY, Style::new().color(Color::parse("bright_cyan").unwrap()));
    t.set(names::TRACEBACK_LOCALS_VALUE, Style::new().color(Color::parse("white").unwrap()));
    t.set(names::TRACEBACK_EXC_TYPE, Style::new().color(Color::parse("bright_red").unwrap()).bold(true));
    t.set(names::TRACEBACK_EXC_VALUE, Style::new().color(Color::parse("white").unwrap()));

    // logging extras
    t.set(names::LOGGING_KEYWORD, Style::new().color(Color::parse("bright_yellow").unwrap()).bold(true));
    t.set(names::LOGGING_LEVEL_NOTSET, Style::new().dim(true).color(Color::parse("white").unwrap()));

    // status & prompt
    t.set(names::STATUS_SPINNER, Style::new().color(Color::parse("bright_cyan").unwrap()));
    t.set(names::STATUS_MESSAGE, Style::new().color(Color::parse("white").unwrap()));
    t.set(names::PROMPT, Style::new().color(Color::parse("bright_cyan").unwrap()));
    t.set(names::PROMPT_CHOICES, Style::new().color(Color::parse("cyan").unwrap()));
    t.set(names::PROMPT_DEFAULT, Style::new().color(Color::parse("bright_black").unwrap()));

    t
}