o q>eH @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.. )rIrPjoin)r(rJrK completionsoutrrDrrs   zShellComplete.complete)r.r/r0r1r3ClassVarr5__annotations__rMutableMappingr4r)propertyrBDictrErTupleListrIrrPrSrrrrrr7s>      r7csleZdZdZdZeZedddZde 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)rbz-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=searchrcdecodegroupsr r)rfoutputmatchmajorminorrrr_check_version0s&     zBashComplete._check_versioncs|tSr')rpsuperrrD __class__rrrJs zBashComplete.sourcecCXttjd}ttjd}|d|}z ||}W||fSty+d}Y||fSwN COMP_WORDS COMP_CWORDrr9r osenvironint IndexErrorr(cwordscwordrJrKrrrrIN   z BashComplete.get_completion_argsrQcCs|jd|jS)N,)r!r rRrrrrSZszBashComplete.format_completion)rN)r.r/r0r1r+ _SOURCE_BASHr8 staticmethodrpr5rr3r_r`rIrrS __classcell__rrrrrra*s  rac@HeZdZdZdZeZdejej e e ffddZ de de fddZ d S) ZshCompletezShell completion for Zsh.zshrcCrtrurxr}rrrrIdrzZshComplete.get_completion_argsrQcCs(|jd|jd|jr|jSdS)NrVr)r!r r"rRrrrrSps(zZshComplete.format_completionN)r.r/r0r1r+ _SOURCE_ZSHr8r3r_r`r5rIrrSrrrrr^  rc@r) FishCompletezShell completion for Fish.fishrcCsHttjd}tjd}|dd}|r |r |d|kr |||fS)Nrvrwr)r ryrzpop)r(r~rKrJrrrrIzs   z FishComplete.get_completion_argsrQcCs2|jr|jd|jd|jS|jd|jS)Nr )r"r!r rRrrrrSszFishComplete.format_completionN)r.r/r0r1r+ _SOURCE_FISHr8r3r_r`r5rIrrSrrrrrtrrShellCompleteType)bound)rbrr_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 rrNparamcCslt|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,nargsget_parameter_sourcer COMMANDLINEtuplelistlen)rNrr rrr_is_incomplete_arguments     rr cCs|sdS|d}||jvS)z5Check if the value looks like the start of an option.Fr) _opt_prefixes)rNr crrr_start_of_options rrJcCsjt|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) rris_flagcount enumeratereversedrropts)rNrJr 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)rallow_extra_argsallow_interspersed_argsr) make_contextcopyprotected_argsrJcommandrrchainresolve_command) r rrrJrNrr+cmdsub_ctxrrrrLsB  $rLrKcCs|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. =r9z--)rrappendr get_paramsrr)rNrJrKr+rrrrrrrM(s"        rMr')1ryr=typingr3rrcorerrrrrr r parserr utilsr r\r5r4r{rrrrrr7rarrTypeVarTyperrr^r[r6rrboolrrr`rrLr_UnionrMrrrrs            && ,j4    9