ane-editor 0.2.0

use ratatui::{
    Frame,
    layout::{Position, Rect},
    style::{Color, Modifier, Style},
    text::{Line, Span},
    widgets::{Clear, Paragraph},
};

use crate::data::buffer::Buffer;
use crate::data::lsp::types::SemanticToken;
use crate::data::state::{EditorState, Mode, Selection};

fn token_style(token_type: &str) -> Style {
    match token_type {
        "keyword" | "modifier" => Style::default().fg(Color::Blue),
        "function" | "method" => Style::default().fg(Color::Yellow),
        "type" | "class" | "struct" | "enum" | "interface" | "typeParameter" => {
            Style::default().fg(Color::Cyan)
        }
        "string" => Style::default().fg(Color::Green),
        "number" => Style::default().fg(Color::Magenta),
        "comment" => Style::default().fg(Color::DarkGray),
        "key" => Style::default().fg(Color::Cyan),
        "variable" | "parameter" | "property" => Style::default().fg(Color::White),
        "macro" => Style::default().fg(Color::LightMagenta),
        "operator" => Style::default().fg(Color::LightRed),
        "namespace" => Style::default().fg(Color::LightCyan),
        "enumMember" => Style::default().fg(Color::Cyan),
        // Markdown token types
        "heading" => Style::default()
            .fg(Color::Yellow)
            .add_modifier(Modifier::BOLD),
        // strong/emphasis intentionally omit fg so they inherit the outer
        // color when nested inside a heading or other styled range.
        "strong" => Style::default().add_modifier(Modifier::BOLD),
        "emphasis" => Style::default().add_modifier(Modifier::ITALIC),
        "code" => Style::default().fg(Color::Green),
        "link" => Style::default()
            .fg(Color::Cyan)
            .add_modifier(Modifier::UNDERLINED),
        "quote" => Style::default().fg(Color::DarkGray),
        "list_marker" => Style::default().fg(Color::Yellow),
        "punctuation" => Style::default().fg(Color::DarkGray),
        _ => Style::default().fg(Color::Gray),
    }
}

fn expand_tabs(s: &str) -> String {
    s.replace('\t', "    ")
}

pub(crate) fn display_col(line: &str, byte_idx: usize) -> usize {
    let mut safe_idx = byte_idx.min(line.len());
    while safe_idx > 0 && !line.is_char_boundary(safe_idx) {
        safe_idx -= 1;
    }
    line[..safe_idx]
        .chars()
        .map(|c| if c == '\t' { 4 } else { 1 })
        .sum()
}

pub(crate) fn byte_col_from_display(line: &str, target_display: usize) -> usize {
    let mut display = 0;
    let mut byte_idx = 0;
    for c in line.chars() {
        let w = if c == '\t' { 4 } else { 1 };
        if display + w > target_display {
            break;
        }
        display += w;
        byte_idx += c.len_utf8();
    }
    byte_idx
}

pub(crate) fn wrap_offsets(line: &str, text_width: usize) -> Vec<usize> {
    if text_width == 0 {
        return vec![0];
    }
    let expanded = expand_tabs(line);
    let chars: Vec<char> = expanded.chars().collect();
    if chars.is_empty() {
        return vec![0];
    }

    let mut offsets = vec![0usize];
    let mut col = 0;
    let mut last_space = None;
    let mut row_start = 0;

    for (i, &ch) in chars.iter().enumerate() {
        if ch == ' ' {
            last_space = Some(i);
        }
        col += 1;

        if col >= text_width && i + 1 < chars.len() {
            if let Some(sp) = last_space
                && sp > row_start
            {
                let break_at = sp + 1;
                offsets.push(break_at);
                row_start = break_at;
                col = (i + 1) - break_at;
                last_space = None;
                continue;
            }
            let break_at = i + 1;
            offsets.push(break_at);
            row_start = break_at;
            col = 0;
            last_space = None;
        }
    }

    offsets
}

pub(crate) fn visual_row_count(line: &str, text_width: usize) -> usize {
    wrap_offsets(line, text_width).len()
}

pub(crate) fn display_col_to_wrap_pos(offsets: &[usize], display_col: usize) -> (usize, usize) {
    let mut row = 0;
    for (i, &off) in offsets.iter().enumerate().rev() {
        if display_col >= off {
            row = i;
            break;
        }
    }
    (row, display_col - offsets[row])
}

pub(crate) fn wrap_row_start(offsets: &[usize], row: usize) -> usize {
    offsets.get(row).copied().unwrap_or(0)
}

pub(crate) fn screen_to_buffer(
    x: u16,
    y: u16,
    area: Rect,
    state: &EditorState,
) -> Option<(usize, usize)> {
    let buf = state.current_buffer()?;
    if x < area.x || y < area.y || x >= area.right() || y >= area.bottom() {
        return None;
    }

    let line_num_width = format!("{}", buf.line_count().saturating_sub(1)).len();
    let gutter_end = area.x + line_num_width as u16 + 1;
    if x < gutter_end {
        return None;
    }

    let text_width = (area.width as usize).saturating_sub(line_num_width + 1);
    let text_col = (x - gutter_end) as usize;
    let target_visual_row = (y - area.y) as usize;

    let mut accumulated_rows = 0;
    let mut logical_line = state.scroll_offset;

    while logical_line < buf.line_count() {
        let line_text = &buf.lines[logical_line];
        let row_count = visual_row_count(line_text, text_width);

        if accumulated_rows + row_count > target_visual_row {
            let wrap_row_within_line = target_visual_row - accumulated_rows;
            let offsets = wrap_offsets(line_text, text_width);
            let target_display_col = wrap_row_start(&offsets, wrap_row_within_line) + text_col;
            let byte_col = byte_col_from_display(line_text, target_display_col);
            return Some((logical_line, byte_col));
        }

        accumulated_rows += row_count;
        logical_line += 1;
    }

    // Click below last buffer line — clamp to end of last line
    if buf.line_count() > 0 {
        let last = buf.line_count() - 1;
        Some((last, buf.lines[last].len()))
    } else {
        Some((0, 0))
    }
}

fn styled_line_with_tokens<'a>(
    line_text: &'a str,
    line_num: usize,
    tokens: &[SemanticToken],
    base_style: Style,
) -> Vec<Span<'a>> {
    let chars: Vec<char> = line_text.chars().collect();
    if chars.is_empty() {
        return vec![Span::styled(expand_tabs(line_text), base_style)];
    }

    let line_tokens: Vec<&SemanticToken> = tokens.iter().filter(|t| t.line == line_num).collect();
    if line_tokens.is_empty() {
        return vec![Span::styled(expand_tabs(line_text), base_style)];
    }

    // Apply outer (longer) tokens first; inner (shorter) tokens patch on top
    // via Style::patch, so e.g. an emphasis token nested in a heading adds
    // ITALIC while inheriting the heading's fg+BOLD.
    let mut sorted = line_tokens;
    sorted.sort_by_key(|t| std::cmp::Reverse(t.length));

    let mut styles: Vec<Style> = vec![base_style; chars.len()];
    for token in &sorted {
        let style = token_style(&token.token_type);
        let start = token.start_col.min(chars.len());
        let end = (token.start_col + token.length).min(chars.len());
        for cell in styles.iter_mut().take(end).skip(start) {
            *cell = cell.patch(style);
        }
    }

    let mut spans: Vec<Span<'a>> = Vec::new();
    let mut run_style = styles[0];
    let mut run = String::new();
    for (i, &ch) in chars.iter().enumerate() {
        if styles[i] != run_style {
            if !run.is_empty() {
                spans.push(Span::styled(expand_tabs(&run), run_style));
                run.clear();
            }
            run_style = styles[i];
        }
        run.push(ch);
    }
    if !run.is_empty() {
        spans.push(Span::styled(expand_tabs(&run), run_style));
    }

    spans
}

fn wrap_spans(spans: Vec<Span<'_>>, text_width: usize) -> Vec<Vec<Span<'static>>> {
    if text_width == 0 {
        let row = spans
            .into_iter()
            .map(|s| Span::styled(s.content.into_owned(), s.style))
            .collect();
        return vec![row];
    }

    let full_text: String = spans.iter().map(|s| s.content.as_ref()).collect();
    let offsets = wrap_offsets(&full_text, text_width);

    let mut span_chars: Vec<(char, Style)> = Vec::new();
    for span in &spans {
        let style = span.style;
        for ch in span.content.chars() {
            span_chars.push((ch, style));
        }
    }

    let mut rows: Vec<Vec<Span<'static>>> = Vec::new();
    for w in 0..offsets.len() {
        let start = offsets[w];
        let end = if w + 1 < offsets.len() {
            offsets[w + 1]
        } else {
            span_chars.len()
        };
        let slice = &span_chars[start..end];

        let mut row: Vec<Span<'static>> = Vec::new();
        if !slice.is_empty() {
            let mut cur_style = slice[0].1;
            let mut cur_text = String::new();
            for &(ch, style) in slice {
                if style != cur_style {
                    if !cur_text.is_empty() {
                        row.push(Span::styled(cur_text, cur_style));
                        cur_text = String::new();
                    }
                    cur_style = style;
                }
                cur_text.push(ch);
            }
            if !cur_text.is_empty() {
                row.push(Span::styled(cur_text, cur_style));
            }
        }
        rows.push(row);
    }

    if rows.is_empty() {
        rows.push(Vec::new());
    }

    rows
}

#[allow(clippy::too_many_arguments)]
fn render_selection_highlight(
    frame: &mut Frame,
    area: Rect,
    buf: &Buffer,
    sel: &Selection,
    scroll_offset: usize,
    line_num_width: usize,
    text_width: usize,
    visible_height: usize,
) {
    let (start_line, start_col, end_line, end_col) = sel.ordered();
    let gutter_x = area.x + line_num_width as u16 + 1;
    let sel_style = Style::default().bg(Color::LightBlue).fg(Color::Black);

    let mut accumulated_rows = 0usize;
    let mut line_idx = scroll_offset;

    while accumulated_rows < visible_height && line_idx < buf.line_count() {
        let line_text = &buf.lines[line_idx];
        let offsets = wrap_offsets(line_text, text_width);
        let row_count = offsets.len();

        if line_idx >= start_line && line_idx <= end_line {
            let sel_start_byte = if line_idx == start_line {
                start_col.min(line_text.len())
            } else {
                0
            };
            let sel_end_byte = if line_idx == end_line {
                end_col.min(line_text.len())
            } else {
                line_text.len()
            };

            let sel_start_dc = display_col(line_text, sel_start_byte);
            let sel_end_dc = display_col(line_text, sel_end_byte);

            for wrap_row in 0..row_count {
                let screen_y = area.y + (accumulated_rows + wrap_row) as u16;
                if screen_y >= area.bottom() {
                    break;
                }

                let row_start_dc = offsets[wrap_row];
                let row_end_dc = if wrap_row + 1 < offsets.len() {
                    offsets[wrap_row + 1]
                } else {
                    display_col(line_text, line_text.len())
                };

                let hl_start = sel_start_dc.max(row_start_dc);
                let hl_end = sel_end_dc.min(row_end_dc);

                if hl_start < hl_end {
                    let sx_start = gutter_x + (hl_start - row_start_dc) as u16;
                    let sx_end = gutter_x + (hl_end - row_start_dc) as u16;

                    for sx in sx_start..sx_end.min(area.right()) {
                        if let Some(cell) = frame.buffer_mut().cell_mut(Position::new(sx, screen_y))
                        {
                            cell.set_style(sel_style);
                        }
                    }
                }
            }
        }

        accumulated_rows += row_count;
        line_idx += 1;
    }
}

pub fn render(
    frame: &mut Frame,
    area: Rect,
    state: &EditorState,
    semantic_tokens: &[SemanticToken],
) {
    frame.render_widget(Clear, area);
    match state.current_buffer() {
        Some(buf) => {
            let visible_height = area.height as usize;
            let line_num_width = format!("{}", buf.line_count().saturating_sub(1)).len();
            let text_width = (area.width as usize).saturating_sub(line_num_width + 1);
            let use_highlighting = !semantic_tokens.is_empty();

            let is_empty_file = buf.line_count() == 1 && buf.lines[0].is_empty() && !buf.dirty;
            let mut visual_lines: Vec<Line> = Vec::new();
            let mut cursor_visual_row: Option<usize> = None;
            let mut cursor_visual_col: Option<usize> = None;
            let mut logical_line = state.scroll_offset;

            while visual_lines.len() < visible_height && logical_line < buf.line_count() {
                let line_text = &buf.lines[logical_line];
                let is_current = logical_line == state.cursor_line;

                let gutter_bg = Style::default().bg(Color::Rgb(40, 40, 50));
                let num_style = if is_empty_file {
                    gutter_bg
                } else if is_current && state.mode == Mode::Edit {
                    Style::default()
                        .fg(Color::Yellow)
                        .add_modifier(Modifier::BOLD)
                } else {
                    Style::default().fg(Color::DarkGray)
                };

                let text_style = if is_current && state.mode == Mode::Edit {
                    Style::default().fg(Color::White)
                } else {
                    Style::default().fg(Color::Gray)
                };

                let content_spans = if use_highlighting {
                    styled_line_with_tokens(line_text, logical_line, semantic_tokens, text_style)
                } else {
                    vec![Span::styled(expand_tabs(line_text), text_style)]
                };

                let wrapped_rows = wrap_spans(content_spans, text_width);

                if is_current {
                    let cursor_display_col = display_col(line_text, state.cursor_col);
                    let offsets = wrap_offsets(line_text, text_width);
                    let (c_wrap_row, c_col_in_row) =
                        display_col_to_wrap_pos(&offsets, cursor_display_col);
                    let max_row = wrapped_rows.len().saturating_sub(1);
                    let (final_row, final_col) = if c_wrap_row > max_row {
                        let last_row_len: usize = wrapped_rows.last().map_or(0, |spans| {
                            spans.iter().map(|s| s.content.chars().count()).sum()
                        });
                        (max_row, last_row_len)
                    } else {
                        (c_wrap_row, c_col_in_row)
                    };
                    cursor_visual_row = Some(visual_lines.len() + final_row);
                    cursor_visual_col = Some(final_col);
                }

                for (wrap_idx, row_spans) in wrapped_rows.into_iter().enumerate() {
                    if visual_lines.len() >= visible_height {
                        break;
                    }

                    let num_str = if is_empty_file {
                        " ".repeat(line_num_width + 1)
                    } else if wrap_idx == 0 {
                        format!("{:>width$} ", logical_line, width = line_num_width)
                    } else {
                        " ".repeat(line_num_width + 1)
                    };

                    let mut spans = vec![Span::styled(num_str, num_style)];
                    spans.extend(row_spans);
                    visual_lines.push(Line::from(spans));
                }

                logical_line += 1;
            }

            if is_empty_file {
                let gutter_bg = Style::default().bg(Color::Rgb(40, 40, 50));
                while visual_lines.len() < visible_height {
                    let gutter_span = Span::styled(" ".repeat(line_num_width + 1), gutter_bg);
                    visual_lines.push(Line::from(vec![gutter_span]));
                }
            }

            let paragraph = Paragraph::new(visual_lines);
            frame.render_widget(paragraph, area);

            if let Some(sel) = &state.selection {
                render_selection_highlight(
                    frame,
                    area,
                    buf,
                    sel,
                    state.scroll_offset,
                    line_num_width,
                    text_width,
                    visible_height,
                );
            }

            if let (Some(vis_row), Some(vis_col)) = (cursor_visual_row, cursor_visual_col) {
                let cursor_x = area.x + line_num_width as u16 + 1 + vis_col as u16;
                let cursor_y = area.y + vis_row as u16;
                if cursor_x < area.right() && cursor_y < area.bottom() {
                    if state.mode == Mode::Edit && !state.focus_tree {
                        frame.set_cursor_position(Position::new(cursor_x, cursor_y));
                    } else if state.mode == Mode::Chord
                        && !state.focus_tree
                        && let Some(cell) = frame
                            .buffer_mut()
                            .cell_mut(Position::new(cursor_x, cursor_y))
                    {
                        cell.set_style(Style::default().bg(Color::Blue).fg(Color::White));
                    }
                }
            }
        }
        None => {
            let welcome = vec![
                Line::from(""),
                Line::from(Span::styled(
                    "  ane — Agent Native Editor",
                    Style::default()
                        .fg(Color::Cyan)
                        .add_modifier(Modifier::BOLD),
                )),
                Line::from(""),
                Line::from(Span::styled(
                    "  Select a file from the tree to begin editing",
                    Style::default().fg(Color::DarkGray),
                )),
                Line::from(Span::styled(
                    "  Ctrl-E to toggle Edit mode | Ctrl-T to open file tree",
                    Style::default().fg(Color::DarkGray),
                )),
            ];
            let paragraph = Paragraph::new(welcome);
            frame.render_widget(paragraph, area);
        }
    }
}

#[cfg(test)]
mod tests {
    use std::io::Write as IoWrite;

    use ratatui::layout::Rect;

    use super::*;
    use crate::data::state::EditorState;

    fn make_state_with_lines(lines: &[&str]) -> (tempfile::NamedTempFile, EditorState) {
        let mut f = tempfile::NamedTempFile::new().unwrap();
        for (i, line) in lines.iter().enumerate() {
            if i > 0 {
                f.write_all(b"\n").unwrap();
            }
            f.write_all(line.as_bytes()).unwrap();
        }
        f.flush().unwrap();
        let state = EditorState::for_file(f.path()).unwrap();
        (f, state)
    }

    #[test]
    fn wrap_offsets_empty_line() {
        assert_eq!(wrap_offsets("", 80), vec![0]);
    }

    #[test]
    fn wrap_offsets_short_line_no_wrap() {
        assert_eq!(wrap_offsets("hello world", 80), vec![0]);
    }

    #[test]
    fn wrap_offsets_breaks_on_space() {
        // "hello world" with width 8: "hello " fits (6 chars), then "wo" fills to 8
        // but word-wrap should break after "hello " to keep "world" whole
        let offsets = wrap_offsets("hello world", 8);
        assert_eq!(offsets, vec![0, 6]);
    }

    #[test]
    fn wrap_offsets_long_word_falls_back_to_char_break() {
        let offsets = wrap_offsets("abcdefghij", 5);
        assert_eq!(offsets, vec![0, 5]);
    }

    #[test]
    fn wrap_offsets_multiple_wraps() {
        let offsets = wrap_offsets("aaa bbb ccc ddd", 8);
        assert_eq!(offsets, vec![0, 8]);
    }

    #[test]
    fn wrap_offsets_exact_fit_no_extra_row() {
        let offsets = wrap_offsets("12345", 5);
        assert_eq!(offsets, vec![0]);
    }

    #[test]
    fn visual_row_count_uses_word_wrap() {
        assert_eq!(visual_row_count("hello world", 8), 2);
        assert_eq!(visual_row_count("hi", 80), 1);
    }

    #[test]
    fn display_col_to_wrap_pos_basic() {
        let offsets = vec![0, 6];
        assert_eq!(display_col_to_wrap_pos(&offsets, 0), (0, 0));
        assert_eq!(display_col_to_wrap_pos(&offsets, 5), (0, 5));
        assert_eq!(display_col_to_wrap_pos(&offsets, 6), (1, 0));
        assert_eq!(display_col_to_wrap_pos(&offsets, 8), (1, 2));
    }

    #[test]
    fn wrap_row_start_basic() {
        let offsets = vec![0, 6, 12];
        assert_eq!(wrap_row_start(&offsets, 0), 0);
        assert_eq!(wrap_row_start(&offsets, 1), 6);
        assert_eq!(wrap_row_start(&offsets, 2), 12);
        assert_eq!(wrap_row_start(&offsets, 5), 0);
    }

    #[test]
    fn token_style_markdown_heading_is_bold_yellow() {
        let style = token_style("heading");
        assert_eq!(style.fg, Some(Color::Yellow));
        assert!(
            style.add_modifier.contains(Modifier::BOLD),
            "heading should be bold"
        );
    }

    #[test]
    fn token_style_markdown_emphasis_is_italic() {
        let style = token_style("emphasis");
        assert!(
            style.add_modifier.contains(Modifier::ITALIC),
            "emphasis should be italic"
        );
    }

    #[test]
    fn token_style_markdown_code_is_green() {
        let style = token_style("code");
        assert_eq!(style.fg, Some(Color::Green));
    }

    #[test]
    fn styled_line_overlap_heading_emphasis_merges_styles() {
        // "## Hello *world*"
        //  0         1
        //  0123456789012345
        // Heading covers 0..16, emphasis covers 9..16.
        // The emphasis range should end up bold+italic+yellow (heading wins on
        // fg/bold, emphasis adds italic via Style::patch).
        let line = "## Hello *world*";
        let tokens = vec![
            SemanticToken {
                line: 0,
                start_col: 0,
                length: 16,
                token_type: "heading".to_string(),
            },
            SemanticToken {
                line: 0,
                start_col: 9,
                length: 7,
                token_type: "emphasis".to_string(),
            },
        ];
        let spans = styled_line_with_tokens(line, 0, &tokens, Style::default().fg(Color::Gray));

        let yellow_bold_only = spans.iter().find(|s| {
            s.style.fg == Some(Color::Yellow)
                && s.style.add_modifier.contains(Modifier::BOLD)
                && !s.style.add_modifier.contains(Modifier::ITALIC)
        });
        assert!(
            yellow_bold_only.is_some(),
            "expected a heading-only segment (yellow+bold, no italic); got {:?}",
            spans
        );

        let yellow_bold_italic = spans.iter().find(|s| {
            s.style.fg == Some(Color::Yellow)
                && s.style.add_modifier.contains(Modifier::BOLD)
                && s.style.add_modifier.contains(Modifier::ITALIC)
        });
        assert!(
            yellow_bold_italic.is_some(),
            "expected a heading+emphasis segment (yellow+bold+italic); got {:?}",
            spans
        );

        let concat: String = spans.iter().map(|s| s.content.as_ref()).collect();
        assert_eq!(
            concat, line,
            "rendered spans should reconstruct the line exactly"
        );
    }

    // --- work item 0007: screen_to_buffer ---

    #[test]
    fn screen_to_buffer_basic() {
        // 1-line buffer → line_num_width=1, gutter_end=2, text_width=28
        // click at x=5: text_col = 5-2 = 3 → byte 3 in "hello world"
        let (_f, state) = make_state_with_lines(&["hello world"]);
        let area = Rect::new(0, 0, 30, 10);
        assert_eq!(screen_to_buffer(5, 0, area, &state), Some((0, 3)));
    }

    #[test]
    fn screen_to_buffer_gutter_click_returns_none() {
        // x=1 < gutter_end=2 → gutter click, no position returned
        let (_f, state) = make_state_with_lines(&["hello world"]);
        let area = Rect::new(0, 0, 30, 10);
        assert_eq!(screen_to_buffer(1, 0, area, &state), None);
    }

    #[test]
    fn screen_to_buffer_below_buffer_clamps_to_end_of_last_line() {
        // visual row 5 is past the single line; result clamps to end of line 0
        let (_f, state) = make_state_with_lines(&["only line"]);
        let area = Rect::new(0, 0, 30, 10);
        assert_eq!(screen_to_buffer(5, 5, area, &state), Some((0, 9)));
    }

    #[test]
    fn screen_to_buffer_soft_wrap_click_on_second_visual_row() {
        // "hello world" with text_width=8 wraps: row 0 = "hello ", row 1 = "world"
        // (wrap_offsets = [0, 6]).  A click at visual row 1, column 0 in the text
        // area maps to display col 6 → byte 6 (the 'w' in "world").
        let (_f, state) = make_state_with_lines(&["hello world"]);
        // area.width=10 → text_width = 10 - 1(gutter) - 1(sep) = 8
        let area = Rect::new(0, 0, 10, 10);
        // x=2 (gutter_end), y=1 → text_col=0, visual_row=1
        assert_eq!(screen_to_buffer(2, 1, area, &state), Some((0, 6)));
    }

    #[test]
    fn screen_to_buffer_tab_expansion() {
        // "\thello": tab = 4 display cols (byte 0), then h(1) e(2) l(3) l(4) o(5).
        // Display col 8 = 'o' at byte 5.  x=10 → text_col = 10-2 = 8.
        let (_f, state) = make_state_with_lines(&["\thello"]);
        let area = Rect::new(0, 0, 30, 10);
        assert_eq!(screen_to_buffer(10, 0, area, &state), Some((0, 5)));
    }

    #[test]
    fn token_style_key_is_cyan() {
        assert_eq!(token_style("key"), Style::default().fg(Color::Cyan));
    }

    #[test]
    fn token_style_key_differs_from_string() {
        assert_ne!(token_style("key"), token_style("string"));
    }
}