From d55c383bc4cbbe457790ef3482eb32893341ee28 Mon Sep 17 00:00:00 2001 From: Stefano Date: Thu, 15 May 2025 17:14:33 -0500 Subject: [PATCH] Added cognito authentication - Fix 10 --- app/main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/app/main.py b/app/main.py index 4d9ebee..b8daa08 100644 --- a/app/main.py +++ b/app/main.py @@ -1,5 +1,6 @@ from fastapi import FastAPI, Depends, HTTPException from fastapi.responses import JSONResponse, RedirectResponse +from requests import request from app.cabletv.utils.auth import exchange_code_for_token, get_current_user, DOMAIN, CLIENT_ID app = FastAPI() @@ -17,15 +18,14 @@ async def protected_route(user = Depends(get_current_user)): @app.get("/auth/callback") async def auth_callback(code: str): try: - tokens = exchange_code_for_token(code) + redirect_uri = str(request.base_url) + "auth/callback" + tokens = exchange_code_for_token(code, redirect_uri) - # Use id_token response = JSONResponse(content={ "message": "Authentication successful", "id_token": tokens["id_token"] # Changed from access_token }) - # Store id_token in cookie response.set_cookie( key="token", value=tokens["id_token"], # Changed from access_token