vtcode_tui/core_tui/app/session/

diff_preview.rs

//! Diff preview rendering for file edit approval
//!
//! Renders a syntax-highlighted diff preview with permission controls.

use ratatui::{
    Frame,
    layout::{Constraint, Layout, Rect},
    style::{Color, Modifier, Style},
    text::{Line, Span},
    widgets::{Block, Borders, Paragraph},
};

use super::Session;
use crate::core_tui::app::types::{DiffPreviewMode, DiffPreviewState, TrustMode};
use crate::core_tui::style::{ratatui_color_from_ansi, ratatui_style_from_ansi};
use crate::ui::markdown::highlight_line_for_diff;
use crate::utils::diff::{DiffBundle, DiffLineKind, DiffOptions, compute_diff_with_theme};
use crate::utils::diff_styles::{
    DiffColorPalette, DiffLineType, content_background, current_diff_render_style_context,
    style_gutter, style_line_bg, style_sign,
};

pub fn render_diff_preview(session: &Session, frame: &mut Frame<'_>, area: Rect) {
    let Some(preview) = session.diff_preview_state() else {
        return;
    };

    let palette = DiffColorPalette::default();
    let diff_bundle = compute_diff_with_theme(
        &preview.before,
        &preview.after,
        DiffOptions {
            context_lines: 3,
            old_label: None,
            new_label: None,
            missing_newline_hint: false,
        },
    );
    let (additions, deletions) = count_diff_changes(&diff_bundle);

    let chunks = Layout::vertical([
        Constraint::Length(2),
        Constraint::Min(5),
        Constraint::Length(4),
    ])
    .split(area);

    render_file_header(frame, chunks[0], preview, &palette, additions, deletions);
    render_diff_content(frame, chunks[1], preview, &diff_bundle);
    render_controls(frame, chunks[2], preview);
}

fn render_file_header(
    frame: &mut Frame<'_>,
    area: Rect,
    preview: &DiffPreviewState,
    palette: &DiffColorPalette,
    additions: usize,
    deletions: usize,
) {
    let header_style = Style::default().fg(ratatui_color_from_ansi(palette.header_fg));
    let header = Line::from(vec![
        Span::styled(header_action_label(preview.mode), header_style),
        Span::styled(&preview.file_path, header_style),
        Span::styled(" (", header_style),
        Span::styled(
            format!("+{}", additions),
            Style::default().fg(ratatui_color_from_ansi(palette.added_fg)),
        ),
        Span::styled(" ", header_style),
        Span::styled(
            format!("-{}", deletions),
            Style::default().fg(ratatui_color_from_ansi(palette.removed_fg)),
        ),
        Span::styled(")", header_style),
    ]);
    frame.render_widget(Paragraph::new(header), area);
}

fn detect_language(file_path: &str) -> Option<&'static str> {
    let ext = file_path.rsplit('.').next()?;
    match ext.to_lowercase().as_str() {
        "rs" => Some("rust"),
        "py" => Some("python"),
        "js" => Some("javascript"),
        "ts" | "tsx" => Some("typescript"),
        "go" => Some("go"),
        "java" => Some("java"),
        "sh" | "bash" => Some("bash"),
        "swift" => Some("swift"),
        "c" | "h" => Some("c"),
        "cpp" | "cc" | "cxx" | "hpp" => Some("cpp"),
        "json" => Some("json"),
        "yaml" | "yml" => Some("yaml"),
        "toml" => Some("toml"),
        "md" => Some("markdown"),
        "html" | "htm" => Some("html"),
        "css" | "scss" => Some("css"),
        _ => None,
    }
}

fn highlight_line_with_bg(
    line: &str,
    language: Option<&str>,
    bg: Option<Color>,
) -> Vec<Span<'static>> {
    let text = line.trim_end_matches('\n');
    if let Some(segments) = highlight_line_for_diff(text, language) {
        segments
            .into_iter()
            .map(|(anstyle, t)| {
                let style = apply_diff_bg_if_missing(ratatui_style_from_ansi(anstyle), bg);
                Span::styled(t, style)
            })
            .collect()
    } else {
        let style = apply_diff_bg_if_missing(Style::default(), bg);
        vec![Span::styled(text.to_string(), style)]
    }
}

fn apply_diff_bg_if_missing(mut style: Style, bg: Option<Color>) -> Style {
    if style.bg.is_none()
        && let Some(bg_color) = bg
    {
        style = style.bg(bg_color);
    }
    style
}

fn count_diff_changes(diff_bundle: &DiffBundle) -> (usize, usize) {
    let mut additions = 0usize;
    let mut deletions = 0usize;

    for hunk in &diff_bundle.hunks {
        for line in &hunk.lines {
            match line.kind {
                DiffLineKind::Addition => additions += 1,
                DiffLineKind::Deletion => deletions += 1,
                DiffLineKind::Context => {}
            }
        }
    }

    (additions, deletions)
}

fn render_diff_content(
    frame: &mut Frame<'_>,
    area: Rect,
    preview: &DiffPreviewState,
    diff_bundle: &DiffBundle,
) {
    let language = detect_language(&preview.file_path);
    let style_context = current_diff_render_style_context();

    let mut lines: Vec<Line> = Vec::new();
    let max_display = area.height.saturating_sub(1) as usize;

    for hunk in &diff_bundle.hunks {
        if lines.len() >= max_display {
            break;
        }

        lines.push(Line::from(Span::styled(
            format!("@@ -{} +{} @@", hunk.old_start, hunk.new_start),
            Style::default().fg(Color::Cyan),
        )));

        for diff_line in &hunk.lines {
            if lines.len() >= max_display {
                break;
            }

            let line_num = match diff_line.kind {
                DiffLineKind::Context => diff_line.new_line.unwrap_or(0),
                DiffLineKind::Addition => diff_line.new_line.unwrap_or(0),
                DiffLineKind::Deletion => diff_line.old_line.unwrap_or(0),
            };
            let line_num_str = format!("{:>4} ", line_num);
            let text = diff_line.text.trim_end_matches('\n');

            let line_type = match diff_line.kind {
                DiffLineKind::Context => DiffLineType::Context,
                DiffLineKind::Addition => DiffLineType::Insert,
                DiffLineKind::Deletion => DiffLineType::Delete,
            };

            let gutter_style = style_gutter(line_type);
            let sign_style = style_sign(line_type);
            let line_bg = style_line_bg(line_type, style_context);
            let content_bg = content_background(line_type, style_context);

            let prefix = match line_type {
                DiffLineType::Insert => "+",
                DiffLineType::Delete => "-",
                DiffLineType::Context => " ",
            };

            let mut spans = vec![
                Span::styled(prefix.to_string(), sign_style),
                Span::styled(line_num_str, gutter_style),
            ];

            // For changed lines with syntax highlighting, apply bg tint
            let highlighted = highlight_line_with_bg(text, language, content_bg);
            spans.extend(highlighted);

            lines.push(Line::from(spans).style(line_bg));
        }
    }

    if lines.is_empty() {
        lines.push(Line::from(Span::styled(
            "(no changes)",
            Style::default().fg(Color::DarkGray),
        )));
    }

    frame.render_widget(
        Paragraph::new(lines).block(Block::default().borders(Borders::NONE)),
        area,
    );
}

fn header_action_label(mode: DiffPreviewMode) -> &'static str {
    match mode {
        DiffPreviewMode::EditApproval => "← Edit ",
        DiffPreviewMode::FileConflict => "← Conflict ",
    }
}

fn render_controls(frame: &mut Frame<'_>, area: Rect, preview: &DiffPreviewState) {
    let lines = control_lines(preview);

    frame.render_widget(
        Paragraph::new(lines).block(
            Block::default()
                .borders(Borders::TOP)
                .border_style(Style::default().fg(Color::DarkGray)),
        ),
        area,
    );
}

fn control_lines(preview: &DiffPreviewState) -> Vec<Line<'static>> {
    match preview.mode {
        DiffPreviewMode::EditApproval => {
            let trust = match preview.trust_mode {
                TrustMode::Once => "Once",
                TrustMode::Session => "Session",
                TrustMode::Always => "Always",
                TrustMode::AutoTrust => "Auto",
            };

            vec![
                Line::from(vec![
                    Span::styled(
                        "Enter",
                        Style::default()
                            .fg(Color::Green)
                            .add_modifier(Modifier::BOLD),
                    ),
                    Span::raw(" Apply  "),
                    Span::styled(
                        "Esc",
                        Style::default().fg(Color::Red).add_modifier(Modifier::BOLD),
                    ),
                    Span::raw(" Reject  "),
                    Span::styled("Tab", Style::default().fg(Color::Yellow)),
                    Span::raw("/"),
                    Span::styled("S-Tab", Style::default().fg(Color::Yellow)),
                    Span::raw(" Nav"),
                ]),
                Line::from(vec![
                    Span::styled("1", Style::default().fg(Color::Cyan)),
                    Span::raw("-Once "),
                    Span::styled("2", Style::default().fg(Color::Cyan)),
                    Span::raw("-Sess "),
                    Span::styled("3", Style::default().fg(Color::Cyan)),
                    Span::raw("-Always "),
                    Span::styled("4", Style::default().fg(Color::Cyan)),
                    Span::raw("-Auto "),
                    Span::styled(
                        format!("[{}]", trust),
                        Style::default()
                            .fg(Color::DarkGray)
                            .add_modifier(Modifier::BOLD),
                    ),
                ]),
            ]
        }
        DiffPreviewMode::FileConflict => vec![
            Line::from(vec![
                Span::styled(
                    "Enter",
                    Style::default()
                        .fg(Color::Green)
                        .add_modifier(Modifier::BOLD),
                ),
                Span::raw(" Proceed  "),
                Span::styled("r", Style::default().fg(Color::Cyan)),
                Span::raw(" Reload  "),
                Span::styled(
                    "Esc",
                    Style::default().fg(Color::Red).add_modifier(Modifier::BOLD),
                ),
                Span::raw(" Abort"),
            ]),
            Line::from(vec![
                Span::styled("Tab", Style::default().fg(Color::Yellow)),
                Span::raw("/"),
                Span::styled("S-Tab", Style::default().fg(Color::Yellow)),
                Span::raw(" Nav"),
            ]),
        ],
    }
}

#[cfg(test)]
mod tests {
    use super::{apply_diff_bg_if_missing, control_lines, header_action_label};
    use crate::core_tui::app::types::{DiffPreviewMode, DiffPreviewState};
    use ratatui::style::{Color, Style};

    #[test]
    fn apply_diff_bg_if_missing_adds_bg_when_none() {
        let styled = apply_diff_bg_if_missing(Style::default(), Some(Color::Green));
        assert_eq!(styled.bg, Some(Color::Green));
    }

    #[test]
    fn apply_diff_bg_if_missing_preserves_existing_bg() {
        let base = Style::default().bg(Color::Blue);
        let styled = apply_diff_bg_if_missing(base, Some(Color::Green));
        assert_eq!(styled.bg, Some(Color::Blue));
    }

    #[test]
    fn conflict_controls_show_proceed_reload_abort_copy() {
        let preview = DiffPreviewState::new_with_mode(
            "src/main.rs".to_string(),
            "before".to_string(),
            "after".to_string(),
            Vec::new(),
            DiffPreviewMode::FileConflict,
        );

        let lines = control_lines(&preview);
        let first_line: String = lines[0]
            .spans
            .iter()
            .map(|span| span.content.clone().into_owned())
            .collect();

        assert!(first_line.contains("Proceed"));
        assert!(first_line.contains("Reload"));
        assert!(first_line.contains("Abort"));
    }

    #[test]
    fn conflict_header_uses_conflict_label() {
        assert_eq!(
            header_action_label(DiffPreviewMode::FileConflict),
            "← Conflict "
        );
        assert_eq!(
            header_action_label(DiffPreviewMode::EditApproval),
            "← Edit "
        );
    }
}