o i@sdZddlmZddlmZddlmZddlmZ ddl m Z ddl m Z ddlmZmZdd lmZdd lmZdd lmZ ddededeed BfddZddZdefddZ ddeefddZdefddZd S)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)FastMCP/Nserverpathrequired_scopescCst|}g}g}d|j|jdd|j|jdd|j|jdd}|dur-|t||n5|dkr;| t |||n'| t d|did || t d |d id || t d |d id ||j ||j |dS)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)toolresourcepromptNrr/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_configsr,y/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/contrib/component_manager/component_manager.pyset_up_component_managersR    r.csdtffdd }|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. requestc s|jddd}z|IdHtddd|iWSty@}z tddd|d |d}~ww) Nr:rmessagezd z: izUnknown ) status_codedetail) path_paramssplitr capitalizer StarletteHTTPException)r/nameeaction componentconfigr,r-endpoint\s zmake_endpoint..endpointr)r;r<r=r>r,r:r- make_endpointQs r?returncCst|||}|dur|dvrd|dd|}n%|dkr1|dur1|d|d|dd|}n d|d|dd|}t||dgd S) 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. N)rrrz/{rz}/rzs/{POST)r>methods)r?r )r;r<r=r root_pathr>rr,r,r- make_routems  rDc Cs:g}|D]}||}dD] }|t|||||q q|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%rDr+rCrcomponent_management_routesr<r=r;r,r,r-r$s r$c CsPg}|D]}||}dD] }|t|||||q qt|tt|d|dS)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. rE)r))app)r%rDr rrrFr,r,r-r&s  r&)rN)N)__doc__typingr&mcp.server.auth.middleware.bearer_authrstarlette.applicationsrstarlette.exceptionsrr7starlette.requestsrstarlette.responsesrstarlette.routingr r 3fastmcp.contrib.component_manager.component_servicer fastmcp.exceptionsr fastmcp.server.serverr strlistr.r?rDr$r&r,r,r,r-s4           =