from time import time
from typing import Optional
from datetime import timedelta

from pydantic import BaseModel, Field, SecretStr

from auth_service.models.proto import auth_pb2
from auth_service.models.exceptions import ServiceException, ServiceErrorCode


class AuthRecord(BaseModel):
    user_id: str
    password: SecretStr  # Hashed Password


class AuthToken(BaseModel):
    token_type: str = "Bearer"
    access_token: str
    expires_in: int = Field(default=int(timedelta(minutes=30).total_seconds()))
    # refresh_token: str  # todo: implement functionality in the future

    @classmethod
    def to_protobuf(cls, auth_token: "AuthToken") -> auth_pb2.AuthToken:
        return auth_pb2.AuthToken(**auth_token.model_dump())


class AccessTokenClaims(BaseModel):
    sub: str
    iat: int = Field(default_factory=lambda: int(time()))
    exp: int = Field(default_factory=lambda: int(time() + timedelta(minutes=30).total_seconds()))


class AuthRepository:
    """Abstract repository interface"""
    async def auth_with_password(self, user_id: str, password: str) -> AuthToken:
        raise ServiceException("unimplemented internal repository method", ServiceErrorCode.SERVICE_ERROR)
    
    async def set_password_auth_method(self, user_id: str, password: str) -> None:
        raise ServiceException("unimplemented internal repository method", ServiceErrorCode.SERVICE_ERROR)

    async def delete_password_auth_method(self, user_id: str) -> None:
        raise ServiceException("unimplemented internal repository method", ServiceErrorCode.SERVICE_ERROR)


class AuthDBRepository(AuthRepository):
    """Abstract database repository interface"""
    async def get_auth_record(self, user_id: str) -> Optional[AuthRecord]:
        raise ServiceException("unimplemented internal repository method", ServiceErrorCode.SERVICE_ERROR)

    async def create_password_auth_method(self, user_id: str, password: str) -> None:
        raise ServiceException("unimplemented internal repository method", ServiceErrorCode.SERVICE_ERROR)
    
    async def update_password_auth_method(self, user_id: str, password: str) -> None:
        raise ServiceException("unimplemented internal repository method", ServiceErrorCode.SERVICE_ERROR)