cship 1.7.0

//! Starship-compatible per-module format string parser.
//!
//! Supported syntax (simplified subset):
//! - `$value`  — module's computed raw value (absent = empty string in renders, None in conditionals)
//! - `$symbol` — module's configured symbol (defaults to empty string if None)
//! - `[content]($style)` — style span: render `content` with module's configured style
//! - `[content](bold red)` — style span: render `content` with the literal style string
//! - `(content)` — conditional group: renders as empty if `$value` is None
//! - Literal text — preserved verbatim
//!
//! Source: architecture.md#Core Architectural Decisions, epics.md#Story 2.5

/// Centralized style/threshold/format rendering for sub-field render functions.
///
/// Resolves style, thresholds, and format strings using sub-field → parent fallback,
/// handling `invert_threshold` for decreasing-health indicators (e.g. remaining_percentage).
///
/// # Arguments
/// - `val_str`: Already-formatted display string (e.g. `"85"`, `"0.0123"`)
/// - `threshold_val`: Numeric value for threshold comparison; `None` for non-threshold fields
/// - `sub_cfg`: The sub-field's own `SubfieldConfig` (may be `None`)
/// - `parent`: Parent config implementing `HasThresholdStyle` for fallback (may be `None`)
///
/// # Returns
/// `None` when the format path renders empty (conditional group with absent `$value`).
/// `Some(styled_string)` otherwise.
pub fn render_styled_value(
    val_str: &str,
    threshold_val: Option<f64>,
    sub_cfg: Option<&crate::config::SubfieldConfig>,
    parent: Option<&dyn crate::config::HasThresholdStyle>,
) -> Option<String> {
    // Resolve all fields with sub → parent fallback
    let style = sub_cfg
        .and_then(|c| c.style.as_deref())
        .or_else(|| parent.and_then(|p| p.style()));
    let mut effective_val = threshold_val;
    let mut warn_threshold = sub_cfg
        .and_then(|c| c.warn_threshold)
        .or_else(|| parent.and_then(|p| p.warn_threshold()));
    let mut warn_style = sub_cfg
        .and_then(|c| c.warn_style.as_deref())
        .or_else(|| parent.and_then(|p| p.warn_style()));
    let mut critical_threshold = sub_cfg
        .and_then(|c| c.critical_threshold)
        .or_else(|| parent.and_then(|p| p.critical_threshold()));
    let mut critical_style = sub_cfg
        .and_then(|c| c.critical_style.as_deref())
        .or_else(|| parent.and_then(|p| p.critical_style()));

    // Inverted thresholds: use sub-only values (negated) and negate the comparison value.
    // Parent thresholds are in the non-inverted domain and must not be inherited.
    if sub_cfg.and_then(|c| c.invert_threshold).unwrap_or(false) {
        effective_val = threshold_val.map(|v| -v);
        warn_threshold = sub_cfg.and_then(|c| c.warn_threshold).map(|t| -t);
        warn_style = sub_cfg.and_then(|c| c.warn_style.as_deref());
        critical_threshold = sub_cfg.and_then(|c| c.critical_threshold).map(|t| -t);
        critical_style = sub_cfg.and_then(|c| c.critical_style.as_deref());
    }

    // Format path: resolve symbol, threshold style, then apply format
    let fmt = sub_cfg
        .and_then(|c| c.format.as_deref())
        .or_else(|| parent.and_then(|p| p.format_str()));
    if let Some(fmt) = fmt {
        let symbol = sub_cfg
            .and_then(|c| c.symbol.as_deref())
            .or_else(|| parent.and_then(|p| p.symbol_str()));
        let effective_style = crate::ansi::resolve_threshold_style(
            effective_val,
            style,
            warn_threshold,
            warn_style,
            critical_threshold,
            critical_style,
        );
        let result = apply_module_format(fmt, Some(val_str), symbol, effective_style);
        if result.is_none() {
            tracing::warn!(
                "render_styled_value: format path returned None (empty conditional group — \
                 $value absent from context)"
            );
        }
        return result;
    }

    // Default path: apply_style_with_threshold handles no-threshold gracefully
    Some(crate::ansi::apply_style_with_threshold(
        val_str,
        effective_val,
        style,
        warn_threshold,
        warn_style,
        critical_threshold,
        critical_style,
    ))
}

/// Apply a per-module format string.
///
/// # Arguments
/// - `fmt`: Format template (e.g. `"[ ctx: $value% ]($style)"`)
/// - `value`: Module's computed raw value; `None` = field absent from context
/// - `symbol`: Module's configured `symbol` field; `None` = no symbol configured
/// - `style`: Module's configured `style` string; used when format references `$style`
///
/// # Returns
/// `None` when the rendered result is empty (conditional group whose `$value` is None).
/// `Some(rendered)` otherwise.
pub fn apply_module_format(
    fmt: &str,
    value: Option<&str>,
    symbol: Option<&str>,
    style: Option<&str>,
) -> Option<String> {
    let rendered = render_fmt(fmt, value, symbol, style);
    if rendered.is_empty() {
        None
    } else {
        Some(rendered)
    }
}

fn render_fmt(s: &str, value: Option<&str>, symbol: Option<&str>, style: Option<&str>) -> String {
    let mut out = String::with_capacity(s.len());
    let mut pos = 0;

    while pos < s.len() {
        let remaining = &s[pos..];

        if remaining.starts_with('[')
            && let Some((content, style_spec, len)) = parse_style_span(remaining)
        {
            let rendered_content = render_fmt(&content, value, symbol, style);
            let applied = if style_spec == "$style" {
                crate::ansi::apply_style(&rendered_content, style)
            } else {
                crate::ansi::apply_style(&rendered_content, Some(style_spec.as_str()))
            };
            out.push_str(&applied);
            pos += len;
            continue;
        }

        if remaining.starts_with('(')
            && let Some((content, len)) = parse_paren_group(remaining)
        {
            let conditional_result = render_conditional(&content, value, symbol, style);
            out.push_str(&conditional_result);
            pos += len;
            continue;
        }

        if let Some(rest) = remaining.strip_prefix('$') {
            let var_end = rest
                .find(|c: char| !c.is_alphanumeric() && c != '_')
                .unwrap_or(rest.len());
            let var_name = &rest[..var_end];
            let subst = match var_name {
                "value" => value.unwrap_or(""),
                "symbol" => symbol.unwrap_or(""),
                _ => "",
            };
            out.push_str(subst);
            pos += 1 + var_end; // skip '$' + var_name bytes
            continue;
        }

        // Literal character — advance by one Unicode scalar
        let ch = remaining.chars().next().unwrap();
        out.push(ch);
        pos += ch.len_utf8();
    }

    out
}

/// Conditional group: renders as empty string if `$value` is referenced AND is `None`.
fn render_conditional(
    content: &str,
    value: Option<&str>,
    symbol: Option<&str>,
    style: Option<&str>,
) -> String {
    if content.contains("$value") && value.is_none() {
        return String::new();
    }
    render_fmt(content, value, symbol, style)
}

/// Parse `[content]($style_spec)` from the start of `s`.
/// Returns `(content, style_spec, bytes_consumed)` or `None` if not a valid span.
/// Handles nested brackets via depth tracking (e.g., `[[nested]](style)` works correctly).
fn parse_style_span(s: &str) -> Option<(String, String, usize)> {
    debug_assert!(s.starts_with('['));
    let close_bracket = find_matching_close(s, 1, '[', ']')?;
    let content = s[1..close_bracket].to_string();
    let after = &s[close_bracket + 1..];
    if !after.starts_with('(') {
        return None;
    }
    let close_paren = find_matching_close(after, 1, '(', ')')?;
    let style_spec = after[1..close_paren].to_string();
    let total_len = close_bracket + 1 + close_paren + 1;
    Some((content, style_spec, total_len))
}

/// Parse `(content)` from the start of `s`.
/// Returns `(content, bytes_consumed)` or `None` if not a valid group.
/// Handles nested parentheses via depth tracking (e.g., `(cost: $value (USD))` works correctly).
fn parse_paren_group(s: &str) -> Option<(String, usize)> {
    debug_assert!(s.starts_with('('));
    let close_paren = find_matching_close(s, 1, '(', ')')?;
    let content = s[1..close_paren].to_string();
    Some((content, close_paren + 1))
}

/// Find the byte offset of the matching closing delimiter, respecting nesting depth.
/// `start` is the byte offset to begin scanning (after the opening delimiter).
fn find_matching_close(s: &str, start: usize, open: char, close: char) -> Option<usize> {
    let mut depth: u32 = 1;
    for (i, ch) in s[start..].char_indices() {
        if ch == open {
            depth += 1;
        } else if ch == close {
            depth -= 1;
            if depth == 0 {
                return Some(start + i);
            }
        }
    }
    None
}

#[cfg(test)]
mod tests {
    use super::*;

    // --- render_styled_value tests ---

    #[test]
    fn test_render_styled_value_with_format_string() {
        let sub = crate::config::SubfieldConfig {
            format: Some("[$value]($style)".to_string()),
            style: Some("bold green".to_string()),
            ..Default::default()
        };
        let result = render_styled_value("85", Some(85.0), Some(&sub), None);
        let s = result.unwrap();
        assert!(s.contains("85"), "value present: {s:?}");
        assert!(s.contains('\x1b'), "ANSI codes present: {s:?}");
    }

    #[test]
    fn test_render_styled_value_no_format_threshold_above_warn() {
        let sub = crate::config::SubfieldConfig {
            warn_threshold: Some(50.0),
            warn_style: Some("yellow".to_string()),
            critical_threshold: Some(90.0),
            critical_style: Some("bold red".to_string()),
            ..Default::default()
        };
        let result = render_styled_value("75", Some(75.0), Some(&sub), None);
        let s = result.unwrap();
        assert!(s.contains("75"), "value present: {s:?}");
        assert!(s.contains('\x1b'), "ANSI codes for warn style: {s:?}");
    }

    #[test]
    fn test_render_styled_value_no_format_no_threshold() {
        let sub = crate::config::SubfieldConfig {
            style: Some("cyan".to_string()),
            ..Default::default()
        };
        let result = render_styled_value("hello", None, Some(&sub), None);
        let s = result.unwrap();
        assert!(s.contains("hello"), "value present: {s:?}");
        assert!(s.contains('\x1b'), "ANSI codes for base style: {s:?}");
    }

    #[test]
    fn test_render_styled_value_invert_threshold() {
        // invert_threshold: value 20 with warn_threshold 30 → inverted: -20 >= -30 → warn fires
        let sub = crate::config::SubfieldConfig {
            invert_threshold: Some(true),
            warn_threshold: Some(30.0),
            warn_style: Some("yellow".to_string()),
            critical_threshold: Some(10.0),
            critical_style: Some("bold red".to_string()),
            ..Default::default()
        };
        let result = render_styled_value("20", Some(20.0), Some(&sub), None);
        let s = result.unwrap();
        assert!(s.contains("20"), "value present: {s:?}");
        assert!(s.contains('\x1b'), "ANSI codes for inverted warn: {s:?}");
    }

    #[test]
    fn test_render_styled_value_parent_fallback_style() {
        let sub = crate::config::SubfieldConfig {
            ..Default::default()
        };
        let parent = crate::config::ContextWindowConfig {
            style: Some("bold magenta".to_string()),
            ..Default::default()
        };
        let result = render_styled_value(
            "42",
            None,
            Some(&sub),
            Some(&parent as &dyn crate::config::HasThresholdStyle),
        );
        let s = result.unwrap();
        assert!(s.contains("42"), "value present: {s:?}");
        assert!(s.contains('\x1b'), "ANSI codes from parent style: {s:?}");
    }

    #[test]
    fn test_render_styled_value_no_sub_no_parent() {
        let result = render_styled_value("plain", None, None, None);
        assert_eq!(result, Some("plain".to_string()));
    }

    #[test]
    fn test_dollar_value_substituted() {
        let result = apply_module_format("$value", Some("35"), None, None);
        assert_eq!(result, Some("35".to_string()));
    }

    #[test]
    fn test_dollar_symbol_substituted() {
        let result = apply_module_format("$symbol$value", Some("35"), Some("🔷"), None);
        assert_eq!(result, Some("🔷35".to_string()));
    }

    #[test]
    fn test_literal_text_preserved() {
        let result = apply_module_format("ctx: $value%", Some("8"), None, None);
        assert_eq!(result, Some("ctx: 8%".to_string()));
    }

    #[test]
    fn test_style_span_applies_ansi() {
        let result = apply_module_format("[$value]($style)", Some("35"), None, Some("bold green"));
        let s = result.unwrap();
        assert!(s.contains("35"), "value present: {s:?}");
        assert!(s.contains('\x1b'), "ANSI codes present: {s:?}");
    }

    #[test]
    fn test_style_span_dollar_style_uses_module_style() {
        // $style in format means "use the module's configured style"
        let result = apply_module_format("[$value]($style)", Some("OK"), None, Some("bold red"));
        let s = result.unwrap();
        assert!(s.contains("OK"));
        assert!(s.contains('\x1b'));
    }

    #[test]
    fn test_style_span_literal_style_string() {
        // literal style in format (not $style)
        let result = apply_module_format("[text](bold cyan)", Some("x"), None, None);
        let s = result.unwrap();
        assert!(s.contains("text"));
        assert!(s.contains('\x1b'));
    }

    #[test]
    fn test_conditional_value_none_returns_none() {
        // AC3: ($value) with value=None → entire group omitted → module returns None
        let result = apply_module_format("($value)", None, None, None);
        assert_eq!(result, None);
    }

    #[test]
    fn test_conditional_value_present_renders_without_parens() {
        // AC4: ($value) with value="NORMAL" → "NORMAL"
        let result = apply_module_format("($value)", Some("NORMAL"), None, None);
        assert_eq!(result, Some("NORMAL".to_string()));
    }

    #[test]
    fn test_complex_format_ac1_style() {
        // AC1 pattern: '[ ctx: $value% ]($style)'
        let result = apply_module_format("[ ctx: $value% ]($style)", Some("8"), None, None);
        // No style configured → no ANSI codes, just literal text
        assert_eq!(result, Some(" ctx: 8% ".to_string()));
    }

    #[test]
    fn test_complex_format_ac1_with_style() {
        let result = apply_module_format(
            "[ ctx: $value% ]($style)",
            Some("8"),
            None,
            Some("bold green"),
        );
        let s = result.unwrap();
        assert!(s.contains("8"), "value in output: {s:?}");
        assert!(s.contains('\x1b'), "ANSI codes: {s:?}");
    }

    #[test]
    fn test_no_variables_returns_literal() {
        let result = apply_module_format("fixed text", None, None, None);
        assert_eq!(result, Some("fixed text".to_string()));
    }

    #[test]
    fn test_empty_format_returns_none() {
        let result = apply_module_format("", None, None, None);
        assert_eq!(result, None);
    }

    #[test]
    fn test_value_none_outside_conditional_renders_empty_string() {
        // $value with None outside conditional → treated as "" (not conditional)
        let result = apply_module_format("prefix $value suffix", None, None, None);
        assert_eq!(result, Some("prefix  suffix".to_string()));
    }

    #[test]
    fn test_nested_parens_in_conditional_group() {
        // "(cost: $value (USD))" — outer group contains $value; inner (USD) is a nested
        // conditional group with no $value ref → renders "USD" (parens are syntax, not output).
        let result = apply_module_format("(cost: $value (USD))", Some("5.00"), None, None);
        assert_eq!(result, Some("cost: 5.00 USD".to_string()));
    }

    #[test]
    fn test_nested_parens_conditional_absent_omits_entire_group() {
        // Outer group references $value which is None → entire group omitted
        let result = apply_module_format("(cost: $value (USD))", None, None, None);
        assert_eq!(result, None);
    }

    #[test]
    fn test_ac2_pattern_value_symbol_style_combined() {
        // AC2 pattern: "[$value $symbol]($style)" with all three variables
        let result = apply_module_format(
            "[$value $symbol]($style)",
            Some("bar"),
            Some("🧠"),
            Some("bold cyan"),
        );
        let s = result.unwrap();
        assert!(s.contains("bar"), "value present: {s:?}");
        assert!(s.contains("🧠"), "symbol present: {s:?}");
        assert!(s.contains('\x1b'), "ANSI codes present: {s:?}");
    }

    #[test]
    fn test_render_styled_value_format_path_none_returns_none() {
        // AC: When apply_module_format returns None on the format path, render_styled_value
        // must return None. The tracing::warn! guard is exercised (best-effort in tests).
        // An empty format string causes apply_module_format to return None.
        let sub_empty_fmt = crate::config::SubfieldConfig {
            format: Some(String::new()),
            ..Default::default()
        };
        let result = render_styled_value("x", None, Some(&sub_empty_fmt), None);
        assert_eq!(
            result, None,
            "empty format string should cause format path to return None"
        );
    }
}