cthiesfork-awspec 1.0.0

data/lib/awspec/helper/finder/account_attributes.rb

@@ -0,0 +1,61 @@
+module Awspec::Helper
+  module Finder
+    module AccountAttributes
+      def find_ec2_account_attributes
+        attributes = {}
+        ec2_client.describe_account_attributes.account_attributes.each do |attr|
+          values = attr[:attribute_values].map do |v|
+            v[:attribute_value]
+          end
+          if attr[:attribute_name] == 'supported-platforms'
+            # supported-platforms
+            attributes[attr[:attribute_name].tr('-', '_').to_sym] = values
+          else
+            value = values.first
+            if value =~ /\A\d+\z/
+              attributes[attr[:attribute_name].tr('-', '_').to_sym] = value.to_i
+            else
+              attributes[attr[:attribute_name].tr('-', '_').to_sym] = value
+            end
+          end
+        end
+        attributes.to_struct
+      end
+
+      def find_rds_account_attributes
+        attributes = {}
+        rds_client.describe_account_attributes.account_quotas.each do |attr|
+          values = {
+            used: attr[:used],
+            max: attr[:max]
+          }
+          attributes[attr[:account_quota_name].to_sym] = values.to_struct
+        end
+        attributes.to_struct
+      end
+
+      def find_lambda_account_settings
+        attributes = {}
+        settings = lambda_client.get_account_settings
+        settings.account_limit.members.each do |key|
+          attributes[key] = { limit: settings.account_limit[key] }
+        end
+        settings.account_usage.members.each do |key|
+          if attributes.include?(key)
+            attributes[key][:usage] = settings.account_usage[key]
+          else
+            attributes[key] = { usage: settings.account_usage[key] }
+          end
+        end
+        attributes.to_struct
+      end
+
+      def find_ses_send_quota
+        ses_client.get_send_quota
+      rescue
+        # Aws::Errors::NoSuchEndpointError
+        nil
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/acm.rb

@@ -0,0 +1,40 @@
+module Awspec::Helper
+  module Finder
+    module Acm
+      def find_certificate(id)
+        selected = []
+        loop do
+          req = {}
+          res = acm_client.list_certificates(req)
+          selected += res.certificate_summary_list.select do |c|
+            c.certificate_arn == id || c.domain_name == id
+          end
+          break if res.next_token.nil?
+          req[:next_token] = res.next_token
+        end
+
+        cert = selected.single_resource(id)
+        acm_client.describe_certificate({ certificate_arn: cert.certificate_arn }).certificate
+      end
+
+      def select_all_certificates
+        certs = []
+        loop do
+          req = {}
+          res = acm_client.list_certificates(req)
+          res.certificate_summary_list.each do |c|
+            certs << c.certificate_arn
+          end
+          break if res.next_token.nil?
+          req[:next_token] = res.next_token
+        end
+
+        certificates = []
+        certs.each do |cert|
+          certificates << acm_client.describe_certificate({ certificate_arn: cert }).certificate
+        end
+        certificates
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/alb.rb

@@ -0,0 +1,68 @@
+module Awspec::Helper
+  module Finder
+    module Alb
+      def find_alb(id)
+        res = elbv2_client.describe_load_balancers({ names: [id] })
+        res.load_balancers.select do |lb|
+          lb.type == 'application'
+        end.single_resource(id)
+      rescue Aws::ElasticLoadBalancingV2::Errors::LoadBalancerNotFound
+        return nil
+      end
+
+      def select_alb_by_vpc_id(vpc_id)
+        res = elbv2_client.describe_load_balancers
+        res.load_balancers.select do |lb|
+          lb.vpc_id == vpc_id && lb.type == 'application'
+        end
+      end
+
+      def find_alb_listener(arn)
+        res = elbv2_client.describe_listeners({ listener_arns: [arn] })
+        res.listeners.single_resource(arn)
+      rescue StandardError
+        return nil
+      end
+
+      def select_alb_listener_by_alb_arn(arn)
+        selected = []
+        next_marker = nil
+        loop do
+          res = elbv2_client.describe_listeners({ marker: next_marker, load_balancer_arn: arn })
+          selected += res.listeners unless res.nil?
+          (res.nil? && next_marker = res.next_marker) || break
+        end
+        selected
+      end
+
+      def find_alb_target_group(id)
+        res = elbv2_client.describe_target_groups({ names: [id] })
+        res.target_groups.select do |tg|
+          %w(HTTP HTTPS).include?(tg.protocol)
+        end.single_resource(id)
+      rescue StandardError
+        return nil
+      end
+
+      def select_rule_by_alb_listener_id(id)
+        selected = []
+        next_marker = nil
+        loop do
+          res = elbv2_client.describe_rules(marker: next_marker, listener_arn: id)
+          selected += res.rules unless res.nil?
+          (res.nil? && next_marker = res.next_marker) || break
+        end
+        selected
+      end
+
+      def select_all_alb_tags(id)
+        res = elbv2_client.describe_tags({ resource_arns: [id] })
+        res.tag_descriptions.select do |resource|
+          resource.resource_arn == id
+        end.first.tags
+      rescue StandardError
+        return nil
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/ami.rb

@@ -0,0 +1,22 @@
+module Awspec::Helper
+  module Finder
+    module Ami
+      def find_ami(image_id)
+        res = ec2_client.describe_images({
+                                           filters: [{ name: 'image-id', values: [image_id] }]
+                                         })
+        resource = res.images.single_resource(image_id)
+        return resource if resource
+        res = ec2_client.describe_images({
+                                           filters: [{ name: 'name', values: [image_id] }]
+                                         })
+        resource = res.images.single_resource(image_id)
+        return resource if resource
+        res = ec2_client.describe_images({
+                                           filters: [{ name: 'tag:Name', values: [image_id] }]
+                                         })
+        res.images.single_resource(image_id)
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/apigateway.rb

@@ -0,0 +1,35 @@
+module Awspec::Helper
+  module Finder
+    module Apigateway
+      def find_apigateway_by_id(id)
+        apis = []
+        apigateway_client.get_rest_apis(limit: 500).each do |response|
+          apis += response.items
+        end
+        apis.each do |api|
+          return api if api.id == id
+        end
+        nil
+      end
+
+      def find_apigateway_by_name(name)
+        apis = []
+        apigateway_client.get_rest_apis(limit: 500).each do |response|
+          apis += response.items
+        end
+        apis.each do |api|
+          return api if api.name == name
+        end
+        nil
+      end
+
+      def find_api_resources_by_id(api_id)
+        all_resources = []
+        apigateway_client.get_resources(rest_api_id: api_id, limit: 500).each do |response|
+          all_resources += response.items
+        end
+        all_resources != [] ? all_resources : nil
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/autoscaling.rb

@@ -0,0 +1,60 @@
+module Awspec::Helper
+  module Finder
+    module Autoscaling
+      def find_autoscaling_group(id)
+        res = autoscaling_client.describe_auto_scaling_groups({
+                                                                auto_scaling_group_names: [id]
+                                                              })
+        res.auto_scaling_groups.single_resource(id)
+      end
+
+      def find_launch_configuration(id)
+        res = autoscaling_client.describe_launch_configurations({
+                                                                  launch_configuration_names: [id]
+                                                                })
+        res.launch_configurations.single_resource(id)
+      end
+
+      def select_alb_target_group_by_autoscaling_group_name(name)
+        res = autoscaling_client.describe_load_balancer_target_groups({
+                                                                        auto_scaling_group_name: name
+                                                                      })
+        res.load_balancer_target_groups
+      end
+
+      def select_lb_target_group_by_autoscaling_group_name(name)
+        res = autoscaling_client.describe_load_balancer_target_groups({
+                                                                        auto_scaling_group_name: name
+                                                                      })
+        res.load_balancer_target_groups
+      end
+
+      def find_block_device_mapping(id, device_id)
+        ret = find_launch_configuration(id).block_device_mappings.select do |device|
+          next true if device.device_name == device_id
+          next true if device.virtual_name == device_id
+        end
+        ret.single_resource(device_id)
+      end
+
+      def select_autoscaling_group_by_vpc_id(vpc_id)
+        subnets = []
+        select_subnet_by_vpc_id(vpc_id).each { |subnet| subnets << subnet.subnet_id }
+        req = {}
+        selected = []
+        loop do
+          res = autoscaling_client.describe_auto_scaling_groups(req)
+          res.auto_scaling_groups.select do |auto_scaling_group|
+            vpc_zone_identifiers = auto_scaling_group.vpc_zone_identifier.split(',')
+            selected.push(auto_scaling_group) if vpc_zone_identifiers.any? do |vpc_zone_identifier|
+              subnets.include?(vpc_zone_identifier)
+            end
+          end
+          break if res.next_token.nil?
+          req[:next_token] = res.next_token
+        end
+        selected
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/batch.rb

@@ -0,0 +1,20 @@
+module Awspec::Helper
+  module Finder
+    module Batch
+      def find_batch_compute_environment(name)
+        res = batch_client.describe_compute_environments(compute_environments: [name], max_results: 1)
+        res.compute_environments[0] if res.compute_environments
+      end
+
+      def find_batch_job_definition(name)
+        res = batch_client.describe_job_definitions(job_definitions: [name], max_results: 1)
+        res.job_definitions[0] if res.job_definitions
+      end
+
+      def find_batch_job_queue(name)
+        res = batch_client.describe_job_queues(job_queues: [name], max_results: 1)
+        res.job_queues[0] if res.job_queues
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/cloudformation.rb

@@ -0,0 +1,12 @@
+module Awspec::Helper
+  module Finder
+    module Cloudformation
+      def find_cloudformation_stack(id)
+        res = cloudformation_client.describe_stacks({
+                                                      stack_name: id
+                                                    })
+        res.stacks.single_resource(id)
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/cloudfront.rb

@@ -0,0 +1,21 @@
+module Awspec::Helper
+  module Finder
+    module Cloudfront
+      def find_cloudfront_distribution(id)
+        selected = []
+        res = cloudfront_client.list_distributions
+        loop do
+          selected += res.distribution_list.items.select do |item|
+            item.id == id || item.domain_name == id
+          end
+          break unless res.distribution_list.is_truncated
+          res = cloudfront_client.list_distributions({
+                                                       marker: res.distribution_list.next_marker
+                                                     })
+        end
+
+        selected.single_resource(id)
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/cloudtrail.rb

@@ -0,0 +1,29 @@
+module Awspec::Helper
+  module Finder
+    module Cloudtrail
+      def select_all_trails
+        cloudtrail_client.describe_trails.trail_list
+      end
+
+      def find_trail(id)
+        ret = cloudtrail_client.describe_trails.trail_list.find do |trail|
+          trail.name == id
+        end
+      end
+
+      def get_trail_status(id)
+        cloudtrail_client.get_trail_status(name: id)
+      end
+
+      def get_trail_tags(arn)
+        cloudtrail_client.list_tags(
+          resource_id_list: [arn]
+        )[:resource_tag_list].first[:tags_list]
+      end
+
+      def is_logging?(id)
+        ret = get_trail_status(id).is_logging
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/cloudwatch.rb

@@ -0,0 +1,33 @@
+module Awspec::Helper
+  module Finder
+    module Cloudwatch
+      def find_cloudwatch_alarm(id)
+        res = cloudwatch_client.describe_alarms({
+                                                  alarm_names: [id]
+                                                })
+        return res.metric_alarms.first if res.metric_alarms.count == 1
+
+        res = cloudwatch_client.describe_alarms
+        alarms = res.metric_alarms.select do |alarm|
+          alarm.alarm_arn == id
+        end
+        alarms.single_resource(id)
+      end
+
+      def select_all_cloudwatch_alarms
+        selected = []
+        res = cloudwatch_client.describe_alarms
+
+        loop do
+          selected += res.metric_alarms
+          break if res.next_token.nil?
+          res = cloudwatch_client.describe_alarms({
+                                                    next_token: res.next_token
+                                                  })
+        end
+
+        selected
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/cloudwatch_event.rb

@@ -0,0 +1,15 @@
+module Awspec::Helper
+  module Finder
+    module CloudwatchEvent
+      def find_cloudwatch_event(id)
+        cloudwatch_event_client.list_rules.rules.find do |rule|
+          rule.name == id || rule.arn == id
+        end
+      end
+
+      def select_all_cloudwatch_events
+        cloudwatch_event_client.list_rules.rules
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/cloudwatch_logs.rb

@@ -0,0 +1,93 @@
+module Awspec::Helper
+  module Finder
+    module CloudwatchLogs
+      def find_cloudwatch_logs_group(id)
+        cloudwatch_logs_client.describe_log_groups({
+                                                     log_group_name_prefix: id
+                                                   }).log_groups.find do |log_group|
+          log_group.log_group_name == id
+        end
+      end
+
+      def last_cloudwatch_logs_stream_by_log_group_name(id)
+        cloudwatch_logs_client.describe_log_streams({ log_group_name: id }).log_streams.last
+      end
+
+      def find_cloudwatch_logs_stream_by_log_group_name(id, stream_name)
+        req = {
+          log_group_name: id,
+          log_stream_name_prefix: stream_name
+        }
+        ret = nil
+        loop do
+          res = cloudwatch_logs_client.describe_log_streams(req)
+          res.log_streams.find do |log_stream|
+            if log_stream[:log_stream_name] == stream_name
+              ret = log_stream[:log_stream_name]
+              break
+            end
+          end
+          break if ret.present?
+          break if res.next_token.nil?
+          req[:next_token] = res.next_token
+        end
+        ret
+      end
+
+      def find_cloudwatch_logs_metric_fileter_by_log_group_name(id, filter_name)
+        cloudwatch_logs_client.describe_metric_filters({
+                                                         log_group_name: id,
+                                                         filter_name_prefix: filter_name
+                                                       }).metric_filters.find do |filter|
+          filter.filter_name == filter_name
+        end
+      end
+
+      def find_cloudwatch_logs_subscription_fileter_by_log_group_name(id, filter_name)
+        cloudwatch_logs_client.describe_subscription_filters({
+                                                               log_group_name: id,
+                                                               filter_name_prefix: filter_name
+                                                             }).subscription_filters.find do |filter|
+          filter.filter_name == filter_name
+        end
+      end
+
+      def select_all_cloudwatch_logs_log_groups
+        req = {}
+        log_groups = []
+        loop do
+          res = cloudwatch_logs_client.describe_log_groups(req)
+          log_groups.push(*res.log_groups)
+          break if res.next_token.nil?
+          req[:next_token] = res.next_token
+        end
+        log_groups
+      end
+
+      def find_tags_by_log_group_name(id)
+        cloudwatch_logs_client.list_tags_log_group(log_group_name: id)[:tags]
+      end
+
+      filter_types = %w(metric subscription)
+      filter_types.each do |type|
+        define_method 'select_all_cloudwatch_logs_' + type + '_filter' do |*args|
+          req = { log_group_name: args.first }
+          method_name = 'describe_' + type + '_filters'
+          resources = []
+          loop do
+            res = cloudwatch_logs_client.send(method_name, req)
+            case type
+            when 'metric' then
+              resources.push(*res.metric_filters)
+            when 'subscription' then
+              resources.push(*res.subscription_filters)
+            end
+            break if res.next_token.nil?
+            req[:next_token] = res.next_token
+          end
+          resources
+        end
+      end
+    end
+  end
+end