KidZddlmZddlmZddlmZddlmZ ddl m Z ddl m Z ddlmZmZdd lmZdd lmZdd lmZ dd ededeed zfdZdZdefdZ ddeefdZdefdZy )z Routes and helpers for managing tools, resources, and prompts in FastMCP. Provides endpoints for enabling/disabling components via HTTP, with optional authentication scopes. )Any)RequireAuthMiddleware) Starlette) HTTPExceptionRequest) JSONResponse)MountRoute)ComponentService) NotFoundError)FastMCPNserverpathrequired_scopescpt|}g}g}d|j|jdd|j|jdd|j |j dd}||jt||n|dk7r|jt|||nc|jtd|did ||jtd |d id ||jtd |d id ||jj||jj|y)aASet up routes for enabling/disabling tools, resources, and prompts. Args: server: The FastMCP server instance path: Path used to mount all component-related routes on the server required_scopes: Optional list of scopes required for these routes. Applies only if authentication is enabled. tool_name)paramenabledisablezuri:path prompt_name)toolresourcepromptN/r/toolsr /resourcesr/prompts) r _enable_tool _disable_tool_enable_resource_disable_resource_enable_prompt_disable_promptextend!build_component_manager_endpointsappendbuild_component_manager_mount_additional_http_routes)rrrserviceroutesmounts route_configss /home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/contrib/component_manager/component_manager.pyset_up_component_managerr/sOv&GFF!**,,  ..00 #,,.. M$ 7 tLM 3; MM-mT?S  MM-]623X  MM-z!:; #  MM-}X67_   ""))&1 ""))&1c(dtffd }|S)ad Factory for creating Starlette endpoint functions for enabling/disabling a component. Args: action: 'enable' or 'disable' component: The component type (e.g., 'tool', 'resource', or 'prompt') config: Dict with param and handler functions for the component Returns: An async endpoint function for Starlette. requestcK|jdjdd} |d{tdjdd|iS7'#t$r}t ddd| |d}~wwxYww) Nr:rmessagezd z: izUnknown ) status_codedetail) path_paramssplitr capitalizer StarletteHTTPException)r2nameeaction componentconfigs r.endpointzmake_endpoint..endpoint\s""6'?#8#8#=a#@A  &.& & &v00232i[4&IJ  ' (!)Btf5  s8%BA!A&A!BA!! B*A>>BBr)r>r?r@rAs``` r. make_endpointrBQs   Or0returnct|||}||dvr d|dd|}n&|dk7r||d|d|dd|}nd|d|dd|}t||dgS) ap Creates a Starlette Route for enabling/disabling a component. Args: action: 'enable' or 'disable' component: The component type config: Dict with param and handler functions required_scopes: Optional list of required auth scopes root_path: The base path for the route Returns: A Starlette Route object. )rrrz/{rz}/rzs/{POST)rAmethods)rBr )r>r?r@r root_pathrArs r. make_routerHmsVY7H"y5( VG_%S1   7[)D0AVHMDykfWo%6c&BD 6( ;;r0c lg}|D],}||}dD] }|jt|||||".|S)a_ Build a list of Starlette Route objects for all components/actions. Args: route_configs: Dict describing component types and their handlers root_path: The base path for the routes required_scopes: Optional list of required auth scopes Returns: List of Starlette Route objects for component management. rr)r'rHr-rGrcomponent_management_routesr?r@r>s r.r&r&sX02" !.y!9+ F ' . .69foyQ   '&r0c g}|D],}||}dD] }|jt|||||".t|tt ||S)a] Build a Starlette Mount with authentication for component management routes. Args: route_configs: Dict describing component types and their handlers root_path: The base path for the mount required_scopes: List of required auth scopes Returns: A Starlette Mount object with authentication middleware. rJ)r+)app)r'rHr rrrKs r.r(r(sw02" !.y!9+ F ' . .69foyQ    + ! 8 9?  r0)rN)N)__doc__typingr&mcp.server.auth.middleware.bearer_authrstarlette.applicationsrstarlette.exceptionsrr;starlette.requestsrstarlette.responsesr starlette.routingr r 3fastmcp.contrib.component_manager.component_servicer fastmcp.exceptionsr fastmcp.server.serverrstrlistr/rBrHr&r(r0r.r]s H,H&,*P,)KO:2 :2:27;Cy47G:2z8<<</3' %['0PUr0