from typing import Callable, Dict, Tuple, List

from server.tasks import task1_json, task2_yaml, task3_dockerfile
from server.tasks import task4_compose, task5_k8s, task6_github_actions, task7_nginx

# CRITICAL: Import from grader_api wrapper layer (with normalization), NOT raw graders
from server.graders.grader_api import (
    grade_task1,
    grade_task2,
    grade_task3,
    grade_task4,
    grade_task5,
    grade_task6,
    grade_task7,
)


class TaskInfo:
    def __init__(self, module, grader_func: Callable):
        self.task_id: str = module.TASK_ID
        self.difficulty: str = module.DIFFICULTY
        self.file_type: str = module.FILE_TYPE
        self.num_bugs: int = module.NUM_BUGS
        self.description: str = module.DESCRIPTION
        self.broken_config: str = module.BROKEN_CONFIG
        self.error_message: str = module.ERROR_MESSAGE
        self.ground_truth: str = module.GROUND_TRUTH
        self.grader: Callable[[str], Tuple[float, str, List[str]]] = grader_func


TASK_ORDER = [
    "task1_json",
    "task2_yaml",
    "task3_dockerfile",
    "task4_compose",
    "task5_k8s",
    "task6_github_actions",
    "task7_nginx",
]

TASK_REGISTRY: Dict[str, TaskInfo] = {
    "task1_json": TaskInfo(task1_json, grade_task1),
    "task2_yaml": TaskInfo(task2_yaml, grade_task2),
    "task3_dockerfile": TaskInfo(task3_dockerfile, grade_task3),
    "task4_compose": TaskInfo(task4_compose, grade_task4),
    "task5_k8s": TaskInfo(task5_k8s, grade_task5),
    "task6_github_actions": TaskInfo(task6_github_actions, grade_task6),
    "task7_nginx": TaskInfo(task7_nginx, grade_task7),
}


def get_task(task_id: str) -> TaskInfo:
    return TASK_REGISTRY[task_id]


def get_all_task_ids() -> List[str]:
    return list(TASK_ORDER)