cmonson_2ndwatch_awspec 0.85.3

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/autoscaling_group.rb

@@ -0,0 +1,46 @@
+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
+  end
+end

data/lib/awspec/type/base.rb

@@ -0,0 +1,47 @@
+require 'aws-sdk'
+require 'awspec/resource_reader'
+require 'awspec/helper/finder'
+
+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 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]
+      else
+        super unless self.respond_to?(:resource)
+        method_missing_via_black_list(name, delegate_to: resource)
+      end
+    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,46 @@
+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
+  end
+end

data/lib/awspec/type/cloudtrail.rb

@@ -0,0 +1,29 @@
+module Awspec::Type
+  class Cloudtrail < ResourceBase
+    aws_resource Aws::CloudTrail
+
+    def resource_via_client
+      @resource_via_client ||= find_trail(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.name if resource_via_client
+    end
+
+    def has_global_service_events_included?
+      resource_via_client.include_global_service_events
+    end
+
+    def multi_region_trail?
+      resource_via_client.is_multi_region_trail
+    end
+
+    def has_log_file_validation_enabled?
+      resource_via_client.log_file_validation_enabled
+    end
+
+    def logging?
+      is_logging?(id)
+    end
+  end
+end

data/lib/awspec/type/cloudwatch_alarm.rb

@@ -0,0 +1,23 @@
+module Awspec::Type
+  class CloudwatchAlarm < ResourceBase
+    def resource_via_client
+      @resource_via_client ||= find_cloudwatch_alarm(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.alarm_arn if resource_via_client
+    end
+
+    def has_ok_action?(name)
+      resource_via_client.ok_actions.include?(name)
+    end
+
+    def has_alarm_action?(name)
+      resource_via_client.alarm_actions.include?(name)
+    end
+
+    def has_insufficient_data_action?(name)
+      resource_via_client.insufficient_data_actions.include?(name)
+    end
+  end
+end

data/lib/awspec/type/cloudwatch_event.rb

@@ -0,0 +1,19 @@
+module Awspec::Type
+  class CloudwatchEvent < ResourceBase
+    def resource_via_client
+      @resource_via_client ||= find_cloudwatch_event(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.arn if resource_via_client
+    end
+
+    def enable?
+      resource_via_client.state == 'ENABLED'
+    end
+
+    def scheduled?(schedule)
+      resource_via_client.schedule_expression == schedule
+    end
+  end
+end

data/lib/awspec/type/cloudwatch_logs.rb

@@ -0,0 +1,31 @@
+module Awspec::Type
+  class CloudwatchLogs < ResourceBase
+    def resource_via_client
+      @resource_via_client ||= find_cloudwatch_logs_group(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.log_group_name if resource_via_client
+    end
+
+    def has_log_stream?(stream_name)
+      ret = find_cloudwatch_logs_stream_by_log_group_name(@id).log_stream_name
+      return true if ret == stream_name
+    end
+
+    def has_metric_filter?(filter_name)
+      ret = find_cloudwatch_logs_metric_fileter_by_log_group_name(@id, filter_name).filter_name
+      return true if ret == filter_name
+    end
+
+    def has_subscription_filter?(filter_name, pattern = nil)
+      ret = find_cloudwatch_logs_subscription_fileter_by_log_group_name(@id, filter_name)
+      if pattern.nil?
+        return true if ret.filter_name == filter_name
+      else
+        return false unless ret.filter_pattern == pattern
+      end
+      return true if ret.filter_name == filter_name
+    end
+  end
+end

data/lib/awspec/type/customer_gateway.rb

@@ -0,0 +1,40 @@
+module Awspec::Type
+  class CustomerGateway < ResourceBase
+    tags_allowed
+
+    def initialize(name)
+      super
+      @display_name = name
+    end
+
+    def resource_via_client
+      @resource_via_client ||= find_customer_gateway(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.customer_gateway_id if resource_via_client
+    end
+
+    STATES = %w(
+      pending available deleting deleted
+    )
+
+    STATES.each do |state|
+      define_method state.tr('-', '_') + '?' do
+        resource_via_client.state == state
+      end
+    end
+
+    def ip_address
+      resource_via_client.ip_address
+    end
+
+    def bpg_asn
+      resource_via_client.bpg_asn
+    end
+
+    def type
+      resource_via_client.type
+    end
+  end
+end

data/lib/awspec/type/directconnect_virtual_interface.rb

@@ -0,0 +1,27 @@
+module Awspec::Type
+  class DirectconnectVirtualInterface < ResourceBase
+    def initialize(name)
+      super
+      @display_name = name
+    end
+
+    def resource_via_client
+      @resource_via_client ||= find_virtual_interface(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.virtual_interface_id if resource_via_client
+    end
+
+    STATES = %w(
+      confirming verifying pending available
+      deleting deleted rejected
+    )
+
+    STATES.each do |state|
+      define_method state + '?' do
+        resource_via_client.virtual_interface_state == state
+      end
+    end
+  end
+end

data/lib/awspec/type/dynamodb_table.rb

@@ -0,0 +1,40 @@
+module Awspec::Type
+  class DynamodbTable < ResourceBase
+    aws_resource Aws::DynamoDB::Table
+
+    def resource_via_client
+      @resource_via_client ||= find_dynamodb_table(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.table_name if resource_via_client
+    end
+
+    STATUSES = %w(
+      CREATING
+      UPDATING
+      DELETING
+      ACTIVE
+    )
+
+    STATUSES.each do |status|
+      define_method status.downcase + '?' do
+        resource_via_client.table_status == status
+      end
+    end
+
+    def has_attribute_definition?(name, type)
+      ret = resource_via_client.attribute_definitions.select { |a| a.attribute_name == name }
+      return false if ret.empty?
+      return false unless ret.single_resource.attribute_type == type
+      return true if ret.single_resource.attribute_name == name
+    end
+
+    def has_key_schema?(name, type)
+      ret = resource_via_client.key_schema.select { |k| k.attribute_name == name }
+      return false if ret.empty?
+      return false unless ret.single_resource.key_type == type
+      return true if ret.single_resource.attribute_name == name
+    end
+  end
+end

data/lib/awspec/type/ebs.rb

@@ -0,0 +1,37 @@
+module Awspec::Type
+  class Ebs < ResourceBase
+    aws_resource Aws::EC2::Volume
+    tags_allowed
+
+    def initialize(name)
+      super
+      @display_name = name
+    end
+
+    def resource_via_client
+      @resource_via_client ||= find_ebs(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.volume_id if resource_via_client
+    end
+
+    STATES = %w(
+      creating available in-use deleting deleted error
+    )
+
+    STATES.each do |state|
+      define_method state.tr('-', '_') + '?' do
+        resource_via_client.state == state
+      end
+    end
+
+    def attached_to?(instance_id)
+      instance = find_ec2(instance_id)
+      return false unless instance
+      return false unless resource_via_client.attachments
+      resource_via_client.attachments.first.instance_id == instance.instance_id && \
+        resource_via_client.attachments.first.state == 'attached'
+    end
+  end
+end

data/lib/awspec/type/ec2.rb

@@ -0,0 +1,145 @@
+module Awspec::Type
+  class Ec2 < ResourceBase
+    aws_resource Aws::EC2::Instance
+    tags_allowed
+
+    def initialize(name)
+      super
+      @display_name = name
+    end
+
+    def resource_via_client
+      @resource_via_client ||= find_ec2(@display_name)
+    end
+
+    def id
+      @id ||= resource_via_client.instance_id if resource_via_client
+    end
+
+    STATES = %w(
+      pending running shutting-down
+      terminated stopping stopped
+    )
+
+    STATES.each do |state|
+      define_method state.tr('-', '_') + '?' do
+        resource_via_client.state.name == state
+      end
+    end
+
+    def disabled_api_termination?
+      ret = find_ec2_attribute(id, 'disableApiTermination')
+      ret.disable_api_termination.value
+    end
+
+    def has_eip?(ip_address = nil)
+      option = {
+        filters: [{ name: 'instance-id', values: [id] }]
+      }
+      option[:public_ips] = [ip_address] if ip_address
+      ret = ec2_client.describe_addresses(option)
+      return ret.addresses.count == 1 if ip_address
+      return ret.addresses.count > 0 unless ip_address
+    end
+
+    def has_security_groups?(sg_ids)
+      return true if match_group_ids?(sg_ids) || match_group_names?(sg_ids)
+
+      group_ids = resource_security_groups.map { |sg| sg.group_id }
+      tags = select_security_group_by_group_id(group_ids).map { |sg| sg.tags }.flatten
+      group_names = tags.select { |tag| tag.key == 'Name' }.map { |tag| tag.value }
+      group_names == sg_ids
+    end
+
+    def has_security_group?(sg_id)
+      sgs = resource_via_client.security_groups
+      ret = sgs.find do |sg|
+        sg.group_id == sg_id || sg.group_name == sg_id
+      end
+      return true if ret
+      sg2 = find_security_group(sg_id)
+      return false unless sg2.tag_name == sg_id
+      sgs.find do |sg|
+        sg.group_id == sg2.group_id
+      end
+    end
+
+    def has_iam_instance_profile?(iam_instance_profile_name)
+      iam = resource_via_client.iam_instance_profile
+      ret = iam.arn.split('/').last == iam_instance_profile_name
+      return true if ret
+    end
+
+    def has_ebs?(volume_id)
+      blocks = resource_via_client.block_device_mappings
+      ret = blocks.find do |block|
+        next false unless block.ebs
+        block.ebs.volume_id == volume_id
+      end
+      return true if ret
+      blocks2 = find_ebs(volume_id)
+      blocks2.attachments.find do |attachment|
+        attachment.instance_id == id
+      end
+    end
+
+    def has_network_interface?(network_interface_id, device_index = nil)
+      res = find_network_interface(network_interface_id)
+      interfaces = resource_via_client.network_interfaces
+      ret = interfaces.find do |interface|
+        next false if device_index && interface.attachment.device_index != device_index
+        interface.network_interface_id == res.network_interface_id
+      end
+    end
+
+    def has_event?(event_code)
+      status = find_ec2_status(id)
+      ret = status.events.find do |event|
+        event.code == event_code
+      end
+    end
+
+    def has_events?
+      status = find_ec2_status(id)
+      return false if status.nil?
+      status.events.count > 0
+    end
+
+    def has_classiclink?(vpc_id = nil)
+      option = {
+        instance_ids: [id]
+      }
+      option[:filters] = [{ name: 'vpc-id', values: [vpc_id] }] if vpc_id
+      ret = ec2_client.describe_classic_link_instances(option)
+      return ret.instances.count == 1 if vpc_id
+    end
+
+    def has_classiclink_security_group?(sg_id)
+      option = {
+        instance_ids: [id]
+      }
+      classic_link_instances = ec2_client.describe_classic_link_instances(option)
+      return false if classic_link_instances.instances.count == 0
+      instances = classic_link_instances[0]
+      sgs = instances[0].groups
+      ret = sgs.find do |sg|
+        sg.group_id == sg_id || sg.group_name == sg_id
+      end
+      return true if ret
+    end
+
+    private
+
+    def match_group_ids?(sg_ids)
+      sg_ids.sort == resource_security_groups.map { |sg| sg.group_id }.sort
+    end
+
+    def match_group_names?(sg_names)
+      sg_names.sort == resource_security_groups.map { |sg| sg.group_name }.sort
+    end
+
+    def resource_security_groups
+      resource_via_client.security_groups
+    end
+  end
+end