cmonson_2ndwatch_awspec 0.85.3

data/lib/awspec/generator/template.rb

@@ -0,0 +1,146 @@
+module Awspec::Generator
+  # rubocop:disable Metrics/ClassLength
+  class Template
+    def self.generate(type)
+      @type = type
+      @account_attribute = false
+      @root_path = File.dirname(__FILE__) + '/../../../'
+      generate_stub
+      generate_type
+      generate_type_spec
+      generate_generator_doc
+      generate_resource_type_doc
+      put_message
+    end
+
+    def self.generate_account_attribute(type)
+      @type = type
+      @account_attribute = true
+      @root_path = File.dirname(__FILE__) + '/../../../'
+      generate_type
+      generate_account_attribute_generator_doc
+      generate_resource_type_doc
+      put_message
+    end
+
+    def self.generate_type
+      path = 'lib/awspec/type/' + @type.underscore + '.rb'
+      base = @account_attribute ? 'AccountAttributeBase' : 'ResourceBase'
+      content = <<-"EOF"
+module Awspec::Type
+  class #{@type.camelize} < #{base}
+    def resource_via_client
+      @resource_via_client ||= # FIXME
+    end
+
+    def id
+     @id ||= # FIXME
+    end
+  end
+end
+EOF
+      self.file_check_and_puts(path, content)
+    end
+
+    def self.generate_stub
+      path = 'lib/awspec/stub/' + @type.underscore + '.rb'
+      content = <<-"EOF"
+# Aws.config[:ec2] = {
+#   stub_responses: true
+# }
+EOF
+      self.file_check_and_puts(path, content)
+    end
+
+    def self.generate_type_spec
+      path = 'spec/type/' + @type.underscore + '_spec.rb'
+      content = <<-"EOF"
+require 'spec_helper'
+Awspec::Stub.load '#{@type.underscore}'
+
+describe #{@type.underscore}('my-#{@type.underscore.tr('_', '-')}') do
+  it { should exist }
+end
+EOF
+      self.file_check_and_puts(path, content)
+    end
+
+    def self.generate_generator_doc
+      path = 'lib/awspec/generator/doc/type/' + @type.underscore + '.rb'
+      content = <<-"EOF"
+module Awspec::Generator
+  module Doc
+    module Type
+      class #{@type.camelize} < Base
+        def initialize
+          super
+          @type_name = '#{@type.camelize}'
+          @type = Awspec::Type::#{@type.camelize}.new('my-#{@type.underscore.tr('_', '-')}')
+          @ret = @type.resource_via_client
+          @matchers = []
+          @ignore_matchers = []
+          @describes = []
+        end
+      end
+    end
+  end
+end
+EOF
+      self.file_check_and_puts(path, content)
+    end
+
+    def self.generate_account_attribute_generator_doc
+      path = 'lib/awspec/generator/doc/type/' + @type.underscore + '.rb'
+      content = <<-"EOF"
+module Awspec::Generator
+  module Doc
+    module Type
+      class #{@type.camelize} < AccountAttributeBase
+        def initialize
+          super
+          @type_name = '#{@type.camelize}'
+          @type = Awspec::Type::#{@type.camelize}.new
+          @ret = @type.resource_via_client
+          @matchers = []
+          @ignore_matchers = []
+          @describes = []
+        end
+      end
+    end
+  end
+end
+EOF
+      self.file_check_and_puts(path, content)
+    end
+
+    def self.generate_resource_type_doc
+      path = 'doc/_resource_types/' + @type.underscore + '.md'
+      content = <<-"EOF"
+### exist
+EOF
+      self.file_check_and_puts(path, content)
+    end
+
+    def self.put_message
+      content = <<-"EOF"
+
+Generate #{@type.camelize} template files.
+
+* !! AND add '#{@type.underscore}' to Awspec::Helper::Type::TYPES in lib/awspec/helper/type.rb *
+* !! AND add '#{@type.underscore}' client to lib/awspec/helper/finder.rb *
+
+EOF
+    end
+
+    def self.file_check_and_puts(path, content)
+      if File.exist? @root_path + path
+        $stderr.puts "!! #{path} already exists"
+      else
+        File.open(@root_path + path, 'w') do |f|
+          f.puts content
+        end
+        puts " + #{path}"
+      end
+    end
+  end
+end

data/lib/awspec/helper.rb

@@ -0,0 +1,8 @@
+require 'awspec/helper/color'
+require 'awspec/helper/type'
+require 'awspec/helper/finder'
+extend Awspec::Helper::Type
+class RSpec::Core::ExampleGroup
+  extend Awspec::Helper::Type
+  include Awspec::Helper::Type
+end

data/lib/awspec/helper/color.rb

@@ -0,0 +1,5 @@
+require 'term/ansicolor'
+
+class Color
+  extend Term::ANSIColor
+end

data/lib/awspec/helper/finder.rb

@@ -0,0 +1,119 @@
+require 'aws-sdk'
+require 'awspec/helper/finder/alb'
+require 'awspec/helper/finder/vpc'
+require 'awspec/helper/finder/subnet'
+require 'awspec/helper/finder/ec2'
+require 'awspec/helper/finder/ecr'
+require 'awspec/helper/finder/ecs'
+require 'awspec/helper/finder/efs'
+require 'awspec/helper/finder/security_group'
+require 'awspec/helper/finder/rds'
+require 'awspec/helper/finder/route53'
+require 'awspec/helper/finder/s3'
+require 'awspec/helper/finder/autoscaling'
+require 'awspec/helper/finder/ebs'
+require 'awspec/helper/finder/elb'
+require 'awspec/helper/finder/lambda'
+require 'awspec/helper/finder/iam'
+require 'awspec/helper/finder/kms'
+require 'awspec/helper/finder/elasticache'
+require 'awspec/helper/finder/elasticsearch'
+require 'awspec/helper/finder/cloudwatch'
+require 'awspec/helper/finder/cloudwatch_event'
+require 'awspec/helper/finder/ses'
+require 'awspec/helper/finder/directconnect'
+require 'awspec/helper/finder/ami'
+require 'awspec/helper/finder/cloudfront'
+require 'awspec/helper/finder/elastictranscoder'
+require 'awspec/helper/finder/cloudtrail'
+require 'awspec/helper/finder/waf'
+require 'awspec/helper/finder/acm'
+require 'awspec/helper/finder/cloudwatch_logs'
+require 'awspec/helper/finder/dynamodb'
+require 'awspec/helper/finder/sqs'
+require 'awspec/helper/finder/cloudformation'
+
+require 'awspec/helper/finder/account_attributes'
+
+module Awspec::Helper
+  module Finder
+    include Awspec::Helper::Finder::Alb
+    include Awspec::Helper::Finder::Vpc
+    include Awspec::Helper::Finder::Subnet
+    include Awspec::Helper::Finder::Ec2
+    include Awspec::Helper::Finder::Ecr
+    include Awspec::Helper::Finder::Ecs
+    include Awspec::Helper::Finder::Efs
+    include Awspec::Helper::Finder::SecurityGroup
+    include Awspec::Helper::Finder::Rds
+    include Awspec::Helper::Finder::Route53
+    include Awspec::Helper::Finder::S3
+    include Awspec::Helper::Finder::Autoscaling
+    include Awspec::Helper::Finder::Ebs
+    include Awspec::Helper::Finder::Elb
+    include Awspec::Helper::Finder::Lambda
+    include Awspec::Helper::Finder::Iam
+    include Awspec::Helper::Finder::Kms
+    include Awspec::Helper::Finder::Elasticache
+    include Awspec::Helper::Finder::Elasticsearch
+    include Awspec::Helper::Finder::Cloudwatch
+    include Awspec::Helper::Finder::CloudwatchEvent
+    include Awspec::Helper::Finder::Ses
+    include Awspec::Helper::Finder::Directconnect
+    include Awspec::Helper::Finder::Ami
+    include Awspec::Helper::Finder::Cloudfront
+    include Awspec::Helper::Finder::Elastictranscoder
+    include Awspec::Helper::Finder::Cloudtrail
+    include Awspec::Helper::Finder::Waf
+    include Awspec::Helper::Finder::Acm
+    include Awspec::Helper::Finder::AccountAttributes
+    include Awspec::Helper::Finder::CloudwatchLogs
+    include Awspec::Helper::Finder::Dynamodb
+    include Awspec::Helper::Finder::Sqs
+    include Awspec::Helper::Finder::Cloudformation
+
+    CLIENTS = {
+      ec2_client: Aws::EC2::Client,
+      ecr_client: Aws::ECR::Client,
+      ecs_client: Aws::ECS::Client,
+      efs_client: Aws::EFS::Client,
+      rds_client: Aws::RDS::Client,
+      route53_client: Aws::Route53::Client,
+      s3_client: Aws::S3::Client,
+      autoscaling_client: Aws::AutoScaling::Client,
+      elb_client: Aws::ElasticLoadBalancing::Client,
+      elbv2_client: Aws::ElasticLoadBalancingV2::Client,
+      lambda_client: Aws::Lambda::Client,
+      iam_client: Aws::IAM::Client,
+      kms_client: Aws::KMS::Client,
+      elasticache_client: Aws::ElastiCache::Client,
+      cloudwatch_client: Aws::CloudWatch::Client,
+      cloudwatch_event_client: Aws::CloudWatchEvents::Client,
+      ses_client: Aws::SES::Client,
+      directconnect_client: Aws::DirectConnect::Client,
+      cloudfront_client: Aws::CloudFront::Client,
+      elastictranscoder_client: Aws::ElasticTranscoder::Client,
+      elasticsearch_client: Aws::ElasticsearchService::Client,
+      cloudtrail_client: Aws::CloudTrail::Client,
+      waf_client: Aws::WAF::Client,
+      sts_client: Aws::STS::Client,
+      acm_client: Aws::ACM::Client,
+      cloudwatch_logs_client: Aws::CloudWatchLogs::Client,
+      dynamodb_client: Aws::DynamoDB::Client,
+      sqs_client: Aws::SQS::Client,
+      cloudformation_client: Aws::CloudFormation::Client
+    }
+
+    CLIENT_OPTIONS = {
+      http_proxy: ENV['http_proxy'] || ENV['https_proxy'] || nil
+    }
+
+    CLIENTS.each do |method_name, client|
+      define_method method_name do
+        unless self.methods.include? "@#{method_name}"
+          instance_variable_set("@#{method_name}", client.new(CLIENT_OPTIONS))
+        end
+      end
+    end
+  end
+end

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,18 @@
+module Awspec::Helper
+  module Finder
+    module Acm
+      def find_certificate(domain)
+        cert = acm_client.list_certificates.certificate_summary_list.find { |c| c[:domain_name] == domain }
+        acm_client.describe_certificate({ certificate_arn: cert.certificate_arn }).certificate
+      end
+
+      def select_all_certificates
+        certificates = []
+        acm_client.list_certificates.certificate_summary_list.each do |cert|
+          certificates << acm_client.describe_certificate({ certificate_arn: cert.certificate_arn }).certificate
+        end
+        certificates
+      end
+    end
+  end
+end

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

@@ -0,0 +1,48 @@
+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
+        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
+        return nil
+      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
+        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
+    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/autoscaling.rb

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