kreuzberg 4.3.0

//! Djot content conversion and HTML rendering APIs.
//!
//! Provides public APIs for converting between different representations:
//! - DjotContent to djot markup
//! - ExtractionResult to djot markup
//! - Djot markup to HTML

use super::rendering::render_block_to_djot;
use jotdown::Parser;
#[cfg(test)]
use std::borrow::Cow;

/// Convert DjotContent back to djot markup.
///
/// This function takes a `DjotContent` structure and generates valid djot markup
/// from it, preserving:
/// - Block structure (headings, code blocks, lists, blockquotes, etc.)
/// - Inline formatting (strong, emphasis, highlight, subscript, superscript, etc.)
/// - Attributes where present ({.class #id key="value"})
///
/// # Arguments
///
/// * `content` - The DjotContent to convert
///
/// # Returns
///
/// A String containing valid djot markup
///
/// # Example
///
/// ```ignore
/// let djot_content = // ... extract from some source
/// let markup = djot_content_to_djot(&djot_content);
/// println!("{}", markup);
/// ```
pub fn djot_content_to_djot(content: &crate::types::DjotContent) -> String {
    let mut output = String::new();

    for block in &content.blocks {
        render_block_to_djot(&mut output, block, 0);
    }

    output
}

/// Convert any ExtractionResult to djot format.
///
/// This function converts an `ExtractionResult` to djot markup:
/// - If `djot_content` is `Some`, uses `djot_content_to_djot` for full fidelity conversion
/// - Otherwise, wraps the plain text content in paragraphs
///
/// # Arguments
///
/// * `result` - The ExtractionResult to convert
///
/// # Returns
///
/// A `Result` containing the djot markup string
///
/// # Example
///
/// ```ignore
/// let result = extractor.extract_bytes(bytes, "text/plain", &config).await?;
/// let djot_markup = extraction_result_to_djot(&result)?;
/// ```
pub fn extraction_result_to_djot(result: &crate::types::ExtractionResult) -> crate::Result<String> {
    if let Some(ref djot_content) = result.djot_content {
        Ok(djot_content_to_djot(djot_content))
    } else {
        // Convert plain text to basic djot paragraphs
        let mut output = String::new();

        // Split content by double newlines to create paragraphs
        let paragraphs: Vec<&str> = result.content.split("\n\n").collect();

        for para in paragraphs {
            let trimmed = para.trim();
            if !trimmed.is_empty() {
                output.push_str(trimmed);
                output.push_str("\n\n");
            }
        }

        Ok(output)
    }
}

/// Render djot content to HTML.
///
/// This function takes djot source text and renders it to HTML using jotdown's
/// built-in HTML renderer.
///
/// # Arguments
///
/// * `djot_source` - The djot markup text to render
///
/// # Returns
///
/// A `Result` containing the rendered HTML string
///
/// # Example
///
/// ```ignore
/// let djot = "# Hello\n\nThis is *bold* and _italic_.";
/// let html = djot_to_html(djot)?;
/// assert!(html.contains("<h1>"));
/// assert!(html.contains("<strong>"));
/// assert!(html.contains("<em>"));
/// ```
pub fn djot_to_html(djot_source: &str) -> crate::Result<String> {
    let parser = Parser::new(djot_source);
    let html = jotdown::html::render_to_string(parser);
    Ok(html)
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::types::{BlockType, DjotContent, ExtractionResult, FormattedBlock, InlineElement, InlineType, Metadata};

    #[test]
    fn test_djot_content_to_djot_heading() {
        let content = DjotContent {
            plain_text: "Test Heading".to_string(),
            blocks: vec![FormattedBlock {
                block_type: BlockType::Heading,
                level: Some(1),
                inline_content: vec![InlineElement {
                    element_type: InlineType::Text,
                    content: "Test Heading".to_string(),
                    attributes: None,
                    metadata: None,
                }],
                attributes: None,
                language: None,
                code: None,
                children: vec![],
            }],
            metadata: Metadata::default(),
            tables: vec![],
            images: vec![],
            links: vec![],
            footnotes: vec![],
            attributes: Default::default(),
        };

        let markup = djot_content_to_djot(&content);
        assert!(markup.contains("# Test Heading"));
    }

    #[test]
    fn test_extraction_result_to_djot_with_djot_content() {
        let result = ExtractionResult {
            content: "Test content".to_string(),
            mime_type: Cow::Borrowed("text/djot"),
            metadata: Metadata::default(),
            tables: vec![],
            detected_languages: None,
            chunks: None,
            images: None,
            pages: None,
            djot_content: Some(DjotContent {
                plain_text: "Test content".to_string(),
                blocks: vec![FormattedBlock {
                    block_type: BlockType::Paragraph,
                    level: None,
                    inline_content: vec![InlineElement {
                        element_type: InlineType::Text,
                        content: "Test content".to_string(),
                        attributes: None,
                        metadata: None,
                    }],
                    attributes: None,
                    language: None,
                    code: None,
                    children: vec![],
                }],
                metadata: Metadata::default(),
                tables: vec![],
                images: vec![],
                links: vec![],
                footnotes: vec![],
                attributes: Default::default(),
            }),
            elements: None,
            ocr_elements: None,
            document: None,
        };

        let markup = extraction_result_to_djot(&result).expect("Should convert");
        assert!(markup.contains("Test content"));
    }

    #[test]
    fn test_extraction_result_to_djot_without_djot_content() {
        let result = ExtractionResult {
            content: "Paragraph one\n\nParagraph two".to_string(),
            mime_type: Cow::Borrowed("text/plain"),
            metadata: Metadata::default(),
            tables: vec![],
            detected_languages: None,
            chunks: None,
            images: None,
            pages: None,
            djot_content: None,
            elements: None,
            ocr_elements: None,
            document: None,
        };

        let markup = extraction_result_to_djot(&result).expect("Should convert");
        assert!(markup.contains("Paragraph one"));
        assert!(markup.contains("Paragraph two"));
    }

    #[test]
    fn test_djot_to_html_heading() {
        let djot = "# Hello";
        let html = djot_to_html(djot).expect("Should render");
        assert!(html.contains("<h1>") || html.contains("<H1>"));
    }

    #[test]
    fn test_djot_to_html_formatting() {
        let djot = "This is *bold* and _italic_.";
        let html = djot_to_html(djot).expect("Should render");
        assert!(html.contains("<strong>") || html.contains("<em>"));
    }
}