from http import HTTPStatus as status

from ninja import Router
from django.shortcuts import get_object_or_404

from api.v1.schemas import NotFoundError, UnauthorizedError, ForbiddenError
from api.v1.ping.schemas import PingOut
from api.v1.task.schemas import (
    TaskOutSchema,
    TaskSubmissionOut,
    TaskSubmissionIn,
)
from apps.task.models import (
    Competition,
    CompetitionTask,
    CompetetionTaskSumbission,
)

router = Router(tags=["competition"])


@router.post(
    "competitions/{competition_id}/start",
    description="Start a competition completing (open access to tasks)",
    response={
        status.OK: PingOut,
        status.UNAUTHORIZED: UnauthorizedError,
        status.NOT_FOUND: NotFoundError,
    },
)
def start_competition(request, competition_id: str) -> PingOut: ...


@router.get(
    "competitions/{competition_id}/tasks",
    description="Get all tasks of competition (works only if user started competition)",
    response={
        status.OK: list[TaskOutSchema],
        status.UNAUTHORIZED: UnauthorizedError,
        status.FORBIDDEN: ForbiddenError,
        status.NOT_FOUND: NotFoundError,
    },
)
def get_competition_tasks(
    request, competition_id: str
) -> list[TaskOutSchema]: ...


@router.get(
    "competitions/{competition_id}/tasks/{task_id}",
    description="Get a task of competition task",
    response={
        status.OK: TaskOutSchema,
        status.UNAUTHORIZED: UnauthorizedError,
        status.FORBIDDEN: ForbiddenError,
        status.NOT_FOUND: NotFoundError,
    },
)
def get_task(request, competition_id: str, task_id: str) -> TaskOutSchema: ...


@router.post(
    "competitions/{competition_id}/tasks/{task_id}/submit",
    description="Submit task solution",
    response={
        status.OK: TaskSubmissionOut,
        status.UNAUTHORIZED: UnauthorizedError,
        status.FORBIDDEN: ForbiddenError,
        status.NOT_FOUND: NotFoundError,
    },
)
def submit_task(
    request, competition_id: str, task_id: str, submission: TaskSubmissionIn
) -> PingOut:
    user = request.auth
    competetion = get_object_or_404(Competition, id=competition_id)
    task = get_object_or_404(
        CompetitionTask, competetion=competetion, id=task_id
    )

    if task.type == CompetitionTask.CompetitionTaskType.INPUT:
        CompetetionTaskSumbission.objects.create(
            user=user,
            task=task,
            status=CompetetionTaskSumbission.StatusChoices.CHECKED,
            result={"correct": submission.content == task.answer_file_path},
        )
    if task.type == CompetitionTask.CompetitionTaskType.REVIEW:
        CompetetionTaskSumbission.objects.create(
            user=user,
            task=task,
            status=CompetetionTaskSumbission.StatusChoices.SENT,
        )
    if task.type == CompetitionTask.CompetitionTaskType.CHECKER:
        CompetetionTaskSumbission.objects.create(
            user=user,
            task=task,
            status=CompetetionTaskSumbission.StatusChoices.CHECKING,
        )

    return TaskSubmissionOut(id=CompetetionTaskSumbission.id)