o

�N�g18�@s
dZd
dl
Z
d
dlZzd
dlZdZWney


dZYn
we
�e�
Z	dd�Z
ddd�
Z							dd
d�
Z						d dd
�
Z
	ddd�
Z	ddd�
Z	ddd�
Z	ddd�
Z						d dd�
Z						d dd�
Zddd�
Z				ddd�
ZdS)!a�
Connection module for Amazon EFS

.. versionadded:: 2017.7.0

:configuration: This module accepts explicit EFS credentials but can also
    utilize IAM roles assigned to the instance through Instance Profiles or
    it can read them from the ~/.aws/credentials file or from these
    environment variables: AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY.
    Dynamic credentials are then automatically obtained from AWS API and no
    further configuration is necessary.  More information available at:

    .. code-block:: text

        http://docs.aws.amazon.com/efs/latest/ug/
            access-control-managing-permissions.html

        http://boto3.readthedocs.io/en/latest/guide/
            configuration.html#guide-configuration

    If IAM roles are not used you need to specify them either in a pillar or
    in the minion's config file

    .. code-block:: yaml

        efs.keyid: GKTADJGHEIQSXMKKRBJ08H
        efs.key: askd+ghsdfjkghWupU/asdflkdfklgjsdfjajkghs

    A region may also be specified in the configuration

    .. code-block:: yaml

        efs.region: us-east-1

    If a region is not specified, the default is us-east-1.

    It's also possible to specify key, keyid, and region via a profile, either
    as a passed in dict, or as a string to pull from pillars or minion config:

    .. code-block:: yaml

        myprofile:
          keyid: GKTADJGHEIQSXMKKRBJ08H
          key: askd+ghsdfjkghWupU/asdflkdfklgjsdfjajkghs
          region: us-east-1

:depends: boto3
�NTFcCstj
jjd
dd�S)zi
    Only load if boto3 libraries exist and if boto3 libraries are greater than
    a given version.
    z1.0.0F)Z	boto3_verZ
check_boto)�salt�utilsZversionsZcheck_boto_reqs�rr�I/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/boto_efs.py�__virtual__AsrcKs�d
}|rt|t
�r|tvrt|}n#|tvrt|}n|s!|
s!|r5i}|r)||d<|
r/|
|d<|r5||d<t|t�rtd|vrJ|d|d<|�dd
�
d|vrZ|d|d<|�dd
�
d|vrj|d|d<|�dd
�
tjd	i|�
�
}|St�d�
}|S)
z1
    Create a boto3 client connection to EFS
    N�key�keyid�regionZregion_nameZaws_secret_access_keyZaws_access_key_id�efs)
r
)�
isinstance�strZ
__pillar__Z__opts__�dict�pop�boto3�client)rr�profiler	�kwargsrrrr�	_get_connIs<







�


















�r�generalPurposecKsb|d
ur|}d|d�}t||||d�}	|	j
||
d�}
d|
vr'|	j|
d|d�
d|
vr/||
d<|
S)a_
    Creates a new, empty file system.

    name
        (string) - The name for the new file system

    performance_mode
        (string) - The PerformanceMode of the file system. Can be either
        generalPurpose or maxIO

    creation_token
        (string) - A unique name to be used as reference when creating an EFS.
        This will ensure idempotency. Set to name if not specified otherwise

    returns
        (dict) - A dict of the data for the elastic file system

    CLI Example:

    .. code-block:: bash

        salt 'my-minion' boto_efs.create_file_system efs-name generalPurpose
    N�Name�ZKeyZValue�rrrr	)�
CreationTokenZPerformanceMode�FileSystemId�r�Tags)r�create_file_system�create_tags)�nameZperformance_moderrrr	�creation_tokenr�tagsr�responserrrros"


�

rc
Kspt||||d
�}	|dur|dur|	j
||
d�S|dur#|	j
||
|d�S|dur/|	j
||
|d�S|	j
||
||d�S)a�
    Creates a mount target for a file system.
    You can then mount the file system on EC2 instances via the mount target.

    You can create one mount target in each Availability Zone in your VPC.
    All EC2 instances in a VPC within a given Availability Zone share a
    single mount target for a given file system.

    If you have multiple subnets in an Availability Zone,
    you create a mount target in one of the subnets.
    EC2 instances do not need to be in the same subnet as the mount target
    in order to access their file system.

    filesystemid
        (string) - ID of the file system for which to create the mount target.

    subnetid
        (string) - ID of the subnet to add the mount target in.

    ipaddress
        (string) - Valid IPv4 address within the address range
                    of the specified subnet.

    securitygroups
        (list[string]) - Up to five VPC security group IDs,
                            of the form sg-xxxxxxxx.
                            These must be for the same VPC as subnet specified.

    returns
        (dict) - A dict of the response data

    CLI Example:

    .. code-block:: bash

        salt 'my-minion' boto_efs.create_mount_target filesystemid subnetid
    rN)r�SubnetId)rr"�SecurityGroups)rr"�	IpAddress)rr"r$r#)r�create_mount_target)
�filesystemidZsubnetid�	ipaddressZsecuritygroupsrrrr	rrrrrr%�s"1


�

�



�r%cKsHt||||d
�}g}|
�
�D]\}	}
|�|	|
d��

q|j||d�
dS)a�

    Creates or overwrites tags associated with a file system.
    Each tag is a key-value pair. If a tag key specified in the request
    already exists on the file system, this operation overwrites
    its value with the value provided in the request.

    filesystemid
        (string) - ID of the file system for whose tags will be modified.

    tags
        (dict) - The tags to add to the file system

    CLI Example:

    .. code-block:: bash

        salt 'my-minion' boto_efs.create_tags
    rrrN)r�items�appendr)r&r rrrr	rrZnew_tags�
k�
vrrrr�s


rcK� t||
||d
�}|j
|d�

dS)a�

    Deletes a file system, permanently severing access to its contents.
    Upon return, the file system no longer exists and you can't access
    any contents of the deleted file system. You can't delete a file system
    that is in use. That is, if the file system has any mount targets,
    you must first delete them.

    filesystemid
        (string) - ID of the file system to delete.

    CLI Example:

    .. code-block:: bash

        salt 'my-minion' boto_efs.delete_file_system filesystemid
    r�
rN)r�delete_file_system)r&rrrr	rrrrrr.
sr.cKr,)aP
    Deletes the specified mount target.

    This operation forcibly breaks any mounts of the file system via the
    mount target that is being deleted, which might disrupt instances or
    applications using those mounts. To avoid applications getting cut off
    abruptly, you might consider unmounting any mounts of the mount target,
    if feasible. The operation also deletes the associated network interface.
    Uncommitted writes may be lost, but breaking a mount target using this
    operation does not corrupt the file system itself.
    The file system you created remains.
    You can mount an EC2 instance in your VPC via another mount target.

    mounttargetid
        (string) - ID of the mount target to delete

    CLI Example:

    .. code-block:: bash

        salt 'my-minion' boto_efs.delete_mount_target mounttargetid
    r�
�
MountTargetIdN)r�delete_mount_target)�
mounttargetidrrrr	rrrrrr1$
sr1cK�"t||||d
�}|j
||
d�
dS)a;

    Deletes the specified tags from a file system.

    filesystemid
        (string) - ID of the file system for whose tags will be removed.

    tags
        (list[string]) - The tag keys to delete to the file system

    CLI Example:

    .. code-block:: bash

        salt 'my-minion' boto_efs.delete_tags
    rrN)r�delete_tags)r&r rrrr	rrrrrr4C
sr4c
Ks�d
}t||
||d�}|r|r|j
||d�}	|	d}|S|r)|j
|d�
}	|	d}|S|r7|j
|d�
}	|	d}|S|�
�}	|	d}d|	vrV|j
|	dd�
}	|�|	d�

d|	vsC|S)	a]
    Get all EFS properties or a specific instance property
    if filesystemid is specified

    filesystemid
        (string) - ID of the file system to retrieve properties

    creation_token
        (string) - A unique token that identifies an EFS.
        If fileysystem created via create_file_system this would
        either be explictitly passed in or set to name.
        You can limit your search with this.

    returns
        (list[dict]) - list of all elastic file system properties

    CLI Example:

    .. code-block:: bash

        salt 'my-minion' boto_efs.get_file_systems efs-id
    Nr)rrZFileSystemsr-)
r�
NextMarker)
�Marker)rZdescribe_file_systems�extend)
r&rrrr	rr�resultrr!rrr�get_file_systems[
s. 


��


�


�

�r9c
Ks|d
}t||||d�}|r0|j
|d�
}	|	d}d|	vr.|j
||	dd�}	|�|	d�

d|	vs|S|
r<|j
|
d�
}	|	d}|S)a|
    Get all the EFS mount point properties for a specific filesystemid or
    the properties for a specific mounttargetid. One or the other must be
    specified

    filesystemid
        (string) - ID of the file system whose mount targets to list
                   Must be specified if mounttargetid is not

    mounttargetid
        (string) - ID of the mount target to have its properties returned
                   Must be specified if filesystemid is not

    returns
        (list[dict]) - list of all mount point properties

    CLI Example:

    .. code-block:: bash

        salt 'my-minion' boto_efs.get_mount_targets
    Nrr-ZMountTargetsr5�rr6r/)rZdescribe_mount_targetsr7)
r&r2rrrr	rr8rr!rrr�get_mount_targets�
s  





��	�

r;c	KsXt||
||d
�}|j
|d�
}|d}d|vr*|j
||dd�}|�|d�

d|vs|S)a%

    Return the tags associated with an EFS instance.

    filesystemid
        (string) - ID of the file system whose tags to list

    returns
        (list) - list of tags as key/value pairs

    CLI Example:

    .. code-block:: bash

        salt 'my-minion' boto_efs.get_tags efs-id
    rr-rr5r:)rZ
describe_tagsr7)	r&rrrr	rrr!r8rrr�get_tags�
s



��r<cKr3)a�

    Modifies the set of security groups in effect for a mount target

    mounttargetid
        (string) - ID of the mount target whose security groups will be modified

    securitygroups
        (list[string]) - list of no more than 5 VPC security group IDs.

    CLI Example:

    .. code-block:: bash

        salt 'my-minion' boto_efs.set_security_groups my-mount-target-id my-sec-group
    r)r0r#N)rZ#modify_mount_target_security_groups)r2Z
securitygrouprrrr	rrrrr�set_security_groups�
s


�r=)NNNN)rNNNNN)NNNNNN)�__doc__�loggingZsalt.utils.versionsrrZ	HAS_BOTO3�ImportError�	getLogger�__name__�logrrrr%rr.r1r4r9r;r<r=rrrr�<module>
sl1


�

(





�9





�H
� 
�
� 
�





�;





�
2 


�