import re
import boto3
from botocore.exceptions import ClientError

from virga.common import VirgaException


class VirgaClient:
    """VirgaClient substitute the standard AWS client for more complex requests."""

    @staticmethod
    def find_certificate(resource_object: dict) -> any:
        """
        Call boto3/acm for finding the certificate for the passed domain.

        :param resource_object: Object filter
        :return: Response from AWS
        """
        client = boto3.client('acm')
        certificates = client.list_certificates()
        try:
            res_certificates = [
                cert for cert in certificates['CertificateSummaryList']
                if cert['DomainName'] == resource_object['domain_name']
            ]
            return client.describe_certificate(CertificateArn=res_certificates[0]['CertificateArn'])
        except ClientError:
            raise VirgaException('Lookup certificates domain_name %s failed' % resource_object['domain_name'])
        except (KeyError, IndexError):
            return None

    @staticmethod
    def find_elb(resource_object: dict) -> any:
        """
        Call boto3/elb for finding the ELB instances by name.

        The result contains:
        - load balancer attributes
        - load balancer tags

        :param resource_object: Object filter
        :return: List of ELB resources
        """
        client = boto3.client('elb')

        def describe_elb(resource: dict) -> dict:
            """
            Add to the ELB resource additional information.

            :param resource: Resource
            :return: ELB resource enriched
            """
            tags = client.describe_tags(LoadBalancerNames=[resource['LoadBalancerName']])
            resource['Tags'] = tags['TagDescriptions'][0]['Tags']

            return resource

        try:
            resources = []
            described_lbs = client.describe_load_balancers()
            lbs_names = [x['LoadBalancerName'] for x in described_lbs['LoadBalancerDescriptions']]
            regex = re.compile(resource_object['name'].replace('*', '.*'))
            lbs_names = filter(regex.search, lbs_names)
            filtered_lbs = [x for x in described_lbs['LoadBalancerDescriptions'] if x['LoadBalancerName'] in lbs_names]
            for load_balancer in filtered_lbs:
                resources.append(describe_elb(load_balancer))
            return {'LoadBalancerDescriptions': resources}
        except ClientError:
            raise VirgaException('Lookup %s %s %s failed' % ('elb', 'name', resource_object['name']))
        except (KeyError, IndexError):
            return None

    @staticmethod
    def find_elbv2(resource_object: dict) -> any:
        """
        Call boto3/elbv2 for finding the ELBv2 instances by name.

        The result contains:
        - load balancer attributes
        - listeners
        - target groups
        - target group attributes
        - load balancer tags

        :param resource_object: Object filter
        :return: List of ELBv2 resources
        """
        client = boto3.client('elbv2')

        def describe_elbv2(resource: dict) -> dict:
            """
            Add to the ELBv2 resource additional information.

            :param resource: Resource
            :return: ELBv2 resource enriched
            """
            attributes = client.describe_load_balancer_attributes(LoadBalancerArn=resource['LoadBalancerArn'])
            resource['Attributes'] = attributes['Attributes']

            listeners = client.describe_listeners(LoadBalancerArn=resource['LoadBalancerArn'])
            resource['Listeners'] = listeners['Listeners']

            target_groups = client.describe_target_groups(LoadBalancerArn=resource['LoadBalancerArn'])

            result_target_group = []
            for target_group in target_groups['TargetGroups']:
                attributes = client.describe_target_group_attributes(
                    TargetGroupArn=target_group['TargetGroupArn'])['Attributes']
                target_group['Attributes'] = attributes
                result_target_group.append(target_group)
            resource['TargetGroups'] = result_target_group

            tags = client.describe_tags(ResourceArns=[resource['LoadBalancerArn']])
            resource['Tags'] = tags['TagDescriptions'][0]['Tags']

            return resource

        try:
            resources = []
            described_lbs = client.describe_load_balancers()
            lbs_names = [x['LoadBalancerName'] for x in described_lbs['LoadBalancers']]
            regex = re.compile(resource_object['name'].replace('*', '.*'))
            lbs_names = filter(regex.search, lbs_names)
            filtered_lbs = [x for x in described_lbs['LoadBalancers'] if x['LoadBalancerName'] in lbs_names]
            for load_balancer in filtered_lbs:
                resources.append(describe_elbv2(load_balancer))
            return {'LoadBalancers': resources}
        except ClientError:
            raise VirgaException('Lookup %s %s %s failed' % ('elbv2', 'name', resource_object['name']))
        except (KeyError, IndexError):
            return None

    @staticmethod
    def find_cloudtrail(resource_object: dict) -> dict:
        """
        Call boto3/cloudtrail for finding the CloudTrail trail by name.

        :param resource_object: Object filter
        :return: Response from AWS
        """
        client = boto3.client('cloudtrail')
        response = client.describe_trails()
        return {'trailList': list(filter(lambda d: d['Name'] == resource_object['name'], response['trailList']))}

    @staticmethod
    def find_autoscaling_groups(resource_object: dict) -> dict:
        """
        Call boto3/autoscaling for finding the AutoScaling group by name.

        :param resource_object: Object filter
        :return: Response from AWS
        """
        client = boto3.client('autoscaling')
        response = client.describe_auto_scaling_groups()
        group_names = [x['AutoScalingGroupName'] for x in response['AutoScalingGroups']]
        regex = re.compile(resource_object['name'].replace('*', '.*'))
        group_names = filter(regex.search, group_names)
        groups = [x for x in response['AutoScalingGroups'] if x['AutoScalingGroupName'] in group_names]
        return {'AutoScalingGroups': groups}

import re
import boto3
from botocore.exceptions import ClientError

from virga.common import VirgaException


class VirgaClient:
    """VirgaClient substitute the standard AWS client for more complex requests."""

    @staticmethod
    def find_certificate(resource_object: dict) -> any:
        """
        Call boto3/acm for finding the certificate for the passed domain.

        :param resource_object: Object filter
        :return: Response from AWS
        """
        client = boto3.client('acm')
        certificates = client.list_certificates()
        try:
            res_certificates = [
                cert for cert in certificates['CertificateSummaryList']
                if cert['DomainName'] == resource_object['domain_name']
            ]
            return client.describe_certificate(CertificateArn=res_certificates[0]['CertificateArn'])
        except ClientError:
            raise VirgaException('Lookup certificates domain_name %s failed' % resource_object['domain_name'])
        except (KeyError, IndexError):
            return None

    @staticmethod
    def find_elb(resource_object: dict) -> any:
        """
        Call boto3/elb for finding the ELB instances by name.

        The result contains:
        - load balancer attributes
        - load balancer tags

        :param resource_object: Object filter
        :return: List of ELB resources
        """
        client = boto3.client('elb')

        def describe_elb(resource: dict) -> dict:
            """
            Add to the ELB resource additional information.

            :param resource: Resource
            :return: ELB resource enriched
            """
            tags = client.describe_tags(LoadBalancerNames=[resource['LoadBalancerName']])
            resource['Tags'] = tags['TagDescriptions'][0]['Tags']

            return resource

        try:
            resources = []
            described_lbs = client.describe_load_balancers()
            lbs_names = [x['LoadBalancerName'] for x in described_lbs['LoadBalancerDescriptions']]
            regex = re.compile(resource_object['name'].replace('*', '.*'))
            lbs_names = filter(regex.search, lbs_names)
            filtered_lbs = [x for x in described_lbs['LoadBalancerDescriptions'] if x['LoadBalancerName'] in lbs_names]
            for load_balancer in filtered_lbs:
                resources.append(describe_elb(load_balancer))
            return {'LoadBalancerDescriptions': resources}
        except ClientError:
            raise VirgaException('Lookup %s %s %s failed' % ('elb', 'name', resource_object['name']))
        except (KeyError, IndexError):
            return None

    @staticmethod
    def find_elbv2(resource_object: dict) -> any:
        """
        Call boto3/elbv2 for finding the ELBv2 instances by name.

        The result contains:
        - load balancer attributes
        - listeners
        - target groups
        - target group attributes
        - load balancer tags

        :param resource_object: Object filter
        :return: List of ELBv2 resources
        """
        client = boto3.client('elbv2')

        def describe_elbv2(resource: dict) -> dict:
            """
            Add to the ELBv2 resource additional information.

            :param resource: Resource
            :return: ELBv2 resource enriched
            """
            attributes = client.describe_load_balancer_attributes(LoadBalancerArn=resource['LoadBalancerArn'])
            resource['Attributes'] = attributes['Attributes']

            listeners = client.describe_listeners(LoadBalancerArn=resource['LoadBalancerArn'])
            resource['Listeners'] = listeners['Listeners']

            target_groups = client.describe_target_groups(LoadBalancerArn=resource['LoadBalancerArn'])

            result_target_group = []
            for target_group in target_groups['TargetGroups']:
                attributes = client.describe_target_group_attributes(
                    TargetGroupArn=target_group['TargetGroupArn'])['Attributes']
                target_group['Attributes'] = attributes
                result_target_group.append(target_group)
            resource['TargetGroups'] = result_target_group

            tags = client.describe_tags(ResourceArns=[resource['LoadBalancerArn']])
            resource['Tags'] = tags['TagDescriptions'][0]['Tags']

            return resource

        try:
            resources = []
            described_lbs = client.describe_load_balancers()
            lbs_names = [x['LoadBalancerName'] for x in described_lbs['LoadBalancers']]
            regex = re.compile(resource_object['name'].replace('*', '.*'))
            lbs_names = filter(regex.search, lbs_names)
            filtered_lbs = [x for x in described_lbs['LoadBalancers'] if x['LoadBalancerName'] in lbs_names]
            for load_balancer in filtered_lbs:
                resources.append(describe_elbv2(load_balancer))
            return {'LoadBalancers': resources}
        except ClientError:
            raise VirgaException('Lookup %s %s %s failed' % ('elbv2', 'name', resource_object['name']))
        except (KeyError, IndexError):
            return None

    @staticmethod
    def find_cloudtrail(resource_object: dict) -> dict:
        """
        Call boto3/cloudtrail for finding the CloudTrail trail by name.

        :param resource_object: Object filter
        :return: Response from AWS
        """
        client = boto3.client('cloudtrail')
        response = client.describe_trails()
        return {'trailList': list(filter(lambda d: d['Name'] == resource_object['name'], response['trailList']))}

    @staticmethod
    def find_autoscaling_groups(resource_object: dict) -> dict:
        """
        Call boto3/autoscaling for finding the AutoScaling group by name.

        :param resource_object: Object filter
        :return: Response from AWS
        """
        client = boto3.client('autoscaling')
        response = client.describe_auto_scaling_groups()
        group_names = [x['AutoScalingGroupName'] for x in response['AutoScalingGroups']]
        regex = re.compile(resource_object['name'].replace('*', '.*'))
        group_names = filter(regex.search, group_names)
        groups = [x for x in response['AutoScalingGroups'] if x['AutoScalingGroupName'] in group_names]
        return {'AutoScalingGroups': groups}

VirgaClient substitute the standard AWS client for more complex requests. Call boto3/acm for finding the certificate for the passed domain.          :param resource_object: Object filter         :return: Response from AWS Call boto3/elb for finding the ELB instances by name.          The result contains:         - load balancer attributes         - load balancer tags          :param resource_object: Object filter         :return: List of ELB resources Add to the ELB resource additional information.              :param resource: Resource             :return: ELB resource enriched Call boto3/elbv2 for finding the ELBv2 instances by name.          The result contains:         - load balancer attributes         - listeners         - target groups         - target group attributes         - load balancer tags          :param resource_object: Object filter         :return: List of ELBv2 resources Add to the ELBv2 resource additional information.              :param resource: Resource             :return: ELBv2 resource enriched Call boto3/cloudtrail for finding the CloudTrail trail by name.          :param resource_object: Object filter         :return: Response from AWS Call boto3/autoscaling for finding the AutoScaling group by name.          :param resource_object: Object filter         :return: Response from AWS