CasperSecurity
o
���a�< � @ s� d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
e
jdkr:dZndZeejddd �e�Zd
d � Zd d
� ZdZG dd� de�ZdS )� )�print_function�absolute_importN)�SqliteDatabase�3�gettext�ugettextzcommand-not-foundT)�fallbackc sv d� �fdd�t t��d �D �}dd� |D �}dd� |D �}� fdd�|D �}� fdd�|D �}t|| | | �S ) zw
return a set with spelling1 distance alternative spellings
based on http://norvig.com/spell-correct.html
z&abcdefghijklmnopqrstuvwxyz-_0123456789c s$ g | ]}� d |� � |d � f�qS �N� )�.0�i)�wordr
�A/usr/lib/python3/dist-packages/CommandNotFound/CommandNotFound.py�
<listcomp># � $ z!similar_words.<locals>.<listcomp>� c S s$ g | ]\}}|r||d d� �qS �r Nr
�r �a�br
r
r r $ r c S s< g | ]\}}t |�d kr||d |d |dd� �qS )r r � N��lenr r
r
r r % s < c s2 g | ]\}}� D ]}|r|| |d d� �qqS r r
�r r r �c��alphabetr
r r & s 2 c s& g | ]\}}� D ]}|| | �qqS r
r
r r r
r r ' s & )�ranger �set)r
�s�deletes�
transposes�replaces�insertsr
)r r
r �
similar_words s r$ c C sD zt �� } t�d�d | v pt�d�d | v W S ty! Y dS w )N�sudor �adminF)�posix� getgroups�grp�getgrnam�KeyError)�groupsr
r
r �
user_can_sudo+ s � �r- z&/var/lib/command-not-found/commands.dbc @ s� e Zd ZdZdZdZdZejZ d*dd�Z
d+d d
�Z d d � Z d
d� Z
dd� Zdd� Zdd� Zd,dd�Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd-d'd(�Zd)S ).�CommandNotFoundz
programs.d� )z /snap/binz/binz/usr/binz/usr/local/binz/sbinz /usr/sbinz/usr/local/sbinz
/usr/gamesz
/usr/bin/snap� /usr/share/command-not-foundc C sN | � � | _tj�t�rt�ttj�rtt�| _ nt
d��t � | _ t �
� | _d S )NzCannot find database)�_getSourcesList� sources_list�os�path�exists�dbpath�access�R_OKr �db�FileNotFoundErrorr- r' �geteuid�euid)�self�data_dirr
r
r �__init__K s
zCommandNotFound.__init__� c C sd g }|t |� kr| jks|S |S t|�D ]}| �|�}|D ]\}}}|�||||f� q q|S )z try to correct the spelling )r �max_lenr$ � get_packages�append) r= r
�min_len�possible_alternatives�w�packages�package�ver�compr
r
r �spelling_suggestionsU s �
�z$CommandNotFound.spelling_suggestionsc C s | j �|�S r )r9 �lookup�r= �commandr
r
r rB ` s z CommandNotFound.get_packagesc
C sl g }g }t j�| j�st�d| j � g g fS z%tt j��}tj | jddd|g|dd�}W d � n1 s6w Y W n tj
yZ } zt�d| � g g fW Y d }~S d }~ww t�d| � zt � |�}W n t j
y� } zt�d | � g g fW Y d }~S d }~ww |D ]'}|d
|kr�|�|d |d
|�d �f� q�|�|d
|d |�d �f� q�||fS )
Nz
%s not existsz advise-snapz
--format=jsonz --commandT)�stderr�universal_newlinesz.calling snap advice-snap returned an error: %sz got %s from snap advise-snapzcannot decoding json: %s�Command�Snap�Version)r3 r4 r5 �snap_cmd�logging�debug�open�devnull�
subprocess� check_output�CalledProcessError�json�loads�JSONDecodeErrorrC �get) r= rN � exact_result�mispell_resultrX �output�e�snaps�snapr
r
r � get_snapsc sD �� ������ zCommandNotFound.get_snapsc C sj z)t tj�t�dd�df���}dd� |D �W d � W S 1 s"w Y W d S ty4 g Y S w )N�HOMEz/rootz .command-not-found.blacklistc S s g | ] }|� � d kr|� � �qS )� )�strip)r �liner
r
r r � s z0CommandNotFound.getBlacklist.<locals>.<listcomp>)rW r3 �sep�join�getenv�IOError)r= � blacklistr
r
r � getBlacklist� s (� �z CommandNotFound.getBlacklistc C sz zdd l }ddlm} |�� W n
ttfy g Y S w tg �}|dd�D ]}|js:|js:|j D ]}|�
|� q2q'|S )Nr )� SourcesListF)� withMatcher) �apt_pkg�aptsources.sourceslistrq �init� SystemError� ImportErrorr �disabled�invalid�comps�add)r= rs rq r2 �source� componentr
r
r r1 � s �
�z CommandNotFound._getSourcesListc C s� dt jvrd S |r[td�}tjdkrt|�}dd� }n||�}tjjr/t|t �r/|�
tjj�}|� � td�kr]| j dkr?d}nd }d
||f }t
d | tjd � tj|�� d
d� d S d S d S )N� COMMAND_NOT_FOUND_INSTALL_PROMPTz Do you want to install it? (N/y)r c S s | S r r
)�xr
r
r �<lambda>� s z0CommandNotFound.install_prompt.<locals>.<lambda>�yr rh �sudo z%sapt install %sz%s��fileF)�shell)r3 �environ�_�sys�version�input�stdin�encoding�
isinstance�str�decode�lowerr< �print�stdoutrY �call�split)r= � package_name�prompt�answer� raw_input�command_prefix�install_commandr
r
r �install_prompt� s&
�
z CommandNotFound.install_prompt� c
C sV t |�t |� |kr ttd�|t |�f | jd� | j�� dS t |�t |� dkrtttd�| | jd� |D ] \}}}|rCd| }nd}ttd�|||f | jd� q7|D ] \}}} }|rcd| }nd}ttd �|||f | jd� qVt |�dkr�ttd
�| jd� n t |�dkr�| jr�ttd �d | jd� n ttd �d
| jd� | j�� dS )z3 print spelling suggestions for packages and snaps z6Command '%s' not found, but there are %s similar ones.r� Nr �%Command '%s' not found, did you mean:z (%s)rh z command '%s' from snap %s%s� command '%s' from deb %s%s�3See 'snap info <snapname>' for additional versions.zTry: %s <deb name>zsudo apt installz apt install)r r� r� � output_fd�flushr- )
r= r
�mispell_packages�
mispell_snaps�max_altrN re rI rH rJ r
r
r �print_spelling_suggestions� s.
z*CommandNotFound.print_spelling_suggestionsc C s t td�d|i | jd� d S )Nz;Command '%(command)s' not found, but can be installed with:rN r� )r� r� r� rM r
r
r �_print_exact_header� s
�
�z#CommandNotFound._print_exact_headerc C s� | � |� |d }| jdkrtd|d | jd� n | jr)td|d | jd� ntd|d | jd� ttd�� | j�� d S )Nr zsnap install %sr� zsudo snap install %s� Please ask your administrator.)r� r< r� r� r- r� r� )r= rN rG rd re r
r
r �advice_single_snap_package� s
z*CommandNotFound.advice_single_snap_packagec C s� | � |� | jdkr!td|d d | jd� | �|d d � nE| jr;td|d d | jd� | �|d d � n+td|d d | jd� ttd�� |d d | jvrfttd�|d d | jd� | j�� d S )Nr zapt install %sr� zsudo apt install %sr� r z1You will have to enable the component called '%s') r� r< r� r� r� r- r� r2 r� )r= rN rG rd r
r
r �advice_single_deb_package� s
z)CommandNotFound.advice_single_deb_packagec C s | j dkr
| jr
dS dS )Nr r� rh )r< r- )r= r
r
r r% � s zCommandNotFound.sudoc C s | � |� tdd� || D ��}t|�D ]W\}}d}|d r5|dkr/t|�dkr/d|d }nd|d }|d | jv rNtd | �� ||d |f | jd
� qtd | �� ||d |f d td �|d d
| jd
� q| j dkr}| j
s}ttd�| jd
� | j� � d S )Nc S � g | ]}t |d ��qS �r r �r r r
r
r r � � z<CommandNotFound.advice_multi_deb_package.<locals>.<listcomp>rh r r � # version %s, or� # version %sr z%sapt install %-*s%sr� z (z-You will have to enable component called '%s'�)z.Ask your administrator to install one of them.) r� �max� enumerater r2 r� r% r� r� r< r- r� )r= rN rG rd �padr rH rI r
r
r �advice_multi_deb_package� s
$<z(CommandNotFound.advice_multi_deb_packagec C s� | � |� tdd� || D ��}t|�D ]2\}}d}|d r5|dkr/t|�dkr/d|d }nd|d }td| �� ||d |f | jd � qttd
�| jd � | j�� d S ) Nc S r� r� r r� r
r
r r r� z>CommandNotFound.advice_multi_snap_packages.<locals>.<listcomp>rh r r r� r� �%ssnap install %-*s%sr� r� ) r� r� r� r r� r% r� r� r� )r= rN rG rd r� r re rI r
r
r �advice_multi_snap_packages s
$z*CommandNotFound.advice_multi_snap_packagesc C s | � |� tdd� || D ��}t|�D ],\}}d}|d r/|dkr)d|d }nd|d }td| �� ||d |f | jd � q|D ] }d}|d
rQd|d
}td | �� ||d |f | jd � qCt|�d
kryttd �|d d | jd � n ttd
�| jd � | j�� d S )Nc S r� r� r r� r
r
r r r� z?CommandNotFound.advice_multi_mixed_packages.<locals>.<listcomp>rh r r r� r� r� r� r z%sapt install %-*s%sz+See 'snap info %s' for additional versions.r� ) r� r� r� r� r% r� r r� r� ) r= rN rG rd r� r re rI rH r
r
r �advice_multi_mixed_packages s$
$ $ z+CommandNotFound.advice_multi_mixed_packagesFc
sP dd� � t ��| jkr
dS ��d�r#tj���r tj���g}ng }n � �fdd�| jD �}|r�|s�t |�dkrNtt d��tj�
|d ��d
� | j d � n tt d �d
�i | j d � |D ]}tdtj�
|�� | j d � q]t t
|�t
t�dd��d�� �}t |�d kr�tt d�d�
|� | j d � dd�
|�v r�tt d�| j d � dS tj�d�s�tj�d�s�dS �| �� v r�dS �dkr�tt d�� | j d � tt d�d | j d � tt d�d | j d � d S | ���}| ���\}}t�d ||f � t |�d k�r t |�d k�r | ���} t | �d k�st |�d k�r | ��| |� nmt |�d k�r6t |�dk�r6| ��||� nWt |�d k�rLt |�d k�rL| ��||� nAt |�dk�rbt |�d k�rb| ��||� n+t |�dk�rxt |�d k�rx| ��||� nt |�d k�r�t |�d k�r�| ��||� t |�d k�p�t |�d k�p�t |�d k�p�t | �d kS ) z7 give advice where to find the given command to stderr c S s. t j�t j�| |��ot j�t j�| |�� S )z? helper that returns if a command is found in the given prefix )r3 r4 r5 rl �isdir)�prefixrN r
r
r �
_in_prefix' s �z*CommandNotFound.advise.<locals>._in_prefixF�/c s g | ] }� |��r|�qS r
r
)r r� �r� rN r
r r 5 s z*CommandNotFound.advise.<locals>.<listcomp>r z1Command '%(command)s' is available in '%(place)s'r )rN �placer� z:Command '%(command)s' is available in the following placesrN z * %s�PATHrh �:z_The command could not be located because '%s' is not included in the PATH environment variable.�sbinzfThis is most likely caused by the lack of administrative privileges associated with your user account.z /usr/bin/aptz/usr/bin/aptitude�pythonr� r� )�python3r� rh )r� zpython-is-python3rh Tzgot debs: %s snaps: %s) r rA �
startswithr3 r4 r5 �dirname�prefixesr� r� rl r� �listr rm r� rp rB rf rU rV rK r� r� r� r� r� r� )
r= rN �ignore_installedr� r� �missingrG rd r� r� r
r� r �advise% sl
, "
�
�
�
�zCommandNotFound.adviseN)r0 )r@ )r� )F)�__name__�
__module__� __qualname__� programs_dirrA r� rT r� rO r� r? rK rB rf rp r1 r� r� r� r� r� r% r� r� r� r� r
r
r
r r. 8 s,
r. )�
__future__r r r r) r\ rU r3 �os.pathr' r� rY �CommandNotFound.db.dbr r� �_gettext_method�getattr� translationr� r$ r- r6 �objectr. r
r
r
r �<module> s&