KidZddlZddlZddlZddlmZddlmZddlZddl m Z ddl m Z ddl mZdd lmZe eZd edzfd Zd efd Zddddddd dededzdedeedzdeedzdeeefdzdedzdedzdedzd efdZddddddddddedeedzej4ddgdfdeeedzej4dd d!"fdeeedzej4d#d$d!"fdeeedzej4d%d&d!"fd'eedzej4d(d)*fd+eedzej4d,d-*fdeedzej4d.d/*fdeedzej4d0d1*fd dfd2Zy)3z:Gemini CLI integration for FastMCP install using Cyclopts.N)Path) Annotated)print) get_logger) UVEnvironment)process_common_argsreturnctjd}|r tj|dgdd|St jdz dz dz t dt jdz d z dz t d g}|D]C}|js tjt|dgddt|cSy #tjt f$rYwxYw#tjt f$rYwxYw) 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_pathspaths j/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/cli/install/gemini_cli.pyfind_gemini_commandrs\\(+N  NN-#  " ! i')H4 $% m#e+h6 '( O  ;;= Y ,#' 4y   9--/@A   2113DE  s#C -C+ C('C(+DDctduS)z!Check if Gemini CLI is available.N)rrcheck_gemini_cli_availabler"@s   ,,r!) with_editable with_packagesenv_varspython_versionwith_requirementsprojectfile server_objectnamer#r$r%r&r'r(c t} | s tdyt||xsgdgz|||} |r|jd|} nt |j} | j dd| g} | ddg} |r0|j D]\}}| jd |d |g| j|| d d g| j| d d tj| dddy#tj$rK}td|d|jr|jjn t |dYd}~yd}~wt$r}td|d|dYd}~yd}~wwxYw)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 rtextz[red]Failed to install '[bold]z[/bold]' in Gemini CLI: z[/red])rrrresolver build_commanditemsextendrrrstderrstrip Exception)r)r*r+r#r$r%r&r'r( gemini_cmd env_config server_spec full_command cmd_partskeyvaluees rinstall_gemini_clirFEs:%&J   [ #)ri[8& J(-9 $,,.) ++Y{,KLLUE*I"..* 7JC   dse1UG$45 6 7dLOT23 \!"%& yTM  ( ( ,TF2J_`_g_g188>>K[mpqrmsJttz {   .tf4LQCvVWs%C33E4AE  E4E//E4) server_namer#r$r%env_filer.r'r(r@rGz--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))rInegativez--withz;Additional packages to install (can be used multiple times)z--envzFEnvironment variables in KEY=VALUE format (can be used multiple times)rHz --env-filez)Load environment variables from .env file)rIr.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| | | || | ||| }|rtd| dytjdy7Fw)zInstall an MCP server in Gemini CLI. Args: server_spec: Python file to install, optionally with :object suffix N) r)r*r+r#r$r%r&r'r(z[green]Successfully installed 'z' in Gemini CLIr)r rFrsysexit)r@rGr#r$r%rHr.r'r(r)r*r+packagesenv_dictsuccesss rgemini_cli_commandrRsJ"'RM!'RM~2H:M[-8;51D-x! # #+ G /v_EF  '5s%A0A.AA0)__doc__rrrMpathlibrtypingrcycloptsrichrfastmcp.utilities.loggingr6fastmcp.utilities.mcp_server_config.v1.environments.uvrsharedr __name__loggerrrboolr"listdictrF ParameterrRr r!rras@  0P' H )S4Z)X-D-(,&*&*!%%)N N:N N :$ N 9t# N38nt#N$JNd{ND[N Nt        {[[ d D!;  [ T T i  [$ S D N  %[4 S D Y  5[D t  <  E[R  d  ;   S[`! t  !A  a[n t  E  o[| }[r!