kreuzberg 4.7.2

//! LaTeX environment processing.
//!
//! This module handles LaTeX environments like itemize, enumerate, description,
//! tabular, and table environments.

use super::commands::process_line;
use super::utilities::{clean_text, collect_environment, extract_braced, extract_env_name};
use crate::types::Table;

/// Processes a list environment (itemize, enumerate, or description).
///
/// Converts LaTeX lists into markdown-style lists with proper nesting.
pub fn process_list(content: &str, list_type: &str, output: &mut String) {
    let lines: Vec<&str> = content.lines().collect();
    let mut item_num = 1;
    let mut i = 0;

    while i < lines.len() {
        let line = lines[i];
        let trimmed = line.trim();

        // Handle nested lists
        if (trimmed.contains("\\begin{") || trimmed.contains("\\begin {"))
            && let Some(env_name) = extract_env_name(trimmed)
            && (env_name == "itemize" || env_name == "enumerate" || env_name == "description")
        {
            let (nested_content, new_i) = collect_environment(&lines, i, &env_name);
            let current_output_len = output.len();
            process_list(&nested_content, &env_name, output);
            let nested_output = output[current_output_len..].to_string();
            output.truncate(current_output_len);
            // Indent nested list
            for nested_line in nested_output.lines() {
                output.push_str("  ");
                output.push_str(nested_line);
                output.push('\n');
            }
            i = new_i;
            continue;
        }

        // Handle \item
        if trimmed.starts_with("\\item")
            && let Some(pos) = trimmed.find("\\item")
        {
            let after = trimmed[pos + 5..].trim();

            // Handle \item[label] for description lists
            if after.starts_with('[')
                && let Some(bracket_end) = after.find(']')
            {
                let label = after[1..bracket_end].to_string();
                let text = after[bracket_end + 1..].trim().to_string();
                if list_type == "description" {
                    let processed_text = process_line(&text);
                    output.push_str(&format!("{}: {}\n", label, processed_text));
                    item_num += 1;
                    i += 1;
                    continue;
                }
            }

            // Regular list item
            let prefix = if list_type == "enumerate" {
                format!("{}. ", item_num)
            } else {
                "- ".to_string()
            };
            output.push_str(&prefix);

            let item_text = process_line(after);
            output.push_str(item_text.trim());
            output.push('\n');
            item_num += 1;
        }

        i += 1;
    }
    output.push('\n');
}

/// Processes a tabular environment.
///
/// Converts LaTeX tables into markdown tables and creates Table structures.
pub fn process_table(content: &str, output: &mut String, tables: &mut Vec<Table>) {
    let lines: Vec<&str> = content.lines().collect();
    let mut rows: Vec<Vec<String>> = Vec::new();

    for line in lines {
        let trimmed = line.trim();
        if trimmed.starts_with("\\hline")
            || trimmed.is_empty()
            || trimmed.contains("\\begin{tabular}")
            || trimmed.contains("\\end{tabular}")
        {
            continue;
        }

        let row_str = trimmed.replace("\\\\", "");
        let cells: Vec<String> = row_str
            .split('&')
            .map(|s| clean_text(s.trim()))
            .filter(|s| !s.is_empty())
            .collect();

        if !cells.is_empty() {
            rows.push(cells);
        }
    }

    if !rows.is_empty() {
        let mut markdown = String::new();
        for (i, row) in rows.iter().enumerate() {
            markdown.push('|');
            for cell in row {
                markdown.push_str(&format!(" {} |", cell));
            }
            markdown.push('\n');

            // Add header separator after first row
            if i == 0 && rows.len() > 1 {
                markdown.push('|');
                for _ in row {
                    markdown.push_str(" --- |");
                }
                markdown.push('\n');
            }
        }

        output.push_str(&markdown);

        let table = Table {
            cells: rows,
            markdown: markdown.clone(),
            page_number: 1,
            bounding_box: None,
        };
        tables.push(table);
    }
}

/// Processes a table environment with caption.
///
/// Extracts the caption and processes the embedded tabular environment.
pub fn process_table_with_caption(content: &str, output: &mut String, tables: &mut Vec<Table>) {
    // Extract and add caption if present
    if content.contains("\\caption{")
        && let Some(caption) = extract_braced(content, "caption")
    {
        output.push_str(&caption);
        output.push('\n');
    }

    // Process the tabular environment inside
    let end_tag = "\\end{tabular}";
    if content.contains("\\begin{tabular}")
        && let Some(start) = content.find("\\begin{tabular}")
        && let Some(end) = content.find(end_tag)
    {
        let tabular_content = &content[start..end + end_tag.len()];
        process_table(tabular_content, output, tables);
    }
}