o iN@sdZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z m Z mZddlmZer>ddlmZdd lmZmZd d d ed efddZddded ed eefddZdedded eefddZdddeded eefddZ    d=d eeed!fe fd"eed!fd#ed ed$eed!fd%eed!fd eefd&d'Zd(ed efd)d*Z d+ed efd,d-Z!ddded eefd.d/Z"ddded efd0d1Z#d2eed!fd3d ded eefd4d5Z$d6e d efd7d8Z%ddded efd9d:Z&d3d ded efd;d<Z'dS)>zZsh completion script generator. Generates static zsh completion scripts using the compsys framework. No runtime Python dependency. N)dedent)indent) TYPE_CHECKING)CompletionActionCompletionDataclean_choice_textextract_completion_dataget_completion_action strip_markup)docstring_parse)App)ArgumentArgumentCollectionappr prog_namereturnc Cs|rtd|std|dt|}d|dd|dddg}|t|d ||jr0t|jnd |jr9t|jnd d |d dgd |d S) aGenerate zsh completion script. Parameters ---------- app : App The Cyclopts application to generate completion for. prog_name : str Program name (alphanumeric with hyphens/underscores). Returns ------- str Complete zsh completion script. Raises ------ ValueError If prog_name contains invalid characters. z^[a-zA-Z0-9_-]+$zInvalid prog_name: z0. Must be alphanumeric with hyphens/underscores.z #compdef _z() {z local line state)r help_flags version_flags} ) rematch ValueErrorrextend_generate_completion_for_pathrtuplerjoin)rrcompletion_datalinesrr]/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/cyclopts/completion/zsh.pygenerate_completion_scripts0  r# argumentsr indent_strcCs4td|d|d}t||}dd|dDS)aAGenerate dynamic completion for the 'run' command. Parameters ---------- arguments : ArgumentCollection Arguments for run command. indent_str : str Indentation string. prog_name : str Program name. Returns ------- list[str] Zsh completion code lines. a local script_path local -a completions local -a remaining_words # If completing first argument (the script path), suggest files if [[ $CURRENT -eq 2 ]]; then _files return fi # Get absolute path to the script file script_path=${words[2]} script_path=${script_path:a} if [[ -f $script_path ]]; then remaining_words=(${words[3,-1]}) local result local cmd if command -v z$ &>/dev/null; then cmd="a" else return fi # Call back into cyclopts to get dynamic completions from the script result=$($cmd _complete run "$script_path" "${remaining_words[@]}" 2>/dev/null) if [[ -n $result ]]; then # Parse and display completion results completions=() while IFS= read -r line; do completions+=($line) done <<< $result _describe 'command' completions fi ficSsg|]}|qSr)rstrip).0linerrr" sz4_generate_run_command_completion..r)rtextwrap_indentsplit)r$r%rtemplateindentedrrr" _generate_run_command_completionPs $r.positional_argsr help_formatc CsFg}dd|D}dd|D}|D]L}d|jpd}t||}|jdd}|r;dd|D} d | } d | d } ntt|j} | rPd |d |d | d nd |d |d } || q|D]?}t||}|jdd}|rdd|D} d | } d | d } ntt|j} | rd|d | d nd|d } || qa|S)aGenerate positional argument specs for nested command context. In nested contexts (after *::arg:->args), word indexing is shifted: - words[1] = subcommand name - words[2] = first positional argument - words[3] = second positional argument, etc. Parameters ---------- positional_args : list[Argument] Positional arguments to generate specs for. help_format : str Help text format. Returns ------- list[str] List of zsh positional argument specs. cSsg|]}|r|qSris_var_positionalr'argrrr"r)z5_generate_nested_positional_specs..cSsg|]}|s|qSrr1r3rrr"r)r5rTforcecSg|]}tt|qSr_escape_completion_choicerr'crrr"r)r5 ()':cSr9rr:r<rrr"r)r5'*:)index_get_description_from_argument get_choicesr_map_completion_action_to_zshr hintappend) r/r0specs variadic_argsnon_variadic_argsr4posdescchoicesescaped_choices choices_stractionspecrrr"!_generate_nested_positional_specss2   .    " rTargumentc Csg}t||}|jdd}|rKdd|D}||d||d|D]}||d|d|d q(||d ||d |St|j}|tjkr_||d |S|tjkrl||d |S)aYGenerate _describe-based completion for a single positional argument. Parameters ---------- argument : Argument Argument to generate completion for. help_format : str Help text format. indent_str : str Indentation string. Returns ------- list[str] Zsh completion code lines. Tr7cSr9rr:r<rrr"r)r5z1_generate_describe_completion..zlocal -a choicesz choices=(z 'rBrAr@z_describe 'argument' choices_files _directories)rErFrIr rHrFILES DIRECTORIES) rUr0r%r!rNrOrPchoicerRrrr"_generate_describe_completions&    r[cycloptsrr . command_pathrrrc Cs||}|j}|j}d|} g} |dkr#|dkr#| t|| || Sg} g} dd|D} dd|D}| jddd |D] }t||j}| |q?t}|D] }td d |j Drrt |j |j |j}| || |j qR|D]}| d r||vrd |d}| |qu|D]}| d r||vrd |d}| |qtdd |D}| r|s|rt| |j} n| D] }t||j}| |q| | } |r| d| d| r|rdnd}| | d|d| ddD] }| | d|dq| | d| d| d|r| | d| | dg}|D]#}|j D]}| d sKt|j |j}|d |d|d q0q+| | d| | d|D] }| | d |qa| | d!| | d"| | d#| | d$| | d%|D]?}|j D]8}| d rq||f}||vr| | d |d&t|||d'|||}| || | d(qq| | d)| | d#| | d*| S)+aGenerate completion code for a specific command path. Parameters ---------- completion_data : dict Extracted completion data. command_path : tuple[str, ...] Command path. indent : int Indentation level. prog_name : str Program name. help_flags : tuple[str, ...] Help flags. version_flags : tuple[str, ...] Version flags. Returns ------- list[str] Zsh code lines. r>)runr]cSs g|] }|jdur|jr|qS)N)rDshowr3rrr"r),s z1_generate_completion_for_path..cSsg|] }|s|jr|qSr)is_positional_onlyr`r3rrr"r)-scSs |jpdS)Nr)rD)arrr"0s z/_generate_completion_for_path..)keycss|]}|dVqdS-N) startswith)r'namerrr" :sz0_generate_completion_for_path..rfrAz![Display this message and exit.]'z[Display application version.]'css(|]}|jD] }|d VqqdSre)namesrg)r'registered_commandcmd_namerrr"riLs   z '1: :->cmds'z'*::arg:->args'z-C rz _arguments \Nz z \zcase $state inz cmds)rBz local -a commandsz commands=(z z )z, _describe -t commands 'command' commandsz ;;z args)z case $words[1] inr@z ;;z esacesac)commandsr$rr.sort_generate_keyword_specsr0setanyrj#_generate_keyword_specs_for_commandrupdatergrIrT_generate_positional_spec_safe_get_description_from_appr)r r^rrrrdatarqr$r%r! args_specspositional_specsr/ keyword_argsrUrJflag_command_namesrkflagrShas_non_flag_commandsc_flagcmd_listrlrNcmdsub_path sub_linesrrr"rs                       rrZcCs|dd}|dd}|dd}|dd}|d d }|d d }|d d}|dd}|dd}|dd}|dd}|dd}|dd}|dd}|S)a{Escape special characters in a completion choice value for zsh. Choice should already be cleaned via clean_choice_text() before calling this function. This function only applies zsh-specific escaping. Parameters ---------- choice : str Cleaned choice value. Returns ------- str Escaped choice value safe for zsh completion. rm\\rA'\''`\`$\$"\"r>z\ r?z\(r@z\)[\[]\];z\;|z\|&z\&rB\:replace)rZrrr"r;s              r;textcCsd|dd}|dd}|dd}|dd}|d d }|d d }|d d}|dd}|S)zEscape special characters in description text for zsh. Parameters ---------- text : str Cleaned description text. Returns ------- str Escaped description safe for zsh completion. rmrrrrrrrrArrBrrrrrr)rrrr"_escape_zsh_descriptions       rc Cs"g}t||}|}d}|jdd}|r*dd|D}d|}d|d}d }ntt|j}|jjD]?} | d s=q5|rK|sKd | d |d } n$|r`d | d |d| d d|d } nd | d |d| d d } | | q5|j D]} | d sqxd | d |d } | | qx|S)aGenerate zsh _arguments specs for a keyword argument. Parameters ---------- argument : Argument Argument object from ArgumentCollection. help_format : str Help text format. Returns ------- list[str] List of zsh argument specs. rTr7cSr9rr:r<rrr"r)r5z+_generate_keyword_specs..r>r?r@FrfrAr]'z]:rB) rEis_flagrFrrGr rH parameterrhrglstriprI negatives) rUr0rJrNrrRrOrPrQrhrSrrr"rss4      &    rscCst||}|jdd}|r dd|D}d|}d|d}ntt|j}|r<|r6d|d |d Sd|d S|jd urJtd |j d |jd}|r]d |d |d |d Sd |d |d S)aGenerate zsh _arguments spec for a positional argument. Parameters ---------- argument : Argument Positional argument object. help_format : str Help text format. Returns ------- str Zsh positional argument spec. Tr7cSr9rr:r<rrr"r)r5z-_generate_positional_spec..r>r?r@rCrBrANzPositional-only argument z missing indexr6) rErFrrGr rHr2rDrrj)rUr0rNrOrPrQrRrMrrr"rxs   "  .rxrjcmd_appcCsBg}t||}|D]}|drd|d|d}||q |S)adGenerate zsh _arguments specs for a command that looks like a flag. Parameters ---------- names : tuple[str, ...] Registered names for the command. cmd_app : App Command app with flag-like names. help_format : str Help text format. Returns ------- list[str] List of zsh argument specs. rfrArr)ryrgrI)rjrr0rJrNrhrSrrr"rv*s   rvrRcCs |tjkrdS|tjkrdSdS)aMap shell-agnostic completion action to zsh-specific completion command. Parameters ---------- action : CompletionAction Shell-agnostic completion action. Returns ------- str Zsh completion command (e.g., "_files", "_directories", or ""). rVrWr)rrXrY)rRrrr"rGFs  rGcCst|jjpd|d}t|S)aHExtract plain text description from Argument, escaping zsh special chars. Parameters ---------- argument : Argument Argument object with parameter help text. help_format : str Help text format. Returns ------- str Escaped plain text description (truncated to 80 chars). rformat)r rhelpr)rUr0rrrr"rEZsrEcCsV|jsdSz t|jd}|jpd}Wnty t|j}Ynwt||d}t|S)a/Extract plain text description from App, escaping zsh special chars. Parameters ---------- cmd_app : App Command app with help text. help_format : str Help text format. Returns ------- str Escaped plain text description (truncated to 80 chars). r plaintextr)rr short_description Exceptionstrr r)rr0parsedrrrr"ryms   ry)r\r]rr)(__doc__rtextwraprrr*typingrcyclopts.completion._baserrrrr r cyclopts.help.helpr r]r cyclopts.argumentr rrr#listr.rTr[dictrintrr;rrsrxrvrGrEryrrrr"s      5 = C 2    !6&%