"""XER table handlers module."""

from xer_mcp.parser.table_handlers.base import TableHandler
from xer_mcp.parser.table_handlers.calendar import CalendarHandler
from xer_mcp.parser.table_handlers.project import ProjectHandler
from xer_mcp.parser.table_handlers.projwbs import ProjwbsHandler
from xer_mcp.parser.table_handlers.task import TaskHandler
from xer_mcp.parser.table_handlers.taskpred import TaskpredHandler

# Registry mapping table names to handlers
TABLE_HANDLERS: dict[str, type[TableHandler]] = {
    "PROJECT": ProjectHandler,
    "TASK": TaskHandler,
    "TASKPRED": TaskpredHandler,
    "PROJWBS": ProjwbsHandler,
    "CALENDAR": CalendarHandler,
}

__all__ = [
    "CalendarHandler",
    "ProjectHandler",
    "ProjwbsHandler",
    "TABLE_HANDLERS",
    "TableHandler",
    "TaskHandler",
    "TaskpredHandler",
]