~hW':dZddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZmZdd lmZdd lmZeeZ d! d"d ZeGd dZ d# d$dZedk(r ddlZddlZeZedededededededededeeedZej<dedej>ej@jBdedd yy)%z OAuth callback server for handling authorization code flows. This module provides a reusable callback server that can handle OAuth redirects and display styled responses to users. ) annotationsN) dataclass) Starlette)Request) HTMLResponse)Route)ConfigServer)find_available_port) get_loggerc d|rdnd}|rdnd}d}|r|rd|d}d|d |d |rd nd d |d|d|d S)z2Create a styled HTML response for OAuth callbacks.u✅u❌z#10b981z#ef4444zM
Connected to: z%
z a ��
z)
z
z
You can safely close this tab now.
)message is_successtitle server_url status_emoji status_color server_infos Y/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/fastmcp/client/oauth_callback.pycreate_callback_htmlrs'5EL *9 LKj''1l3   wBD%~&,%-,1'b(4n5"")+ MAG Gc`eZdZUdZded<dZded<dZded<dZded<ed dZ d dZ y) CallbackResponseN str | Nonecodestateerrorerror_descriptionc |di|jDcic]\}}||jvs||c}}Scc}}w)Nr)items__annotations__)clsdatakvs r from_dictzCallbackResponse.from_dicts9Qtzz|Ptq!qCF)ri) status_codez)OAuth callback missing authorization codez3FastMCP OAuth Error: No authorization code receivedzFastMCP OAuth login complete!)r)dict query_paramsrr(rr done set_exception RuntimeErrorrrr set_resultr)requestr8callback_response error_descresponse_futurers rcallback_handlerz6create_oauth_callback_server..callback_handlersUG001 ,66|D  " "*<<OJ';';'=-- '(9(?(?'@J<P  $+,=,C,C+DD U$  !%%';';'=-- !LM $I$   ?#7#7#9  & &"''):)@)@A  !@Z X  sE E )routes 127.0.0.1offwarning)apphostportlifespan log_level)r=r)rrr r )rH callback_pathrr@rArFs `` rcreate_oauth_callback_serverrLsD&/ b E-1ABC DC    r__main__u🎭 OAuth Callback Test Serveru📍 Test URLs:z Success: http://localhost:z /callback?code=test123&state=xyzz Error: http://localhost:z=/callback?error=access_denied&error_description=User%20deniedz Missing: http://localhost:u🛑 Press Ctrl+C to stopz'https://fastmcp-test-server.example.com)rHrzhttp://localhost:rCrEF)rGrHrJ access_log)Tz FastMCP OAuthN) rstrrboolrrOrrr.rO)r3NN) rHintrKrOrrr@zasyncio.Future | Noner.r )"__doc__ __future__rasyncio dataclassesrstarlette.applicationsrstarlette.requestsrstarlette.responsesrstarlette.routingruvicornr r fastmcp.utilities.httpr fastmcp.utilities.loggingr r/loggerrrrL webbrowserrHprintserveropenrunconfigrFrrrrds#!,&,#"60 H   ! Z ZZ Z Z  Zz  I I  I %!-1 N NNN+ N  Nb z3  D +,  (.N OP &tf,ij (i 89 %& G* GF JOO'v-MNOGKK    5r