o 96dH @sBUddlZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZdd lmZd edejeejfdedededef ddZGdddZdZdZdZGdddZGdddeZGdddeZGd d!d!eZejd"ej ed#Z!eeed$Z"ej#eej efe$d%< d&sz*ShellComplete.complete.. )rFrMjoin)r'rGrHZ completionsoutrrArrs   zShellComplete.complete)r-r.r/r0r2ClassVarr4__annotations__rMutableMappingr3r(propertyr@DictrBrTupleListrFrrMrPrrrrrr6s>      r6csleZdZdZdZeZedddZde ffdd Z de j e j e e ffd d Zd ede fd d ZZS) BashCompletezShell completion for Bash.bashrNcCsddl}|jgd|jd}td|j}|dur;|\}}|dks-|dkr7|dkr9tt ddddSdSdStt d dddS) Nr)r^z-czecho "${BASH_VERSION}")stdoutz^(\d+)\.(\d+)\.\d+4zCShell completion is not supported for Bash versions older than 4.4.T)errz@Couldn't detect Bash version, shell completion is not supported.) subprocessrunPIPEr;searchr_decodegroupsr r)rboutputmatchmajorminorrrr_check_version0s&     zBashComplete._check_versioncs|tSr&)rlsuperrrA __class__rrrJs zBashComplete.sourcecCXttjd}ttjd}|d|}z ||}W||fSty+d}Y||fSwN COMP_WORDS COMP_CWORDrr8r osenvironint IndexErrorr'cwordscwordrGrHrrrrFN   z BashComplete.get_completion_argsrNcCs|jd|jS)N,)r rrOrrrrPZszBashComplete.format_completion)rN)r-r.r/r0r* _SOURCE_BASHr7 staticmethodrlr4rr2r[r\rFrrP __classcell__rrrnrr]*s  r]c@HeZdZdZdZeZdejej e e ffddZ de de fddZ d S) ZshCompletezShell completion for Zsh.zshrcCrprqrtryrrrrFdr|zZshComplete.get_completion_argsrNcCs(|jd|jd|jr|jSdS)NrSr)r rr!rOrrrrPps(zZshComplete.format_completionN)r-r.r/r0r* _SOURCE_ZSHr7r2r[r\r4rFrrPrrrrr^  rc@r) FishCompletezShell completion for Fish.fishrcCsHttjd}tjd}|dd}|r |r |d|kr |||fS)Nrrrsr)r rurvpop)r'rzrHrGrrrrFzs   z FishComplete.get_completion_argsrNcCs2|jr|jd|jd|jS|jd|jS)Nr} )r!r rrOrrrrPszFishComplete.format_completionN)r-r.r/r0r* _SOURCE_FISHr7r2r[r\r4rFrrPrrrrrtrrShellCompleteType)bound)r^rr_available_shellsclsr*cCs|dur|j}|t|<|S)amRegister a :class:`ShellComplete` subclass under the given name. The name will be provided by the completion instruction environment variable during completion. :param cls: The completion class that will handle completion for the shell. :param name: Name to register the class under. Defaults to the class's ``name`` attribute. N)r*r)rr*rrradd_completion_classs rrcCs t|S)zLook up a registered :class:`ShellComplete` subclass by the name provided by the completion instruction environment variable. If the name isn't registered, returns ``None``. :param shell: Name the class is registered under. )rr+)rrrrrs rrKparamcCslt|tsdS|jdusJ|j|j}|jdkp5||jtjup5|jdko5t|t t fo5t ||jkS)zDetermine if the given parameter is an argument that can still accept values. :param ctx: Invocation context for the command represented by the parsed complete args. :param param: Argument object being checked. FNrr) isinstancerr*paramsr+nargsZget_parameter_sourcer Z COMMANDLINEtuplelistlen)rKrrrrr_is_incomplete_arguments     rrcCs|sdS|d}||jvS)z5Check if the value looks like the start of an option.Fr)Z _opt_prefixes)rKrcrrr_start_of_options rrGcCsjt|tsdS|js |jrdSd}tt|D]\}}|d|jkr$nt||r+|}q|duo4||jvS)zDetermine if the given parameter is an option that needs a value. :param args: List of complete args before the incomplete value. :param param: Option object being checked. FNr) rrZis_flagcount enumeratereversedrropts)rKrGrZ last_optionindexargrrr_is_incomplete_options   rc Csd|d<|j||fi|}|j|j}|ru|j}t|trp|jsB|||\}}}|dur2|S|j|||dd}|j|j}n1|}|re|||\}}}|durU|S|j|||dddd}|j}|sF|}g|j|j}n |S|s|S)a`Produce the context hierarchy starting with the command and traversing the complete arguments. This only follows the commands, it doesn't trigger input prompts or callbacks. :param cli: Command being called. :param prog_name: Name of the executable in the shell. :param args: List of complete args before the incomplete value. Tresilient_parsingN)parentrF)rZallow_extra_argsallow_interspersed_argsr) Z make_contextcopyZprotected_argsrGcommandrrchainZresolve_command) r rrrGrKrr*cmdZsub_ctxrrrrIsB  $rIrHcCs|dkrd}nd|vrt||r|d\}}}||d|vr+t||r+|j|fS|j|}|D]}t|||rA||fSq3|D] }t||rQ||fSqD|j|fS)ahFind the Click object that will handle the completion of the incomplete value. Return the object and the incomplete value. :param ctx: Invocation context for the command represented by the parsed complete args. :param args: List of complete args before the incomplete value. :param incomplete: Value being completed. May be empty. =r8z--)rrappendr get_paramsrr)rKrGrHr*rrrrrrrJ(s"        rJr&)1rur;typingr2rrcorerrrrrr r parserr utilsr rXr4r3rwrrr~rrr6r]rrTypeVarTyperrrZrWr5rrboolrrr\rrIr[UnionrJrrrrs            && ,j4    9