o i@sdZddlZddlZddlZddlmZddlmZddlZddl m Z ddl m Z ddl mZdd lmZe eZd edBfd d Zd efd dZddddddddededBdedeedBdeedBdeeefdBdedBdedBdedBd efddZddddddddddedeedBejddgd d!fdeeedBejd"d#d$d%fdeeedBejd&d'd$d%fdeeedBejd(d)d$d%fd*eedBejd+d,d-fd.eedBejd/d0d-fdeedBejd1d2d-fdeedBejd3d4d-fd dfd5d6ZdS)7z:Gemini CLI integration for FastMCP install using Cyclopts.N)Path) Annotated)print) get_logger) UVEnvironment)process_common_argsreturnc Cstd}|r!z tj|dgddd|WStjtfy Ynwtdddtdtdd dtd g}|D]&}|rcztjt |dgdddt |WStjtfybYq=wq=d S) zFind the Gemini CLI command.geminiz --versionT)checkcapture_outputz.geminilocalz/usr/local/bin/geminiz .npm-globalbinz/opt/homebrew/bin/geminiN) shutilwhich subprocessrunCalledProcessErrorFileNotFoundErrorrhomeexistsstr)gemini_in_pathpotential_pathspathrd/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/cli/install/gemini_cli.pyfind_gemini_commands>    rcCs tduS)z!Check if Gemini CLI is available.N)rrrrrcheck_gemini_cli_available@s r) with_editable with_packagesenv_varspython_versionwith_requirementsprojectfile server_objectnamerr r!r"r#r$c Csjt} | s tddSt||pgdg|||d} |r%|d|} nt|} | dd| g} | ddg} |rO|D]\}}| d |d |gq>| || d d g| | d dz tj | ddddWdStj y}ztd|d|j r|j nt|dWYd}~dSd}~wt y}ztd|d|dWYd}~dSd}~ww)aInstall FastMCP server in Gemini CLI. Args: file: Path to the server file server_object: Optional server object name (for :object suffix) name: Name for the server in Gemini CLI 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]Gemini CLI not found.[/red] [blue]Please ensure Gemini CLI is installed. Try running 'gemini --version' to verify.[/blue] [blue]You can install it using 'npm install -g @google/gemini-cli'.[/blue] Ffastmcp)python dependencies requirementsr$editable:rmcpaddz-e=rz--rNT)r r textz[red]Failed to install '[bold]z[/bold]' in Gemini CLI: z[/red])rrrresolver build_commanditemsextendrrrstderrstrip Exception)r%r&r'rr r!r"r#r$ gemini_cmd env_config server_spec full_command cmd_partskeyvalueerrrinstall_gemini_cliEsJ    $rA) server_namerr r!env_filer)r#r$r;rBz--namez-nz(Custom name for the server in Gemini CLI)r'helpz--with-editablezVDirectory with pyproject.toml to install in editable mode (can be used multiple times))rDnegativez--withz;Additional packages to install (can be used multiple times)z--envzFEnvironment variables in KEY=VALUE format (can be used multiple times)rCz --env-filez)Load environment variables from .env file)rDr)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 sz|pg}|pg}|p g}t|||||IdH\} } } } } t| | | || | |||d }|r6td| ddStddS)zInstall an MCP server in Gemini CLI. Args: server_spec: Python file to install, optionally with :object suffix N) r%r&r'rr r!r"r#r$z[green]Successfully installed 'z' in Gemini CLIr)rrArsysexit)r;rBrr r!rCr)r#r$r%r&r'packagesenv_dictsuccessrrrgemini_cli_commands*E  rL)__doc__rrrGpathlibrtypingrcycloptsrichrfastmcp.utilities.loggingr6fastmcp.utilities.mcp_server_config.v1.environments.uvrsharedr__name__loggerrrboolrlistdictrA ParameterrLrrrrs      ,      Z   ")07>