cthiesfork-awspec 1.0.0

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

@@ -0,0 +1,24 @@
+module Awspec::Helper
+  module Finder
+    module Codebuild
+      def find_codebuild_project(id)
+        projects = select_all_codebuild_projects.select do |project|
+          project == id
+        end
+        projects.single_resource(id)
+      end
+
+      def select_all_codebuild_projects
+        req = {}
+        projects = []
+        loop do
+          res = codebuild_client.list_projects(req)
+          projects.push(*res.projects)
+          break if res.next_token.nil?
+          req[:next_token] = res.next_token
+        end
+        projects
+      end
+    end
+  end
+end

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

@@ -0,0 +1,15 @@
+module Awspec::Helper
+  module Finder
+    module Codedeploy
+      def find_codedeploy_app(name)
+        res = codedeploy_client.get_application(application_name: name)
+        res.application
+      end
+
+      def find_codedeploy_deployment_group(application, group)
+        res = codedeploy_client.get_deployment_group(application_name: application, deployment_group_name: group)
+        res.deployment_group_info
+      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,43 @@
+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 }
+          break if res.next_token.nil?
+          res = ec2_client.describe_volumes({
+                                              next_token: res.next_token
+                                            })
+        end
+
+        selected
+      end
+    end
+  end
+end

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

@@ -0,0 +1,203 @@
+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_credit_specifications(id)
+        res = ec2_client.describe_instance_credit_specifications({
+                                                                   instance_ids: [id]
+                                                                 })
+        res.instance_credit_specifications.first if res.instance_credit_specifications.count == 1
+      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.send(
+            method_name,
+            { filters: [{ name: type + '-gateway-id', values: [gateway_id] }] }
+          )
+          resource = res[type + '_gateways'].single_resource(gateway_id)
+          return resource if resource
+          res = ec2_client.send(
+            method_name,
+            { 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)
+        # nat_gateway_id or tag:Name
+        begin
+          res = ec2_client.describe_nat_gateways({
+                                                   nat_gateway_ids: [gateway_id]
+                                                 })
+        rescue
+          res = ec2_client.describe_nat_gateways({
+                                                   filter: [{ name: 'tag:Name', values: [gateway_id] }]
+                                                 })
+        end
+        if res.nat_gateways.count == 1
+          res.nat_gateways.single_resource(gateway_id)
+        elsif res.nat_gateways.count > 1
+          raise Awspec::DuplicatedResourceTypeError, "Duplicate nat_gateways matching id or tag #{gateway_id}"
+        end
+      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
+
+      def find_launch_template(id)
+        # launch_template_id or launch_template_name
+        begin
+          res = ec2_client.describe_launch_templates({
+                                                       launch_template_ids: [id]
+                                                     })
+        rescue
+          res = ec2_client.describe_launch_templates({
+                                                       launch_template_names: [id]
+                                                     })
+        end
+        res.launch_templates.single_resource(id)
+      end
+
+      def find_launch_template_versions(id)
+        # launch_template_id or launch_template_name
+        res = ec2_client.describe_launch_template_versions({
+                                                             launch_template_id: id
+                                                           })
+      rescue
+        res = ec2_client.describe_launch_template_versions({
+                                                             launch_template_name: id
+                                                           })
+      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(cluster, service)
+        res = ecs_client.describe_services(cluster: cluster, 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

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

@@ -0,0 +1,63 @@
+module Awspec::Helper
+  module Finder
+    module Efs
+      def find_efs(id)
+        begin
+          res = efs_client.describe_file_systems({
+                                                   file_system_id: id,
+                                                   max_items: 1
+                                                 })
+        rescue
+          # Aws::EFS::Errors::BadRequest (invalid file system ID: my-efs)
+          file_system_id = get_id_by_name_tag(id)
+          res = efs_client.describe_file_systems({
+                                                   file_system_id: file_system_id,
+                                                   max_items: 1
+                                                 })
+        end
+        res.file_systems.single_resource(id)
+      end
+
+      def find_efs_tags(id, tag_key)
+        begin
+          tag_set = efs_client.describe_tags({ file_system_id: id })
+        rescue
+          # Aws::EFS::Errors::BadRequest (invalid file system ID: my-efs)
+          file_system_id = get_id_by_name_tag(id)
+          tag_set = efs_client.describe_tags({
+                                               file_system_id: file_system_id
+                                             })
+        end
+        tag_query = tag_set.tags.find { |tag| tag.key == tag_key }
+        return tag_query if tag_query
+      end
+
+      def get_id_by_name_tag(name)
+        # takes a name tag and iterates inventory and returns file_system obj
+        inventory = efs_client.describe_file_systems
+        file_systems = inventory.file_systems.select do |fs|
+          tag_query = efs_client.describe_tags({
+                                                 file_system_id: fs.file_system_id
+                                               })
+          name_tag = tag_query.tags.find { |tag| tag.key == 'Name' } if tag_query.tags
+          name_tag && name_tag.value == name
+        end
+        file_systems.single_resource(name).file_system_id
+      end
+
+      def get_name_by_id(id)
+        tag_query = efs_client.describe_tags({
+                                               file_system_id: id
+                                             })
+        name_tag = tag_query.tags.find { |tag| tag.key == 'Name' }
+        return name_tag.value if name_tag
+      end
+
+      def select_all_file_systems
+        file_systems_query = efs_client.describe_file_systems
+        file_systems = file_systems_query.file_systems
+        file_systems
+      end
+    end
+  end
+end

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

@@ -0,0 +1,10 @@
+module Awspec::Helper
+  module Finder
+    module Eks
+      def find_eks_cluster(name)
+        res = eks_client.describe_cluster({ name: name })
+        res.cluster
+      end
+    end
+  end
+end