o

�Ad��@s�dZd
dl
m
m
mZ
d
dlmZ
d
dlmZ	
d
dlm
Z
mZmZ
d
dlm
Z

d
dlmZ
e	�e�
ZdZed	ed
d�ZdZd
Zd
ZGdd�dej�Zdd�Zeejf
fg
Zdd�ZdS)z>Hetzner Cloud API Documentation
   https://docs.hetzner.cloud/�N)
�dmi)
�log)�net�sources�util)
�NoDHCPLeaseError)
�EphemeralDHCPv4z!http://169.254.169.254/hetzner/v1z	/metadataz	/userdata)�metadata_url�userdata_url�<�c@s4eZ
dZd
Zdd�Zdd�Zdd�Zedd	��
Zd
S)�DataSourceHetzner�HetznercCs�tj
�||
||�
||_t�|_t�t�|
d
dgi�t	g�
|_
|j
d|_|j
d|_|j
�
dt�|_|j
�
dt�|_|j
�
dt�|_tj|_tj|_dS)N�
datasourcerr	r
�retries�timeout�
wait_retry)r�
DataSource�__init__�distro�dict�metadatar�
mergemanydict�get_cfg_by_path�BUILTIN_DS_CONFIG�ds_cfg�metadata_address�userdata_address�get�
MD_RETRIESr�
MD_TIMEOUTr�
MD_WAIT_RETRYr�UNSET�_network_config�DSMODE_NETWORK�dsmode)�self�sys_cfgr�paths�r)�E/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceHetzner.pyr"s




��





zDataSourceHetzner.__init__c

Cs*
t�\}
}|
s	d
Sz<t
t��dtdi
|j��d��"
tj|j	|j
|j|jd�}tj
|j|j
|j|jd�}Wd�
n1s?w



Y
WntyX
}
zt�d|�
�d}~w
wt�|�
|_||_|d|jd<|d|jd	<|�d
d�|jd
<|�dd�|jd<|�dd�|_|��|kr�td
�
�
dS)NF�urlz/metadata/instance-id)�iface�connectivity_url_data�tmp_dir)r�sec_betweenrzBailing, DHCP Exception: %szinstance-id�hostnamezlocal-hostname�network-configzpublic-keys�vendor_dataz6SMBIOS serial does not match instance ID from metadataT)�get_hcloud_datarr�find_fallback_nic�BASE_URL_V1r�get_tmp_exec_path�	hc_helper�
read_metadatarrrr�
read_userdatarr�LOG�error�maybe_b64decode�userdata_raw�
metadata_fullrr�vendordata_raw�get_instance_id�RuntimeError)r&�
on_hetzner�serial�md�ud�
er)r)r*�	_get_data4sR




��



�



���

��






�zDataSourceHetzner._get_datacCst�
|��d
�S)N�system-serial-number)r�instance_id_matches_system_uuidr@)r&r'r)r)r*�check_instance_idms

�z#DataSourceHetzner.check_instance_idc
CsT|jd
urt
�dtj�
tj|_|jtjkr|jS|jd}
|
s$td�
�
|
|_|jS)z�Configure the networking. This needs to be done each boot, since
        the IP information may have changed due to snapshot and/or
        migration.
        Nz5Found None as cached _network_config. Resetting to %sr1z'Unable to get meta-data from server....)r#r:�warningrr"r�	Exception)r&�_net_configr)r)r*�network_configrs



�



z DataSourceHetzner.network_configN)	�__name__�
__module__�__qualname__�dsnamerrGrJ�propertyrNr)r)r)r*r
s9
r
cCs@t�
d
�
}|dkrdSt�
d�
}
|
rt�d|
�
d|
fStd�
�
)Nzsystem-manufacturerr)FNrHz#Running on Hetzner Cloud: serial=%sz+Hetzner Cloud detected, but no serial foundT)r�
read_dmi_datar:�debugrA)�vendor_namerCr)r)r*r3�s






�r3c

Cst�
|t�S)
N)r�list_from_depends�datasources)
�dependsr)r)r*�get_datasource_list�s
rZ)�__doc__�!cloudinit.sources.helpers.hetznerr�helpers�hetznerr7�	cloudinitrr�loggingrr�cloudinit.net.dhcpr�cloudinit.net.ephemeralr�	getLoggerrOr:r5rrr r!rr
r3�DEP_FILESYSTEMrXrZr)r)r)r*�<module>
s(






�

n
�