use crate::mem;
use crate::shapes::{GrowType, RawTextData, Type};
use crate::textlayout::{
    auto_height, build_paragraphs_with_width, paragraph_builder_group_from_text,
};
use crate::{with_current_shape, with_current_shape_mut, STATE};

#[no_mangle]
pub extern "C" fn clear_shape_text() {
    with_current_shape_mut!(state, |shape: &mut Shape| {
        shape.clear_text();
    });
}

#[no_mangle]
pub extern "C" fn set_shape_text_content() {
    let bytes = mem::bytes();
    with_current_shape_mut!(state, |shape: &mut Shape| {
        let raw_text_data = RawTextData::from(&bytes);
        shape
            .add_paragraph(raw_text_data.paragraph)
            .expect("Failed to add paragraph");
    });
    mem::free_bytes();
}

#[no_mangle]
pub extern "C" fn set_shape_grow_type(grow_type: u8) {
    with_current_shape_mut!(state, |shape: &mut Shape| {
        if let Type::Text(text_content) = &mut shape.shape_type {
            text_content.set_grow_type(GrowType::from(grow_type));
        }
    });
}

#[no_mangle]
pub extern "C" fn get_text_dimensions() -> *mut u8 {
    let mut width = 0.01;
    let mut height = 0.01;
    let mut m_width = 0.01;

    with_current_shape!(state, |shape: &Shape| {
        width = shape.selrect.width();
        height = shape.selrect.height();

        if let Type::Text(content) = &shape.shape_type {
            // 1. Reset Paragraphs
            let paragraph_width = content.width();
            let mut paragraphs = paragraph_builder_group_from_text(content, None);
            let built_paragraphs = build_paragraphs_with_width(&mut paragraphs, paragraph_width);

            // 2. Max Width Calculation
            m_width = built_paragraphs
                .iter()
                .flatten()
                .fold(0.0, |max_width, p| f32::max(p.max_width(), max_width));

            // 3. Width and Height Calculation
            match content.grow_type() {
                GrowType::AutoHeight => {
                    let mut paragraph_height = paragraph_builder_group_from_text(content, None);
                    height = auto_height(&mut paragraph_height, paragraph_width).ceil();
                }
                GrowType::AutoWidth => {
                    width = paragraph_width;
                    let mut paragraph_height = paragraph_builder_group_from_text(content, None);
                    height = auto_height(&mut paragraph_height, paragraph_width).ceil();
                }
                GrowType::Fixed => {}
            }
        }
    });

    let mut bytes = vec![0; 12];
    bytes[0..4].clone_from_slice(&width.to_le_bytes());
    bytes[4..8].clone_from_slice(&height.to_le_bytes());
    bytes[8..12].clone_from_slice(&m_width.to_le_bytes());
    mem::write_bytes(bytes)
}