File: //lib/mysqlsh/python-packages/mysqlsh/plugin_manager/__pycache__/repositories.cpython-39.pyc
a
��kfk6 � @ s� d dl Zd dlZd dlZd dlmZmZ ddlmZ ddddd �gZ d
d � Z
d d
d�Z d dd�Z ed�dd� �Z
ed�d dd��Zdd� Zed�dd� �ZdS ) � N)�plugin�plugin_function� )�generalz&Official MySQL Shell Plugin RepositoryzRThe official MySQL Shell Plugin Repository maintained by the MySQL Team at Oracle.z6https://cdn.mysql.com/mysql_shell/plugins_manifest.zipz*manifest/mysql-shell-plugins-manifest.json)�name� description�urlZ manifestPathc C sX d}d}| D ]F}||d�d|� dd�� d|� dd �� d|� d
d�� d �7 }|d7 }q |S ) z�Returns a formatted list of repositories.
Args:
repositories (list): A list of repositories.
Returns:
The formated list as string
� r z>4� r z??z
r �-r z
��get)� repositories�out�i�r� r �G/usr/lib/mysqlsh/python-packages/mysqlsh/plugin_manager/repositories.py�format_repository_listing. s
�
�
r Fc C s t j�t�� d�}t j�|�� r�z@t|d�� }t�|�� d�}W d � n1 sR0 Y |W S tj
y� } z>t d|� dt |j
�� d|j� d|j� �� | r�� W Y d}~dS d}~0 t� y� } z*t d |� dt |�� �� | r� W Y d}~n
d}~0 0 ng S dS )
z�Fetches the registered user repositories
Args:
raise_exceptions (bool): Whether exceptions are raised
Returns:
The registered repositories as list
�plugin-repositories.jsonr r Nz"ERROR: Could not parse JSON file '�'.
�, line: �, col: zERROR: Error reading file ')�os�path�joinr �get_shell_user_dir�isfile�open�json�loadr
�JSONDecodeError�print�str�msg�lineno�colno�OSError)�raise_exceptions�user_repo_file_path�user_repo_file�
user_repos�er r r �get_user_repositoriesB s8 � .����
�r- c
C s� dd| d�}t j�t�� d�}z@t|d�� }tj||dd� W d � n1 sP0 Y W d S ty� } z,t d
|� d t
|�� �� |r�� W Y d}~d S d}~0 0 dS )
z�Stores the registered user repositories
Args:
user_repos (list): The list of
raise_exceptions (bool): Whether exceptions are raised
Returns:
True on success
z MySQL Shell Plugin Repositoriesz0.0.1)ZfileType�versionr r �w� )�indentNTz ERROR: Error writing to file 'r F) r r r r r r r �dumpr' r" r# )r+ r( Zuser_repo_file_contentr) r* r, r r r �set_user_repositoriesc s$ � � .
�r3 zplugins.repositories.listc K sz | � dd�}| � dd�}t�� }t|d�}|r8|�|� |s@|S tdt|d�� dt|�� d t|�d
krjd nd � d
�� dS )a� Lists all registered plugin repositories
This function will list all registered MySQL Shell plugin repositories.
To add a repository use the plugins.repositories.add() function.
Args:
**kwargs: Optional parameters
Keyword Args:
return_formatted (bool): If set to true, a list object is returned.
raise_exceptions (bool): Whether exceptions are raised
Returns:
None or a list of dicts representing the plugins
�return_formattedTr( F�r( z-Registered MySQL Shell Plugin Repositories.
)r z Total of z
repositorr �yZiesz.
N)r
�DEFAULT_PLUGIN_REPOSITORIES�copyr- �extendr" r �len)�kwargsr4 r( Zreposr+ r r r �get_plugin_repositories� s
���r< zplugins.repositories.addc K s |� dd�}|� dd�}| s@|r@td� tjj�dddi��� } | sPtd� d S | �� �d
�r�| � d �}t
|�d kr�td
� d S |d }t
|�dkr�d}n|d }d|� d |� d�} nH| �d�s�| �d�s�d| � �} | � d�s�| � d �r�| � d�} n
| � d�} �z�t �
| �}|�sW d S t�|�}|� d�} | � d�}
|
�sJtd� W d S | � d�} |
�shtd� W d S |� d �} | �r�t
| �d k�r�td � W d S td � td � d!|
� d"d#� d!| � d$| � d%�
� | D ] }
td&|
� d'�� d"�� �q�tjj�d(|
� d)�dd*i��� �� }|d+k�r td,� W d S td-� t|d.�}|d u �rBW d S |D ]6}|� d/��� | �� k�rFtd0|
� d1�� W d S �qFtd2|
� d3�� |�|
| | d4�� t|��r�td5|
� d6�� W nV tj�y } z:td7t|j�� d8|j� d9|j� �� |�r�� W Y d }~d S d }~0 0 d S ):ae Adds a new plugin repository
By calling this function a new plugin repository can be added. The url
parameter allows for shortcuts.
'domain.com' expands to https://domain.com/mysql-shell-plugins-manifest.json
'domain.com/plugins' expands to https://domain.com/plugins/mysql-shell-plugins-manifest.json
'github/username' expands to https://raw.githubusercontent.com/username/mysql-shell-plugins/master/mysql-shell-plugins-manifest.json
'github/username/repo' expands to https://raw.githubusercontent.com/username/repo/master/mysql-shell-plugins-manifest.json
Args:
url (str): The url of a MySQL Shell plugins repository.
**kwargs: Optional parameters
Keyword Args:
interactive (bool): Whether user input is accepted
raise_exceptions (bool): Whether exceptions are raised
Returns:
None
� interactiveTr( z�To add a new MySQL Shell plugin repository the URL of the repository manifest file
needs to be specified.
Examples:
domain.com
domain.com/plugins
github/username
github/username/repository
z/Please enter the URL of the plugin repository: Z defaultValuer z"No URL given. Cancelling OperationNzgithub/�/r z#A github user needs to be specified� zmysql-shell-pluginsz"https://raw.githubusercontent.com/z)/master/mysql-shell-plugins-manifest.jsonzhttps://zhttp://z.jsonz!mysql-shell-plugins-manifest.jsonz"/mysql-shell-plugins-manifest.json�
repositoryr z0The given repository misses the 'name' property.r z7The given repository misses the 'description' property.�pluginsr zAThe repository does not contain any plugins. Operation cancelled.a�
WARNING:
You are about to add an external MySQL Shell plugin repository.
External plugin repositories and their plugins complement
the functionality of MySQL Shell and can contain system
level software that could be potentially harmful to your
system. Please review the description below and only proceed
if you have obtained the external plugin repository URL from
a trusted source.
Oracle and its affiliates cannot be held responsible for
any potential harm caused by using plugins from external sources.
z Repository :r
�
z Description:z
URL: z1
The repository contains the following plugins:
z - Zcaptionz-Are you sure you want to add the repository 'z ' [yes/NO]: �no�yeszOperation cancelled.z%Fetching current user repositories...r5 r �The repository 'z' has already been added.
zAdding repository '�'...)r r r z Repository 'z' successfully added.
z"ERROR: Could not parse JSON file. r r )r
r" �mysqlsh�globals�shell�prompt�strip�lower�
startswith�splitr: �endswithr �
download_filer �loadsr- �appendr3 r! r# r$ r% r& )r r; r= r( Z
github_strZ github_userZ github_repoZmanifestZmanifest_contentr@ r r rA r rJ r+ r r, r r r �add_plugin_repository� s� ��
�
���
��
�
��rS c C s@ d}d}|D ]"}|� d�| kr&d} q0|d7 }q |r8|S d S d S )Nr Fr Tr r )r r+ �nr�foundr r r r �search_user_repoD s
rV zplugins.repositories.removec K s� | � d�}| � dd�}| � dd�}t|d�}|s<td� dS d}|rft||�}|du rftd |� d
�� |du �r<|�r<td t|�� �� |du �r<tjj�d �� � }|�r<z`z.t
|�}|d
kr�|t |�kr�|d }nt �W n, t
�y t||�}|du �rt
�Y n0 W n( t
t f�y2 td|� d�� Y n0 q��q<q�|du �rPtd� n8|| � dd�}td|� d�� ||= t|��r�td� dS )a� Removes a registered plugin repository
This function will remove a shell plugin repository previously registered.
Args:
**kwargs: Optional parameters
Keyword Args:
url (str): The url of a MySQL Shell plugins repository.
interactive (bool): Whether user input is accepted
raise_exceptions (bool): Whether exceptions are raised
Returns:
None
r r= Tr( Fr5 z6No custom MySQL Shell plugin repositories registered.
Nz No user repository matches '�'z Removing a plugin repository.
z/Please enter the index or URL of a repository: r r rE zJ' was not found. Please try again or leave empty to cancel the operation.
z,No valid URL provided. Cancelling operation.r z
Removing repository 'rF z!Repository successfully removed.
)r
r- r" rV r rG rH rI rJ rK �intr: �
IndexError�
ValueErrorr3 )r; r r= r( r+ Z
repo_indexrJ rT r r r �remove_plugin_repositoryT sR
�
�
r[ )F)F)N)�os.pathr r rG �mysqlsh.plugin_managerr r r r r7 r r- r3 r< rS rV r[ r r r r �<module> s( �
!
%