from cv_analysis.utils.rectangle import Rectangle
from synthesis.segment.content_rectangle import ContentRectangle
from synthesis.segment.plot import RandomPlot
from synthesis.segment.text_block import TextBlock
from synthesis.text.font import pick_random_font_available_on_system


def generate_random_plot(rectangle: Rectangle) -> ContentRectangle:
    block = RandomPlot(*rectangle.coords)
    block.content = rectangle.content if isinstance(rectangle, ContentRectangle) else None  # TODO: Refactor
    block.generate_random_plot(rectangle)
    return block


def generate_recursive_random_table(rectangle: Rectangle, **kwargs) -> ContentRectangle:
    from synthesis.segment.table.table import RecursiveRandomTable

    block = RecursiveRandomTable(*rectangle.coords, **kwargs)
    if isinstance(rectangle, RecursiveRandomTable):
        block.content = rectangle.content if rectangle.content else None  # TODO: Refactor
    block.generate_random_table()
    return block


def generate_text_block(rectangle: Rectangle, text) -> ContentRectangle:
    block = TextBlock(
        *rectangle.coords,
        font=pick_random_font_available_on_system(
            includes=("serif", "sans-serif", "bold"),
            excludes=("mono", "italic", "oblique", "cursive"),
        ),
        font_size=30,  # TODO: De-hardcode font size... Seems to have no effect on top of that
    )
    block.content = rectangle.content if isinstance(rectangle, ContentRectangle) else None  # TODO: Refactor
    block.put_text(text, rectangle)
    return block