KidZddlZddlZddlmZddlmZddlZddlm Z ddl m Z m Z ddl mZddlmZd d lmZeeZd edzfd Zddddddd dededzdedeedzdeedzdeeefdzdedzdedzdedzd efdZddddddddddedeedzej6ddgdfdeeedzej6dd d!"fdeeedzej6d#d$d!"fdeeedzej6d%d&d!"fd'eedzej6d(d)*fd+eedzej6d,d-*fdeedzej6d.d/*fdeedzej6d0d1*fd dfd2Zy)3z>Claude Desktop integration for FastMCP install using Cyclopts.N)Path) Annotated)print)StdioMCPServerupdate_config_file) get_logger) UVEnvironment)process_common_argsreturnctjdk(r!ttjddd}ntjdk(r!ttjddd}n`tjj dr@tt j jd tjd z d}ny |jr|Sy ) z2Get the Claude config directory based on platform.win32AppDataRoamingClaudedarwinLibraryzApplication SupportlinuxXDG_CONFIG_HOMEz.configN) sysplatformrhome startswithosenvirongetexists)paths n/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/cli/install/claude_desktop.pyget_claude_config_pathr s ||wDIIKIx@  !DIIK,A8L   ) JJNN,diikI.E F  {{} ) with_editable with_packagesenv_varspython_versionwith_requirementsprojectfile server_objectnamer"r#r$r%r&r'ct} | s tdy| dz } t||xsgdgz|||} |r|jd|} nt |j} | j dd| g} t | d| d d |xsi } | jrjdd l}| jj}|rF|j|}d |vr1||d vr*|d |jd i}|ri||}n|}||_ t| ||td|dy#t$r}td|dYd }~yd }~wwxYw)aInstall FastMCP server in Claude Desktop. Args: file: Path to the server file server_object: Optional server object name (for :object suffix) name: Name for the server in Claude's config with_editable: Optional list of directories to install in editable mode with_packages: Optional list of additional packages to install env_vars: Optional dictionary of environment variables python_version: Optional Python version to use with_requirements: Optional requirements file to install from project: Optional project directory to run within Returns: True if installation was successful, False otherwise z[red]Claude Desktop config directory not found.[/red] [blue]Please ensure Claude Desktop is installed and has been run at least once to initialize its config.[/blue]Fzclaude_desktop_config.jsonfastmcp)python dependencies requirementsr'editable:runrr N)commandargsenv mcpServersr5z[green]Successfully installed 'z' in Claude Desktop[/green]Tz[red]Failed to install server: z[/red])r rr resolvestr build_commandrrjson read_textstriploadsrr5r Exception)r(r)r*r"r#r$r%r&r' config_dir config_file env_config server_spec full_command server_configr:contentconfig existing_env merged_enves rinstall_claude_desktoprJ&s8()J   ~ ;;K#)ri[8& J(-9 $,,.) ++Y{,KLL#Q !"  NM      !++-335GG,6)df\6J.J#),#7#=#A#A%#LL%A %A%A %1 (2M% ;m< /v5PQR  /s&9:sBD++ E 4EE ) server_namer"r#r$env_filer-r&r'rBrKz--namez-nz5Custom name for the server in Claude Desktop's config)r*helpz--with-editablezVDirectory with pyproject.toml to install in editable mode (can be used multiple times))rMnegativez--withz;Additional packages to install (can be used multiple times)z--envzFEnvironment variables in KEY=VALUE format (can be used multiple times)rLz --env-filez)Load environment variables from .env file)rMr-z--pythonz(Python version to use (e.g., 3.10, 3.11)z--with-requirementsz.Requirements file to install dependencies fromz --projectz2Run the command within the given project directoryc K|xsg}|xsg}|xsg}t|||||d{\} } } }} t| | | ||| ||| } | stjdyy77w)zInstall an MCP server in Claude Desktop. Args: server_spec: Python file to install, optionally with :object suffix N) r(r)r*r"r#r$r%r&r'r )r rJrexit)rBrKr"r#r$rLr-r&r'r(r)r*env_dictsuccesss rclaude_desktop_commandrT}sJ"'RM!'RM~2H?R[-8@:6D-}h% # ##+ G    !:s%A!A8A!)__doc__rrpathlibrtypingrcycloptsrichrfastmcp.mcp_configrrfastmcp.utilities.loggingr6fastmcp.utilities.mcp_server_config.v1.environments.uvr sharedr __name__loggerr r8listdictboolrJ ParameterrTr!rresD A0P' H t .(,&*&*!%%)T T:T T :$ T 9t# T38nt#T$JTd{TD[T T@        {YY d D!H  Y T T i  Y$ S D N  %Y4 S D Y  5YD t  <  EYR  d  ;   SY`! t  !A  aYn t  E  oY| }Yr!