eix>dZddlmZddlmZddlmZddlZddlZddlZddlm Z ddZ dd Z dd Z dd Z dd ZdZddZdZddZdZdZdS)z?Provides tab completion functionality for CLIs built with Fire.)absolute_import)division)print_functionN) inspectutilsbashc|dkrt|t||St|t||S)Nfish) _FishScript _Commands _BashScript)name componentdefault_optionsshells S/home/div/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fire/completion.pyScriptrs? f__ tYy11? C CC T9Y// A AAc |p t}t||\}}}d}d}d}ddfd} g} t} | | t|} | t|} | D]} | | dt|| || } | || | d | }|| }||d| d d  dd  dd d|S)aXReturns a Bash script registering a completion function for the commands. Args: name: The first token in the commands, also the name of the command. commands: A list of all possible commands that tab completion can complete to. Each command is a list or tuple of the string tokens that make up that command. default_options: A dict of options that can be used with any command. Use this if there are flags that can always be appended to a command. Returns: A string which is the Bash script. Source the bash script to enable tab completion in Bash. a# bash completion support for {name} # DO NOT EDIT. # This script is autogenerated by fire/completion.py. _complete-{identifier}() {{ local cur prev opts lastcommand COMPREPLY=() prev="${{COMP_WORDS[COMP_CWORD-1]}}" cur="${{COMP_WORDS[COMP_CWORD]}}" lastcommand=$(get_lastcommand) opts="{default_options}" GLOBAL_OPTIONS="{global_options}" {checks} COMPREPLY=( $(compgen -W "${{opts}}" -- ${{cur}}) ) return 0 }} get_lastcommand() {{ local lastcommand i lastcommand= for ((i=0; i < ${{#COMP_WORDS[@]}}; ++i)); do if [[ ${{COMP_WORDS[i]}} != -* ]] && [[ -n ${{COMP_WORDS[i]}} ]] && [[ ${{COMP_WORDS[i]}} != $cur ]]; then lastcommand=${{COMP_WORDS[i]}} fi done echo $lastcommand }} filter_options() {{ local opts opts="" for opt in "$@" do if ! option_already_entered $opt; then opts="$opts $opt" fi done echo $opts }} option_already_entered() {{ local opt for opt in ${{COMP_WORDS[@]:0:$COMP_CWORD}} do if [ $1 == $opt ]; then return 0 fi done return 1 }} is_prev_global() {{ local opt for opt in $GLOBAL_OPTIONS do if [ $opt == $prev ]; then return 0 fi done return 1 }} complete -F _complete-{identifier} {command} z; case "${{lastcommand}}" in {lastcommand_checks} esaczQ {command}) {opts_assignment} opts=$(filter_options $opts) ;;z if is_prev_global; then opts="${{GLOBAL_OPTIONS}}" else opts="{options} ${{GLOBAL_OPTIONS}}" fiz, opts="{options} ${{GLOBAL_OPTIONS}}" c|krSSN)commandr %opts_assignment_main_command_template#opts_assignment_subcommand_templates r_GetOptsAssignmentTemplatez/_BashScript.._GetOptsAssignmentTemplates$ 22 00r )options)ropts_assignment )lastcommand_checks/.,)r rchecksr identifierglobal_options) set_GetMapsaddunionkeysformatjoinsortedappendreplace)r commandsrr' options_mapsubcommands_mapbash_completion_template check_wrapperlastcommand_check_templaterlines commands_setrrr r%rrs` @@rr r "s<$,suu/19 Ho22..+KZ -   ) %+/'1111111 %,4##C(<(<(>(>$?$?@@,##C (8(8(:(:$;$;<<,  g0099@@ ;w'--og.FGG H H  AO  LL"))+ * - - yy''   +   & %%((?33\\#r**223;;CCCLL.11 & rc h|p t}t|||\}}}d}d}d}d} t|t|D]} || D]} |||| | z }|| |D];} | |k} |||| | r| nd| dz }<|d d |D S) aXReturns a Fish script registering a completion function for the commands. Args: name: The first token in the commands, also the name of the command. commands: A list of all possible commands that tab completion can complete to. Each command is a list or tuple of the string tokens that make up that command. default_options: A dict of options that can be used with any command. Use this if there are flags that can always be appended to a command. Returns: A string which is the Fish script. Source the fish script to enable tab completion in Fish. aLfunction __fish_using_command set cmd (commandline -opc) for i in (seq (count $cmd) 1) switch $cmd[$i] case "-*" case "*" if [ $cmd[$i] = $argv[1] ] return 0 else return 1 end end end return 1 end function __option_entered_check set cmd (commandline -opc) for i in (seq (count $cmd)) switch $cmd[$i] case "-*" if [ $cmd[$i] = $argv[1] ] return 1 end end end return 0 end function __is_prev_global set cmd (commandline -opc) set global_options {global_options} set prev (count $cmd) for opt in $global_options if [ "--$opt" = $cmd[$prev] ] echo $prev return 0 end end return 1 end zJcomplete -c {name} -n '__fish_using_command {command}' -f -a {subcommand} z}complete -c {name} -n '__fish_using_command {command};{prev_global_check} and __option_entered_check --{option}' -l {option} z and __is_prev_global;)r r subcommandr"--)r rprev_global_checkoptionrc3"K|] }d|dV dS)"Nr).0r>s r z_FishScript..s*IIm&mmmIIIIIIr)r')r(r)r,r+r-lstripr.)r r2rr'r3r4 fish_sourcesubcommand_template flag_templater=rr;r> check_neededs rr r s$,suu/19 Ho22..+++Z;F-/_))++,,223{7G7G7I7I3J3JKKg%g. (//0kk g&,,^<<_l]))1=E--2t$$ *kk   XXII.IIIII   rFct|tr|drdS|rdS|tus|tus |t urdSt|t trdSg}tj|r||vrdStj |rt|tj |pi}| |}|rE|j dvrdSttdt d}t|j|rdSt|tr|d SdS)aReturns whether a member should be included in auto-completion or help. Determines whether a member of an object with the specified name should be included in auto-completion or help text(both usage and detailed help). If the member name starts with '__', it will always be excluded. If it starts with only one '_', it will be included for all non-string types. If verbose is True, the members, including the private members, are included. When not in verbose mode, some modules and functions are excluded as well. Args: component: The component containing the member. name: The name of the member. member: The member itself. class_attrs: (optional) If component is a class, provide this as: inspectutils.GetClassAttrsDict(component). If not provided, it will be computed. verbose: Whether to include private members. Returns A boolean value indicating whether the member should be included. __FTN)methodproperty _tuplegetter_) isinstancestr startswithrrrtypeinspectismoduleisclassrGetClassAttrsDictgetkindgetattr collectionsobject)rr member class_attrsverbosemodules_to_hide class_attr tuplegetters r MemberVisiblerasa.ctt44 5  4  8   > ! ! 5_--.. 5/ f&O";"; 5 _Y 29==Ck&&J  2 2 2uKdDDk J%{ 3 3uc$s## ## rcttr}ntj}t jfd|DS)aReturns a list of the members of the given component. If verbose is True, then members starting with _ (normally ignored) are included. Args: component: The component whose members to list. class_attrs: (optional) If component is a class, you may provide this as: inspectutils.GetClassAttrsDict(component). If not provided, it will be computed. If provided, this determines how class members will be treated for visibility. In particular, methods are generally hidden for non-instantiated classes, but if you wish them to be shown (e.g. for completion scripts) then pass in a different class_attr for them. verbose: Whether to include private members. Returns: A list of tuples (member_name, member) of all members of the component. Nc Fg|]\}}t||||fS)r\r])ra)rA member_namer[r\rr]s r z"VisibleMembers..osR    3 V y+v;& ( ( ( F   r)rNdictitemsrR getmembersrrU)rr\r]memberss``` rVisibleMembersrkUs$ 4  ,ooGG ++G0;;K      7>   rcpg}|D]0}|dd}|d|1|S)zTakes a list of fn args and returns a list of the fn's completion strings. Args: fn_args: A list of the args accepted by a function. Returns: A list of possible completion strings for that function. rM-r<)r1r0)fn_args completionsargs r_CompletionsFromArgsrqvsN+ ##c ++c3  CzCzz"""" rctj|stj|r0tj|}t |j|jzSt|ttfr&dtt|DStj |rgSdt||DS)aGives possible Fire command completions for the component. A completion is a string that can be appended to a command to continue that command. These are used for TAB-completions in Bash for Fire CLIs. Args: component: The component whose completions to list. verbose: Whether to include all completions, even private members. Returns: A list of completions for a command that would so far return the component. c,g|]}t|Sr)rO)rAindexs rrfzCompletions..s : : :5CJJ : : :rc2g|]\}}t|Sr)_FormatForCommand)rArerMs rrfzCompletions..s4    +q $$   rr])rR isroutinerTrGetFullArgSpecrqargs kwonlyargsrNtuplelistrangelen isgeneratorrk)rr]specs r Completionsrs y!!=W_Y%?%?=  &y 1 1D  DO ; < << E4=)); : :E#i..$9$9 : : :: ## I  *9gFFF   rct|tst|}|dr|S|ddS)aReplaces underscores with hyphens, unless the token starts with a token. This is because we typically prefer hyphens to underscores at the command line, but we reserve hyphens at the start of a token for flags. This becomes relevant when --verbose is activated, so that things like __str__ don't get transformed into --str--, which would get confused for a flag. Args: token: The token to transform. Returns: The transformed token. rMrm)rNrOrPr1)tokens rrvrvsP E3   JJE c L sC  rc#\Ktj|stj|rt|dD]}|fVtj|rdS|dkrdSt |idD]7\}}t |}|fVt ||dz D] }|f|zV 8dS)a Yields tuples representing commands. To use the command from Python, insert '.' between each element of the tuple. To use the command from the command line, insert ' ' between each element of the tuple. Args: component: The component considered to be the root of the yielded commands. depth: The maximum depth with which to traverse the member DAG for commands. Yields: Tuples, each tuple representing one possible command for this CLI. Only traverses the member DAG up to a depth of depth. FrwNrd)rRrxrTrrkrvr )rdepth completionrer[rs rr r s  y!!W_Y%?%?!)U;;; M y!!  F QYY F,I249;;;%%k6$K00K .VUQY//%% NW $$$$$%%%rc,|dS)Nrm)rP)rps r _IsOptionrs   rcftj}tjfd}tjt}|D]}t |dkrSt |dr||dF|||dh|r{|d}t|d}t |r|} n|} | ||| |dd||||fS)aReturns sets of subcommands and options for each command. Args: name: The first token in the commands, also the name of the command. commands: A list of all possible commands that tab completion can complete to. Each command is a list or tuple of the string tokens that make up that command. default_options: A dict of options that can be used with any command. Use this if there are flags that can always be appended to a command. Returns: global_options: A set of all options of the first token of the command. subcommands_map: A dict storing set of subcommands for each command/subcommand. options_map: A dict storing set of options for each subcommand. c,tjSr)copy)rsrz_GetMaps..s /0J0JrrrrMrm) rrY defaultdictr(rrr*rvr1) r r2rr'r3r4rr;rpargs_maps ` rr)r)s9 9_--.'(J(J(J(JKK++C00/66g 7||q 71:  .71:&&&&!!'!*----  62;j gbk * *c 3#"zs###z!!#s++,00555 o 55r)Nrr)NF)F)r)__doc__ __future__rrrrYrrRfirerrr r rarkrqrrvr rr)rrrrsWFE&&&&&&%%%%%% BBBB ZZZZzZZZZz6666rB   :!!!, % % % %F'6'6'6'6'6r