File: //usr/local/CyberPanel/lib64/python3.10/site-packages/asyncssh/__pycache__/editor.cpython-310.pyc
o
�h�v � @ s d Z ddlZddlmZ ddlmZmZmZmZ ddlm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZ er@dd lmZ dd
lmZ eeef Zed gdf Zeeege ee eef f f Zd Z d
Z dedefdd�Z G dd � d �Z G dd� d�Z G dd� d�Z!dS )�Input line editor� N)�partial)�
TYPE_CHECKING�Callable�Dict�List)�Optional�Set�Tuple�Union�cast��east_asian_width� )�DataType)�SSHServerChannel)�SSHServerSession�
SSHLineEditor�P )�ansi�cygwin�linux�putty�screen�teraterm�cit80�vt100�vt102�vt220�vt320�xtermz xterm-colorz
xterm-16colorzxterm-256color�rxvtz
rxvt-color�ch�returnc C s t | �dv S )z!Return display width of character�WFr
�r" � r& �E/usr/local/CyberPanel/lib/python3.10/site-packages/asyncssh/editor.py�_is_wide4 � r( c @ s e Zd ZdZdddddededed ed
efd d �Zd
ededdfdd�Z d
eddfdd�Z
d~dd�Z dedefdd�Z ddedede
e deeef fd d �Zddede
e ddfd d �Zd~d d!�Zd"edefd#d$�Zdeddfd%d&�Zdeddfd'd(�Zd~d)d*�Zd~d+d,�Zd~d-d.�Zd"ed/eddfd0d1�Zdeded"eddfd2d3�Zd~d4d5�Zd~d6d7�Zdeddfd8d9�Z d~d:d;�Z d~d<d=�Z d~d>d?�Z d~d@dA�Z d~dBdC�Z!d~dDdE�Z"d
edFe#ddfdGdH�Z$d~dIdJ�Z%d~dKdL�Z&d~dMdN�Z'd~dOdP�Z(d~dQdR�Z)d~dSdT�Z*d~dUdV�Z+d~dWdX�Z,d~dYdZ�Z-e d[fe d\fe d]fe d^fe!d_fe"d`fe%dafe&dbfe'dcfe(ddfe)defe*dffe+dgfe,dhfe-diffZ.d
edFe#ddfdjdk�Z/d
eddfdldm�Z0dnededdfdodp�Z1dqeddfdrds�Z2dteddfdudv�Z3d
eddfdwdx�Z4dedye5ddfdzd{�Z6deddfd|d}�Z7dS )�r r �chan�SSHServerChannel[str]�session�SSHServerSession[str]� line_echo� history_size�max_line_length� term_type�widthc C s� || _ || _|| _d| _|dkr|nd| _|| _|tv | _|p t| _ d| _
d| _ d| _ d| _
d| _d| _d| _d| _d| _d| _t� | _g | _i | _| j| _d| _g | _d| _| j D ]\}} | D ]}
| � |
|� q`qZ| � � d S )NFr T� ) �_chan�_session�
_line_echo�
_line_pending�
_history_size�_max_line_length�_ansi_terminals�_wrap�_DEFAULT_WIDTH�_width�
_line_mode�_echo�
_start_column� _end_column�_cursor� _left_pos�
_right_pos�_pos�_line�
_bell_rung�set� _early_wrap�_outbuf�_keymap�
_key_state�_erased�_history�_history_index�_keylist�_add_key�_build_printable) �selfr* r, r. r/ r0 r1 r2 �func�keys�keyr& r& r' �__init__= s<
� zSSHLineEditor.__init__rV rT r# Nc C sF | j }|dd� D ]}||vri ||< tt|| �}q |||d < dS )zAdd a key to the keymapN���)rK r � _CharDict)rS rV rT �keymapr" r& r& r' rQ a s zSSHLineEditor._add_keyc C sH | j }|dd� D ]}||vr dS tt|| �}q |�|d d� dS )z Delete a key from the keymapNrX )rK r rY �pop)rS rV rZ r" r& r& r' �_del_keyn s zSSHLineEditor._del_keyc s� dt dtfdd�}dtdtf� fdd� }g }| j�� � td�}d }||k ro||k r=|t|��s=|d
7 }||k r=|t|��r/|}|t|��rO|d
7 }|t|��sE|�||�� ||d
krg|�d ||d
� � |d
}||k s%t� d d
�
|� d �| _ dS )z4Build a regex of printable ASCII non-registered keys�cr# c S s t | �}|dv r d| S d| S )z<Backslash escape special characters in regex character rangez-&|[]\^~�\r3 )�chr)r] r" r& r& r' �_escape~ s z/SSHLineEditor._build_printable.<locals>._escaper" c s | � � o| � vS )z3Return if character is printable and has no handler)� isprintabler% �rU r& r' �
_is_printable� � z5SSHLineEditor._build_printable.<locals>._is_printable� i r �-�[r3 z]*N) �int�str�boolrK rU �ordr_ �append�re�compile�join�
_printable)rS r` rc �pat�start�limit�endr& rb r' rR { s(
� � � z SSHLineEditor._build_printable�posc C s dt | j| � |d | jv S )z/Return width of character at specified positionr )r( rF rI �rS ru r& r& r' � _char_width� s zSSHLineEditor._char_width�data�columnc C s< d}|}|}g }|D ]m}|dkr|d8 }nZ|dkr d}nS|r%|dkr$d}nJt |�r]|| j | jd kr]|d7 }|dur\|dusAJ �|dusGJ �|�||| || � � |}| j�|� n
|durg| j�|� |dt |� 7 }|durw|d7 }q
|dur�|dus�J �|dus�J �|�||| d� � d�|�|fS ||fS ) z/Determine new output column after output occursF�r �T�mNre )r( r= rl rI �add�discardro ) rS rx ry ru �escaped�offset�
last_wrap_pos� wrapped_datar" r& r& r' �_determine_column� sH
� � � � z SSHLineEditor._determine_columnc C s� |� d�}|dkr!| j�|d|d � � ||d d� }d| _n|}| �|| j|�\}| _| j�|� | jrG| j| j dkrI| j�d� dS dS dS )z/Generate output and calculate new output column�
r Nr � )�rfindrJ rl rB r� r= )rS rx ru �idx�tailr& r& r' �_output� s
�zSSHLineEditor._outputc C s | j s| j�d� d| _ dS dS )zRing the terminal bell�TN)rG rJ rl �rS r& r& r' �
_ring_bell� s
�zSSHLineEditor._ring_bell�new_posc C s: t | j�}|| jk r|| _n`||k r|d7 }| j}| j}||k r/|| �|�7 }|d7 }||k s || jkrN|| jkrM|| �| j�8 }| jd7 _|| jks9n | jdkrn|| �| j�7 }|| jk rh| jd8 _nn| jdksS| j}| j| _| j|k r�| �| j�}|| | jk r�| jd7 _||7 }n |S | j|k sz|S )zAUpdate visible input window when not wrapping onto multiple linesr r ) �lenrF rC rE rB rw r= r@ rD )rS r� �line_lenru ry �ch_widthr& r& r' �_update_input_window� sD
�
��
�
� z"SSHLineEditor._update_input_windowc C s� | j | j }| j | j }|| j }|| j }||k r)| j�dt|| � d � n||kr;| j�dt|| � d � ||krN| j�dt|| � d � n||k r`| j�dt|| � d � || _ dS )z2Move the cursor to selected position in input linez[�A�B�C�DN)rB r= rJ rl ri )rS ry � start_row� start_col�end_row�end_colr& r& r' � _move_cursor s
zSSHLineEditor._move_cursorc C s4 | j r
| �|� dS | j�d| j| � || _dS )z;Move the cursor backward to selected position in input linerz N)r; r� rJ rl rB �rS ry r& r& r' �
_move_back1 s
zSSHLineEditor._move_backc C s` | j }| j| }|dkr&| j�d| � | j| _ | j | j dkr&| j�d� | �|� || _dS )z7Clear any remaining characters from previous input liner re r� N)rB rA rJ rl r= r� )rS ry � remainingr& r& r' �
_clear_to_end: s
zSSHLineEditor._clear_to_endc C s" | � | j� | �� | j�� dS )zErase current input lineN)r� r@ r� rI �clearr� r& r& r' � _erase_inputJ s zSSHLineEditor._erase_inputc C s� | j rR| jrT| jr&| �| j d| j� d� | j}| �| j | jd� | j� n!| �| j� | �| j | j| j� � | j}| �| j | j| j� � | j| _ | �
|� dS dS dS )zDraw current input lineNr ) rF r? r; r� rE rB r� rC rD rA r� r� r& r& r' � _draw_inputQ s �zSSHLineEditor._draw_input�
new_columnc C s4 | j r| jr | �|� n | �| j| j|� || _dS )z3Reposition the cursor to selected position in inputN)r? r; r� �
_update_inputrE rB )rS r� r� r& r& r' � _repositionb s
zSSHLineEditor._repositionc C s� | j ri| jr<|| jv r|d8 }| �|� || jv }| �| j||� |� | j}| �| j|d� |� ||| jv | 7 }n$| �|� | �| j � | �| j| j
|� � | j}| �| j|| j � � | � � | �|� || _
dS )z-Update selected portion of current input liner N)r? r; rI r� r� rF rB r� r� r@ rC rD r� rE )rS ru ry r� � prev_wrapr& r& r' r� m s$
zSSHLineEditor._update_inputc C s, d| _ d| _d| _d| _| j| _| j| _dS )zReset input line to emptyr3 r N)rF rC rD rE rB r@ rA r� r& r& r' � _reset_line� s zSSHLineEditor._reset_linec C s$ | j r| �� | �� d| _ dS dS )z"Reset a pending echoed line if anyFN)r7 r� r� r� r& r& r' �_reset_pending� s
�z SSHLineEditor._reset_pendingc C s� t | j�}t |�}| jr"|| | jkr"| �� | j| }|d|� }|rF| j}|| }| jd|� | | j|d� | _| �|| j|� dS dS )z Insert data into the input lineN)r� rF r9 r� rE r� rB )rS rx r� �data_lenru r� r& r& r' �_insert_printable� s
"�z SSHLineEditor._insert_printablec C s� | j }| jo| j o| jdkp| jt|�k }| js|r9|r+| �d| j| j | � n| �
� | �d� | � � n| �
� d| _ | jrX| j
rX|rX| j�|� | j| j
d� | _t| j�| _| j�|d d� dS )z5End the current input line and send it to the sessionr rz �
TNr� )rF r? r; rC rD r� r6 r� rB r@ � _move_to_endr� r7 r8 rN rl rO r5 �
data_received)rS �line� need_wrapr& r& r' � _end_line� s" �
zSSHLineEditor._end_linec C s | j s
| j�� dS | �� dS )z@Erase character to the right, or send EOF if input line is emptyN)rF r5 �soft_eof_received� _erase_rightr� r& r& r' �_eof_or_delete� s z SSHLineEditor._eof_or_deletec C sd | j dkr,| j d }| j| �|� }| jd|� | j|d d� | _| �|||� dS | �� dS )zErase character to the leftr r N)rE rB rw rF r� r� �rS ru ry r& r& r' � _erase_left� s
" zSSHLineEditor._erase_leftc C sX | j t| j�k r&| j }| jd|� | j|d d� | _| �|| j|� dS | �� dS )z Erase character to the rightNr )rE r� rF r� rB r� rv r& r& r' r� � s
" zSSHLineEditor._erase_rightc C s" | j | _d| _ | �d| jd� dS )zErase entire input liner3 r N)rF rM r� r@ r� r& r& r' � _erase_line� s zSSHLineEditor._erase_linec C s: | j }| j|d� | _| jd|� | _| �|| j|� dS )zErase to end of input lineN)rE rF rM r� rB rv r& r& r' �
_erase_to_end� s zSSHLineEditor._erase_to_end�handlerc C s� || j | j�}|du r |�� r| �|� dS | �� dS |du r&| �� dS ttttf |�\}}|dk r=| j �
|� dS || _ | � d| j |� dS )z Call an external key handlerTFr N)
rF rE ra r� r� r r
ri rh r5 �signal_receivedr� r@ )rS rV r� �resultr� r� r& r& r' � _handle_key� s zSSHLineEditor._handle_keyc C sL | j dkr | j d8 _ | j| j | _| �d| jt| j�� dS | �� dS )z+Replace input with previous line in historyr r N)rO rN rF r� r@ r� r� r� r& r& r' �
_history_prev s
zSSHLineEditor._history_prevc C sj | j t| j�k r/| j d7 _ | j t| j�k r | j| j | _nd| _| �d| jt| j�� dS | �� dS )z'Replace input with next line in historyr r3 r N)rO r� rN rF r� r@ r� r� r& r& r' �
_history_next s zSSHLineEditor._history_nextc C s@ | j dkr| j d }| j| �|� }| �||� dS | �� dS )zMove left in input liner r N)rE rB rw r� r� r� r& r& r' �
_move_left" s
zSSHLineEditor._move_leftc C sF | j t| j�k r | j }| j| �|� }| �|d |� dS | �� dS )zMove right in input liner N)rE r� rF rB rw r� r� r� r& r& r' � _move_right, s
zSSHLineEditor._move_rightc C s | � d| j� dS )zMove to start of input liner N)r� r@ r� r& r& r' �_move_to_start6 s z SSHLineEditor._move_to_startc C s | � t| j�| j� dS )zMove to end of input lineN)r� r� rF rA r� r& r& r' r� ; s zSSHLineEditor._move_to_endc C s | � � | �� dS )zRedraw input lineN)r� r� r� r& r& r' �_redraw@ s zSSHLineEditor._redrawc C s | � | j� dS )z Insert previously erased inputN)r� rM r� r& r& r' �_insert_erasedF rd z SSHLineEditor._insert_erasedc C s | j �d� dS )zSend break to sessionr N)r5 �break_receivedr� r& r& r' � _send_breakK rd zSSHLineEditor._send_break)r� �
zOM)�)rz �)z[3~)�)� )�z[AzOA)�z[BzOB)�z[DzOD)�z[CzOC)�z[Hz[1~)�z[Fz[4~)�)�)�z[33~c C s$ | � |ttj||d�� | �� dS )z5Register a handler to be called when a key is pressed)rV r� N)rQ r r r� rR �rS rV r� r& r& r' � register_key` s � zSSHLineEditor.register_keyc C s | � |� | �� dS )z(Remove the handler associated with a keyN)r\ rR �rS rV r& r& r' �unregister_keyg s
z SSHLineEditor.unregister_keyr� c C s" | � � || _| �d| j|� dS )z"Set input line and cursor positionr N)r� rF r� r@ �rS r� ru r& r& r' � set_inputm s zSSHLineEditor.set_input� line_modec C s> | � � | jr|s| j}| �� d| _| j�|d� || _dS )z!Enable/disable input line editingr3 N)r� rF r� r5 r� r> )rS r� rx r& r& r' �
set_line_modeu s
zSSHLineEditor.set_line_mode�echoc C sH | � � | jr|s| �� d| _dS |r | js"d| _| �� dS dS dS )z,Enable/disable echoing of input in line modeFTN)r� r? r� r� �rS r� r& r& r' �set_echo� s
�zSSHLineEditor.set_echoc C s< | � � |pt| _| jr| �| j| jd�\}| _| �� dS )zSet terminal line widthr N) r� r< r= r; r� rF r@ rB r� )rS r2 �_r& r& r' � set_width� s
� zSSHLineEditor.set_width�datatypec C sD | j r�t|�}d}||k r�| �� || }|d7 }|| jv rA| j| }t|�r:z
tt|�| � W | j| _nG| j| _w tt|�| _n;| j| jkrt|� � rt| j
� ||d �}|dusZJ �|d }|rn| � |� |t|�d 7 }n| � |� n| j| _| �
� ||k s
d| _| jr�| j�d�| j�� | j�� dS dS | j�||� dS )zProcess input from channelr r NFr3 )r> r� r� rL �callabler � _CharHandlerrK rY ra rp �matchr� r� rG rJ r4 �writero r� r5 r� )rS rx r� r� r� r" � key_stater� r& r& r' �
process_input� s>
� �zSSHLineEditor.process_inputc C s� | j r"|�| j�r| j| _|t| j�d� }n| �� | �� d| _ |�dd�}| �� | � |� | j
s<| j| j ; _| j| _| j| _ | �
� | j�d�| j�� | j�� dS )zProcess output to channelNFr� r� r3 )r7 �
startswithrF rB r@ r� r� r� �replacer� r; r= rA r� r4 r� ro rJ r� )rS rx r& r& r' �process_output� s"
z SSHLineEditor.process_output�r# N�N)8�__name__�
__module__� __qualname__�__doc__rj rh ri rW r� rQ r\ rR rw r r
r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� � _KeyHandlerr� r� r� r� r� r� r� r� r� r� rP r� r� r� r� r� r� r r� r� r& r& r& r' r : s� �����
�$
% ��
� 0
-
�
.c
@ s� e Zd ZdZdddddededef
d d
�Zd efd d
�Zde e
fdd�Z dede ddfdd�Z
deddfdd�Zd(dd�Zdededdfd d �Zd eddfd d �Zd!eddfd"d#�Zd)d$ed%eddfd&d'�ZdS )*�SSHLineEditorChannela� Input line editor channel wrapper
When creating server channels with `line_editor` set to `True`,
this class is wrapped around the channel, providing the caller with
the ability to enable and disable input line editing and echoing.
.. note:: Line editing is only available when a pseudo-terminal
is requested on the server channel and the character
encoding on the channel is not set to `None`.
� orig_chanr+ � orig_sessionr- r. r/ r0 c C s( || _ || _|| _|| _|| _d | _d S r� )�
_orig_chan�
_orig_sessionr6 r8 r9 �_editor)rS r� r� r. r/ r0 r& r& r' rW � s
z SSHLineEditorChannel.__init__�attrc C � t | j|�S )z3Delegate most channel functions to original channel)�getattrr� �rS r� r& r& r' � __getattr__� r) z SSHLineEditorChannel.__getattr__r# c C sT | j �� \}}| j �� }| j �� d }|r'|r't| j | j| j| j| j||�| _ | j S )z>Create input line editor if encoding and terminal type are setr )
r� � get_encoding�get_terminal_type�get_terminal_sizer r� r6 r8 r9 r� )rS �encodingr� r1 r2 r& r& r' �
create_editor s
�z"SSHLineEditorChannel.create_editorrV r� Nc C � | j dusJ �| j �||� dS )a Register a handler to be called when a key is pressed
This method registers a handler function which will be called
when a user presses the specified key while inputting a line.
The handler will be called with arguments of the current
input line and cursor position, and updated versions of these
two values should be returned as a tuple.
The handler can also return a tuple of a signal name and
negative cursor position to cause a signal to be delivered
on the channel. In this case, the current input line is left
unchanged but the signal is delivered before processing any
additional input. This can be used to define "hot keys" that
trigger actions unrelated to editing the input.
If the registered key is printable text, returning `True` will
insert that text at the current cursor position, acting as if
no handler was registered for that key. This is useful if you
want to perform a special action in some cases but not others,
such as based on the current cursor position.
Returning `False` will ring the bell and leave the input
unchanged, indicating the requested action could not be
performed.
:param key:
The key sequence to look for
:param handler:
The handler function to call when the key is pressed
:type key: `str`
:type handler: `callable`
N)r� r� r� r& r& r' r� s $z!SSHLineEditorChannel.register_keyc C s | j dusJ �| j �|� dS )a� Remove the handler associated with a key
This method removes a handler function associated with
the specified key. If the key sequence is printable,
this will cause it to return to being inserted at the
current position when pressed. Otherwise, it will cause
the bell to ring to signal the key is not understood.
:param key:
The key sequence to look for
:type key: `str`
N)r� r� r� r& r& r' r� 8 s z#SSHLineEditorChannel.unregister_keyc C s | j dusJ �| j �dd� dS )zQClear input line
This method clears the current input line.
Nr3 r �r� r� r� r& r& r' � clear_inputJ s z SSHLineEditorChannel.clear_inputr� ru c C r )a+ Clear input line
This method sets the current input line and cursor position.
:param line:
The new input line
:param pos:
The new cursor position within the input line
:type line: `str`
:type pos: `int`
Nr r� r& r& r' r� T s z SSHLineEditorChannel.set_inputr� c C �6 | j j�d|r dnd� | jdusJ �| j�|� dS )av Enable/disable input line editing
This method enabled or disables input line editing. When set,
only full lines of input are sent to the session, and each
line of input can be edited before it is sent.
:param line_mode:
Whether or not to process input a line at a time
:type line_mode: `bool`
z%s line editor�Enabling� DisablingN)r� �logger�infor� r� )rS r� r& r& r' r� e s
�z"SSHLineEditorChannel.set_line_moder� c C r )a! Enable/disable echoing of input in line mode
This method enables or disables echoing of input data when
input line editing is enabled.
:param echo:
Whether or not input to echo input as it is entered
:type echo: `bool`
z%s echor r N)r� r r r� r� r� r& r& r' r� x s
�z SSHLineEditorChannel.set_echorx r� c C s0 | j r|du r| j �|� dS | j�||� dS )z#Process data written to the channelN)r� r� r� r� �rS rx r� r& r& r' r� � s zSSHLineEditorChannel.writer� r� )r� r� r� r� rj rh rW ri r� r r r� r� r� r� r r� r� r� r r� r& r& r& r' r� � s( ���
�
'
r� c
@ s� e Zd ZdZdeddfdd�Zdefdd �Zdd d
�Zde de de de d
d f
dd�Z
dede d
d fdd�Z d
e
e fdd�Zd S )�SSHLineEditorSessionz!Input line editor session wrapperr* r� r- c C s || _ || _d | _d S r� )r4 r� r� )rS r* r� r& r& r' rW � s
z SSHLineEditorSession.__init__r� c C r� )z3Delegate most channel functions to original session)r� r� r� r& r& r' r� � r) z SSHLineEditorSession.__getattr__r# Nc C s | j �� | _| j�� dS )z4Start a session for this newly opened server channelN)r4 r� r� r� �session_startedr� r& r& r' r
� s z$SSHLineEditorSession.session_startedr2 �height�pixwidth� pixheightc C s( | j r | j �|� | j�||||� dS )z The terminal size has changedN)r� r� r� �terminal_size_changed)rS r2 r r r
r& r& r' r � s
�z*SSHLineEditorSession.terminal_size_changedrx r� c C s* | j r | j �||� dS | j�||� dS )z&Process data received from the channelN)r� r� r� r� r r& r& r' r� � s z"SSHLineEditorSession.data_receivedc C s | j r | j �d� | j�� S )z%Process EOF received from the channelF)r� r� r� � eof_receivedr� r& r& r' r � s
z!SSHLineEditorSession.eof_receivedr� )r� r� r� r� r� rW ri r� r
rh r r r� r rj r r& r& r& r' r � s
�
��
�
r )"r� rm � functoolsr �typingr r r r r r r
r r � unicodedatar r, r �channelr r ri �objectrY r� rh rj r� r< r: r( r r� r r& r& r& r' �<module> s2 2 -