kKjcdZddlmZddlmZddlZddlmZddlm 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 n/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/client/oauth_callback.pycreate_callback_htmlr#"s(2#7N Kj%ZL )DD % dTT  ]O ".store_result_once}sE  #|';|?R?R?T  $!&!&r$cdKt|j}tj|}|jrQ|j xsd}|jdk(rd}nd|}t |tt|dd S|js,d }t |tt|dd S|j,d }t |tt|dd S|j|j ttd dSw)z:Handle OAuth callback requests with proper HTML responses.z Unknown error access_deniedz.Access was denied by the authorization server.zAuthorization failed: )r*Fri) status_codez3No authorization code was received from the server.z=The OAuth server did not return the expected state parameter.)r(r)rT)rr) dict query_paramsr&r4r*r+ RuntimeErrorrr#r(r))requestrPcallback_response error_desc user_messagerrKs r"callback_handlerz6create_oauth_callback_server..callback_handlersLG001 ,66|D  " "*<<OJ!&&/9O !7 |D  L$> ?.$ $  !%%PL L$> ?.$ $    " " *O  L$> ?.$ $   "''#)) +  L  sD-D0)routesoffwarningzwebsockets-sansio)apphostportlifespan log_levelws)r(r'r)r'r*rAr:None)rRr)rrr r) r\r[ callback_pathrrIrJrVrZrKs ``` @r"create_oauth_callback_serverrbgs~0  "&      @ D 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)r\rzhttp://localhost:rYF)r[r\r^ access_log)Tz FastMCP OAuthN) rstrrboolrrerr'r:re)rCrDNNN)r\intr[rerarerr'rIzOAuthCallbackResult | NonerJzanyio.Event | Noner:r )*rB __future__r dataclassesranyiostarlette.applicationsrstarlette.requestsrstarlette.routingruvicornrr fastmcp.utilities.httpr fastmcp.utilities.loggingr fastmcp.utilities.uir r rrrrrrr;loggerr#r&r@rb webbrowserr\printserveropenrunconfigrZr-r$r"rys#! ,&#"60    H   ! * ** * *  *Z  I I  I ## #$!37'+ r r rr r 1 r % r rj z3  D +,  (.N OP &tf,ij (i 89 %& G* GF JOO'v-MNOGKK    5r$