CasperSecurity
o
�+caQF � @ s, U d dl Z d dlZd dlZd dlmZ ddlmZ ddlmZ ddlm Z ddlm
Z
ddlm Z dd lm Z dd
lm
Z
dd lmZ dd lmZ d
edejeejf dedededef dd�ZG dd� d�ZdZdZdZG dd� d�ZG d d � d e�Z G d d � d e�Z G d d!� d!e�Z e e e d"�Z ejeej e f e!d#< d:d$ej e d%ej"e ddfd&d'�Z#d(edej"ej e fd)d*�Z$d+e d,e de%fd-d.�Z&d/ede%fd0d1�Z'd2ej(e d,e de%fd3d4�Z)d
edejeejf ded2ej(e de f
d5d6�Z*d+e d2ej(e d7edej+ej,ee f ef fd8d9�Z-dS );� N)�gettext� )�Argument)� BaseCommand)�Context)� MultiCommand)�Option)� Parameter)�ParameterSource)�split_arg_string)�echo�cli�ctx_args� prog_name� complete_var� instruction�returnc C sf |� d�\}}}t|�}|du rdS || |||�}|dkr%t|�� � dS |dkr1t|�� � dS dS )a Perform shell completion for the given CLI program.
:param cli: Command being called.
:param ctx_args: Extra arguments to pass to
``cli.make_context``.
:param prog_name: Name of the executable in the shell.
:param complete_var: Name of the environment variable that holds
the completion instruction.
:param instruction: Value of ``complete_var`` with the completion
instruction and shell, in the form ``instruction_shell``.
:return: Status code to exit with.
�_Nr �sourcer �complete)� partition�get_completion_classr r r ) r
r r r r �shellr �comp_cls�comp� r �8/usr/lib/python3/dist-packages/click/shell_completion.py�shell_complete s r c @ sV e Zd ZdZdZ ddejdedeje dejd df
d
d �Z d ed ejfd
d�Z
dS )�CompletionItema) Represents a completion value and metadata about the value. The
default metadata is ``type`` to indicate special shell handling,
and ``help`` if a shell supports showing a help string next to the
value.
Arbitrary parameters can be passed when creating the object, and
accessed using ``item.attr``. If an attribute wasn't passed,
accessing it returns ``None``.
:param value: The completion suggestion.
:param type: Tells the shell script to provide special completion
support for the type. Click uses ``"dir"`` and ``"file"``.
:param help: String shown next to the value if supported.
:param kwargs: Arbitrary metadata. The built-in implementations
don't use this, but custom type completions paired with custom
shell support could use it.
��value�type�help�_info�plainNr r! r" �kwargsr c K � || _ || _|| _|| _d S �Nr )�selfr r! r" r% r r r �__init__L �
zCompletionItem.__init__�namec C s | j �|�S r' )r# �get)r( r+ r r r � __getattr__X s zCompletionItem.__getattr__)r$ N) �__name__�
__module__� __qualname__�__doc__� __slots__�t�Any�str�Optionalr) r- r r r r r 7 s" �����
� r a� %(complete_func)s() {
local IFS=$'\n'
local response
response=$(env COMP_WORDS="${COMP_WORDS[*]}" COMP_CWORD=$COMP_CWORD %(complete_var)s=bash_complete $1)
for completion in $response; do
IFS=',' read type value <<< "$completion"
if [[ $type == 'dir' ]]; then
COMREPLY=()
compopt -o dirnames
elif [[ $type == 'file' ]]; then
COMREPLY=()
compopt -o default
elif [[ $type == 'plain' ]]; then
COMPREPLY+=($value)
fi
done
return 0
}
%(complete_func)s_setup() {
complete -o nosort -F %(complete_func)s %(prog_name)s
}
%(complete_func)s_setup;
a� #compdef %(prog_name)s
%(complete_func)s() {
local -a completions
local -a completions_with_descriptions
local -a response
(( ! $+commands[%(prog_name)s] )) && return 1
response=("${(@f)$(env COMP_WORDS="${words[*]}" COMP_CWORD=$((CURRENT-1)) %(complete_var)s=zsh_complete %(prog_name)s)}")
for type key descr in ${response}; do
if [[ "$type" == "plain" ]]; then
if [[ "$descr" == "_" ]]; then
completions+=("$key")
else
completions_with_descriptions+=("$key":"$descr")
fi
elif [[ "$type" == "dir" ]]; then
_path_files -/
elif [[ "$type" == "file" ]]; then
_path_files -f
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
}
compdef %(complete_func)s %(prog_name)s;
a� function %(complete_func)s;
set -l response;
for value in (env %(complete_var)s=fish_complete COMP_WORDS=(commandline -cp) COMP_CWORD=(commandline -t) %(prog_name)s);
set response $response $value;
end;
for completion in $response;
set -l metadata (string split "," $completion);
if test $metadata[1] = "dir";
__fish_complete_directories $metadata[2];
else if test $metadata[1] = "file";
__fish_complete_path $metadata[2];
else if test $metadata[1] = "plain";
echo $metadata[2];
end;
end;
end;
complete --no-files --command %(prog_name)s --arguments "(%(complete_func)s)";
c
@ s� e Zd ZU dZeje ed< eje ed< dedej eej
f dededd f
d
d �Z e defd d
��Z
dej eej
f fdd�Zdefdd�Zdejeje ef fdd�Zdeje dedeje fdd�Zdedefdd�Zdefdd �Zd S ) �
ShellCompletea� Base class for providing shell completion support. A subclass for
a given shell will override attributes and methods to implement the
completion instructions (``source`` and ``complete``).
:param cli: Command being called.
:param prog_name: Name of the executable in the shell.
:param complete_var: Name of the environment variable that holds
the completion instruction.
.. versionadded:: 8.0
r+ �source_templater
r r r r Nc C r& r' )r
r r r )r( r
r r r r r r r) � r* zShellComplete.__init__c C s( t �dd| j�dd�t j�}d|� d�S )zQThe name of the shell function defined by the completion
script.
z\W*� �-r � _completion)�re�subr �replace�ASCII)r( � safe_namer r r � func_name� s zShellComplete.func_namec C s | j | j| jd�S )z�Vars for formatting :attr:`source_template`.
By default this provides ``complete_func``, ``complete_var``,
and ``prog_name``.
)�
complete_funcr r )rA r r �r( r r r � source_vars� s �zShellComplete.source_varsc C s | j | �� S )z�Produce the shell script that defines the completion
function. By default this ``%``-style formats
:attr:`source_template` with the dict returned by
:meth:`source_vars`.
)r8 rD rC r r r r � s zShellComplete.sourcec C � t �)z�Use the env vars defined by the shell script to return a
tuple of ``args, incomplete``. This must be implemented by
subclasses.
��NotImplementedErrorrC r r r �get_completion_args� s z!ShellComplete.get_completion_args�args�
incompletec C s0 t | j| j| j|�}t|||�\}}|�||�S )aT Determine the context and last complete command or parameter
from the complete args. Call that object's ``shell_complete``
method to get the completions for the incomplete value.
:param args: List of complete args before the incomplete value.
:param incomplete: Value being completed. May be empty.
)�_resolve_contextr
r r �_resolve_incompleter )r( rI rJ �ctx�objr r r �get_completions s
z ShellComplete.get_completions�itemc C rE )z�Format a completion item into the form recognized by the
shell script. This must be implemented by subclasses.
:param item: Completion item to format.
rF �r( rP r r r �format_completion s z ShellComplete.format_completionc s4 � � � \}}� �||�}� fdd�|D �}d�|�S )z�Produce the completion data to send back to the shell.
By default this calls :meth:`get_completion_args`, gets the
completions, then calls :meth:`format_completion` for each
completion.
c s g | ]}� � |��qS r )rR )�.0rP rC r r �
<listcomp>$ s z*ShellComplete.complete.<locals>.<listcomp>�
)rH rO �join)r( rI rJ � completions�outr rC r r s
zShellComplete.complete)r. r/ r0 r1 r3 �ClassVarr5 �__annotations__r �Dictr4 r) �propertyrA rD r �Tuple�ListrH r rO rR r r r r r r7 � s>
����
� ��
�r7 c sh e Zd ZdZdZeZddd�Zdef� fdd� Z de
j e
j e ef fd d
�Z
d edefd d
�Z� ZS )� BashCompletezShell completion for Bash.�bashr Nc C sz dd l }|jg d�|jd�}t�d|j�� �}|d ur7|�� \}}|dk s-|dkr3|dk r5tt d���d S d S tt d���)Nr )r` z-czecho ${BASH_VERSION})�stdoutz^(\d+)\.(\d+)\.\d+�4zCShell completion is not supported for Bash versions older than 4.4.z@Couldn't detect Bash version, shell completion is not supported.)
�
subprocess�run�PIPEr< �searchra �decode�groups� RuntimeErrorr )r( rc �output�match�major�minorr r r �_check_version. s"
� ����zBashComplete._check_versionc s | � � t� �� S r' )rn �superr rC �� __class__r r r E s
zBashComplete.sourcec C �X t tjd �}ttjd �}|d|� }z || }W ||fS ty+ d}Y ||fS w �N�
COMP_WORDS�
COMP_CWORDr r9 �r �os�environ�int�
IndexError�r( �cwords�cwordrI rJ r r r rH I �
��z BashComplete.get_completion_argsrP c C s |j � d|j� �S )N�,)r! r rQ r r r rR U s z BashComplete.format_completion)r N)r. r/ r0 r1 r+ � _SOURCE_BASHr8 rn r5 r r3 r] r^ rH r rR �
__classcell__r r rp r r_ ( s
r_ c @ �H e Zd ZdZdZeZdejej e
e
f fdd�Z de de
fdd�Z
d S )
� ZshCompletezShell completion for Zsh.�zshr c C rr rs rv r{ r r r rH _ r~ z ZshComplete.get_completion_argsrP c C s( |j � d|j� d|jr|j� �S d� �S )NrU r )r! r r" rQ r r r rR k s (z ZshComplete.format_completionN)r. r/ r0 r1 r+ � _SOURCE_ZSHr8 r3 r] r^ r5 rH r rR r r r r r� Y � r� c @ r� )
� FishCompletezShell completion for Fish.�fishr c C sH t tjd �}tjd }|dd � }|r |r |d |kr |�� ||fS )Nrt ru r ���)r rw rx �pop)r( r| rJ rI r r r rH u s
z FishComplete.get_completion_argsrP c C s2 |j r|j� d|j� d|j � �S |j� d|j� �S )Nr � )r" r! r rQ r r r rR � s z FishComplete.format_completionN)r. r/ r0 r1 r+ � _SOURCE_FISHr8 r3 r] r^ r5 rH r rR r r r r r� o r� r� )r` r� r� �_available_shells�clsr+ c C s |du r| j }| t|< dS )am Register 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� )r� r+ r r r �add_completion_class� s r� r c C s
t �| �S )z�Look 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.
)r� r, )r r r r r � s
r rM �paramc C sj t |t�sdS |jdusJ �| j|j }|jdkp4| �|j�tjup4|jdko4t |tt f�o4t
|�|jk S )z�Determine 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.
FNr� r ) �
isinstancer r+ �params�nargs�get_parameter_sourcer
� COMMANDLINE�tuple�list�len)rM r� r r r r �_is_incomplete_argument� s
�
� �r� r c C s" | sdS | d }|� � o|dkS )z5Check if the value looks like the start of an option.Fr �/)�isalnum)r �cr r r �_start_of_option� s r� rI c C sb t |t�sdS |jr dS d}tt| ��D ]\}}|d |jkr! nt|�r'|}q|duo0||jv S )z�Determine 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 )r� r �is_flag� enumerate�reversedr� r� �opts)rI r� � last_option�index�argr r r �_is_incomplete_option� s
�r� c C s� d|d< | j ||�� fi |��}|j|j }|rs|j}t|t�rn|jsB|�||�\}}}|du r2|S |j |||dd�}|j|j }n/|rc|�||�\}}}|du rS|S |j |||dddd�}|j}|sD|}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.
T�resilient_parsingN)�parentr� F)r� �allow_extra_args�allow_interspersed_argsr� ) � make_context�copy�protected_argsrI �commandr� r �chain�resolve_command) r
r r rI rM r� r+ �cmd�sub_ctxr r r rK � s@
���"rK rJ c C s� |dkrd}nd|v r t |�r |�d�\}}}|�|� d|vr)t |�r)| j|fS | j�| �}|D ]
}t||�r>||f S q1|D ]
}t| |�rN||f S qA| j|fS )ah Find 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.
�=r9 z--)r� r �appendr� �
get_paramsr� r� )rM rI rJ r+ r r� r� r r r rL s"
�
�
rL r' ).rw r< �typingr3 r r �corer r r r r r r
�parserr �utilsr r[ r5 r4 ry r r r� r� r� r7 r_ r� r� r� �TyperZ r6 r� r �boolr� r� r^ r� rK r] �UnionrL r r r r �<module> s�
�����
�&& &j1 ����
�
����
�4����