cmonson_2ndwatch_awspec 0.85.3

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,18 @@
+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
+          (res.next_page? && res = res.next_page) || break
+        end
+
+        selected.single_resource(id)
+      end
+    end
+  end
+end

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

@@ -0,0 +1,23 @@
+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 is_logging?(id)
+        ret = get_trail_status(id).is_logging
+      end
+    end
+  end
+end

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

@@ -0,0 +1,30 @@
+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
+          (res.next_page? && res = res.next_page) || break
+        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,68 @@
+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 find_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_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
+
+      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.method(method_name).call(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

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

@@ -0,0 +1,19 @@
+module Awspec::Helper
+  module Finder
+    module Directconnect
+      def find_virtual_interface(virtual_interface_id)
+        res = directconnect_client.describe_virtual_interfaces
+        ret = res.virtual_interfaces.select do |virtual_interface|
+          next true if virtual_interface.virtual_interface_id == virtual_interface_id
+          next true if virtual_interface.virtual_interface_name == virtual_interface_id
+        end
+        ret.single_resource(virtual_interface_id)
+      end
+
+      def select_virtual_interfaces
+        res = directconnect_client.describe_virtual_interfaces
+        res.virtual_interfaces
+      end
+    end
+  end
+end

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

@@ -0,0 +1,10 @@
+module Awspec::Helper
+  module Finder
+    module Dynamodb
+      def find_dynamodb_table(id)
+        resource = dynamodb_client.describe_table({ table_name: id }).table
+        return resource if resource.table_name == id
+      end
+    end
+  end
+end

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

@@ -0,0 +1,40 @@
+module Awspec::Helper
+  module Finder
+    module Ebs
+      def find_ebs(volume_id)
+        res = ec2_client.describe_volumes({
+                                            filters: [{ name: 'volume-id', values: [volume_id] }]
+                                          })
+        resource = res.volumes.single_resource(volume_id)
+        return resource if resource
+        res = ec2_client.describe_volumes({
+                                            filters: [{ name: 'tag:Name', values: [volume_id] }]
+                                          })
+        res.volumes.single_resource(volume_id)
+      end
+
+      def select_ebs_by_instance_id(id)
+        res = find_ec2(id)
+        volumes = []
+        return volumes unless res
+        res.block_device_mappings.each do |block|
+          volume = find_ebs(block.ebs.volume_id)
+          volumes.push(volume) if volume
+        end
+        volumes
+      end
+
+      def select_all_attached_ebs
+        selected = []
+        res = ec2_client.describe_volumes
+
+        loop do
+          selected += res.volumes.select { |v| v.attachments.count > 0 }
+          (res.next_page? && res = res.next_page) || break
+        end
+
+        selected
+      end
+    end
+  end
+end

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

@@ -0,0 +1,158 @@
+module Awspec::Helper
+  module Finder
+    module Ec2
+      def find_ec2(id)
+        # instance_id or tag:Name
+        begin
+          res = ec2_client.describe_instances({
+                                                instance_ids: [id]
+                                              })
+        rescue
+          # Aws::EC2::Errors::InvalidInstanceIDMalformed
+          # Aws::EC2::Errors::InvalidInstanceIDNotFound
+          res = ec2_client.describe_instances({
+                                                filters: [{ name: 'tag:Name', values: [id] }]
+                                              })
+        end
+        # rubocop:enable Style/GuardClause
+        if res.reservations.count == 1
+          res.reservations.first.instances.single_resource(id)
+        elsif res.reservations.count > 1
+          raise Awspec::DuplicatedResourceTypeError, "Duplicate instances matching id or tag #{id}"
+        end
+      end
+
+      def find_ec2_attribute(id, attribute)
+        res = ec2_client.describe_instance_attribute({
+                                                       instance_id: id, attribute: attribute
+                                                     })
+      end
+
+      def find_ec2_status(id)
+        res = ec2_client.describe_instance_status({
+                                                    instance_ids: [id]
+                                                  })
+        res.instance_statuses.first if res.instance_statuses.count == 1
+      end
+
+      # find_internet_gateway find_vpn_gateway find_customer_gateway
+      gateway_types = %w(internet vpn customer)
+      gateway_types.each do |type|
+        define_method 'find_' + type + '_gateway' do |*args|
+          gateway_id = args.first
+          method_name = 'describe_' + type + '_gateways'
+          res = ec2_client.method(method_name).call({
+                                                      filters: [{ name: type + '-gateway-id', values: [gateway_id] }]
+                                                    })
+          resource = res[type + '_gateways'].single_resource(gateway_id)
+          return resource if resource
+          res = ec2_client.method(method_name).call({
+                                                      filters: [{ name: 'tag:Name', values: [gateway_id] }]
+                                                    })
+          res[type + '_gateways'].single_resource(gateway_id)
+        end
+      end
+
+      def find_vpn_connection(vpn_connection_id)
+        res = ec2_client.describe_vpn_connections({
+                                                    filters: [
+                                                      {
+                                                        name: 'vpn-connection-id',
+                                                        values: [vpn_connection_id]
+                                                      }
+                                                    ]
+                                                  })
+        resource = res.vpn_connections.single_resource(vpn_connection_id)
+        return resource if resource
+        res = ec2_client.describe_vpn_connections({
+                                                    filters: [
+                                                      {
+                                                        name: 'tag:Name',
+                                                        values: [vpn_connection_id]
+                                                      }
+                                                    ]
+                                                  })
+        res.vpn_connections.single_resource(vpn_connection_id)
+      end
+
+      def find_nat_gateway(gateway_id)
+        res = ec2_client.describe_nat_gateways({
+                                                 filter: [{ name: 'nat-gateway-id', values: [gateway_id] }]
+                                               })
+        res.nat_gateways.single_resource(gateway_id)
+      end
+
+      def find_network_interface(interface_id)
+        res = ec2_client.describe_network_interfaces({
+                                                       filters: [
+                                                         {
+                                                           name: 'network-interface-id',
+                                                           values: [interface_id]
+                                                         }
+                                                       ]
+                                                     })
+        resource = res.network_interfaces.single_resource(interface_id)
+        return resource if resource
+        res = ec2_client.describe_network_interfaces({
+                                                       filters: [{ name: 'tag:Name', values: [interface_id] }]
+                                                     })
+        res.network_interfaces.single_resource(interface_id)
+      end
+
+      def select_ec2_by_vpc_id(vpc_id)
+        res = ec2_client.describe_instances({
+                                              filters: [{ name: 'vpc-id', values: [vpc_id] }]
+                                            })
+        instances = []
+        res.reservations.each do |reservation|
+          reservation.instances.each do |instance|
+            instances.push(instance)
+          end
+        end
+        instances
+      end
+
+      def select_eip_by_instance_id(id)
+        res = ec2_client.describe_addresses({
+                                              filters: [{ name: 'instance-id', values: [id] }]
+                                            })
+        res.addresses
+      end
+
+      def select_eip_by_public_ip(id)
+        res = ec2_client.describe_addresses({
+                                              filters: [{ name: 'public-ip', values: [id] }]
+                                            })
+        res.addresses
+      end
+
+      def select_network_interface_by_instance_id(id)
+        res = ec2_client.describe_network_interfaces({
+                                                       filters: [{ name: 'attachment.instance-id', values: [id] }]
+                                                     })
+        res.network_interfaces
+      end
+
+      def select_nat_gateway_by_vpc_id(vpc_id)
+        res = ec2_client.describe_nat_gateways({
+                                                 filter: [{ name: 'vpc-id', values: [vpc_id] }]
+                                               })
+        res.nat_gateways
+      end
+
+      def select_internet_gateway_by_vpc_id(vpc_id)
+        res = ec2_client.describe_internet_gateways({
+                                                      filters: [{ name: 'attachment.vpc-id', values: [vpc_id] }]
+                                                    })
+        res.internet_gateways
+      end
+
+      def select_network_interface_by_vpc_id(vpc_id)
+        res = ec2_client.describe_network_interfaces({
+                                                       filters: [{ name: 'vpc-id', values: [vpc_id] }]
+                                                     })
+        res.network_interfaces
+      end
+    end
+  end
+end

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

@@ -0,0 +1,10 @@
+module Awspec::Helper
+  module Finder
+    module Ecr
+      def find_ecr_repository(repository_name)
+        res = ecr_client.describe_repositories(repository_names: [repository_name])
+        res.repositories.first if res.repositories.count == 1
+      end
+    end
+  end
+end

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

@@ -0,0 +1,45 @@
+module Awspec::Helper
+  module Finder
+    module Ecs
+      def find_ecs_cluster(cluster)
+        res = ecs_client.describe_clusters(clusters: [cluster])
+        res.clusters.single_resource(cluster)
+      end
+
+      def find_ecs_container_instance(cluster, uuid)
+        res = ecs_client.describe_container_instances(cluster: cluster, container_instances: [uuid])
+        res.container_instances.single_resource(uuid)
+      end
+
+      def find_ecs_task_definition(taskdef)
+        res = ecs_client.describe_task_definition(task_definition: taskdef)
+        res.task_definition
+      end
+
+      def find_ecs_service(service)
+        res = ecs_client.describe_services(services: [service])
+        res.services.single_resource(service)
+      end
+
+      def select_ecs_container_instance_arn_by_cluster_name(cluster)
+        req = { cluster: cluster }
+        arns = []
+        loop do
+          res = ecs_client.list_container_instances(req)
+          arns.push(*res.container_instance_arns)
+          break if res.next_token.nil?
+          req[:next_token] = res.next_token
+        end
+        arns
+      end
+
+      # deprecated method
+      def find_ecs_container_instances(cluster, container_instances)
+        res = ecs_client.describe_container_instances(cluster: cluster, container_instances: container_instances)
+        res.container_instances if res.container_instances
+      end
+
+      alias_method :list_ecs_container_instances, :select_ecs_container_instance_arn_by_cluster_name # deprecated method
+    end
+  end
+end