KizdZddlmZddlmZddlZddlmZddlm Z ddl m Z ddl m Z mZdd lmZdd lmZdd lmZmZmZmZmZmZmZmZeeZ d# d$d ZeGd dZeGddZ d% d&dZ!edk(r ddl"Z"ddl Z eZ#e$de$de$de#de$de#de$de#de$de$e!e#dZ%e"jLde#de jNe%jPjRde#d d!"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. ) annotations) dataclassN) Starlette)Request)Route)ConfigServer)find_available_port) get_logger)HELPER_TEXT_STYLESINFO_BOX_STYLESSTATUS_MESSAGE_STYLEScreate_info_box create_logo create_pagecreate_secure_html_responsecreate_status_messagec|rdnd}d}|r|rtd|dd}n|st|ddd}dtd t|| d |d }ttzt z}t ||| S) z2Create a styled HTML response for OAuth callbacks.zAuthentication successfulzAuthentication failedzConnected to: T)centered monospace)is_errorrrz-
z  is_successz
You can safely close this tab now.
)contenttitleadditional_styles)rrrrr r r)messagerr server_url status_title detail_inforrs i/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/client/oauth_callback.pycreate_callback_htmlr#"s(2#7N Kj%ZL )DD % dTT  ]O "d }" t |_jtt|ddS|j>d }" t |_jtt|ddS42|j_ |j_ jttd d  Sw)z:Handle OAuth callback requests with proper HTML responses.z Unknown error access_deniedz.Access was denied by the authorization server.zAuthorization failed: Fri) status_codez3No authorization code was received from the server.z=The OAuth server did not return the expected state parameter.rT)rr) dict query_paramsr&r4r*r+ RuntimeErrorsetrr#r(r))requestrHcallback_response error_desc user_messageresult_container result_readyrs r"callback_handlerz6create_oauth_callback_server..callback_handler|sG001 ,66|D  " "*<<OJ!&&/9O !7 |D  + 0H)5l)C &  ".$ $  !%%PL + 0H)5l)C &  ".$ $    " " *O   + 0H)5l)C &  ".$ $    'L,D$5$:$:  !%6%<%<  "    *  L  sE;E>)routes 127.0.0.1offwarningzwebsockets-sansio)apphostportlifespan log_levelws)rKr)rrr r)rX callback_pathrrOrPrQrVs ``` r"create_oauth_callback_serverr]gsH*F P 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)rXrzhttp://localhost:rSrUF)rWrXrZ access_log)Tz FastMCP OAuthN) rstrrboolrr`rr'r:r`)rBNNN) rXintr\r`rr'rOzOAuthCallbackResult | NonerPzanyio.Event | Noner:r )*rA __future__r dataclassesranyiostarlette.applicationsrstarlette.requestsrstarlette.routingruvicornrr fastmcp.utilities.httpr fastmcp.utilities.loggingr fastmcp.utilities.uir r rrrrrrr;loggerr#r&r@r] webbrowserrXprintserveropenrunconfigrVr-r$r"rts#! ,&#"60    H   ! * ** * *  *Z  I I  I ## #%!37'+ h hhh1 h % h  hV z3  D +,  (.N OP &tf,ij (i 89 %& G* GF JOO'v-MNOGKK    5r$