_0@sddlZddlZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z ydd l mZWnek rddl ZYnXd Zd Zd Zd ZdededeiZejdZddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Z d&d'Z!d(d)Z"dS)*N)Argument) MultiCommand)Option)split_arg_string)Choice)echo)abc=a %(complete_func)s() { local IFS=$' ' COMPREPLY=( $( env COMP_WORDS="${COMP_WORDS[*]}" \ COMP_CWORD=$COMP_CWORD \ %(autocomplete_var)s=complete $1 ) ) return 0 } %(complete_func)setup() { local COMPLETION_OPTIONS="" local BASH_VERSION_ARR=(${BASH_VERSION//./ }) # Only BASH version 4.4 and later have the nosort option. if [ ${BASH_VERSION_ARR[0]} -gt 4 ] || ([ ${BASH_VERSION_ARR[0]} -eq 4 ] && [ ${BASH_VERSION_ARR[1]} -ge 4 ]); then COMPLETION_OPTIONS="-o nosort" fi complete $COMPLETION_OPTIONS -F %(complete_func)s %(script_names)s } %(complete_func)setup a #compdef %(script_names)s %(complete_func)s() { local -a completions local -a completions_with_descriptions local -a response (( ! $+commands[%(script_names)s] )) && return 1 response=("${(@f)$( env COMP_WORDS="${words[*]}" \ COMP_CWORD=$((CURRENT-1)) \ %(autocomplete_var)s="complete_zsh" \ %(script_names)s )}") for key descr in ${(kv)response}; do if [[ "$descr" == "_" ]]; then completions+=("$key") else completions_with_descriptions+=("$key":"$descr") fi done if [ -n "$completions_with_descriptions" ]; then _describe -V unsorted completions_with_descriptions -U fi if [ -n "$completions" ]; then compadd -U -V unsorted -a completions fi compstate[insert]="automenu" } compdef %(complete_func)s %(script_names)s zcomplete --no-files --command %(script_names)s --arguments "(env %(autocomplete_var)s=complete_fish COMP_WORDS=(commandline -cp) COMP_CWORD=(commandline -t) %(script_names)s)"bashzshfishz [^a-zA-Z0-9_]cCs]tjd|jdd}tj|t}|ddj|d|d|ijdS) N-_Z complete_funcz_{}_completionZ script_namesZautocomplete_var;)_invalid_ident_char_resubreplace_completion_scriptsgetCOMPLETION_SCRIPT_BASHformatstrip) prog_name complete_varshellZcf_namescriptr4/tmp/pip-build-5gj8f0j9/click/click/_bashcomplete.pyget_completion_script_sr c Cs>|j||dd}|j|j}x|r9t|jtr5|jjs|jj||\}}}|dkr}|S|j||d|dd}|j|j}q6xk|r|jj||\}}}|dkr|S|j||d|dddddd}|j}qW|}|j|j}q+Pq+W|S)aParse into a hierarchy of contexts. Contexts are connected through the parent variable. :param cli: command definition :param prog_name: the program that is running :param args: full list of args :return: the final context/command parsed Zresilient_parsingTNparentZallow_extra_argsallow_interspersed_argsF)Z make_contextprotected_argsargs isinstancecommandrchainZresolve_command)clirr$ctxcmd_namecmdZsub_ctxrrr resolve_ctxls6        r,cCs|o|dddkS)z :param param_str: param_str to check :return: whether or not this is the start of an option declaration (i.e. starts "-" or "--") Nrrr)Z param_strrrrstart_of_optionsr-cCst|tsdS|jr dSd}xSttdd|DD]2\}}|d|jkrfPt|rF|}qFW|r||jkrdSdS)aU :param all_args: the full original list of args supplied :param cmd_param: the current command paramter :return: whether or not the last option declaration (i.e. starts "-" or "--") is incomplete and corresponds to this cmd_param. In other words whether this cmd_param option can still accept values FNcSs"g|]}|tkr|qSr) WORDBREAK).0argrrr s z(is_incomplete_option..rT)r%rZis_flag enumeratereversednargsr-opts)all_args cmd_paramZ last_optionindexZarg_strrrris_incomplete_options  &  r9cCst|tsdS||j}|dkr0dS|jdkrCdSt|tjr}|jdkr}t||jkr}dSdS)a` :param current_params: the current params and values for this argument as already entered :param cmd_param: the current command parameter :return: whether or not the last argument is incomplete and corresponds to this cmd_param. In other words whether or not the this cmd_param argument can still accept values FNTr)r%rnamer4r Iterablelen)Zcurrent_paramsr7Zcurrent_param_valuesrrris_incomplete_arguments   r>cs~g}t|jtr:fdd|jjD}n@|jdk rz|jd|d|d}dd|D}|S)a :param ctx: context associated with the parsed command :param args: full list of args :param incomplete: the incomplete text to autocomplete :param cmd_param: command definition :return: all the possible user-specified completions for the param cs1g|]'}t|jr|dfqS)N)str startswith)r/c) incompleterrr1s z,get_user_autocompletions..Nr)r$rBcSs1g|]'}t|tr!|n |dfqS)N)r%tuple)r/rArrrr1s )r%typerchoicesZautocompletion)r)r$rBr7resultsZdynamic_completionsr)rBrget_user_autocompletionss" rGccsSxL|jj|D]8}|j|r|jj||}|js|VqWdS)z :param ctx: context associated with the parsed command :starts_with: string that visible commands must start with. :return: all visible (not hidden) commands that start with starts_with. N)r&Z list_commandsr@Z get_commandhidden)r)Z starts_withrAr&rrr"get_visible_commands_starting_withs  rIcstjtr5|jddt|Dxvjdk rjtjtr8jjr8fddt|D}|jdd|Dq8WdS)NcSs%g|]}|j|jfqSr)r;get_short_help_str)r/rArrrr1s z.add_subcommand_completions..cs(g|]}|jjkr|qSr)r;r#)r/rA)r)rrr1s cSs%g|]}|j|jfqSr)r;rJ)r/rArrrr1 s )r%r&rextendrIr!r')r)rBZcompletions_outZremaining_commandsr)r)radd_subcommand_completionss  rLc stj|t|||}|dkr1gSdk}trtkrjt}j|d|dntkrdg}| r*tr*xv|jjD]ht t rj rfddj j D}|jfdd|DqW|Sx6|jjD](tr7t|Sq7Wx9|jjD]+t|jrpt|SqpWt||t|S) z :param cli: command definition :param prog_name: the program that is running :param args: full list of args :param incomplete: the incomplete text to autocomplete :return: all the possible completions for the incomplete Nz--rrcs+g|]!}|ks!jr|qSr)multiple)r/Z param_opt)r6paramrrr1/s zget_choices..cs.g|]$}|jr|jfqSr)r@help)r/o)rBrOrrr14s )copydeepcopyr,r-r. partitionappendr&paramsr%rrHr5Zsecondary_optsrKr9rGr>rLsorted) r(rr$rBr)Zhas_double_dashZpartition_incompleteZ completionsZ param_optsr)r6rBrOr get_choicess8    !rXc Csttjd}ttjd}|d|}y||}Wntk r^d}YnXxOt||||D]8}t|d|rut|dr|dndquWdS)N COMP_WORDS COMP_CWORDrrrrT)rosenvironint IndexErrorrXr)r(rZinclude_descriptionscwordscwordr$rBitemrrr do_completeEs  "rbcCsttjd}tjd}|dd}x\t||||D]E}|dr}tdjd|dd|dqFt|dqFWdS) NrYrZrz {arg} {desc}r0rdescT)rr[r\rXrr)r(rr_rBr$rarrrdo_complete_fishYs  'rdcCsd|kr'|jdd\}}n |}d}|dkrYtt|||dS|dkr|dkr~t||S|d krt|||dkSd S) Nrrr sourceTZcompleter r F>r r )splitrr rdrb)r(rrZcomplete_instrr&rrrr bashcompletegs      rg)#rRr[recorerrrparserrtypesrutilsr collectionsr ImportErrorr.rZCOMPLETION_SCRIPT_ZSHZCOMPLETION_SCRIPT_FISHrcompilerr r,r-r9r>rGrIrLrXrbrdrgrrrrs@    #  +     5