awspec-api_gateway_extended 1.2.3

data/lib/awspec/type/alb.rb

@@ -0,0 +1,54 @@
+module Awspec::Type
+  class Alb < ResourceBase
+    tags_allowed
+    def resource_via_client
+      @resource_via_client ||= find_alb(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.load_balancer_name if resource_via_client
+    end
+
+    STATES = %w(
+      active provisioning failed
+    )
+
+    STATES.each do |state|
+      define_method state + '?' do
+        check_existence
+        resource_via_client.state.code == state
+      end
+    end
+
+    def has_security_group?(sg_id)
+      check_existence
+      sgs = resource_via_client.security_groups
+      ret = sgs.find do |sg|
+        sg == sg_id
+      end
+      return true if ret
+      sg2 = find_security_group(sg_id)
+      return true if sgs.include? sg2.group_id
+      false
+    end
+
+    def has_subnet?(subnet_id)
+      check_existence
+      azs = resource_via_client.availability_zones
+      ret = azs.find do |az|
+        az.subnet_id == subnet_id
+      end
+      return true if ret
+      subnet2 = find_subnet(subnet_id)
+      subnet2.subnet_id = subnet_id
+    end
+
+    def has_tag?(tag_key, tag_value)
+      alb_arn = resource_via_client.load_balancer_arn
+      tag_set = select_all_alb_tags(alb_arn)
+      tag_set.find do |tag|
+        tag.key == tag_key && tag.value == tag_value
+      end
+    end
+  end
+end

data/lib/awspec/type/alb_listener.rb

@@ -0,0 +1,43 @@
+module Awspec::Type
+  class AlbListener < ResourceBase
+    def resource_via_client
+      @resource_via_client ||= find_alb_listener(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.listener_arn if resource_via_client
+    end
+
+    def has_rule?(rule_id = nil, priority = nil, conditions = nil, actions = nil)
+      rules = select_rule_by_alb_listener_id(id)
+      ret = rules.select do |rule|
+        conditions = [conditions] if conditions.is_a?(Hash)
+        actions = [actions] if actions.is_a?(Hash)
+        next false if !rule_id.nil? && rule.rule_arn != rule_id
+        next false if !priority.nil? && rule.priority != priority
+        unless conditions.nil?
+          next false unless conditions.map do |condition|
+            rule_conds = rule.conditions.select { |rule_cond| rule_cond[:field] == condition[:field] }
+            next if rule_conds.empty?
+            next unless rule_conds.reject { |rule_cond| condition[:values] == rule_cond[:values] }.empty?
+            condition
+          end.compact.size == conditions.size
+        end
+        unless actions.nil?
+          actions = actions.map do |action|
+            if action.key?(:target_group_name)
+              target_group = find_alb_target_group(action[:target_group_name])
+              action[:target_group_arn] = target_group.target_group_arn
+              action.delete(:target_group_name)
+            end
+            action
+          end
+          next false if rule.actions.map(&:to_h).sort_by(&:to_s) != actions.sort_by(&:to_s)
+        end
+        true
+      end
+      ret.single_resource("rule_id = #{rule_id}, priority = #{priority}, \
+conditions = #{conditions}, actions = #{actions}")
+    end
+  end
+end

data/lib/awspec/type/alb_target_group.rb

@@ -0,0 +1,22 @@
+module Awspec::Type
+  class AlbTargetGroup < ResourceBase
+    def resource_via_client
+      @resource_via_client ||= find_alb_target_group(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.target_group_name if resource_via_client
+    end
+
+    def has_ec2?(id)
+      ec2 = find_ec2(id)
+      return nil unless ec2
+      descriptions = elbv2_client.describe_target_health(
+        target_group_arn: resource_via_client.target_group_arn
+      ).target_health_descriptions
+      descriptions.find do |description|
+        description.target.id == ec2.instance_id
+      end
+    end
+  end
+end

data/lib/awspec/type/ami.rb

@@ -0,0 +1,25 @@
+module Awspec::Type
+  class Ami < ResourceBase
+    aws_resource Aws::EC2::Image
+    tags_allowed
+
+    def resource_via_client
+      @resource_via_client ||= find_ami(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.image_id if resource_via_client
+    end
+
+    STATES = %w(
+      pending available invalid deregistered
+      transient failed error
+    )
+
+    STATES.each do |state|
+      define_method state + '?' do
+        resource_via_client.state == state
+      end
+    end
+  end
+end

data/lib/awspec/type/apigateway.rb

@@ -0,0 +1,65 @@
+require 'addressable/uri'
+
+module Awspec::Type
+  class Apigateway < ResourceBase
+    aws_resource Aws::APIGateway::Client
+
+    def resource_via_client
+      @resource_via_client ||= find_apigateway_by_id(@display_name)
+      return @resource_via_client if @resource_via_client
+      @resource_via_client = find_apigateway_by_name(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.id if resource_via_client
+    end
+
+    def api_resources
+      @api_resources.nil? ? @api_resources = find_api_resources_by_id(@id) : @api_resources
+    end
+
+    def has_path?(path)
+      check_existence
+      self.api_resources.each do |resource|
+        return resource if resource.path == path
+      end
+      nil
+    end
+
+    def has_integration_path?(path)
+      check_existence
+      self.api_resources.each do |resource|
+        next if resource.resource_methods.nil?
+        resource.resource_methods.each do |_, method|
+          if method.method_integration.http_method == 'AWS'
+            aws_path = method.method_integration.uri.match(%r{(\/[^\?]+)\??.*$}).captures[0]
+            return resource if aws_path == path
+          end
+          uri = Addressable::URI.parse(method.method_integration.uri)
+          return resource if uri.path == path
+        end
+      end
+      nil
+    end
+
+    def has_method?(path, http_method)
+      check_existence
+      resource_to_check = has_path?(path)
+      return nil if resource_to_check.nil?
+      resource_to_check.resource_methods.each do |_, method|
+        return resource_to_check if method.http_method == http_method
+      end
+      nil
+    end
+
+    def has_integration_method?(integration_path, http_method)
+      check_existence
+      integration_resource_to_check = has_integration_path?(integration_path)
+      return nil if integration_resource_to_check.nil?
+      integration_resource_to_check.resource_methods.each do |_, method|
+        return integration_resource_to_check if method.method_integration.http_method == http_method
+      end
+      nil
+    end
+  end
+end

data/lib/awspec/type/autoscaling_group.rb

@@ -0,0 +1,54 @@
+module Awspec::Type
+  class AutoscalingGroup < ResourceBase
+    aws_resource Aws::AutoScaling::AutoScalingGroup
+    tags_allowed
+
+    def resource_via_client
+      @resource_via_client ||= find_autoscaling_group(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.auto_scaling_group_arn if resource_via_client
+    end
+
+    def has_elb?(name)
+      resource_via_client.load_balancer_names.one? do |lb_name|
+        lb_name == name
+      end
+    end
+
+    def has_ec2?(id)
+      ec2 = find_ec2(id)
+      return nil unless ec2
+      resource_via_client.instances.one? do |instance|
+        instance.instance_id == ec2.instance_id
+      end
+    end
+
+    def has_suspended_process?(id)
+      resource_via_client.suspended_processes.one? do |process|
+        process.process_name == id
+      end
+    end
+
+    def has_launch_configuration?(name)
+      @resource_via_client.launch_configuration_name == name
+    end
+
+    def has_alb_target_group?(id)
+      target_group = find_alb_target_group(id)
+      target_groups = select_alb_target_group_by_autoscaling_group_name(@resource_via_client.auto_scaling_group_name)
+      target_groups.one? do |tg|
+        tg.load_balancer_target_group_arn == target_group.target_group_arn
+      end
+    end
+
+    def has_nlb_target_group?(id)
+      target_group = find_nlb_target_group(id)
+      target_groups = select_lb_target_group_by_autoscaling_group_name(@resource_via_client.auto_scaling_group_name)
+      target_groups.one? do |tg|
+        tg.load_balancer_target_group_arn == target_group.target_group_arn
+      end
+    end
+  end
+end

data/lib/awspec/type/base.rb

@@ -0,0 +1,63 @@
+require 'aws-sdk'
+require 'awspec/resource_reader'
+require 'awspec/helper/finder'
+require 'awspec/error'
+
+module Awspec::Type
+  class Base
+    include Awspec::Helper::Finder
+    include Awspec::BlackListForwardable
+    attr_accessor :account
+
+    def resource_via_client
+      raise 'this method must be override!'
+    end
+
+    def to_s
+      type = self.class.name.demodulize.underscore
+      "#{type} '#{@display_name}'"
+    end
+
+    def inspect
+      to_s
+    end
+
+    def respond_to_missing?(method, include_private = false)
+      resource_via_client.respond_to?(method) || super unless resource_via_client.nil?
+    end
+
+    def self.tags_allowed
+      define_method :has_tag? do |key, value|
+        begin
+          tags = resource_via_client.tags
+        rescue NoMethodError
+          tags = resource_via_client.tag_set
+        end
+        return false unless tags
+        tags.any? { |t| t['key'] == key && t['value'] == value }
+      end
+    end
+
+    def method_missing(name)
+      name_str = name.to_s if name.class == Symbol
+      describe = name_str.tr('-', '_').to_sym
+
+      if !resource_via_client.nil? && resource_via_client.members.include?(describe)
+        resource_via_client[describe]
+      elsif resource_via_client.nil?
+        raise Awspec::NoExistingResource.new(self.class, @display_name)
+      else
+        super unless self.respond_to?(:resource)
+        method_missing_via_black_list(name, delegate_to: resource)
+      end
+    end
+
+    undef :timeout
+
+    private
+
+    def check_existence
+      raise Awspec::NoExistingResource.new(self.class, @display_name) if resource_via_client.nil?
+    end
+  end
+end

data/lib/awspec/type/batch_compute_environment.rb

@@ -0,0 +1,32 @@
+module Awspec::Type
+  class BatchComputeEnvironment < ResourceBase
+    def initialize(compenv)
+      super
+      @display_name = compenv
+    end
+
+    def resource_via_client
+      @resource_via_client ||= find_batch_compute_environment(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.compute_environment_name if resource_via_client
+    end
+
+    STATES = %w(ENABLED DISABLED)
+
+    STATES.each do |state|
+      define_method state.downcase + '?' do
+        resource_via_client.state == state
+      end
+    end
+
+    TYPES = %w(MANAGED UNMANAGED)
+
+    TYPES.each do |type|
+      define_method type.downcase + '?' do
+        resource_via_client.type == type
+      end
+    end
+  end
+end

data/lib/awspec/type/batch_job_definition.rb

@@ -0,0 +1,16 @@
+module Awspec::Type
+  class BatchJobDefinition < ResourceBase
+    def initialize(jobdef)
+      super
+      @display_name = jobdef
+    end
+
+    def resource_via_client
+      @resource_via_client ||= find_batch_job_definition(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.job_definition_name if resource_via_client
+    end
+  end
+end

data/lib/awspec/type/batch_job_queue.rb

@@ -0,0 +1,31 @@
+module Awspec::Type
+  class BatchJobQueue < ResourceBase
+    def initialize(jobqueue)
+      super
+      @display_name = jobqueue
+    end
+
+    def resource_via_client
+      @resource_via_client ||= find_batch_job_queue(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.job_queue_name if resource_via_client
+    end
+
+    STATES = %w(ENABLED DISABLED)
+
+    STATES.each do |state|
+      define_method state.downcase + '?' do
+        resource_via_client.state == state
+      end
+    end
+
+    def has_compute_environment_order?(arn, order)
+      resource_via_client.compute_environment_order.each do |res|
+        return true if res.compute_environment == arn && res.order == order
+      end
+      false
+    end
+  end
+end

data/lib/awspec/type/cloudformation_stack.rb

@@ -0,0 +1,11 @@
+module Awspec::Type
+  class CloudformationStack < ResourceBase
+    def resource_via_client
+      @resource_via_client ||= find_cloudformation_stack(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.stack_name if resource_via_client
+    end
+  end
+end

data/lib/awspec/type/cloudfront_distribution.rb

@@ -0,0 +1,60 @@
+module Awspec::Type
+  class CloudfrontDistribution < ResourceBase
+    def resource_via_client
+      @resource_via_client ||= find_cloudfront_distribution(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.id if resource_via_client
+    end
+
+    STATUSES = %w(
+      InProgress Deployed
+    )
+
+    STATUSES.each do |status|
+      define_method status.underscore + '?' do
+        resource_via_client.status == status
+      end
+    end
+
+    def has_origin?(origin_id,
+                    domain_name: nil,
+                    origin_path: nil,
+                    origin_access_identity: nil)
+      return false unless [origin_id, domain_name].any?
+      resource_via_client.origins.items.find do |origin|
+        next false if !origin_id.nil? && origin.id != origin_id
+        next false if !domain_name.nil? && origin.domain_name != domain_name
+        next false if !origin_path.nil? && origin.origin_path != origin_path
+        next false if !origin_access_identity.nil? && \
+                      origin.s3_origin_config.origin_access_identity != origin_access_identity
+        true
+      end
+    end
+
+    def has_origin_domain_name?(domain_name)
+      has_origin?(nil, domain_name: domain_name)
+    end
+
+    def has_origin_domain_name_and_path?(domain_name_and_path)
+      domain_name = domain_name_and_path.gsub(%r(/.*\z), '')
+      origin_path = domain_name_and_path.gsub(%r(\A[^/]*), '')
+      has_origin?(nil, domain_name: domain_name, origin_path: origin_path)
+    end
+
+    def has_custom_response_error_code?(error_code,
+                                        response_page_path: nil,
+                                        response_code: nil,
+                                        error_caching_min_ttl: nil)
+      return false unless [error_code, domain_name].any?
+      resource_via_client.custom_error_responses.items.find do |error|
+        next false if !error_code.nil? && error.error_code != error_code
+        next false if !response_page_path.nil? && error.response_page_path != response_page_path
+        next false if !response_code.nil? && error.response_code != response_code.to_s
+        next false if !error_caching_min_ttl.nil? && error.error_caching_min_ttl != error_caching_min_ttl
+        true
+      end
+    end
+  end
+end