�Ww � @ s� d d l Z d d l Z d d l m Z d d l m Z d d l m Z m Z d Z e j
d � Z d d � Z d
d � Z
d d
� Z d d � Z d d � Z d S)� N� )�echo)�split_arg_string)� MultiCommand�Optionz�
%(complete_func)s() {
COMPREPLY=( $( env COMP_WORDS="${COMP_WORDS[*]}" \
COMP_CWORD=$COMP_CWORD \
%(autocomplete_var)s=complete $1 ) )
return 0
}
complete -F %(complete_func)s -o default %(script_names)s
z
[^a-zA-Z0-9_]c C sF t j d | j d d � � } t d d | d | d | i j � d S) N� �-�_Z
complete_funcz_%s_completionZ script_namesZautocomplete_var�;)�_invalid_ident_char_re�sub�replace�COMPLETION_SCRIPT�strip)� prog_name� complete_varZcf_name� r �/_bashcomplete.py�get_completion_script s
r c C s� | j | | d d �} x� | j | j r� t | j t � r� | j | j } | j j | | d � } | d k rv d S| j | d | d d � d | d d �} q W| S)NZresilient_parsingTr r �parent)Z make_context�argsZprotected_args�
isinstance�commandr Z get_command)�clir r �ctx�a�cmdr r r � resolve_ctx s % 0r c c s� t | | | � } | d k r" d Sg } | r� | d d � j � r� xt | j j D]8 } t | t � sj qR | j | j � | j | j � qR Wn+ t | j t � r� | j | j j
| � � x" | D] } | j | � r� | Vq� Wd S)Nr ) r �isalnumr Zparamsr r �extendZoptsZsecondary_optsr Z
list_commands�
startswith)r r r �
incompleter �choicesZparam�itemr r r � get_choices* s
r$ c
C s� t t j d � } t t j d � } | d | � } y | | } Wn t k
r^ d } Yn Xx' t | | | | � D] } t | � qu Wd S)NZ
COMP_WORDSZ
COMP_CWORDr r T)r �os�environ�int�
IndexErrorr$ r )r r ZcwordsZcwordr r! r# r r r � do_complete> s
r) c C s@ | d k r# t t | | � � d S| d k r< t | | � Sd S)N�sourceTZcompleteF)r r r) )r r r Zcomplete_instrr r r � bashcompleteM s
r+ )r% �reZutilsr �parserr Zcorer r r �compiler r r r$ r) r+ r r r r �<module> s