CasperSecurity
o
&�J^� � @ s> d dl Z d dlZd dlZd dlZzd dlmZ W n ey! Y nw dZe �dd�Z e �d�Z
e �de
d �Z e �d d
�Z d Z
g Zg Ze j�d e �Zg d
�Zdd� Zdd� Zdd� Zdd� Ze� Ze j�de �Zee�dksrer�e�d� e�de � e�d� e�de � ee�dk�r?ej� d� d Z e d k �r>dZ eD ]Z ej� d e e f � e d7 Z q�zNz
ed e d �Z W n ey� d Z Y nw e r�e d kr�dZ
W njze!e �Z
W n ey� e!e"e ��Z
Y nw e
dkr�e
e k r�W nJe d7 Z d Z
ej#� d � W n8 e$�y ej� d!� e�%d � Y n$ e�y9 e
d k�s'e
d k�r+dZ
Y ne d7 Z d Z
ej#� d � Y nw e d k s�n ee�dk�rHdZ
e
dk�r�ee
d dk�rne d
k�ree �&ddd"e g� n"e �&dde g� nee
d dk�re �&e e g� neee
d � e d
k�r�e �&d
d
g� dS e �&d#d#d$g� dS )%� N)�input�byobu�SHELLz /bin/bash�HOME�BYOBU_CONFIG_DIRz/.byobu�
BYOBU_BACKEND�tmux���z%s/.reuse-session)�DISPLAY�DBUS_SESSION_BUS_ADDRESS�SESSION_MANAGER�GPG_AGENT_INFO�XDG_SESSION_COOKIE�XDG_SESSION_PATH�GNOME_KEYRING_CONTROL�GNOME_KEYRING_PIDr
� SSH_ASKPASS�
SSH_AUTH_SOCK�
SSH_AGENT_PID�WINDOWID� UPSTART_JOB�UPSTART_EVENTS�UPSTART_SESSION�UPSTART_INSTANCEc
C s� g } d}d}t dkr�ztjddgtjd��� d }W n tjy6 } z|jdkr)� |j}W Y d }~nd }~ww tj j
d u rC|� d�}n|� tj j
�}|r�|� � D ]E}t
�dd |�}|r�|d kr�|�d �dkr�t|�dkr�|�d
�dkr�|�d �dkr�t�d |�� � |�d �}| �d
|d � |d7 }qPt dkr�tjddgtjd��� d }tj j
d u r�|� d�}n|� tj j
�}|r�|� � D ])}|r�|�d�s�|�d�dkr�t�d|�� � | �d|�d�d � |d7 }q�| S )Nr F�screenz-ls��stdout� �UTF-8z\s+� z..z._z
screen: %sz screen____%sr �
list-sessions�_�-r ztmux: %sz
tmux____%s�:)r �
subprocess�Popen�PIPE� communicate�CalledProcessError�
returncode�output�sysr �encoding�decode�
splitlines�re�sub�find�len�count�text�append�strip�split�
startswith)�sessions�ir* �cpe�s�items� r> �(/usr/lib/byobu/include/select-session.py� get_sessions0 sJ
�� B
� �r@ c C s� t dkr[tjddgtjd��� d }tjjd u r |�d�}n|�tjj�}|s)d S d| }t �
||t j �}|s9d S d| |� d�f }t �
||t j �D ]}t�dd d
|�d �d g� qJd S d S ) Nr r r r r z^%s:.+\((group [^\)]+)\).*$z^_%s-\d+:.+\(%s\)$r z kill-session�-tr# )r r$ r% r& r' r+ r r, r- r/ �search� MULTILINE�group�findallr7 )� session_namer* �pattern�masterr< r> r> r? � cull_zombiesZ s �rI c C sn | � dd�\}}tD ]*}t�|�}|r4|dkr ddd|||g}n dd|dd||g}tj|ttjd �d
� q
d S ) N�____� r �setenvrA r z-Sz-X�wr )r7 �BYOBU_UPDATE_ENVVARS�os�getenvr$ �call�open�devnull)�session�backendrF �var�value�cmdr> r> r? �update_environmentu s
��rY c C s| t | � | �dd�\}}t|� |dkr3tr&t�ddddd|ddd g� d S t�dddd
d|g� d S t�d d d |g� d S )
NrJ rK r z-u� new-sessionrA �;z
set-optionzdestroy-unattached�attachr �-AOxRR)rY r7 rI �reuse_sessionsrO �execvp)rT rU rF r> r> r? �attach_session� s r` z%s/.always-selectr �NEWz Create a new Byobu session (%s)z Run a shell without Byobu (%s)z
Byobu sessions...
� z %d. %s
z
Choose 1-%d [1]: � z
ERROR: Invalid input
�
rZ r r] )'rO r/ r+ r$ �builtinsr � Exception�PKGrP r r r r �choicer9 r4 �path�existsr^ rN r@ rI rY r` �
show_shellr2 r5 r �write�triesr: r< �
user_input�int�eval�stderr�KeyboardInterrupt�exitr_ r> r> r> r? �<module> s� �
*
� � ���!