U
    -_gx	                     @   sD   d dl Z d dlmZmZmZ G dd deeZG dd deeZdS )    N)PluginUbuntuPluginRedHatPluginc                   @   s$   e Zd ZdZdZdZdZdd ZdS )AzurezMicrosoft Azure clientZazure)Zvirt)ZWALinuxAgentc              	   C   sj   |  dddddddg | d}td	D ]*\}}}|D ]}| j | |||d
 q:q,| jddd d S )Nz/var/log/waagent*z/var/lib/cloudz/etc/default/kv-kvp-daemon-initz/etc/waagent.confz*/sys/module/hv_netvsc/parameters/ring_sizez9/sys/module/hv_storvsc/parameters/storvsc_ringbuffer_sizez/var/lib/AzureEnhancedMonitorZlog_sizez/var/log/azure)Z	sizelimitz|curl -s -H Metadata:true --noproxy "*" "http://169.254.169.254/metadata/instance/compute?api-version=2021-01-01&format=json"zinstance_metadata.json)Zsuggest_filename)Zadd_copy_specZ
get_optionoswalkZ	path_joinadd_cmd_output)selflimitpath_filesname r   :/usr/lib/python3/dist-packages/sos/report/plugins/azure.pysetup   s"    
zAzure.setupN)__name__
__module____qualname__Z
short_descZplugin_nameZprofilesZpackagesr   r   r   r   r   r      s
   r   c                       s   e Zd Z fddZ  ZS )RedHatAzurec                    sV   t    | dr4d}| |d |d |d g d}| |rR| d| g d S )Nz/etc/yum.repos.d/rh-cloud.repoz=curl -s -m 5 -vvv https://rhui-%s.microsoft.com/pulp/repos/%s)1microsoft-azure-rhel7)2r   )3r   z!/etc/pki/rhui/product/content.crtzopenssl x509 -noout -text -in )superr   Zpath_isfiler   )r	   Zcurl_cmdZcrt_path	__class__r   r   r   4   s    


zRedHatAzure.setup)r   r   r   r   __classcell__r   r   r   r   r   2   s   r   )r   Zsos.report.pluginsr   r   r   r   r   r   r   r   r   <module>   s   #