mirror of
https://github.com/hexolan/panels.git
synced 2026-03-26 12:40:21 +00:00
55 lines
2.2 KiB
Python
55 lines
2.2 KiB
Python
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)
|
|
|