roam-sdk 0.3.0

pub mod header;
pub mod main_area;
pub mod status_bar;

use ratatui::layout::{Constraint, Layout, Rect};
use ratatui::style::{Color, Style};
use ratatui::text::{Line, Span};
use ratatui::widgets::{Block as WidgetBlock, BorderType, Borders, Clear};
use ratatui::Frame;

use crate::app::slash::SlashMenuState;
use crate::app::{
    AppState, AutocompleteState, InputMode, LinkPickerState, QuickSwitcherState, SearchState,
    ViewMode,
};
use crate::error::ErrorPopup;

use header::Header;
use main_area::{EditInfo, MainArea};
use status_bar::StatusBar;

pub fn render(frame: &mut Frame, state: &AppState) {
    let chunks = Layout::vertical([
        Constraint::Length(1),
        Constraint::Min(1),
        Constraint::Length(1),
    ])
    .split(frame.area());

    let view_label = match &state.view_mode {
        ViewMode::DailyNotes => state.date_display.clone(),
        ViewMode::Page { title } => title.clone(),
    };
    let header = Header {
        graph_name: &state.graph_name,
        date: &view_label,
    };
    frame.render_widget(header, chunks[0]);

    let edit_info = match &state.input_mode {
        InputMode::Insert { buffer, .. } => Some(EditInfo {
            buffer,
            block_index: state.selected_block,
        }),
        InputMode::Normal => None,
    };

    let main = MainArea {
        days: &state.days,
        selected_block: state.selected_block,
        cursor_col: state.cursor_col,
        loading: state.loading,
        loading_more: state.loading_more,
        edit_info,
        block_ref_cache: &state.block_ref_cache,
        linked_refs: &state.linked_refs,
    };
    frame.render_widget(main, chunks[1]);

    if let Some(ac) = &state.autocomplete {
        render_autocomplete_popup(frame, ac, chunks[1]);
    }

    if let Some(sm) = &state.slash_menu {
        render_slash_menu_popup(frame, sm, chunks[1]);
    }

    if let Some(lp) = &state.link_picker {
        render_link_picker_popup(frame, lp, chunks[1]);
    }

    if let Some(qs) = &state.quick_switcher {
        render_quick_switcher_popup(frame, qs, chunks[1]);
    }

    if let Some(search) = &state.search {
        render_search_popup(frame, search, chunks[1]);
    }

    if state.show_help {
        render_help_popup(frame, &state.help_hints, chunks[1]);
    }

    if let Some(err) = &state.error_popup {
        render_error_popup(frame, err, chunks[1]);
    }

    let insert_mode = !matches!(state.input_mode, InputMode::Normal);
    let nav_back_hint = if state.can_nav_back() {
        Some("Shift+\u{2190}")
    } else {
        None
    };
    let nav_forward_hint = if state.can_nav_forward() {
        Some("Shift+\u{2192}")
    } else {
        None
    };
    let status = StatusBar {
        hints: &state.hints,
        message: state.status_message.as_deref(),
        insert_mode,
        nav_back_hint,
        nav_forward_hint,
    };
    frame.render_widget(status, chunks[2]);
}

fn render_autocomplete_popup(frame: &mut Frame, ac: &AutocompleteState, area: Rect) {
    let max_items = if ac.results.is_empty() {
        1 // room for "No results" message
    } else {
        10.min(ac.results.len())
    };
    let popup_height = (max_items + 2) as u16; // +2 for borders
    let popup_width = (area.width * 60 / 100).max(30).min(area.width);
    let x = area.x + (area.width.saturating_sub(popup_width)) / 2;
    let y = (area.y + area.height / 2).min(area.y + area.height.saturating_sub(popup_height));

    let popup_area = Rect::new(x, y, popup_width, popup_height);

    frame.render_widget(Clear, popup_area);

    let title = if ac.query.is_empty() {
        " Block ref ".to_string()
    } else {
        format!(" Search: {} ", ac.query)
    };

    let block = WidgetBlock::default()
        .borders(Borders::ALL)
        .border_type(BorderType::Rounded)
        .border_style(Style::default().fg(Color::Gray))
        .title(title);

    let inner = block.inner(popup_area);
    frame.render_widget(block, popup_area);

    if ac.results.is_empty() {
        let style = Style::default().fg(Color::DarkGray);
        let line = Line::from(vec![Span::styled("No results", style)]);
        let line_area = Rect::new(inner.x, inner.y, inner.width, 1);
        frame.render_widget(line, line_area);
        return;
    }

    for (i, (_, text)) in ac.results.iter().take(max_items).enumerate() {
        if i as u16 >= inner.height {
            break;
        }
        let is_selected = i == ac.selected;
        let style = if is_selected {
            Style::default().fg(Color::White).bg(Color::DarkGray)
        } else {
            Style::default().fg(Color::Gray)
        };

        let max_text_width = inner.width as usize;
        let display: String = text.chars().take(max_text_width).collect();
        let padding = max_text_width.saturating_sub(display.chars().count());
        let padded = format!("{}{}", display, " ".repeat(padding));

        let line = Line::from(vec![Span::styled(padded, style)]);
        let line_area = Rect::new(inner.x, inner.y + i as u16, inner.width, 1);
        frame.render_widget(line, line_area);
    }
}

fn render_link_picker_popup(frame: &mut Frame, lp: &LinkPickerState, area: Rect) {
    let max_items = 10.min(lp.links.len());
    let popup_height = (max_items + 2) as u16; // +2 for borders
    let popup_width = (area.width * 50 / 100).max(20).min(area.width);
    let x = area.x + (area.width.saturating_sub(popup_width)) / 2;
    let y = (area.y + area.height / 2).min(area.y + area.height.saturating_sub(popup_height));

    let popup_area = Rect::new(x, y, popup_width, popup_height);
    frame.render_widget(Clear, popup_area);

    let block = WidgetBlock::default()
        .borders(Borders::ALL)
        .border_type(BorderType::Rounded)
        .border_style(Style::default().fg(Color::Cyan))
        .title(" Follow link ");

    let inner = block.inner(popup_area);
    frame.render_widget(block, popup_area);

    let scroll_offset = if lp.selected >= max_items {
        lp.selected - max_items + 1
    } else {
        0
    };

    for (i, title) in lp
        .links
        .iter()
        .skip(scroll_offset)
        .take(max_items)
        .enumerate()
    {
        if i as u16 >= inner.height {
            break;
        }
        let is_selected = (i + scroll_offset) == lp.selected;
        let style = if is_selected {
            Style::default().fg(Color::White).bg(Color::DarkGray)
        } else {
            Style::default().fg(Color::Cyan)
        };

        let max_text_width = inner.width as usize;
        let display: String = title.chars().take(max_text_width).collect();
        let padding = max_text_width.saturating_sub(display.chars().count());
        let padded = format!("{}{}", display, " ".repeat(padding));

        let line = Line::from(vec![Span::styled(padded, style)]);
        let line_area = Rect::new(inner.x, inner.y + i as u16, inner.width, 1);
        frame.render_widget(line, line_area);
    }
}

fn render_help_popup(frame: &mut Frame, hints: &[(String, &str)], area: Rect) {
    let line_count = hints.len();
    let popup_height = (line_count + 3).min(area.height as usize) as u16; // +2 borders +1 footer
    let popup_width = (area.width * 60 / 100).max(30).min(area.width);
    let x = area.x + (area.width.saturating_sub(popup_width)) / 2;
    let y = area.y + (area.height.saturating_sub(popup_height)) / 2;

    let popup_area = Rect::new(x, y, popup_width, popup_height);
    frame.render_widget(Clear, popup_area);

    let block = WidgetBlock::default()
        .borders(Borders::ALL)
        .border_type(BorderType::Rounded)
        .border_style(Style::default().fg(Color::Cyan))
        .title(" Help ");

    let inner = block.inner(popup_area);
    frame.render_widget(block, popup_area);

    for (i, (key_str, action_name)) in hints.iter().enumerate() {
        if i as u16 >= inner.height.saturating_sub(1) {
            break;
        }
        let key_span = Span::styled(
            format!("{:>12}", key_str),
            Style::default().fg(Color::Yellow),
        );
        let sep = Span::styled("  ", Style::default());
        let action_span = Span::styled(*action_name, Style::default().fg(Color::White));
        let line = Line::from(vec![key_span, sep, action_span]);
        let line_area = Rect::new(inner.x, inner.y + i as u16, inner.width, 1);
        frame.render_widget(line, line_area);
    }

    // Footer
    if inner.height > 0 {
        let footer_y = inner.y + inner.height - 1;
        let footer = Line::styled(
            "Press any key to close",
            Style::default().fg(Color::DarkGray),
        );
        let footer_area = Rect::new(inner.x, footer_y, inner.width, 1);
        frame.render_widget(footer, footer_area);
    }
}

fn render_error_popup(frame: &mut Frame, popup: &ErrorPopup, area: Rect) {
    let popup_width = (area.width * 50 / 100).max(30).min(area.width);
    let inner_width = popup_width.saturating_sub(2) as usize; // -2 for borders

    // Word-wrap message
    let msg_lines = wrap_text(&popup.message, inner_width);
    // 2 borders + 1 blank top + msg lines + 1 blank + 1 hint + 1 blank + 1 footer
    let content_height = 1 + msg_lines.len() + 1 + 1 + 1 + 1;
    let popup_height = (content_height + 2).min(area.height as usize) as u16; // +2 borders

    let x = area.x + (area.width.saturating_sub(popup_width)) / 2;
    let y = area.y + (area.height.saturating_sub(popup_height)) / 2;

    let popup_area = Rect::new(x, y, popup_width, popup_height);
    frame.render_widget(Clear, popup_area);

    let title = format!(" ! {} ", popup.title);
    let block = WidgetBlock::default()
        .borders(Borders::ALL)
        .border_type(BorderType::Rounded)
        .border_style(Style::default().fg(Color::Red))
        .title(title);

    let inner = block.inner(popup_area);
    frame.render_widget(block, popup_area);

    let mut row: u16 = 0;

    // Blank line
    row += 1;

    // Message lines
    for line_text in &msg_lines {
        if row >= inner.height.saturating_sub(1) {
            break;
        }
        let line = Line::from(Span::styled(
            line_text.clone(),
            Style::default().fg(Color::White),
        ));
        let line_area = Rect::new(inner.x, inner.y + row, inner.width, 1);
        frame.render_widget(line, line_area);
        row += 1;
    }

    // Blank line
    row += 1;

    // Hint
    if row < inner.height.saturating_sub(1) {
        let hint = Line::from(Span::styled(
            popup.hint.clone(),
            Style::default().fg(Color::DarkGray),
        ));
        let hint_area = Rect::new(inner.x, inner.y + row, inner.width, 1);
        frame.render_widget(hint, hint_area);
        row += 1;
    }

    // Blank line
    row += 1;

    // Footer
    if row < inner.height {
        let footer = Line::styled(
            "Press any key to close",
            Style::default().fg(Color::DarkGray),
        );
        let footer_area = Rect::new(inner.x, inner.y + row, inner.width, 1);
        frame.render_widget(footer, footer_area);
    }
}

fn wrap_text(text: &str, max_width: usize) -> Vec<String> {
    if max_width == 0 {
        return vec![text.to_string()];
    }
    let mut lines = Vec::new();
    let mut current = String::new();

    for word in text.split_whitespace() {
        if current.is_empty() {
            current = word.to_string();
        } else if current.len() + 1 + word.len() <= max_width {
            current.push(' ');
            current.push_str(word);
        } else {
            lines.push(current);
            current = word.to_string();
        }
    }
    if !current.is_empty() {
        lines.push(current);
    }
    if lines.is_empty() {
        lines.push(String::new());
    }
    lines
}

fn render_slash_menu_popup(frame: &mut Frame, sm: &SlashMenuState, area: Rect) {
    let max_visible = 10;
    let visible_count = if sm.commands.is_empty() {
        1
    } else {
        max_visible.min(sm.commands.len())
    };
    let popup_height = (visible_count + 2) as u16;
    let popup_width = (area.width * 65 / 100).max(30).min(area.width);
    let x = area.x + (area.width.saturating_sub(popup_width)) / 2;
    let y = (area.y + area.height / 2).min(area.y + area.height.saturating_sub(popup_height));

    let popup_area = Rect::new(x, y, popup_width, popup_height);
    frame.render_widget(Clear, popup_area);

    let title = if sm.query.is_empty() {
        " / Commands ".to_string()
    } else {
        format!(" /{} ", sm.query)
    };

    let block = WidgetBlock::default()
        .borders(Borders::ALL)
        .border_type(BorderType::Rounded)
        .border_style(Style::default().fg(Color::Green))
        .title(title);

    let inner = block.inner(popup_area);
    frame.render_widget(block, popup_area);

    if sm.commands.is_empty() {
        let style = Style::default().fg(Color::DarkGray);
        let line = Line::from(vec![Span::styled("No matching commands", style)]);
        let line_area = Rect::new(inner.x, inner.y, inner.width, 1);
        frame.render_widget(line, line_area);
        return;
    }

    let scroll_offset = if sm.selected >= max_visible {
        sm.selected - max_visible + 1
    } else {
        0
    };

    let name_col_width = 16;

    for (i, cmd) in sm
        .commands
        .iter()
        .skip(scroll_offset)
        .take(visible_count)
        .enumerate()
    {
        if i as u16 >= inner.height {
            break;
        }
        let is_selected = (i + scroll_offset) == sm.selected;
        let (name_style, desc_style) = if is_selected {
            (
                Style::default().fg(Color::White).bg(Color::DarkGray),
                Style::default().fg(Color::Gray).bg(Color::DarkGray),
            )
        } else {
            (
                Style::default().fg(Color::Green),
                Style::default().fg(Color::DarkGray),
            )
        };

        let name_display: String = format!("/{}", cmd.name)
            .chars()
            .take(name_col_width)
            .collect();
        let name_padding = name_col_width.saturating_sub(name_display.chars().count());
        let name_padded = format!("{}{}", name_display, " ".repeat(name_padding));

        let desc_width = (inner.width as usize).saturating_sub(name_col_width + 1);
        let desc_display: String = cmd.description.chars().take(desc_width).collect();
        let desc_padding = desc_width.saturating_sub(desc_display.chars().count());
        let desc_padded = format!("{}{}", desc_display, " ".repeat(desc_padding));

        let line = Line::from(vec![
            Span::styled(name_padded, name_style),
            Span::styled(
                " ",
                if is_selected {
                    Style::default().bg(Color::DarkGray)
                } else {
                    Style::default()
                },
            ),
            Span::styled(desc_padded, desc_style),
        ]);
        let line_area = Rect::new(inner.x, inner.y + i as u16, inner.width, 1);
        frame.render_widget(line, line_area);
    }
}

fn render_quick_switcher_popup(frame: &mut Frame, qs: &QuickSwitcherState, area: Rect) {
    let max_visible = 10;
    let visible_count = if qs.filtered.is_empty() {
        1
    } else {
        max_visible.min(qs.filtered.len())
    };
    let popup_height = (visible_count + 2) as u16;
    let popup_width = (area.width * 80 / 100).max(30).min(area.width);
    let x = area.x + (area.width.saturating_sub(popup_width)) / 2;
    let y = (area.y + area.height / 2).min(area.y + area.height.saturating_sub(popup_height));

    let popup_area = Rect::new(x, y, popup_width, popup_height);
    frame.render_widget(Clear, popup_area);

    let title = if qs.query.is_empty() {
        " Quick Switcher ".to_string()
    } else {
        format!(" Quick Switcher: {} ", qs.query)
    };

    let block = WidgetBlock::default()
        .borders(Borders::ALL)
        .border_type(BorderType::Rounded)
        .border_style(Style::default().fg(Color::Magenta))
        .title(title);

    let inner = block.inner(popup_area);
    frame.render_widget(block, popup_area);

    if qs.filtered.is_empty() {
        let msg = if qs.fetching {
            "Searching..."
        } else if qs.query.is_empty() {
            "Type to search..."
        } else {
            "No results"
        };
        let style = Style::default().fg(Color::DarkGray);
        let line = Line::from(vec![Span::styled(msg, style)]);
        let line_area = Rect::new(inner.x, inner.y, inner.width, 1);
        frame.render_widget(line, line_area);
        return;
    }

    let scroll_offset = if qs.selected >= max_visible {
        qs.selected - max_visible + 1
    } else {
        0
    };

    for (i, (title, uid)) in qs
        .filtered
        .iter()
        .skip(scroll_offset)
        .take(visible_count)
        .enumerate()
    {
        if i as u16 >= inner.height {
            break;
        }
        let is_selected = (i + scroll_offset) == qs.selected;
        let is_create = uid.is_empty();
        let display_text = if is_create {
            format!("+ Create: {}", title)
        } else {
            title.clone()
        };
        let style = if is_selected {
            Style::default().fg(Color::White).bg(Color::DarkGray)
        } else if is_create {
            Style::default().fg(Color::Green)
        } else {
            Style::default().fg(Color::Gray)
        };

        let max_text_width = inner.width as usize;
        let display: String = display_text.chars().take(max_text_width).collect();
        let padding = max_text_width.saturating_sub(display.chars().count());
        let padded = format!("{}{}", display, " ".repeat(padding));

        let line = Line::from(vec![Span::styled(padded, style)]);
        let line_area = Rect::new(inner.x, inner.y + i as u16, inner.width, 1);
        frame.render_widget(line, line_area);
    }
}

fn render_search_popup(frame: &mut Frame, search: &SearchState, area: Rect) {
    let max_visible = 10;
    let visible_count = if search.results.is_empty() {
        1
    } else {
        max_visible.min(search.results.len())
    };
    let popup_height = (visible_count + 2) as u16;
    let popup_width = (area.width * 80 / 100).max(30).min(area.width);
    let x = area.x + (area.width.saturating_sub(popup_width)) / 2;
    let y = (area.y + area.height / 2).min(area.y + area.height.saturating_sub(popup_height));

    let popup_area = Rect::new(x, y, popup_width, popup_height);
    frame.render_widget(Clear, popup_area);

    let title = if search.query.is_empty() {
        " Search ".to_string()
    } else {
        format!(" Search: {} ", search.query)
    };

    let block = WidgetBlock::default()
        .borders(Borders::ALL)
        .border_type(BorderType::Rounded)
        .border_style(Style::default().fg(Color::Yellow))
        .title(title);

    let inner = block.inner(popup_area);
    frame.render_widget(block, popup_area);

    if search.results.is_empty() {
        let style = Style::default().fg(Color::DarkGray);
        let line = Line::from(vec![Span::styled("No results", style)]);
        let line_area = Rect::new(inner.x, inner.y, inner.width, 1);
        frame.render_widget(line, line_area);
        return;
    }

    // Scroll window: keep selected item visible
    let scroll_offset = if search.selected >= max_visible {
        search.selected - max_visible + 1
    } else {
        0
    };

    for (i, (_, text)) in search
        .results
        .iter()
        .skip(scroll_offset)
        .take(visible_count)
        .enumerate()
    {
        if i as u16 >= inner.height {
            break;
        }
        let is_selected = (i + scroll_offset) == search.selected;
        let style = if is_selected {
            Style::default().fg(Color::White).bg(Color::DarkGray)
        } else {
            Style::default().fg(Color::Gray)
        };

        let max_text_width = inner.width as usize;
        let display: String = text.chars().take(max_text_width).collect();
        let padding = max_text_width.saturating_sub(display.chars().count());
        let padded = format!("{}{}", display, " ".repeat(padding));

        let line = Line::from(vec![Span::styled(padded, style)]);
        let line_area = Rect::new(inner.x, inner.y + i as u16, inner.width, 1);
        frame.render_widget(line, line_area);
    }
}