from loguru import logger
from typing import Type, TypeVar
from asyncpg.connection import Connection


class BaseRepo:
    def __init__(self, conn: Connection):
        self.conn = conn


T = TypeVar("T", bound=BaseRepo)


class Repos(BaseRepo):
    def get_repo(self, Repo: Type[T]) -> T:
        return Repo(self.conn)


class UserRepo(BaseRepo):
    async def add(self, user_id: int, chat_id: int):
        await self.conn.execute(
            "INSERT INTO users (user_id, chat_id) "
            "VALUES ($1, $2) ON CONFLICT (user_id) DO NOTHING",
            user_id,
            chat_id,
        )

    async def get(self, user_id: int):
        return await self.conn.fetchval(
            "SELECT user_id FROM users WHERE user_id = $1", user_id
        )


class ProjectRepo(BaseRepo):
    async def add(self, name: str, description: str, *args):
        await self.conn.execute(
            "INSERT INTO projects "
            "(name, description, contributors, status) "
            "VALUES ($1, $2, $3, $4)",
            name,
            description,
            *args
        )

    async def get(self, user_id: int):
        return await self.conn.fetchval(
            "SELECT name, description FROM projects WHERE creator = $1", user_id
        )