itamae-spec 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 32c0a5e4298c6d97786f447e64e6edea3071ed874f8f347532e33bfa92330fdf
-  data.tar.gz: d69af9290654e7d104d9decb50ff9f6c8afb80587af0189ab1e933264abec152
+  metadata.gz: c331c48127fb1a98218a1a3c3558ced75f5be9c8ca8498a5a5cc90d1f317314c
+  data.tar.gz: 61b144fafd16ca07d70edbfd545021433a47db68872ddf61d44ad7a42c93b752
 SHA512:
-  metadata.gz: 1234cfda80bca59a5e53251046f7c743587d692097f80a63033c9bbe150eb18dcc161b703bd3019d416527c327703e02ea8065b486f0f7ab44beca652f46343a
-  data.tar.gz: a3a82b93122ab5b95eddb874f09b14fb0c919e888008b834d32e501a6bb7ad070cefb88bdc11fb2cee752c75b5cda54174d7ec394cc8824cce32442de312c4c3
+  metadata.gz: a66032f85bcee1eb44d9fad26a0a1ccffab6d76eda4f6cb8965f1c1f22fd121519fe6347b4e3b1cfb72c04087227b4f905714865019b384841c1b8527abc4627
+  data.tar.gz: 51d791fe0021ca63c798441d214d018d0f74301b92cb91ccece9ce1b05edf7f00902777fa6242bf8820f0f9802332e9d662c5b90be8a1f22e58612706b0d8de2

data/lib/itamae-spec/generators/project.rb

@@ -5,6 +5,12 @@ module Itamae
       def self.source_root
         File.dirname(__FILE__) + '/templates/project'
       end
+
+      def bundle
+        Dir.chdir(destination_root) do
+          run 'bundle install'
+        end
+      end
     end
   end
 end

data/lib/itamae-spec/resource.rb

@@ -1,2 +1,4 @@
 require 'itamae-spec/resource/http_request'
+require 'itamae-spec/resource/file'
 require 'itamae-spec/resource/s3_file'
+require 'itamae-spec/resource/route53_record'

data/lib/itamae-spec/resource/file.rb

@@ -0,0 +1,44 @@
+
+module Itamae
+  module Resource
+    class File
+      def send_tempfile
+        if !attributes.content && !content_file
+          @temppath = nil
+          return
+        end
+
+        begin
+          src = if content_file
+                  content_file
+                else
+                  f = Tempfile.open('itamae')
+                  f.write(attributes.content)
+                  f.close
+                  f.path
+                end
+
+          @temppath = ::File.join(runner.tmpdir, Time.now.to_f.to_s)
+
+          if backend.is_a?(Itamae::Backend::Docker)
+            run_command(["mkdir", @temppath])
+            backend.send_file(src, @temppath)
+            @temppath = ::File.join(@temppath, ::File.basename(src))
+          elsif backend.is_a?(Itamae::Backend::Local)
+            run_command(["touch", @temppath])
+            run_specinfra(:change_file_mode, @temppath, '0600')
+            run_specinfra(:copy_file, src, @temppath)
+          else
+            run_command(["touch", @temppath])
+            run_specinfra(:change_file_mode, @temppath, '0600')
+            backend.send_file(src, @temppath)
+          end
+
+          run_specinfra(:change_file_mode, @temppath, '0600')
+        ensure
+          f.unlink if f
+        end
+      end
+    end
+  end
+end

data/lib/itamae-spec/resource/route53_record.rb

@@ -0,0 +1,145 @@
+require 'multi_json'
+
+module Itamae
+  module Resource
+    class Route53Record < File
+      define_attribute :action, default: :create
+      define_attribute :region, type: String, required: true
+      define_attribute :profile, type: String, default: 'default'
+      define_attribute :hosted_zone_id, type: String, required: true
+      define_attribute :comment, type: String
+      define_attribute :record_name, type: String, default_name: true
+      define_attribute :type, type: String, required: true
+      define_attribute :ttl, type: Integer, required: true
+      define_attribute :value, type: [ String, Array ], required: true
+      # define_attribute :set_identifier, type: String
+      # define_attribute :weight, type: String
+      # define_attribute :failover, type: String
+      # define_attribute :health_check_id, type: String
+      # define_attribute :traffic_policy_instance_id, type: String
+
+      def pre_action
+        attributes.record_name = attributes.record_name + '.' unless attributes.record_name[-1] == '.'
+        credentials = Aws::SharedCredentials.new(profile_name: attributes.profile)
+        @route53 = Aws::Route53::Client.new(region: attributes.region, credentials: credentials)
+
+        @change_batch = define_change_batch
+        resource_record_set = @change_batch[:changes][0][:resource_record_set]
+        attributes.content = MultiJson.dump(resource_record_set, pretty: true)
+
+        @resource_record_set = compare_record_values(fetch_record)
+
+        case @current_action
+        when :create
+          attributes.exist = true
+        when :upsert
+          attributes.exist = true
+        when :delete
+          attributes.exist = false
+        end
+
+        send_tempfile
+        compare_file if @current_action == :upsert
+      end
+
+      def set_current_attributes
+        current.modified = false
+      end
+
+      def action_create
+        return if current.exist
+        @route53.change_resource_record_sets(
+          change_batch: @change_batch,
+          hosted_zone_id: attributes.hosted_zone_id
+        )
+      end
+
+      def action_upsert
+        @route53.change_resource_record_sets(
+          change_batch: @change_batch,
+          hosted_zone_id: attributes.hosted_zone_id
+        )
+      end
+
+      def action_delete
+        return unless current.exist
+        @route53.change_resource_record_sets(
+          change_batch: @change_batch,
+          hosted_zone_id: attributes.hosted_zone_id
+        )
+      rescue Aws::Route53::Errors::InvalidChangeBatch => e
+        Itamae.logger.warn e.inspect
+      end
+
+      private
+
+      def define_change_batch
+        if attributes.value.class == Array
+          resource_records = attributes.value.map do |v|
+            { value: v }
+          end
+        elsif attributes.value.class == String
+          resource_records = Array.new(1, { value: attributes.value })
+        end
+
+        resource_record_set = {
+          name: attributes.record_name,
+          type: attributes.type,
+          ttl: attributes.ttl,
+          resource_records: resource_records
+        }
+
+        changes = {
+          action: attributes.action.to_s.upcase,
+          resource_record_set: resource_record_set
+        }
+
+        {
+          changes: [ changes ],
+          comment: attributes.comment
+        }
+      end
+
+      def fetch_record
+        resp = @route53.list_resource_record_sets(
+          hosted_zone_id: attributes.hosted_zone_id,
+          start_record_name: attributes.record_name,
+          start_record_type: attributes.type,
+          start_record_identifier: attributes.set_identifier,
+          max_items: 1
+        )
+
+        resp.resource_record_sets[0]
+      end
+
+      def compare_record_values(resource_record_set)
+        if attributes.record_name == resource_record_set.name && attributes.type == resource_record_set.type
+          current.exist = true
+          resource_record_set.to_h
+        else
+          current.exist = false
+          {}
+        end
+      end
+
+      def compare_to
+        if current.exist
+          f = Tempfile.open('itamae')
+          f.write(MultiJson.dump(@resource_record_set, pretty: true))
+          f.close
+          f.path
+        else
+          '/dev/null'
+        end
+      end
+
+      def show_content_diff
+        if attributes.modified
+          Itamae.logger.info 'Convert resource record set to JSON and display the difference.'
+        end
+
+        super
+      end
+    end
+  end
+end

data/lib/itamae-spec/resource/s3_file.rb

@@ -3,7 +3,7 @@ require 'aws-sdk'
 module Itamae
   module Resource
     class S3File < File
-      define_attribute :object_key, type: String, default_name: true
+      define_attribute :object_key, type: String, required: true
       define_attribute :region, type: String, required: true
       define_attribute :bucket, type: String, required: true
       define_attribute :profile, type: String, default: 'default'

data/lib/itamae-spec/version.txt

@@ -1 +1 @@
-0.0.2
+0.0.3

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: itamae-spec
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - kammy1231
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-03-14 00:00:00.000000000 Z
+date: 2018-03-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk
@@ -225,7 +225,9 @@ files:
 - lib/itamae-spec/generators/templates/project/tmp-nodes/.keep
 - lib/itamae-spec/logger.rb
 - lib/itamae-spec/resource.rb
+- lib/itamae-spec/resource/file.rb
 - lib/itamae-spec/resource/http_request.rb
+- lib/itamae-spec/resource/route53_record.rb
 - lib/itamae-spec/resource/s3_file.rb
 - lib/itamae-spec/task/base.rb
 - lib/itamae-spec/task/base_task.rb