from typing import List, Optional

from fastapi import APIRouter, Depends, status, Response
from sqlalchemy.ext.asyncio import AsyncSession

from app.database.session import get_db
from app.domain.entities.users import UserRead, UserUpdate, PasswordChangeRequest, UserCreate, UserCheckLessonRead
from app.domain.models import User
from app.infrastructure.dependencies import require_auth_user, require_admin
from app.infrastructure.register_service import RegisterService
from app.infrastructure.user_check_lessons_service import UserCheckLessonsService
from app.infrastructure.users_service import UsersService

users_router = APIRouter()


@users_router.get(
    '/',
    response_model=List[UserRead],
    summary='Return all users',
    description='Return all users',
)
async def get_all_users(
        db: AsyncSession = Depends(get_db),
        user: User = Depends(require_admin)
):
    users_service = UsersService(db)
    return await users_service.get_all()


@users_router.get(
    '/me/',
    response_model=Optional[UserRead],
    summary='Returns current authenticated user data',
    description='Returns current authenticated user data',
)
async def get_authenticated_user_data(
        db: AsyncSession = Depends(get_db),
        user: User = Depends(require_auth_user)
):
    users_service = UsersService(db)
    return await users_service.get_by_id(user.id)


@users_router.put(
    '/{user_id}/',
    response_model=Optional[UserRead],
    summary='Updates user data',
    description='Updates user data',
)
async def update_user(
        user_id: int,
        user_data: UserUpdate,
        db: AsyncSession = Depends(get_db),
        user: User = Depends(require_auth_user)
):
    users_service = UsersService(db)
    return await users_service.update(user_id, user_data, user)


@users_router.post(
    '/change-password/{user_id}/',
    response_model=Optional[UserRead],
    summary='Updates user data',
    description='Updates user data',
)
async def change_password(
        user_id: int,
        password_data: PasswordChangeRequest,
        db: AsyncSession = Depends(get_db),
        user: User = Depends(require_auth_user)
):
    users_service = UsersService(db)
    return await users_service.change_password(user_id, password_data, user)


@users_router.post(
    '/create/',
    response_model=Optional[UserRead],
    summary='Creates a new user',
    description='Creates a new user',
)
async def create_user(
        user: UserCreate,
        db: AsyncSession = Depends(get_db),
        current_user: User = Depends(require_admin)
):
    register_service = RegisterService(db)
    return await register_service.create_user(user)


@users_router.get(
    '/role/{role_name}/',
    response_model=List[UserRead],
    summary='Return all users with given role',
    description='Return all users with given role',
)
async def get_users_by_role_name(
        role_name: str,
        db: AsyncSession = Depends(get_db),
        current_user: User = Depends(require_auth_user),
):
    users_service = UsersService(db)
    return await users_service.get_by_role_name(role_name)


@users_router.get(
    '/check-my-lessons/{course_id}/',
    response_model=Optional[List[UserCheckLessonRead]],
    summary='Return all users with given lessons',
    description='Return all users with given lessons',
)
async def get_all_users_in_lessons(
        course_id: int,
        db: AsyncSession = Depends(get_db),
        user: User = Depends(require_auth_user)
):
    user_check_lessons_service = UserCheckLessonsService(db)
    return await user_check_lessons_service.get_by_course_id_and_user_id(course_id, user.id)


@users_router.post(
    '/check-lesson/{lesson_id}/',
    status_code=status.HTTP_201_CREATED,
    summary='Mark lesson as checked',
    description='Mark lesson as checked',
)
async def mark_lesson_checked(
        lesson_id: int,
        db: AsyncSession = Depends(get_db),
        user: User = Depends(require_auth_user)
):
    user_check_lessons_service = UserCheckLessonsService(db)
    await user_check_lessons_service.create(lesson_id, user.id)


@users_router.get(
    '/my-progress/{course_id}/',
    summary='Return user progress',
    description='Return user progress',
)
async def get_user_progress(
        course_id: int,
        db: AsyncSession = Depends(get_db),
        user: User = Depends(require_auth_user),
):
    user_check_lessons_service = UserCheckLessonsService(db)
    return await user_check_lessons_service.calculate_user_progress(course_id, user.id)