clankerdiff_ratatui/markdown_review/

render.rs

use super::{
    layout::{MarkdownVisualLayout, MarkdownVisualRow},
    state::{MarkdownFocusPane, MarkdownHitRegion, MarkdownReviewState},
};
use crate::{
    KeyBinding, RatatuiTheme,
    annotation::render_annotation_line,
    theme_picker::render_theme_picker,
    ui::{
        ActionBar, ActionLabel, AppFrame, ButtonVariant, EmptyState, Modal, ModalSize, NoticeTone,
        SelectionState, render_modal_text,
    },
    widgets::{render_vertical_scrollbar, rows_and_track},
};
use ratatui::{
    buffer::Buffer,
    layout::{Constraint, Layout, Position, Rect},
    style::Style,
    text::{Line, Span},
    widgets::{Paragraph, StatefulWidget, Widget},
};

const GUTTER_SEPARATOR_WIDTH: u16 = 3;
const OUTLINE_WIDTH: u16 = 28;
const OUTLINE_BREAKPOINT: u16 = 90;

/// Stateful Ratatui Markdown review widget.
#[derive(Debug, Clone)]
pub struct MarkdownReviewWidget {
    title: String,
    borders: bool,
}

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

impl MarkdownReviewWidget {
    /// Creates a bordered widget titled “Markdown Review”.
    #[must_use]
    pub fn new() -> Self {
        Self {
            title: "Markdown Review".to_owned(),
            borders: true,
        }
    }

    /// Sets the outer title.
    #[must_use]
    pub fn title(mut self, title: impl Into<String>) -> Self {
        self.title = title.into();
        self
    }

    /// Enables or disables the outer border.
    #[must_use]
    pub const fn borders(mut self, borders: bool) -> Self {
        self.borders = borders;
        self
    }
}

impl StatefulWidget for MarkdownReviewWidget {
    type State = MarkdownReviewState;

    fn render(self, area: Rect, buffer: &mut Buffer, state: &mut Self::State) {
        state.set_cursor(None);
        state.clear_hit_regions();
        let theme = RatatuiTheme::from(&state.theme);
        let regions = AppFrame::new(&self.title, self.borders, &theme)
            .footer(state.options.footer)
            .render(area, buffer);
        if regions.body.is_empty() {
            state.dirty = false;
            return;
        }
        let body = regions.body;
        let footer = regions.footer;
        render_body(body, buffer, state, &theme);
        render_footer(footer, buffer, state, &theme);
        if state.help {
            render_help(area, buffer, state, &theme);
        }
        if let Some(picker) = &state.theme_picker {
            render_theme_picker(area, buffer, picker, &theme);
        }
        state.dirty = false;
    }
}

fn render_body(
    area: Rect,
    buffer: &mut Buffer,
    state: &mut MarkdownReviewState,
    theme: &RatatuiTheme,
) {
    state.clear_hit_regions();
    if area.is_empty() {
        return;
    }
    let outline_width = state.options.navigation.width(
        area.width,
        OUTLINE_BREAKPOINT,
        OUTLINE_WIDTH.min(area.width / 3),
    );
    let wide = outline_width > 0 && !state.document().outline().is_empty();
    let (outline, separator, document) = if wide {
        let [outline, separator, document] = Layout::horizontal([
            Constraint::Length(outline_width),
            Constraint::Length(1),
            Constraint::Min(1),
        ])
        .areas(area);
        (outline, separator, document)
    } else {
        (Rect::default(), Rect::default(), area)
    };
    if wide {
        buffer.set_style(
            separator,
            Style::new().fg(theme.ui.border).bg(theme.ui.canvas),
        );
        render_outline(outline, buffer, state, theme);
    }
    let (rows, track) = rows_and_track(document, true);
    state.last_height = usize::from(rows.height).max(1);
    let gutter_width = source_gutter_width(state);
    let content_width = rows.width.saturating_sub(gutter_width).max(1);
    let layout = state.ensure_layout(content_width);
    state.follow_selection(&layout);
    let last = layout.len().saturating_sub(state.last_height);
    state.scroll = state.scroll.min(last);
    if layout.is_empty() {
        EmptyState::new("No Markdown content to review", NoticeTone::Neutral, theme)
            .render(rows, buffer);
    } else {
        render_rows(rows, track, buffer, state, theme, &layout, gutter_width);
    }
}

fn source_gutter_width(state: &MarkdownReviewState) -> u16 {
    let line_count = state.document().source().split('\n').count().max(1);
    let digits = line_count.checked_ilog10().unwrap_or(0).saturating_add(1);
    u16::try_from(digits)
        .unwrap_or(u16::MAX)
        .saturating_add(GUTTER_SEPARATOR_WIDTH)
}

fn render_outline(
    area: Rect,
    buffer: &mut Buffer,
    state: &mut MarkdownReviewState,
    theme: &RatatuiTheme,
) {
    if area.is_empty() {
        return;
    }
    let heading_count = state.document().outline().len();
    state.outline_selected = state.outline_selected.min(heading_count.saturating_sub(1));
    let height = usize::from(area.height);
    if state.follow_pending {
        if state.outline_selected < state.outline_scroll {
            state.outline_scroll = state.outline_selected;
        } else if state.outline_selected >= state.outline_scroll.saturating_add(height) {
            state.outline_scroll = state
                .outline_selected
                .saturating_sub(height.saturating_sub(1));
        }
    }
    let max_scroll = heading_count.saturating_sub(height.max(1));
    state.outline_scroll = state.outline_scroll.min(max_scroll);
    let headings = state.document().outline().to_vec();
    for (offset, heading) in headings
        .iter()
        .skip(state.outline_scroll)
        .take(height)
        .enumerate()
    {
        let row = Rect::new(
            area.x,
            area.y + u16::try_from(offset).unwrap_or(u16::MAX),
            area.width,
            1,
        );
        let index = state.outline_scroll + offset;
        let selected = state.focus == MarkdownFocusPane::Outline && index == state.outline_selected;
        let indent = "  ".repeat(usize::from(heading.level.saturating_sub(1)));
        let selection = if selected {
            SelectionState::Focused
        } else {
            SelectionState::None
        };
        Paragraph::new(Line::from(format!("{indent}{}", heading.title)))
            .style(theme.ui.selection_style(selection))
            .render(row, buffer);
        state.hit_regions.push(MarkdownHitRegion {
            area: row,
            target: Some(heading.target_id),
            outline: true,
        });
    }
}

fn render_rows(
    area: Rect,
    track: Rect,
    buffer: &mut Buffer,
    state: &mut MarkdownReviewState,
    theme: &RatatuiTheme,
    layout: &MarkdownVisualLayout,
    gutter_width: u16,
) {
    let selected = state.selected_target();
    let focused = state.focus == MarkdownFocusPane::Document;
    for (drawn, index) in (state.scroll..layout.len()).enumerate() {
        let y = area
            .y
            .saturating_add(u16::try_from(drawn).unwrap_or(u16::MAX));
        if y >= area.bottom() {
            break;
        }
        let row_area = Rect::new(area.x, y, area.width, 1);
        let row_gutter_width = gutter_width.min(row_area.width.saturating_sub(1));
        let [gutter_area, content_area] =
            Layout::horizontal([Constraint::Length(row_gutter_width), Constraint::Min(1)])
                .areas(row_area);
        let Some(row) = layout.row(index) else {
            break;
        };
        if let MarkdownVisualRow::Annotation {
            annotation,
            line,
            target,
        } = row
        {
            render_annotation_line(content_area, buffer, theme, annotation, line);
            if let Some(column) = annotation.cursor_column(line) {
                state.set_cursor(Some(Position::new(
                    content_area
                        .x
                        .saturating_add(column)
                        .min(content_area.right().saturating_sub(1)),
                    y,
                )));
            }
            if target.is_some() {
                state.hit_regions.push(MarkdownHitRegion {
                    area: row_area,
                    target,
                    outline: false,
                });
            }
            continue;
        }
        let is_selected = focused && row.target().is_some_and(|target| Some(target) == selected);
        let background = if is_selected {
            theme.ui.surface_selected
        } else {
            theme.ui.canvas
        };
        buffer.set_style(row_area, Style::new().fg(theme.ui.text).bg(background));
        let gutter = row.source_line().map_or_else(
            || " ".repeat(usize::from(row_gutter_width)),
            |line| {
                let number_width =
                    usize::from(row_gutter_width.saturating_sub(GUTTER_SEPARATOR_WIDTH));
                format!("{line:>number_width$} │ ")
            },
        );
        Paragraph::new(gutter)
            .style(Style::new().fg(theme.gutter).bg(background))
            .render(gutter_area, buffer);
        if let MarkdownVisualRow::Content(content) = row {
            let mut line = content.line.clone();
            if is_selected {
                line.style = line.style.bg(background);
                for span in &mut line.spans {
                    span.style = span.style.bg(background);
                }
            }
            Paragraph::new(line).render(content_area, buffer);
        }
        if let Some(target) = row.target() {
            state.hit_regions.push(MarkdownHitRegion {
                area: row_area,
                target: Some(target),
                outline: false,
            });
        }
    }
    render_vertical_scrollbar(
        track,
        buffer,
        layout.len(),
        usize::from(area.height),
        state.scroll,
    );
}

fn render_footer(
    area: Rect,
    buffer: &mut Buffer,
    state: &MarkdownReviewState,
    theme: &RatatuiTheme,
) {
    if area.is_empty() {
        return;
    }
    let count = state.review().len();
    let status = format!("  {count} comment{}", if count == 1 { "" } else { "s" });
    let mut actions = if state.session.draft().is_some() {
        vec![
            ActionLabel::new("Enter", "save", theme)
                .variant(ButtonVariant::Primary)
                .into_span(),
            ActionLabel::new("Shift-Enter", "newline", theme).into_span(),
            ActionLabel::new("Esc", "cancel", theme).into_span(),
        ]
    } else {
        vec![Span::styled(
            state.footer_hint(usize::from(area.width).saturating_sub(status.len())),
            Style::new().fg(theme.ui.text_muted),
        )]
    };
    actions.push(Span::styled(status, Style::new().fg(theme.ui.accent)));
    ActionBar::new(Line::from(actions), theme).render(area, buffer);
}

fn render_help(area: Rect, buffer: &mut Buffer, state: &MarkdownReviewState, theme: &RatatuiTheme) {
    let content = Modal::new("Markdown shortcuts", ModalSize::Wide, theme)
        .hint("j/k scroll · ? / Esc close")
        .render(area, buffer);
    render_modal_text(
        content,
        buffer,
        state
            .help_bindings()
            .skip(state.help_scroll)
            .map(KeyBinding::hint)
            .collect::<Vec<_>>()
            .join("\n"),
        theme,
    );
}