o i@sdZddlmZddlmZddlZddlmZddlm Z ddl m Z ddl m Z mZdd lmZdd lmZdd lmZmZmZmZmZmZmZmZeeZ d7d8ddZeGdddZeGdddZ    d9d:d%d&Z!ed'kr ddl"Z"ddl Z eZ#e$d(e$d)e$d*e#d+e$d,e#d-e$d.e#de$d/e$e!e#d0d1Z%e"&d2e#d+e j'e%j(j)d3e#d4d5d6dSdS);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_messageT FastMCP OAuthmessagestr is_successbooltitle server_url str | NonereturncCs~|rdnd}d}|r|rtd|ddd}n |s!t|dddd}dtd t||d d |d }ttt}t|||d S) z2Create a styled HTML response for OAuth callbacks.zAuthentication successfulzAuthentication failedzConnected to: T)centered monospace)is_errorrrz-
z rz
You can safely close this tab now.
)contentradditional_styles)rrrr r r r)rrrr status_title detail_infor!r"r%c/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/client/oauth_callback.pycreate_callback_html"s0    r'c@sVeZdZUdZded<dZded<dZded<dZded<edd d Z dd d Z dS)CallbackResponseNrcodestateerrorerror_descriptiondatadict[str, str]rcs difdd|DS)Ncs i|] \}}|jvr||qSr%)__annotations__.0kvclsr%r& Xs z.CallbackResponse.from_dict..r%)items)r5r-r%r4r& from_dictVs zCallbackResponse.from_dictcCsdd|jDS)NcSsi|] \}}|dur||qS)Nr%r0r%r%r&r6[sz,CallbackResponse.to_dict..)__dict__r7)selfr%r%r&to_dictZszCallbackResponse.to_dict)r-r.rr()rr.) __name__ __module__ __qualname__r)r/r*r+r, classmethodr8r;r%r%r%r&r(Os     r(c@s6eZdZUdZdZded<dZded<dZded<dS)OAuthCallbackResultzSContainer for OAuth callback results, used with anyio.Event for async coordination.Nrr)r*zException | Noner+)r<r=r>__doc__r)r/r*r+r%r%r%r&r@^s   r@ /callbackportint callback_pathresult_containerOAuthCallbackResult | None result_readyanyio.Event | Nonerc s<d fdd }tt||gd}tt|d|ddd d S) a Create an OAuth callback server. Args: port: The port to run the server on callback_path: The path to listen for OAuth redirects on server_url: Optional server URL to display in success messages result_container: Optional container to store callback results result_ready: Optional event to signal when callback is received Returns: Configured uvicorn Server instance (not yet running) requestrcs4t|j}t|}|jr;|jpd}|jdkrd}nd|}dur1dur1t|_tt |dddd S|j s[d }durQdurQt|_tt |dddd S|j dur}d }dursdurst|_tt |dddd Sdurdur|j _ |j _ tt d d dS)z:Handle OAuth callback requests with proper HTML responses.z Unknown error access_deniedz.Access was denied by the authorization server.zAuthorization failed: NFr i) 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(r8r+r, RuntimeErrorsetrr'r)r*)rJrNcallback_response error_desc user_messagerFrHrr%r&callback_handler|sd           z6create_oauth_callback_server..callback_handler)routes 127.0.0.1offwarningzwebsockets-sansio)apphostrClifespan log_levelwsN)rJr)rrrr)rCrErrFrHrUrZr%rTr&create_oauth_callback_servergsHr___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)rCrzhttp://localhost:rWrYF)r[rCr] access_log)TrN) rrrrrrrrrr)rBNNN) rCrDrErrrrFrGrHrIrr)*rA __future__r dataclassesranyiostarlette.applicationsrstarlette.requestsrstarlette.routingruvicornrrfastmcp.utilities.httpr fastmcp.utilities.loggingr fastmcp.utilities.uir r r rrrrrr<loggerr'r(r@r_ webbrowserrCprintserveropenrunconfigrZr%r%r%r&sf       (  -  k