Home/iptv-manager-service
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added cognito authentication - Fix 10
All checks were successful
AWS Deploy on Push / build (push) Successful in 1m9s
Details

Browse Source
This commit is contained in:
Stefano Fiorini
2025-05-15 17:14:33 -05:00
parent 5c17e4b1e9
commit d55c383bc4
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app/main.py
View File

@@ -1,5 +1,6 @@
from fastapi import FastAPI, Depends, HTTPException from fastapi import FastAPI, Depends, HTTPException
from fastapi.responses import JSONResponse, RedirectResponse 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 from app.cabletv.utils.auth import exchange_code_for_token, get_current_user, DOMAIN, CLIENT_ID
app = FastAPI() app = FastAPI()
@@ -17,15 +18,14 @@ async def protected_route(user = Depends(get_current_user)):
@app.get("/auth/callback") @app.get("/auth/callback")
async def auth_callback(code: str): async def auth_callback(code: str):
try: 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={ response = JSONResponse(content={
"message": "Authentication successful", "message": "Authentication successful",
"id_token": tokens["id_token"] # Changed from access_token "id_token": tokens["id_token"] # Changed from access_token
}) })
# Store id_token in cookie
response.set_cookie( response.set_cookie(
key="token", key="token",
value=tokens["id_token"], # Changed from access_token value=tokens["id_token"], # Changed from access_token