from __future__ import annotations

from typing import TYPE_CHECKING

from fastapi import Depends
from sqlalchemy import ForeignKey, Integer, String
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.future import select
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.utils.db import Base, get_async_session

if TYPE_CHECKING:
    from .user import (
        User,
    )


class Profile(Base):
    __tablename__ = "profiles"

    id: Mapped[int] = mapped_column(Integer, primary_key=True)
    user_id: Mapped[int] = mapped_column(
        ForeignKey("users.id"), unique=True, nullable=False
    )

    avatar_file: Mapped[str | None] = mapped_column(String(255), nullable=True)
    banner_file: Mapped[str | None] = mapped_column(String(255), nullable=True)
    description: Mapped[str | None] = mapped_column(String(500), nullable=True)

    publications_count: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
    collections_count: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
    subscriptions_count: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
    followers_count: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
    following_count: Mapped[int] = mapped_column(Integer, default=0, nullable=False)

    user: Mapped["User"] = relationship(
        "User", back_populates="profile", uselist=False, lazy="selectin"
    )

    @classmethod
    async def get_by_user_id(
        cls, user_id: int, session: AsyncSession = Depends(get_async_session)
    ):
        result = await session.execute(select(cls).where(cls.user_id == user_id))
        return result.scalars().first()

    @classmethod
    async def get_by_id(
        cls, profile_id: int, session: AsyncSession = Depends(get_async_session)
    ):
        result = await session.execute(select(cls).where(cls.id == profile_id))
        return result.scalars().first()