' ddZddlZddlmZddlmZmZmZmZm Z m Z er ddl m Z ddl mZddd ed efd Zd ed efd Zd ed efdZd edeeedfefd eefdZded efdZdeeedfefd edeedfdeedfd eef dZdeeedfefdeedfd edeedfdeedfd eef dZdedeedfd eded eef dZdeedfd eefdZd ededeedfdeedfd eef dZded ededed eef d Zd!ed ededed eef d"Zdeedfd ed efd#Zd$d%ded efd&Z y)'zFish completion script generator. Generates static fish completion scripts using `complete -c COMMAND` statements. Completions auto-load from ~/.config/fish/completions/PROGNAME.fish. N) TYPE_CHECKING)CompletionActionCompletionDataclean_choice_textextract_completion_dataget_completion_action strip_markup)App) CommandSpecappr prog_namereturnc|rtjd|std|dt|}d|ddg}t d|j D}|r,|j t|||jd|jrt|jnd}|jrt|jnd}|j t||||d j|d zS) aGenerate fish completion script. Parameters ---------- app : App The Cyclopts application to generate completion for. prog_name : str Program name for completion (alphanumeric with hyphens/underscores). Returns ------- str Complete fish 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# Fish completion for z# Generated by Cycloptsc38K|]}t|dkDyw)rN)len).0paths ~/home/jay/workspace/.worktrees/task-2057-dev2/scripts/.codegraph-venv/lib/python3.12/site-packages/cyclopts/completion/fish.py z-generate_completion_script..7sOc$i!mOs )rematch ValueErrorranykeysextend_generate_helper_functionsappend help_flagstuple version_flags_generate_completionsjoin)r r completion_datalineshas_nested_commandsr!r#s rgenerate_completion_scriptr)s( BHH%8)D.ym;klmm-c2O ! ,!  E O8L8L8NOO / ?KL R*-..s~~&bJ030A0AE#++,rM LL& :}]^ 99U d ""textc&|jddS)z&Escape single quotes for fish strings.'z'\'')replacer+s r_escape_fish_stringr0Ds <<W %%r*c`|jdd}|jdd}t|S)z!Escape description text for fish.r  )r.r0r/s r_escape_fish_descriptionr4Is- <<c "D <<c "D t $$r*r&.c t}|jD]y}|jD]h}|jr|jj s+|jj D]%}|j ds|j|'j{d|d}dd|ddg}|r7djd t|D}|jd |d n|jd |jgd |S)a3Generate helper function for command path detection. Parameters ---------- prog_name : str Program name. completion_data : dict Completion data used to identify options that take values. Returns ------- list[str] Lines defining the helper function. -__fish__using_commandz6# Helper function to check exact command path sequencez function z! set -l cmd (commandline -opc)z set -l subcommandsr2c32K|]}t|ywNr0)ropts rrz-_generate_helper_functions..ss`S 3C 8`z set -l options_with_values 'r-z! set -l options_with_values '')z set -l skip_next 0z; # Extract non-option words (commands) from command linez! for i in (seq 2 (count $cmd))z set -l word $cmd[$i]z if test $skip_next -eq 1z set skip_next 0z continue endz) if string match -qr -- '^-' $wordz> # Check if this option takes a value (exact match)zF if string match -q -- "* $word *" " $options_with_values "z set skip_next 1z endz elsez* # Non-option word is a commandz$ set -a subcommands $wordr> endz8 # Check if subcommand sequence matches expected pathz2 if test (count $subcommands) -ne (count $argv)z return 1r?z" for i in (seq 1 (count $argv))z- if test $subcommands[$i] != $argv[$i]z return 1r>r?z return 0end) setvalues argumentsis_flag parametername startswithaddr%sortedr r) r r&options_with_valuesdataargumentrF func_namer' escaped_optss rrrPs $%&&(6 6H##%(*<*<*A*A$..336Ds++//56 66)N3I@ I;+  Exx`FK^D_``  7 ~QGH 89 LL  B Lr*actioncT|tjk(ry|tjk(ryy)a Map completion action to fish flags. Parameters ---------- action : CompletionAction Completion action type. Returns ------- str Fish completion flags ("-r -F" for files, "-r -a '(...)'" for directories, "" otherwise). z-r -Fz%-r -a '(__fish_complete_directories)'r)rFILES DIRECTORIES)rOs r_map_completion_action_to_fishrSs+!''' !---6 r*r!r#c g}t|jD]W\}}|jt||||||t |j t k7sG|jdY|S)akGenerate all fish completion commands. Parameters ---------- completion_data : dict Extracted completion data. prog_name : str Program name. help_flags : tuple[str, ...] Help flags. version_flags : tuple[str, ...] Version flags. Returns ------- list[str] Completion command lines. )keyr)rIitemsr_generate_completions_for_pathmaxrrr )r&r r!r#r' command_path_datas rr$r$s{0 E%o&;&;&=>  e *    33353? ? LL   Lr*rYc||vrgS||}g}t||}|jt|||||jDcgc]"}|j r|j s!|$} }| s|s|r|jt ||jt|||||jt| |||j|jt|j|||j|Scc}w)aGenerate completions for a specific command path. Parameters ---------- completion_data : dict Extracted completion data. command_path : tuple[str, ...] Command path. prog_name : str Program name. help_flags : tuple[str, ...] Help flags. version_flags : tuple[str, ...] Version flags. Returns ------- list[str] Completion command lines. ) _get_condition_for_pathr _generate_subcommand_completionsrCis_positional_onlyshow_generate_option_section_header"_generate_help_version_completions!_generate_keyword_arg_completions help_format$_generate_command_option_completionscommands) r&rYr r!r#rKr' conditionarg keyword_argss rrWrWs6?* < (D E' i@I LL1$ iQZ[\#'>>_C9O9O9QVYV^V^C_L_z] 4\BC 7 9jZghi 6|YPY[_[k[klm 9$--T]_c_o_opq L`sD D(DrKrfc |jDcgc]'}|jD]}|jdr|)}}}|sgSg}|r$|jddj |n|jd|jD]y}|jD]h}|jdrt |j |j} t| } t|} |jd|d|d| d| d j{|Scc}}w) aTGenerate completions for subcommands. Parameters ---------- data : CompletionData Completion data. command_path : tuple[str, ...] Command path. prog_name : str Program name. condition : str Fish condition. Returns ------- list[str] Completion command lines. r6z# Subcommands for: r2z# Root-level commands complete -c z -a '' -d 'r-) renamesrGr r%_get_description_from_appr rcr4r0) rKrYr rfregistered_commandrFrer'cmd_namedesc escaped_desc escaped_cmds rr]r]s)2(,}}#ASAYAY9=aeapapqtau H  E *388L+A*BCD ,-"mm h*00 hH""3',-?-C-CTEUEUVD3D9L-h7K LL< {!I;eK=PVWcVddef g h h L/s %DDc6|rddj|gSdgS)zGenerate section header comment for options. Parameters ---------- command_path : tuple[str, ...] Command path. Returns ------- list[str] Comment line. z# Options for: r2z# Root-level optionsr%)rYs rr`r`:s*!#((<"8!9:;; " ##r*c g}|D]u}|jdr!|dd}|jd|d|d|d5|jdsGt|dk(sV|d }|jd|d|d |dw|D]u}|jdr!|dd}|jd|d|d|d 5|jdsGt|dk(sV|d }|jd|d|d |d w|S) acGenerate completions for help and version flags. Parameters ---------- prog_name : str Program name. condition : str Fish condition. help_flags : tuple[str, ...] Help flags. version_flags : tuple[str, ...] Version flags. Returns ------- list[str] Completion command lines. --Nrjr2 -l z$ -d 'Display this message and exit.'r6 -s z" -d 'Display application version.')rGr r)r rfr!r#r'flag long_name short_names rraraLs(0 Eu ??4 QRI LL< {!I;d9+Mqr s __S !c$i1naJ LL< {!I;d:,Nrs t us ??4 QRI LL< {!I;d9+Mop q __S !c$i1naJ LL< {!I;d:,Npq r s Lr*rhrcc tg}|D]*}t|jjxsd|}t|}|j }|j d} t |j} |jjxsgD]} | jds| jdr| dd} d |d |d | g} n!t| dk(r| d }d |d |d |g} nZ|r| jd|dn| rK| Dcgc]}tt|}}d j|}| jd|d|dn:t| }|r| j|d|dn| jd|d|jd j| |j D]{} | jds| jdr$| dd} |jd |d |d | d|d Jt| dk(sY| d }|jd |d |d |d|d }-|Scc}w)aQGenerate completions for keyword arguments. Parameters ---------- keyword_args : list Keyword arguments. prog_name : str Program name. condition : str Fish condition. help_format : str Help text format. Returns ------- list[str] Completion command lines. rformatT)forcer6rvrwNrjr2rxryrzz-d 'r-z-x -a 'rk -d 'z-r -d ')r rEhelpr4rD get_choicesrhintrFrGrr r0rr%rS negatives)rhr rfrcr'rLrprqrDchoicesrOrFr| line_partsr}cescaped_choices choices_str action_flagss rrbrbysw0 E -iH..339r+N/5 ""$&&T&2&x}}5&&++1r /D??3't$ H ,YKq 4 {ST Ta!!W ,YKq 4 |TU !!Da"89V]"^QR#67H7K#L"^"^!hh7 !!GK=|nA"NO=fE %%eL>&KL%% ~Q&?@ LL*- .5 /8&& iD??3't$ H  |I;a {$ykQVWcVddefgTa!!W  |I;a {$zlRWXdWeefgh iI-i^ L/#_sH5rec ng}|D]}|jD]}|jdst|j|}t |}|jdr$|dd} |j d|d|d| d|d kt |dk(sz|d } |j d|d|d | d|d |S) ajGenerate completions for commands that look like options. Parameters ---------- commands : list List of RegisteredCommand tuples. prog_name : str Program name. condition : str Fish condition. help_format : str Help text format. Returns ------- list[str] Completion command lines. r6rvrwNrjr2rxrr-ryrz)rlrGrmr r4r r) rer rfrcr'rnrorprqr|r}s rrdrds0 E& i*00 iH&&s+,-?-C-C[QD3D9L""4($QRL  |I;a {$ykQVWcVddefgX!#%a[  |I;a {$zlRWXdWeefgh i i Lr*cV|syd|d}djd|D}d|d|dS)aGenerate fish condition string for a command path. Parameters ---------- command_path : tuple[str, ...] Command path (empty for root). prog_name : str Program name. Returns ------- str Fish condition flag. z-n __fish_use_subcommandr7r8r2c32K|]}t|ywr:r;)rcmds rrz*_get_condition_for_path..sQS 3C 8Qr=z-n 'r-rt)rYr rMescaped_commandss rr\r\sD ))N3IxxQLQQ )A./q 11r*cmd_appzApp | CommandSpeccddlm} ||jd}|jxsd}t ||S#t$rt |jxsd}Y1wxYw)zExtract description from App. Parameters ---------- cmd_app : App | CommandSpec Command app or spec. help_format : str Help text format. Returns ------- str Description text. r)docstring_parse plaintextrr)cyclopts.help.helprrshort_description Exceptionstrr )rrcrparsedr+s rrmrms_3' {;''-2 [ 11 '7<<%2&'s#8"AA)!__doc__rtypingrcyclopts.completion._baserrrrrr cycloptsr cyclopts.command_specr rr)r0r4dictr"listrrSr$rWr]r`rarbrdr\rmrr*rrs  1)#E)#c)#c)#X&c&c& %3%3%II%S/>9:I #YIX+;('%S/>9:''c3h'c? '  #Y 'T+%S/>9:+S/++c3h + c? +  #Y +\/ /S/// /  #Y /d$%S/$d3i$$***c3h*c? *  #Y *ZIIII I  #Y IX)))) )  #Y )X2%S/2c2c2.2':22QT2r*