lKjdZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZerdd lmZdd lmZmZdd lmZd d dedefdZdddededeefdZdeddedeefdZdddededeefdZ d:de!e"edfe fde"edfde#dede"edfd e"edfdeefd!Z$d"edefd#Z%d$edefd%Z&d&edefd'Z'd(edefd)Z(d*edefd+Z)d(edefd,Z*d-ededeefd.Z+dddedeefd/Z,dddedefd0Z-d1e"edfd2d3dedeefd4Z.d5e defd6Z/dddedefd7Z0dddedefd8Z1d2d3dedefd9Z2y);zZsh completion script generator. Generates static zsh completion scripts using the compsys framework. No runtime Python dependency. N)dedent)indent) TYPE_CHECKING)is_iterable_type)CompletionActionCompletionDataclean_choice_textescape_for_shell_patternextract_completion_dataget_completion_action strip_markup)docstring_parse)App)ArgumentArgumentCollection) CommandSpecappr prog_namereturnc |rtjd|std|dt|}d|dd|dddg}|j t |d ||j rt|j nd |jrt|jnd |j d dgd j|d zS) 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 _cyclopts_z() {z local line state)r help_flags version_flags} ) rematch ValueErrorr extend_generate_completion_for_pathrtuplerjoin)rrcompletion_dataliness h/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/cyclopts/completion/zsh.pygenerate_completion_scriptr(s( BHH%8)D.ym;klmm-c2O I; YKu%  E LL%  03uS^^,B696G6G% 1 12R   LL    99U d "" argumentsr indent_strctd|d|d}t||}|jdDcgc]}|jc}Scc}w)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 fir)rtextwrap_indentsplitrstrip)r*r+rtemplateindentedlines r' _generate_run_command_completionr3Xsf*&# $ )""HHx4H&.nnT&: ;&:dDKKM&: ;; ;sA positional_argsr help_formatc g}|Dcgc]*}|jst|js)|,}}|Dcgc]+}|jrt|jr*|-}}dtdtffd }|D]6}d|jxsdz}|j ||t|8t d|Dd}||r|d}||j ||d|Scc}wcc}w) 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. positionrcd|jd}|rP|Dcgc]}tt|}}dj|}d|d}t t | }d}n,t t|j}t| }d}|r||d|d||S||d||Scc}w) NTforce ()"':) get_choices_escape_choice_for_dq_specr r$_escape_zsh_description_dq_description_text_map_completion_action_to_zshr hint_get_description_from_argument) argr7choicescescaped_choices choices_stractiondescquoter5s r'_buildz1_generate_nested_positional_specs.._builds///- Y`aY`TU9:KA:NOY`Oa((?3K Q'F-.?[.QRDE23H3RSF1#{CDE>D% !D66(5':lUGT\S]]^_c^dejdkJllbsB-rc3BK|]}|js|ywNis_var_positional.0as r' z4_generate_nested_positional_specs..sEmq/B/B/D1mN*)rUrrFstrindexappendnext) r4r5specsrH variadic_argsnon_variadic_argsrPposchosens ` r'!_generate_nested_positional_specsres . E%4mOSs7L7L7NRbcfckckRlSOMm&&c.C.C.EN^_b_g_gNh mc mc m!399>" VCS*+ !EmEt LF ~-q!  VFC() LOns*C4C4C9C9)C9argumentc Fg}t||}|jd}|r|Dcgc]}tt|}}|j |d|j |d|D]}|j |d|d|d|j |d|j |d |St |j } | tjk(r|j |d |S| tjk(r|j |d |Scc}w) 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. Tr9zlocal -a choicesz choices=(z 'r@r?r=z_describe 'argument' choices_files _directories) rGrA_escape_completion_choicer r^r rFrFILES DIRECTORIES) rfr5r+r&rNrIrJrKchoicerMs r'_generate_describe_completionrns7* E )(K @D""".GT[\T[q45Fq5IJT[\ |#345 |9-.%F LLJz/_generate_completion_for_path..9sqww|!|r))keyc3>K|]}|jdyw-N) startswith)rWnames r'rYz0_generate_completion_for_path..CsI0Hts#0Hsrxr?z![Display this message and exit.]'z[Display application version.]'c3bK|]'}|jD]}|jd )ywrw)namesry)rWregistered_commandcmd_names r'rYz0_generate_completion_for_path..Us4 ?G);XjXpXpHH   $$Xp$xs-/Fc3BK|]}|js|ywrSrTrVs r'rYz0_generate_completion_for_path..gsR=aAcmds'z'*::arg:->args'z-C rz _arguments \ z \zcase $state inz cmds)r@z local -a commandsz commands=(z z )z, _describe -t commands 'command' commands ;;z args)z case $words[1] inr=z ;;z esacesac)commandsr*r!r3r]showis_positional_onlysort_generate_keyword_specsr5setanyr|#_generate_keyword_specs_for_commandrupdateryr^rerUrrFr__generate_positional_spec_generate_eq_form_prepass_safe_get_description_from_apprj_escape_command_name_for_caser")&r%rorrrrdatarr*r+r& args_specspositional_specsrHr4 keyword_argsrfr`flag_command_namesr}flagspechas_non_flag_commands seen_restrX iterable_args chosen_restis_rest eq_prepassc_flagcmd_listr~rNescaped_cmd_namecmdsub_pathescaped_case_name sub_liness& r'r"r"s< < (D}}HIvJ ExI$; 5iYWX J'0Vis3993HSXXsiOV#,Z9CC4J4J4LQTQYQYC9LZ34!'$2B2BC% ! & I0B0H0HI I7"((*<*@*@$BRBRE   e $  % %&8&>&> ? ' ??3 D0B$Btf=>D   d #  ??3 D0B$Btf;23 |7+,"* .44**3/9:L:P:PRVRbRbcD'@'J$OOa(8'94&$BC 5#+  |#89: |>23C LLJU> U>!U> *V7Vsc2d|jddzdzS)uWrap ``s`` in POSIX-safe single quotes for embedding as a shell argument. The only character that can't appear inside a single-quoted shell string is ``'`` itself, which is handled with the ``'\''`` end-and-restart trick. Everything else (spaces, parens, ``$``, backticks, etc.) is literal — no backslash-escaping is needed, and adding any would just surface as visible backslashes in the resulting argument. r?'\''replace)rs r'_shell_single_quoters 3( (3 ..r)rmc|jdd}|jdd}|jdd}|jdd}|jd d }|jd d }|jd d}|jdd}|jdd}|jdd}|jdd}|jdd}|jdd}|jdd}|S)aEscape a choice value for embedding in a single-quoted shell context. Used for ``_describe`` array elements (``'value:desc'``) where the only parser the value passes through is the array-element parser, not the ``_arguments`` choice-list eval. Choice should already be cleaned via ``clean_choice_text()``. Parameters ---------- choice : str Cleaned choice value. Returns ------- str Escaped choice value safe for zsh completion. r\\r?r`\`$\$r>\"r;z\ r<z\(r=z\)[\[]\];z\;|z\|&z\&r@\:r)rms r'rjrjs$^^D& )F ^^C )F ^^C 'F ^^C 'F ^^C 'F ^^C 'F ^^C 'F ^^C 'F ^^C 'F ^^C 'F ^^C 'F ^^C 'F ^^C 'F ^^C 'F Mr)valuec|jdd}dD]}|j|d|z}|jdd}|jdd}|jdd}|jdd }|S) a^Escape a choice value for ``_arguments``' parenthesized choice list. The value passes through *two* parsers: 1. zsh's outer double-quoted-string parser, which interprets ``\``, ``"``, ``$`` and backtick. 2. ``_arguments``' choice-list parser, which whitespace-tokenizes and reads ``\X`` as a literal X. A literal ``'`` cannot be embedded in a single-quoted spec string (``'\''`` ends/restarts the quoting and the parser then sees an unbalanced ``'``), so choice-bearing specs are emitted with a *double*- quoted outer string and routed through this helper. rrz '"()[]:;|&$`r>rrrrrr)rrchs r'rBrBsy(  dF#A IIb$) $ $A #uA #uA #uA Hr)textc|jdd}|jdd}|jdd}|jdd}|jd d }|jd d }|jd d}|S)zEscape a description for embedding in a double-quoted spec string. Same as ``_escape_zsh_description`` but skips the ``'`` -> ``'\\''`` substitution: ``'`` is literal in a double-quoted context. rrrrrrr>rr@rrrrrrrs r'rCrC sy <<f %D <<U #D <<U #D <<U #D <<U #D <<U #D <<U #D Kr)rzct|dS)aEscape special characters in command name for zsh case patterns. In zsh case patterns, glob characters need to be escaped to match literally. Colons also need escaping because zsh's completion system may treat them specially when populating the $words array after _describe completion. Parameters ---------- name : str Command name. Returns ------- str Escaped command name safe for zsh case patterns. z*?[]()|:)chars)r )rzs r'rrs& $D ;;r)c&|jdd}|jdd}|jdd}|jdd}|jd d }|jd d }|jd d}|jdd}|S)zEscape special characters in description text for zsh. Parameters ---------- text : str Cleaned description text. Returns ------- str Escaped description safe for zsh completion. rrrrrrr>rr?rr@rrrrrrrs r'_escape_zsh_descriptionr1s <<f %D <<U #D <<U #D <<U #D <<W %D <<U #D <<U #D <<U #D Kr)rc Dg}|D]}|js|jjr*|jjxsgDcgc]}|j ds|}}|sf|j d}|r6|Dcgc]}t t|}}ddj|z} n1t|j} t| } | dk(rd} n | dk(rd} n|D]}|j|| f|sgS|dg} |D]q\} } | j|d | d | j|d | d | j|d | | j|d| j|ds| j|d| Scc}wcc}w)uEmit a ``--opt=value`` pattern dispatcher to run before ``_arguments``. For each keyword argument with a long name and a value action, emits a ``--opt=*)`` case branch that strips the ``--opt=`` prefix via ``compset -P`` and then dispatches to the value action (choice list, ``_files``, or ``_directories``). The natural-TAB experience on the option *name* is preserved by leaving the underlying ``_arguments`` spec without an ``=`` suffix; this pre-pass only catches the user explicitly typing the eq form. Skipped for arguments whose ``Parameter.requires_equals`` is True — those already emit the ``=`` spec, which handles eq-form completion via ``_arguments``. Parameters ---------- keyword_args : list Keyword argument objects from ArgumentCollection (already filtered to ``arg.show``). indent_str : str Leading indentation. Returns ------- list[str] Zsh code lines (empty if no eligible options). --Tr9z compadd -- r;rhrizcase ${words[CURRENT]} inrz=*)z compset -P 'z='z z returnrr) is_flag parameterrequires_equalsrzryrArr r$r rFrEr^)rr+casesrfrz long_namesrIrJquoted action_linerM zsh_actionr&opt_names r'rrIs8$&E     !3!3!C!C (0(:(:(?(?(E2(E`(Et$//Z^J_d(E ` &&T&2  JQQA)*;A*>?FQ'#((6*::K*8==9F6v>JX%& ~-, D LL$ , -3!8   ,12 E"'+ |2hZs34 |#3H:R@A |4 }56 |:./ |6*+ "'  LLJr?rxr=r[rrz]:r@]')rrArBr r$rEr rFrCrDrGboolrrrzrylstripr^ negatives)rfr5r`rrM has_choicesrIrJrKrLrNrO requires_eq takes_valuerz accepts_eq spec_name repeat_prefixrdesc_sqs r'rrs/ E    D FK""".G U\]U\PQ56G6JKU\]hh/ [M#./DX]]/ST  ##4X{#KL +Hk B CCE4x))99:K (K""''s#  JT__T%:J{ ",tfAJ$ * W]OI;avQugFD W]OI;avR C@P?QQRSYRZ[`ZabDW]OI;avR C@P?QRWQXYD T("-X{CG""s# 4&'"% T # LK^sGcB|jd}|rP|Dcgc]}tt|}}dj|}d|d}t t ||}d}n,t t|j}t||}d}|jst|jr|r |d|d ||S|d||S|jtd |jd |jd z} |r|| d |d ||S|| d ||Scc}w) 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. Tr9r;r<r=r>r?z*:r@zPositional-only argument z missing indexrQ)rArBr r$rCrDrEr rFrGrUrr]r r|) rfr5rIrJrKrLrMrNrOrcs r'rrsW"""".GU\]U\PQ56G6JKU\]hh/ [M#)*;Hk*RS./DX]]/ST-h D!!#'7 'F5;%4&&%1ZE7"TFSXRYAZZ~~4X^^4DNSTT ..1 C5;eWSE4&&% 1^E73%qQUPVW\V]A^^%^sDr|cmd_appzApp | CommandSpeccg}t||}|D].}|jdsd|d|d}|j|0|S)aeGenerate zsh _arguments specs for a command that looks like a flag. Parameters ---------- names : tuple[str, ...] Registered names for the command. cmd_app : App | CommandSpec Command app or spec. help_format : str Help text format. Returns ------- list[str] List of zsh argument specs. rxr?rr)rryr^)r|rr5r`rNrzrs r'rrsR& E )'; ?D ??3 tfAdV2&D LL  Lr)rMcT|tjk(ry|tjk(ryy)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 ""). rhrir)rrkrl)rMs r'rErE;s+!''' #// / r)c,tt||S)aExtract 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). Falls back to argument name if help text is empty, since zsh _arguments requires a non-empty description for positional specs to work correctly. )rrD)rfr5s r'rGrGOs" ##4X{#K LLr)ct|jjxsd|}|s|jr|jdnd}|S)zDPlain-text description for an argument with the empty-help fallback.rformatrrf)r rhelpr|)rfr5rs r'rDrDcs? **//52k JD %-NNx~~a  Kr)c t|jd}|jxsd}t ||}t |S#t$rt |jxsd}Yrs  . 1/>17#E7#c7#c7#t:<#:<:<:< #Y :9:pS/p p p c3h p c? p #Ypf /3 /3 / c c F c c B S S  <<<,##0FDFcFd3iFR`j`s`tCy`F%_ %_%_%_P c?%8GJ #Y<*:s(MZMcMcM( ),?)c)VY)r)