awspec 0.0.2 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8c81642a5b9b297d827bead3be1bf63946948cf8
-  data.tar.gz: ec37738f21f3047036e55876a6386406edd33220
+  metadata.gz: 174d933837b6f18c123157d8d39636cc8151f2a1
+  data.tar.gz: 80eb495b6e34d683d09122f2f3235b95e60ddf91
 SHA512:
-  metadata.gz: 2af5ebabeece857dbf0eeb260b5541ef925448ddab272f382c74cd1234f0e3097062620ef3bac6bf71879f2f72e36e0a3fe80e3fb644193e4074b183daeb2965
-  data.tar.gz: 6011008a79a69fe3f8fd2184374ee03670f27e550d1162e5b4f64cd1f55f124ed9c0bb1ea657c623e30129d877d6c6f0737d3f18f7630a25b892bc8802f5f7da
+  metadata.gz: 699c01823b05271e2ac4a1c91e9f0e83edaf3f169a16d8a5fd1a54b9fc121e1e7219c0fd29903a3bd36eb598b24c0a9fd16fa58fef4b7dbc0760f0d2596e1d53
+  data.tar.gz: 101535a6b96aad8949196bc35e865ca0980c019d01411a6240ffa3ca876ec8f934ba54c6cd57002975818d3d14ebb3a5dd56236e08caca9b82ae8a8f118ba45a

data/README.md

@@ -46,11 +46,13 @@ end
     - [X] RDS DB Parameter Group  (`rds_db_parameter_group`)
 - [X] Security Group (`security_group`)
 - [X] VPC (`vpc`)
+- [X] S3 (`s3`)
+- Route53
+    - [X] Route53 Hosted Zone (`route53_hosted_zone`)
 
 ### Next..
 
-- [ ] S3
-- [ ] Route53
+- [ ] AutoScaling
 - ...
 
 ## Contributing

data/lib/awspec/helper/type.rb

@@ -2,7 +2,8 @@ module Awspec
   module Helper
     module Type
       types = %w(
-        base ec2 rds rds_db_parameter_group security_group vpc
+        base ec2 rds rds_db_parameter_group security_group
+        vpc s3 route53_hosted_zone
       )
 
       types.each { |type| require "awspec/type/#{type}" }

data/lib/awspec/matcher/have_record_set.rb

@@ -0,0 +1,32 @@
+RSpec::Matchers.define :have_record_set do |name|
+  match do |hosted_zone|
+    hosted_zone.has_record_set?(name, @type, @value, @options)
+  end
+
+  %w(soa a txt ns cname mx ptr srv spf aaaa).each do |type|
+    chain type do |value|
+      @type = type
+      @value = value
+      @options = {} if @options.nil?
+    end
+  end
+
+  chain :alias do |dns_name, hosted_zone_id|
+    @type = 'a'
+    @options = {} if @options.nil?
+    @options[:alias_dns_name] = dns_name
+    @options[:alias_hosted_zone_id] = hosted_zone_id
+  end
+
+  chain :aaaa_alias do |dns_name, hosted_zone_id|
+    @type = 'aaaa'
+    @options = {} if @options.nil?
+    @options[:alias_dns_name] = dns_name
+    @options[:alias_hosted_zone_id] = hosted_zone_id
+  end
+
+  chain :ttl do |ttl|
+    @options = {} if @options.nil?
+    @options[:ttl] = ttl
+  end
+end

data/lib/awspec/matcher.rb

@@ -7,3 +7,6 @@ require 'awspec/matcher/belong_to_db_subnet_group'
 
 # SecurityGroup
 require 'awspec/matcher/be_opened'
+
+# Route53
+require 'awspec/matcher/have_record_set'

data/lib/awspec/type/base.rb

@@ -4,10 +4,16 @@ module Awspec::Type
   class Base
     attr_reader :id, :ec2_client
 
+    # rubocop:disable all
     def initialize(id = nil)
-      @tmp = id
+      @id = nil
       @ec2_client = Aws::EC2::Client.new
     end
+    # rubocop:enable all
+
+    def exists?
+      @id
+    end
 
     def find_vpc(id)
       res = @ec2_client.describe_vpcs({
@@ -59,5 +65,37 @@ module Awspec::Type
 
       return res[:security_groups][0] if res[:security_groups].count == 1
     end
+
+    def find_ec2(id)
+      if id.is_a?(Array)
+        # Aws::EC2::Client.describe_instances native filters format
+        res = @client.describe_instances({
+                                           filters: id
+                                         })
+      elsif id.is_a?(Hash)
+        # syntax sugar
+        filters = []
+        id.each do |k, v|
+          filters.push({ name: k, values: Array(v) })
+        end
+        res = @client.describe_instances({
+                                           filters: filters
+                                         })
+      else
+        # instance_id or tag:Name
+        begin
+          res = @client.describe_instances({
+                                             instance_ids: [id]
+                                           })
+        rescue
+          # Aws::EC2::Errors::InvalidInstanceIDMalformed
+          # Aws::EC2::Errors::InvalidInstanceIDNotFound
+          res = @client.describe_instances({
+                                             filters: [{ name: 'tag:Name', values: [id] }]
+                                           })
+        end
+      end
+      return res[:reservations][0][:instances][0] if res[:reservations][0][:instances].count == 1
+    end
   end
 end

data/lib/awspec/type/ec2.rb

@@ -5,36 +5,8 @@ module Awspec::Type
     def initialize(id)
       super
       @client = @ec2_client
-      if id.is_a?(Array)
-        # Aws::EC2::Client.describe_instances native filters format
-        res = @client.describe_instances({
-                                           filters: id
-                                         })
-      elsif id.is_a?(Hash)
-        # syntax sugar
-        filters = []
-        id.each do |k, v|
-          filters.push({ name: k, values: Array(v) })
-        end
-        res = @client.describe_instances({
-                                           filters: filters
-                                         })
-      else
-        # instance_id or tag:Name
-        begin
-          res = @client.describe_instances({
-                                             instance_ids: [id]
-                                           })
-        rescue
-          # Aws::EC2::Errors::InvalidInstanceIDMalformed
-          # Aws::EC2::Errors::InvalidInstanceIDNotFound
-          res = @client.describe_instances({
-                                             filters: [{ name: 'tag:Name', values: [id] }]
-                                           })
-        end
-      end
-      @id = res[:reservations][0][:instances][0][:instance_id]
-      @instance = res[:reservations][0][:instances][0]
+      @instance = find_ec2(id)
+      @id = @instance[:instance_id] if @instance
     end
 
     states = %w(

data/lib/awspec/type/rds.rb

@@ -9,8 +9,8 @@ module Awspec::Type
       res = @client.describe_db_instances({
                                             db_instance_identifier: id
                                           })
-      @id = res[:db_instances][0][:db_instance_identifier]
-      @instance = res[:db_instances][0]
+      @instance = res[:db_instances][0] if res[:db_instances].count == 1
+      @id = @instance[:db_instance_identifier] if @instance
     end
 
     states = %w(

data/lib/awspec/type/rds_db_parameter_group.rb

@@ -3,6 +3,7 @@ module Awspec::Type
     attr_reader :parameters
 
     def initialize(name)
+      super
       @client = Aws::RDS::Client.new
       @parameters = {}
 
@@ -17,6 +18,7 @@ module Awspec::Type
           @parameters[param.parameter_name] = param.parameter_value
         end
       end
+      @id = name unless @parameters.empty?
     end
 
     def method_missing(name)

data/lib/awspec/type/route53_hosted_zone.rb

@@ -0,0 +1,64 @@
+module Awspec::Type
+  class Route53HostedZone < Base
+    attr_reader :hosted_zone, :resource_record_sets
+
+    def initialize(id)
+      super
+      @client = Aws::Route53::Client.new
+      @hosted_zone = find_hosted_zone(id)
+      @id = @hosted_zone[:id] if @hosted_zone
+      return unless @id
+      res = @client.list_resource_record_sets({
+                                                hosted_zone_id: @id
+                                              })
+      @resource_record_sets = res.resource_record_sets
+    end
+
+    def has_record_set?(name, type, value, options = {})
+      ret = @resource_record_sets.find do |record_set|
+        next if record_set.type != type.upcase
+        options[:ttl] = record_set.ttl unless options[:ttl]
+        if !record_set.resource_records.empty?
+          v = record_set.resource_records.map { |r| r.value }.join("\n")
+          record_set.name == name && \
+          value == v && \
+          record_set.ttl == options[:ttl]
+        else
+          # ALIAS
+          record_set.name == name && \
+          record_set.alias_target.dns_name == options[:alias_dns_name] && \
+          record_set.alias_target.hosted_zone_id == options[:alias_hosted_zone_id]
+        end
+      end
+    end
+
+    def find_hosted_zone(id)
+      hosted_zones = {}
+      marker = nil
+      loop do
+        res = @client.list_hosted_zones({
+                                          marker: marker
+                                        })
+        marker = res.marker
+        break if res.hosted_zones.empty?
+        res.hosted_zones.each do |hosted_zone|
+          hosted_zones[hosted_zone[:name]] = hosted_zones
+          if hosted_zone[:name] == id || hosted_zone[:id] == '/hostedzone/' + id || hosted_zone[:id] == id
+            return hosted_zone
+          end
+        end
+
+        break if marker.nil?
+      end
+    end
+
+    def method_missing(name)
+      describe = name.to_s
+      if @hosted_zone.key?(describe)
+        @hosted_zone[describe]
+      else
+        super
+      end
+    end
+  end
+end

data/lib/awspec/type/s3.rb

@@ -0,0 +1,26 @@
+module Awspec::Type
+  class S3 < Base
+    attr_reader :client, :bucket
+
+    def initialize(id)
+      super
+      @client = Aws::S3::Client.new
+      res = @client.list_buckets
+      ret = @client.list_buckets[:buckets].find do |bucket|
+        bucket.name == id
+      end
+      @id = id if ret
+      @bucket = ret
+    end
+
+    def has_object?(key)
+      res = @client.head_object({
+                                  bucket: @id,
+                                  key: key.sub(%r(\A/), '')
+                                })
+      res
+      rescue
+        false
+    end
+  end
+end

data/lib/awspec/type/security_group.rb

@@ -7,7 +7,7 @@ module Awspec::Type
       @client = @ec2_client
       @inbound = true
       @sg = find_security_group(id)
-      @id = @sg[:group_id]
+      @id = @sg[:group_id] if @sg
     end
 
     def method_missing(name)

data/lib/awspec/type/vpc.rb

@@ -6,7 +6,7 @@ module Awspec::Type
       super
       @client = @ec2_client
       @vpc = find_vpc(id)
-      @id = @vpc[:vpc_id]
+      @id = @vpc[:vpc_id] if @vpc
     end
 
     states = %w(

data/lib/awspec/version.rb

@@ -1,3 +1,3 @@
 module Awspec
-  VERSION = '0.0.2'
+  VERSION = '0.1.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: awspec
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.1.0
 platform: ruby
 authors:
 - k1LoW
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2015-07-29 00:00:00.000000000 Z
+date: 2015-07-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -135,11 +135,14 @@ files:
 - lib/awspec/matcher/belong_to_db_subnet_group.rb
 - lib/awspec/matcher/belong_to_subnet.rb
 - lib/awspec/matcher/belong_to_vpc.rb
+- lib/awspec/matcher/have_record_set.rb
 - lib/awspec/setup.rb
 - lib/awspec/type/base.rb
 - lib/awspec/type/ec2.rb
 - lib/awspec/type/rds.rb
 - lib/awspec/type/rds_db_parameter_group.rb
+- lib/awspec/type/route53_hosted_zone.rb
+- lib/awspec/type/s3.rb
 - lib/awspec/type/security_group.rb
 - lib/awspec/type/vpc.rb
 - lib/awspec/version.rb