branchdiff 0.62.0

use ratatui::{
    style::{Color, Style},
    text::{Line, Span},
};
use unicode_width::UnicodeWidthStr;

use super::ScreenRowInfo;
use crate::diff::DiffLine;

/// Image dimensions for height calculation.
///
/// Tuple of (before_dims, after_dims) where each is `Some((width, height))` in pixels
/// if the image exists in the cache, or `None` if not loaded yet.
///
/// Used by `wrapped_line_height` to calculate the display height for image diff markers.
pub type ImageDimensions = (Option<(u32, u32)>, Option<(u32, u32)>);

/// Calculate how many screen rows a line will take when wrapped.
///
/// This is the single source of truth for line height calculation, used by both
/// `App` (for navigation) and `FrameContext` (for visible range calculation).
///
/// For image markers, `image_dims` should contain the cached dimensions if available.
/// For text lines, handles mixed inline changes (deletions + insertions on same line).
pub fn wrapped_line_height(
    line: &DiffLine,
    content_width: usize,
    image_dims: Option<ImageDimensions>,
    panel_width: u16,
    font_size: (u16, u16),
) -> usize {
    if content_width == 0 {
        return 1;
    }

    // Image markers use dimension-based height
    if line.is_image_marker() {
        if let Some((before, after)) = image_dims
            && (before.is_some() || after.is_some())
        {
            return crate::ui::image_view::calculate_image_height_for_images(
                before,
                after,
                panel_width,
                font_size,
            ) as usize;
        }
        // Fallback: minimal height for images not yet loaded
        return 1;
    }

    // Pure deletions always render as split del/ins lines to show removed text
    if has_pure_deletion_inline(line) {
        let del_width = line
            .old_content
            .as_ref()
            .map(|s| content_display_width(s))
            .unwrap_or(0);
        let ins_width = content_display_width(&line.content);
        let del_height = if del_width == 0 {
            0
        } else {
            del_width.div_ceil(content_width).max(1)
        };
        let ins_height = ins_width.div_ceil(content_width).max(1);
        return del_height + ins_height;
    }

    // Mixed inline changes (deletions + insertions) may render on separate rows
    if has_mixed_inline_changes(line) && content_display_width(&line.content) > content_width {
        let del_width = line
            .old_content
            .as_ref()
            .map(|s| content_display_width(s))
            .unwrap_or(0);
        let ins_width = content_display_width(&line.content);
        let del_height = if del_width == 0 {
            0
        } else {
            del_width.div_ceil(content_width)
        };
        let ins_height = ins_width.div_ceil(content_width);
        return del_height + ins_height;
    }

    // Standard text wrapping
    let width = content_display_width(&line.content);
    if width <= content_width {
        1
    } else {
        width.div_ceil(content_width)
    }
}

/// Check if a line has both deletions and insertions in its inline spans.
fn has_mixed_inline_changes(line: &DiffLine) -> bool {
    if line.inline_spans.is_empty() {
        return false;
    }
    let has_deletions = line.inline_spans.iter().any(|s| s.is_deletion);
    let has_insertions = line
        .inline_spans
        .iter()
        .any(|s| !s.is_deletion && s.source.is_some());
    has_deletions && has_insertions
}

/// Check if a line has only deletions (no insertions) in its inline spans.
fn has_pure_deletion_inline(line: &DiffLine) -> bool {
    if line.inline_spans.is_empty() {
        return false;
    }
    let has_deletions = line.inline_spans.iter().any(|s| s.is_deletion);
    let has_insertions = line
        .inline_spans
        .iter()
        .any(|s| !s.is_deletion && s.source.is_some());
    has_deletions && !has_insertions
}

/// Compute the display width of content accounting for tab expansion and
/// control character replacement. Must match `sanitize_for_display` behavior
/// so that height estimation agrees with actual rendering.
pub fn content_display_width(s: &str) -> usize {
    use unicode_width::UnicodeWidthChar;
    s.chars()
        .map(|ch| {
            if ch == '\t' {
                4
            } else {
                UnicodeWidthChar::width(ch).unwrap_or(1)
            }
        })
        .sum()
}

/// Replace characters that cause terminal rendering artifacts.
/// Tabs expand to 4 spaces; other control characters (unicode-width None)
/// become spaces so they have predictable 1-column display width.
fn sanitize_for_display(s: &str) -> Option<String> {
    use unicode_width::UnicodeWidthChar;
    if !s
        .chars()
        .any(|ch| ch == '\t' || UnicodeWidthChar::width(ch).is_none())
    {
        return None;
    }
    let mut result = String::with_capacity(s.len());
    for ch in s.chars() {
        if ch == '\t' {
            result.push_str("    ");
        } else if UnicodeWidthChar::width(ch).is_none() {
            result.push(' ');
        } else {
            result.push(ch);
        }
    }
    Some(result)
}

/// Find the byte offset where accumulated display width reaches `max_width`.
/// Returns `s.len()` if the entire string fits within `max_width`.
fn display_width_split(s: &str, max_width: usize) -> usize {
    let mut width = 0;
    for (i, ch) in s.char_indices() {
        let cw = unicode_width::UnicodeWidthChar::width(ch).unwrap_or(0);
        if width + cw > max_width {
            return i;
        }
        width += cw;
    }
    s.len()
}

/// Wrap content spans into multiple lines if needed, returning Lines and ScreenRowInfo entries
pub fn wrap_content(
    content_spans: Vec<Span<'static>>,
    content: &str,
    prefix_str: String,
    prefix_char: String,
    style: Style,
    content_width: usize,
    prefix_width: usize,
) -> (Vec<Line<'static>>, Vec<ScreenRowInfo>) {
    // Sanitize control characters that cause terminal rendering artifacts:
    // tabs expand to 4 spaces, other control chars become spaces.
    let content_spans: Vec<Span<'static>> = content_spans
        .into_iter()
        .map(|span| match sanitize_for_display(&span.content) {
            Some(sanitized) => Span::styled(sanitized, span.style),
            None => span,
        })
        .collect();
    let content = sanitize_for_display(content)
        .unwrap_or_else(|| content.to_string());

    let content_display_width: usize = content_spans.iter().map(|s| s.content.width()).sum();

    // If content fits, no wrapping needed
    if content_display_width <= content_width {
        let mut spans = Vec::new();
        spans.push(Span::styled(prefix_str, Style::default().fg(Color::DarkGray)));
        spans.push(Span::styled(prefix_char, style));
        spans.extend(content_spans);

        let row_info = ScreenRowInfo {
            content,
            is_file_header: false,
            file_path: None,
            is_continuation: false,
        };

        return (vec![Line::from(spans)], vec![row_info]);
    }

    // Need to wrap - split content into chunks
    let mut result_lines = Vec::new();
    let mut row_infos = Vec::new();
    let mut current_line_spans: Vec<Span<'static>> = Vec::new();
    let mut current_width = 0;
    let mut is_first_line = true;
    let mut current_content = String::new();

    // Continuation line indent (same width as "123 + ")
    let continuation_indent = " ".repeat(prefix_width);

    // Helper closure to emit the current line and reset state
    let emit_line = |current_line_spans: &mut Vec<Span<'static>>,
                         current_content: &mut String,
                         is_first_line: &mut bool,
                         row_infos: &mut Vec<ScreenRowInfo>,
                         result_lines: &mut Vec<Line<'static>>| {
        let mut line_spans = Vec::new();
        if *is_first_line {
            line_spans.push(Span::styled(prefix_str.clone(), Style::default().fg(Color::DarkGray)));
            line_spans.push(Span::styled(prefix_char.clone(), style));
            *is_first_line = false;
        } else {
            line_spans.push(Span::styled(continuation_indent.clone(), Style::default().fg(Color::DarkGray)));
        }
        line_spans.append(current_line_spans);
        result_lines.push(Line::from(line_spans));

        row_infos.push(ScreenRowInfo {
            content: std::mem::take(current_content),
            is_file_header: false,
            file_path: None,
            is_continuation: !row_infos.is_empty(),
        });
    };

    for span in content_spans {
        let span_text = span.content.to_string();
        let span_style = span.style;
        let mut remaining = span_text.as_str();

        while !remaining.is_empty() {
            let space_available = content_width.saturating_sub(current_width);

            if space_available == 0 {
                emit_line(
                    &mut current_line_spans,
                    &mut current_content,
                    &mut is_first_line,
                    &mut row_infos,
                    &mut result_lines,
                );
                current_width = 0;
                continue;
            }

            let remaining_display_width = remaining.width();

            if remaining_display_width <= space_available {
                // Entire remaining text fits
                current_line_spans.push(Span::styled(remaining.to_string(), span_style));
                current_content.push_str(remaining);
                current_width += remaining_display_width;
                remaining = "";
            } else {
                // Split at display width boundary, taking at least one char
                let split_at = display_width_split(remaining, space_available)
                    .max(remaining.ceil_char_boundary(1));
                let (chunk, rest) = remaining.split_at(split_at);
                current_line_spans.push(Span::styled(chunk.to_string(), span_style));
                current_content.push_str(chunk);
                remaining = rest;

                emit_line(
                    &mut current_line_spans,
                    &mut current_content,
                    &mut is_first_line,
                    &mut row_infos,
                    &mut result_lines,
                );
                current_width = 0;
            }
        }
    }

    // Emit any remaining content
    if !current_line_spans.is_empty() || is_first_line {
        emit_line(
            &mut current_line_spans,
            &mut current_content,
            &mut is_first_line,
            &mut row_infos,
            &mut result_lines,
        );
    }

    (result_lines, row_infos)
}

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

    #[test]
    fn test_wrapped_line_widths_do_not_exceed_available_width() {
        let content_width = 40;
        let prefix_width = 8;

        let content = "a".repeat(content_width + 20);
        let content_spans = vec![Span::styled(content.clone(), Style::default())];

        // '±' prefix: 2 bytes, 1 display column — historically caused overflow
        let prefix_str = "    ".to_string();
        let prefix_char = "± C ".to_string();

        let (lines, _) = wrap_content(
            content_spans,
            &content,
            prefix_str,
            prefix_char,
            Style::default(),
            content_width,
            prefix_width,
        );

        assert!(lines.len() > 1, "Content should wrap");

        for (i, line) in lines.iter().enumerate() {
            let display_width: usize = line
                .spans
                .iter()
                .map(|s| s.content.width())
                .sum();
            assert!(
                display_width <= prefix_width + content_width,
                "Line {} has display width {} but max is {} (prefix {} + content {})",
                i,
                display_width,
                prefix_width + content_width,
                prefix_width,
                content_width
            );
        }
    }

    #[test]
    fn test_wrapped_unicode_content_correct_width() {
        let content_width = 20;
        let prefix_width = 6;

        // Mix of ASCII and multi-byte chars (→ is 1 display col, 3 bytes)
        let content = "→ hello → world → foo → bar → baz";
        let content_spans = vec![Span::styled(content.to_string(), Style::default())];

        let (lines, _) = wrap_content(
            content_spans,
            content,
            "  ".to_string(),
            "+ C ".to_string(),
            Style::default(),
            content_width,
            prefix_width,
        );

        for (i, line) in lines.iter().enumerate() {
            let display_width: usize = line
                .spans
                .iter()
                .map(|s| s.content.width())
                .sum();
            assert!(
                display_width <= prefix_width + content_width,
                "Line {} display width {} exceeds max {}",
                i,
                display_width,
                prefix_width + content_width
            );
        }
    }

    #[test]
    fn test_no_wrap_when_content_fits() {
        let content_width = 40;
        let prefix_width = 6;

        let content = "short line";
        let content_spans = vec![Span::styled(content.to_string(), Style::default())];

        let (lines, row_infos) = wrap_content(
            content_spans,
            content,
            "  ".to_string(),
            "+ C ".to_string(),
            Style::default(),
            content_width,
            prefix_width,
        );

        assert_eq!(lines.len(), 1);
        assert_eq!(row_infos.len(), 1);
        assert!(!row_infos[0].is_continuation);
    }

    #[test]
    fn test_continuation_lines_marked_correctly() {
        let content_width = 10;
        let prefix_width = 6;

        let content = "a".repeat(25);
        let content_spans = vec![Span::styled(content.clone(), Style::default())];

        let (lines, row_infos) = wrap_content(
            content_spans,
            &content,
            "  ".to_string(),
            "+ C ".to_string(),
            Style::default(),
            content_width,
            prefix_width,
        );

        assert!(lines.len() >= 3, "Should produce 3+ wrapped lines");
        assert!(!row_infos[0].is_continuation);
        for info in &row_infos[1..] {
            assert!(info.is_continuation);
        }
    }

    #[test]
    fn test_wide_cjk_characters_wrap_correctly() {
        let content_width = 10;
        let prefix_width = 6;

        // Each CJK char is 2 display columns, so 5 chars = 10 columns = fills exactly
        // 6 chars = 12 columns = should wrap
        let content = "你好世界你好";
        let content_spans = vec![Span::styled(content.to_string(), Style::default())];

        let (lines, _) = wrap_content(
            content_spans,
            content,
            "  ".to_string(),
            "+ C ".to_string(),
            Style::default(),
            content_width,
            prefix_width,
        );

        for (i, line) in lines.iter().enumerate() {
            let display_width: usize = line
                .spans
                .iter()
                .map(|s| s.content.width())
                .sum();
            assert!(
                display_width <= prefix_width + content_width,
                "Line {} display width {} exceeds max {}",
                i,
                display_width,
                prefix_width + content_width
            );
        }
    }

    #[test]
    fn test_tab_characters_expanded_before_wrapping() {
        let content_width = 30;
        let prefix_width = 6;

        // Two tabs + text: without expansion, unicode-width sees tabs as 0-width
        // and thinks the line fits. With expansion (4 spaces each), width is correct.
        let content = "\t\tsome_long_identifier = value;";
        let content_spans = vec![Span::styled(content.to_string(), Style::default())];

        let (lines, row_infos) = wrap_content(
            content_spans,
            content,
            "  ".to_string(),
            "+ C ".to_string(),
            Style::default(),
            content_width,
            prefix_width,
        );

        // Content after tab expansion: "        some_long_identifier = value;" = 38 chars
        // This exceeds content_width (30), so wrapping should occur
        assert!(lines.len() > 1, "Tab-containing content should wrap when expanded width exceeds limit");

        // Verify no tabs remain in rendered spans
        for line in &lines {
            for span in &line.spans {
                assert!(
                    !span.content.contains('\t'),
                    "Rendered spans should not contain tab characters"
                );
            }
        }

        // Verify no tabs in ScreenRowInfo content
        for info in &row_infos {
            assert!(
                !info.content.contains('\t'),
                "ScreenRowInfo content should not contain tabs"
            );
        }

        // Verify display widths don't exceed limits
        for (i, line) in lines.iter().enumerate() {
            let display_width: usize = line
                .spans
                .iter()
                .map(|s| s.content.width())
                .sum();
            assert!(
                display_width <= prefix_width + content_width,
                "Line {} display width {} exceeds max {}",
                i,
                display_width,
                prefix_width + content_width
            );
        }
    }

    #[test]
    fn test_tabs_without_wrapping_still_expanded() {
        let content_width = 40;
        let prefix_width = 6;

        // Short content with a tab - fits even after expansion
        let content = "\thi";
        let content_spans = vec![Span::styled(content.to_string(), Style::default())];

        let (lines, row_infos) = wrap_content(
            content_spans,
            content,
            "  ".to_string(),
            "+ C ".to_string(),
            Style::default(),
            content_width,
            prefix_width,
        );

        assert_eq!(lines.len(), 1);

        // Verify tab is expanded in spans
        for span in &lines[0].spans {
            assert!(!span.content.contains('\t'), "Tab should be expanded to spaces");
        }

        // Verify tab is expanded in row info content
        assert_eq!(row_infos[0].content, "    hi");
    }

    #[test]
    fn test_control_characters_sanitized() {
        let content_width = 40;
        let prefix_width = 6;

        // \x01 and \x7f are control chars with unicode-width None
        let content = "hello\x01world\x7f!";
        let content_spans = vec![Span::styled(content.to_string(), Style::default())];

        let (lines, row_infos) = wrap_content(
            content_spans,
            content,
            "  ".to_string(),
            "+ C ".to_string(),
            Style::default(),
            content_width,
            prefix_width,
        );

        assert_eq!(lines.len(), 1);

        // Control chars replaced with spaces in rendered spans
        let rendered: String = lines[0].spans.iter().map(|s| s.content.to_string()).collect();
        assert!(rendered.contains("hello world !"), "Control chars should become spaces, got: {:?}", rendered);

        // Same in row info
        assert_eq!(row_infos[0].content, "hello world !");
    }

    #[test]
    fn test_content_display_width_matches_sanitized_width() {
        let test_cases = [
            "hello world",
            "\t\tindented",
            "has\x01control\x7fchars",
            "mixed\t\x01both",
            "你好世界",
        ];

        for input in &test_cases {
            let width = content_display_width(input);
            let sanitized = sanitize_for_display(input)
                .unwrap_or_else(|| input.to_string());
            assert_eq!(
                width,
                sanitized.width(),
                "content_display_width and sanitized width disagree for {:?}",
                input
            );
        }
    }

    // Tests for wrapped_line_height function

    #[test]
    fn test_wrapped_line_height_short_line() {
        use crate::diff::LineSource;

        let line = DiffLine::new(LineSource::Base, "short line".to_string(), ' ', None);
        let height = wrapped_line_height(&line, 80, None, 100, (8, 16));
        assert_eq!(height, 1);
    }

    #[test]
    fn test_wrapped_line_height_long_line() {
        use crate::diff::LineSource;

        let line = DiffLine::new(LineSource::Base, "x".repeat(150), ' ', None);
        let height = wrapped_line_height(&line, 80, None, 100, (8, 16));
        assert_eq!(height, 2); // 150 chars / 80 = 2 rows (rounded up)
    }

    #[test]
    fn test_wrapped_line_height_image_with_dimensions() {
        let line = DiffLine::image_marker("test.png");

        // 192x192 image with 8x16 font, panel_width=100
        let dims: ImageDimensions = (Some((192, 192)), None);
        let height = wrapped_line_height(&line, 80, Some(dims), 100, (8, 16));

        // Should use calculate_image_height_for_images, not fallback
        let expected = crate::ui::image_view::calculate_image_height_for_images(
            Some((192, 192)),
            None,
            100,
            (8, 16),
        ) as usize;
        assert_eq!(height, expected);
        assert!(height > 1, "Image with dimensions should be taller than 1 row");
    }

    #[test]
    fn test_wrapped_line_height_image_without_dimensions() {
        let line = DiffLine::image_marker("test.png");

        // No dimensions - should use fallback height of 1
        let height = wrapped_line_height(&line, 80, None, 100, (8, 16));
        assert_eq!(height, 1);
    }

    #[test]
    fn test_wrapped_line_height_image_with_empty_dimensions_tuple() {
        let line = DiffLine::image_marker("test.png");

        // Cache exists but has no actual dimensions (both None)
        let dims: ImageDimensions = (None, None);
        let height = wrapped_line_height(&line, 80, Some(dims), 100, (8, 16));

        assert_eq!(height, 1, "Image with no actual dimensions should fallback to 1");
    }

    #[test]
    fn test_wrapped_line_height_image_with_both_dimensions() {
        let line = DiffLine::image_marker("test.png");

        // Both before and after images
        let dims: ImageDimensions = (Some((100, 100)), Some((200, 200)));
        let height = wrapped_line_height(&line, 80, Some(dims), 100, (8, 16));

        let expected = crate::ui::image_view::calculate_image_height_for_images(
            Some((100, 100)),
            Some((200, 200)),
            100,
            (8, 16),
        ) as usize;
        assert_eq!(height, expected);
    }

    #[test]
    fn test_wrapped_line_height_zero_content_width() {
        use crate::diff::LineSource;

        let line = DiffLine::new(LineSource::Base, "x".repeat(100), ' ', None);
        let height = wrapped_line_height(&line, 0, None, 100, (8, 16));
        assert_eq!(height, 1, "Zero content width should return 1");
    }

    #[test]
    fn test_wrapped_line_height_mixed_inline_changes() {
        use crate::diff::{InlineSpan, LineSource};

        let mut line = DiffLine::new(LineSource::Unstaged, "x".repeat(100), '+', None);
        line.old_content = Some("y".repeat(80));
        line.inline_spans = vec![
            InlineSpan {
                text: "deleted".to_string(),
                source: Some(LineSource::Unstaged),
                is_deletion: true,
            },
            InlineSpan {
                text: "inserted".to_string(),
                source: Some(LineSource::Unstaged),
                is_deletion: false,
            },
        ];

        let height = wrapped_line_height(&line, 50, None, 100, (8, 16));
        // del_height = 80/50 = 2, ins_height = 100/50 = 2, total = 4
        assert_eq!(height, 4);
    }

    #[test]
    fn test_wrapped_line_height_pure_deletion_splits_into_two_lines() {
        use crate::diff::{InlineSpan, LineSource};

        let mut line = DiffLine::new(LineSource::Base, "hello wrld".to_string(), ' ', None);
        line.old_content = Some("hello world".to_string());
        line.inline_spans = vec![
            InlineSpan {
                text: "hello w".to_string(),
                source: None,
                is_deletion: false,
            },
            InlineSpan {
                text: "o".to_string(),
                source: Some(LineSource::DeletedBase),
                is_deletion: true,
            },
            InlineSpan {
                text: "rld".to_string(),
                source: None,
                is_deletion: false,
            },
        ];

        let height = wrapped_line_height(&line, 80, None, 100, (8, 16));
        assert_eq!(height, 2, "pure deletion should render as two lines (del + ins)");
    }

    #[test]
    fn test_wrapped_line_height_pure_deletion_long_lines() {
        use crate::diff::{InlineSpan, LineSource};

        let prefix = "a".repeat(90);
        let old = format!("{}Xend", prefix);
        let new = format!("{}end", prefix);

        let mut line = DiffLine::new(LineSource::Base, new, ' ', None);
        line.old_content = Some(old);
        line.inline_spans = vec![
            InlineSpan {
                text: prefix,
                source: None,
                is_deletion: false,
            },
            InlineSpan {
                text: "X".to_string(),
                source: Some(LineSource::DeletedBase),
                is_deletion: true,
            },
            InlineSpan {
                text: "end".to_string(),
                source: None,
                is_deletion: false,
            },
        ];

        let height = wrapped_line_height(&line, 50, None, 100, (8, 16));
        // del: 94 chars / 50 = 2 rows, ins: 93 chars / 50 = 2 rows, total = 4
        assert_eq!(height, 4);
    }

    #[test]
    fn test_has_pure_deletion_inline_true() {
        use crate::diff::{InlineSpan, LineSource};

        let mut line = DiffLine::new(LineSource::Base, "test".to_string(), ' ', None);
        line.inline_spans = vec![
            InlineSpan {
                text: "tes".to_string(),
                source: None,
                is_deletion: false,
            },
            InlineSpan {
                text: "x".to_string(),
                source: Some(LineSource::DeletedBase),
                is_deletion: true,
            },
            InlineSpan {
                text: "t".to_string(),
                source: None,
                is_deletion: false,
            },
        ];
        assert!(has_pure_deletion_inline(&line));
    }

    #[test]
    fn test_has_pure_deletion_inline_false_for_mixed() {
        use crate::diff::{InlineSpan, LineSource};

        let mut line = DiffLine::new(LineSource::Base, "test".to_string(), ' ', None);
        line.inline_spans = vec![
            InlineSpan {
                text: "old".to_string(),
                source: Some(LineSource::DeletedBase),
                is_deletion: true,
            },
            InlineSpan {
                text: "new".to_string(),
                source: Some(LineSource::Committed),
                is_deletion: false,
            },
        ];
        assert!(!has_pure_deletion_inline(&line));
    }

    #[test]
    fn test_has_pure_deletion_inline_false_for_empty() {
        use crate::diff::LineSource;

        let line = DiffLine::new(LineSource::Base, "test".to_string(), ' ', None);
        assert!(!has_pure_deletion_inline(&line));
    }
}