o i= @sDdZddlZddlmZddlmZmZmZmZm Z m Z er$ddl m Z ddde d e fd d Zd e d e fd dZd e d e fddZde deee dfefd ee fddZded e fddZdeee dfefde dee dfdee dfd ee f ddZdeee dfefdee dfde dee dfdee dfd ee f ddZdedee dfde d e d ee f d!d"Zdee dfd ee fd#d$Zde d e dee dfdee dfd ee f d%d&Zd'ede d e d(e d ee f d)d*Zd+ede d e d(e d ee f d,d-Zdee dfde d e fd.d/Zd0dd(e d e fd1d2ZdS)3zFish 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)Appappr prog_namereturncCs|rtd|std|dt|}d|ddg}tdd|D}|r6|t|||d|j r>t |j nd }|j rHt |j nd }|t ||||d |d S) 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 Cycloptscss|] }t|dkVqdS)rN)len).0pathr^/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/cyclopts/completion/fish.py 6sz-generate_completion_script..r )rematch ValueErrorranykeysextend_generate_helper_functionsappend help_flagstuple version_flags_generate_completionsjoin)r r completion_datalineshas_nested_commandsrrrrrgenerate_completion_scripts r%textcCs |ddS)z&Escape single quotes for fish strings.'z'\'')replacer&rrr_escape_fish_stringCs r*cCs |dd}|dd}t|S)z!Escape description text for fish.r  )r(r*r)rrr_escape_fish_descriptionHs  r-r".c Cst}|D]!}|jD]}|s'|jjr'|jjD] }|dr&||qq qd|d}dd|ddg}|rPdd d t |D}| d |d n| d | gd|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 subcommandsr+cs|]}t|VqdSNr*)roptrrrrrz-_generate_helper_functions..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 $wordr6 endz8 # Check if subcommand sequence matches expected pathz2 if test (count $subcommands) -ne (count $argv)z return 1r7z" for i in (seq 1 (count $argv))z- if test $subcommands[$i] != $argv[$i]z return 1r6r7z return 0end) setvalues argumentsis_flag parametername startswithaddr!sortedrr) r r"options_with_valuesdataargumentr> func_namer# escaped_optsrrrrOs.       !ractioncCs |tjkrdS|tjkrdSdS)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)rGrrr_map_completion_action_to_fishs  rJrrc CsRg}t|D]\}}|t||||||t|tdkr&|dq|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 )rAitemsr_generate_completions_for_pathmaxrrr)r"r rrr# command_path_datarrrr s r rOc Cs||vrgS||}g}t||}|t||||dd|jD}|s)|s)|rQ|t||t|||||t||||j|t|j |||j|S)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. cSsg|] }|s|jr|qSr)is_positional_onlyshow)rargrrr sz2_generate_completions_for_path..) _get_condition_for_pathr _generate_subcommand_completionsr;_generate_option_section_header"_generate_help_version_completions!_generate_keyword_arg_completions help_format$_generate_command_option_completionscommands) r"rOr rrrCr# condition keyword_argsrrrrMs  rMrCr]c Csdd|jD}|s gSg}|r|dd|n|d|jD]/}|jD])}|dr1q)t|j|j}t|} t |} |d|d|d| d | d q)q$|S) 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. cSs&g|]}|jD] }|ds|qqS)r.)namesr?)rregistered_commandr>rrrrTs z4_generate_subcommand_completions..z# Subcommands for: r+z# Root-level commandsr. complete -c z -a '' -d 'r') r\rr!r_r?_get_description_from_appr rZr-r*) rCrOr r]r\r#r`cmd_namedesc escaped_desc escaped_cmdrrrrVs&    & rVcCs|r dd|gSdgS)zGenerate section header comment for options. Parameters ---------- command_path : tuple[str, ...] Command path. Returns ------- list[str] Comment line. z# Options for: r+z# Root-level optionsr!)rOrrrrW9s rWc Csg}|D];}|dr!|dd}|d|d|d|dq|dr?t|dkr?|d }|d|d|d |dq|D];}|dr_|dd}|d|d|d|d qB|dr}t|dkr}|d }|d|d|d |d qB|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. --Nrar+ -l z$ -d 'Display this message and exit.'r. -s z" -d 'Display application version.')r?rr)r r]rrr#flag long_name short_namerrrrXKs$      rXr^rZc Csg}|D]}t|jjp d|d}t|}|}|jdd} t|j} |jjp(gD]} | ds1q)| drI| dd} d |d |d | g} nt | dkr`| d }d |d |d |g} nq)|rm| d|dn6| rdd| D}d |}| d|d|dnt | }|r| |d|dn | d|d| d | q)|jD]B} | dsq| dr| dd} | d |d |d | d|d qt | dkr| d }| d |d |d |d|d qq|S)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. r formatT)forcer.rirjNrar+rkrlrmz-d 'r'cSsg|]}tt|qSr)r*r)rcrrrrTsz5_generate_keyword_arg_completions..z-x -a 'rb -d 'z-r -d ')rr=helpr-r< get_choicesrhintr>r?rrr!rJ negatives)r^r r]rZr#rDrerfr<choicesrGr>ro line_partsrpescaped_choices choices_str action_flagsrrrrYxsP           & $ rYr\c Csg}|D]R}|jD]L}|dsq t|j|}t|}|dr9|dd} |d|d|d| d|d q t|dkrU|d } |d|d|d | d|d q q|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. r.rirjNrar+rkrur'rlrm)r_r?rcr r-rr) r\r r]rZr#r`rdrerfrorprrrr[s      & $r[cCs:|sdSd|d}ddd|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_subcommandr/r0r+csr1r2r3)rcmdrrrrr5z*_get_condition_for_path..z-n 'r'rh)rOr rEescaped_commandsrrrrUs  rUcmd_appcCsZddlm}|js dSz ||jd}|jpd}Wnty&t|j}Ynwt||dS)zExtract description from App. Parameters ---------- cmd_app : App Command app. help_format : str Help text format. Returns ------- str Description text. r)docstring_parser plaintextrq)cyclopts.help.helprrvshort_description Exceptionstrr)rrZrparsedr&rrrrcs    rc)__doc__rtypingrcyclopts.completion._baserrrrrrcycloptsr rr%r*r-dictrlistrrJr rMrVrWrXrYr[rUrcrrrrs   , L   *    .  2   - L ,