import os import re from starlette.endpoints import HTTPEndpoint from starlette.exceptions import HTTPException from starlette.requests import Request from starlette.responses import JSONResponse from orm.author import Author from services.db import local_session class WebhookEndpoint(HTTPEndpoint): async def post(self, request: Request) -> JSONResponse: try: data = await request.json() if not data: raise HTTPException(status_code=400, detail="Request body is empty") auth = request.headers.get("Authorization") if not auth or auth != os.environ.get("WEBHOOK_SECRET"): raise HTTPException( status_code=401, detail="Invalid Authorization header" ) # logger.debug(data) user = data.get("user") if not isinstance(user, dict): raise HTTPException( status_code=400, detail="User data is not a dictionary" ) # name: str = ( f"{user.get('given_name', user.get('slug'))} {user.get('middle_name', '')}" + f"{user.get('family_name', '')}".strip() ) or "Аноним" email: str = user.get("email", "") pic: str = user.get("picture", "") with local_session() as session: author = session.query(Author).filter(Author.email == email).first() if not author: user_id: str = user.get("id", "") # If the author does not exist, create a new one slug: str = email.split("@")[0].replace(".", "-").lower() slug: str = re.sub("[^0-9a-z]+", "-", slug) while True: author = ( session.query(Author).filter(Author.slug == slug).first() ) if not author: break slug = f"{slug}-{len(session.query(Author).filter(Author.email == email).all()) + 1}" author = Author(user=user_id, slug=slug, name=name, pic=pic) session.add(author) session.commit() return JSONResponse({"status": "success"}) except HTTPException as e: return JSONResponse( {"status": "error", "message": str(e.detail)}, status_code=e.status_code ) except Exception as e: import traceback traceback.print_exc() return JSONResponse({"status": "error", "message": str(e)}, status_code=500)