from enum import StrEnum

from pydantic import BaseModel
from unique_sdk import CellRendererTypes


class ExampleColumnNames(StrEnum):
    QUESTION = "Question"
    SECTION = "Section"
    ANSWER = "Answer"
    CRITICAL_CONSISTENCY = "Critical Consistency"
    STATUS = "Status"
    REVIEWER = "Reviewer"

class ColumnDefinition(BaseModel):
    """
    Defines a single table column's structure and styling.

    Attributes:
        order: Column position (0-indexed)
        name: Column header text
        width: Column width in pixels
        renderer: Optional cell renderer type (dropdown, checkbox, etc.)
        editable: Whether the column is editable
    """

    order: int
    name: ExampleColumnNames
    width: int
    renderer: CellRendererTypes | None = None
    editable: bool = True


class ColumnDefinitions(BaseModel):
    """
    Container for all column definitions in the table.

    Provides helper methods to access columns by name.
    """

    columns: list[ColumnDefinition]

    @property
    def column_map(self) -> dict[str, ColumnDefinition]:
        """Map of column names to their definitions."""
        return {column.name: column for column in self.columns}

    def get_column_by_name(self, name: str) -> ColumnDefinition:
        """Get column definition by name."""
        return self.column_map[name]

    def get_column_name_by_order(self, order: int) -> ExampleColumnNames:
        """Get column definition by order."""
        return self.columns[order].name

    def get_column_names(self) -> list[str]:
        """Get list of all column names."""
        return list(self.column_map.keys())





example_configuration = {
    "columns": [
        {
            "order": 0,
            "name": ExampleColumnNames.QUESTION,
            "width": 300,
            "renderer": None,
            "editable": False,
        },
        {
            "order": 1,
            "name": ExampleColumnNames.SECTION,
            "width": 150,
            "renderer": None,
            "editable": False,
        },
        {
            "order": 2,
            "name": ExampleColumnNames.ANSWER,
            "width": 400,
            "renderer": CellRendererTypes.SELECTABLE_CELL_RENDERER,
            "editable": True,
        },
        {
            "order": 3,
            "name": ExampleColumnNames.CRITICAL_CONSISTENCY,
            "width": 200,
            "renderer": None,
            "editable": True,
        },
        {
            "order": 4,
            "name": ExampleColumnNames.STATUS,
            "width": 150,
            "renderer": CellRendererTypes.REVIEW_STATUS_DROPDOWN,
            "editable": True,
        },
        {
            "order": 5,
            "name": ExampleColumnNames.REVIEWER,
            "width": 150,
            "renderer": CellRendererTypes.COLLABORATOR_DROPDOWN,
            "editable": True,
        },
    ]
}

example_column_definitions = ColumnDefinitions.model_validate(example_configuration)