from typing import overload

from sqlalchemy import Select
from sqlalchemy.orm import selectinload

from models.project import Project
from repositories.base import BaseRepository
from repositories.mixins import (
    RepDeleteMixin,
    RepCreateMixin,
    GetByIdMixin,
    GetAllMixin,
)
from schemas.project import CreateProjectSchema, UpdateProjectSchema


class ProjectRepository(
    BaseRepository,
    GetAllMixin[Project],
    RepDeleteMixin[Project],
    RepCreateMixin[Project, CreateProjectSchema],
    GetByIdMixin[Project],
):
    entity_class = Project

    def _process_get_all_stmt(self, stmt: Select) -> Select:
        return stmt.order_by(Project.id)

    def _process_get_by_id_stmt(self, stmt: Select) -> Select:
        return stmt.options(selectinload(Project.boards))

    async def update(self, project: Project, data: UpdateProjectSchema) -> Project:
        project.name = data.name if data.name else project.name

        self.session.add(project)
        await self.session.commit()
        await self.session.refresh(project)
        return project