zergrush_cf 0.0.14 → 0.0.17

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    Y2MxMTBlMWMzNDIwODZlYzcyMzI2ZTE5ZWI1YjMyZjFhMjc5Mzk3ZA==
+    MDU0MGY0MTA4MGVkNDAwNjEzNTg1YzlhZWZhMDFiOTg2ODBmMGZjNw==
   data.tar.gz: !binary |-
-    NTMxZGI0MTMzNTU2ZjRiZjRlMmJlNDgwZjljMDI2MjE3NjM2YjJhMA==
+    NmZiZDEwZjY0NmQwOWRiNDNjMTZhZTkxMDViMjYzMzdjNGNiYTg3MA==
 SHA512:
   metadata.gz: !binary |-
-    Yjc0MjNjMDdhOTYxNDZhZTNiNWRhNGY4NWU5NDg4MTBkYTA2MmM0NDZhN2U0
-    YTBjN2QwYTJlNTk0M2MxODYxODZjZWEzZTBiNGNjYTllMDdjMzQyN2EyYjE5
-    NTUxOTU0NjQwYThlZWQ2YmRmMWM4NTlhMjMzMzE3NTY0YzUyYjM=
+    NzFmYmM5MDJmODA4ZTY0ZWFmZjE4YTNhN2NmYzVjYmE0MjM4OWMzZjJjYTE3
+    ZDBmNGFjNTQ1ODQzMTIxZDA0YzM4OGVkMWE1YzZjMGYzOTAyZGYwNDIzZDZl
+    ODMyYTU3OGQ5NWM5M2ExOTM3ZDllOTI2ZDg4NDJlNDEyNjdkNWQ=
   data.tar.gz: !binary |-
-    ZWU1MzcyZDc0OWM0ZTRlZDE0NzQxNjQ4Zjg3ZWEzZmIyNGU2MzBjZjFlYTRh
-    MTdiZGNhNjhjNDVlMWY3NGE1Y2EzYmRiZTMyOTg5YWQ2ZDVhOTVmZTgyZjlj
-    ODM1MGM2OGQ1YWM2YTI0ZDA0YTUzYTg5NDY1ZTRmODg2M2VjNGM=
+    YjZiMTE2NGZlNzQ5MWZjZDdiYzRjZjAxYTM0MDU0NDE4MWM2ZjFjMmM4ODRi
+    N2RiYjhlYmFhN2NkNzVlZGU1NWU2ZDI1NTcxNzY4MmNjMTE1NzUwOWI4MjE5
+    ZjVkZGRiYWUyZjJjYTM1ZGQ1NWQ1YmM5ZWJmM2IwOWY5NmVmZGE=

data/lib/zergrush_cf/init.rb

@@ -274,6 +274,80 @@ class CloudFormation < ZergGemPlugin::Plugin "/driver"
             abort ("ERROR: AWS error: #{fog_cf_error.ai}")
     end
 
+    def snapshot hive_location, task_name, task_hash, base
+        aws_key_id = task_hash["vm"]["driver"]["driveroptions"][0]["access_key_id"] 
+        aws_secret = task_hash["vm"]["driver"]["driveroptions"][0]["secret_access_key"]
+
+        # eval possible environment variables
+        if aws_key_id =~ /^ENV\['.+'\]$/
+            aws_key_id = eval(aws_key_id)
+        end
+
+        if aws_secret =~ /^ENV\['.+'\]$/
+            aws_secret = eval(aws_secret)
+        end 
+
+        abort("AWS key id is not specified in task") unless aws_key_id != nil
+        abort("AWS secret is not specified in task") unless aws_secret != nil
+
+        base_name = (base.nil?) ? "#{task_name}" : base
+
+        puts("creating AMI Snapshots of all EC2 instances. Will use base name of #{base_name}")
+
+        cf = Fog::AWS::CloudFormation.new(
+            :aws_access_key_id => aws_key_id,
+            :aws_secret_access_key => aws_secret
+        )
+
+        fogCompute = Fog::Compute.new(
+            :provider => 'AWS',
+            :aws_access_key_id => aws_key_id,
+            :aws_secret_access_key => aws_secret
+        )
+
+        rabbit_objects = initRabbitConnection(task_hash["vm"]["driver"]["driveroptions"][0]["rabbit"])
+
+        # create the cloudformation stack
+        stack_name = "#{task_name}"
+        processStack(stack_name, base_name, cf, fogCompute)
+        return 0
+
+        rescue Fog::Errors::Error => fog_cf_error
+            abort ("ERROR: AWS error: #{fog_cf_error.message}")
+    end
+
+    def processStack stack_name, base_name, fogCF, fogCompute
+        fogCF.describe_stack_resources({ 'StackName' => stack_name })[:body]['StackResources'].each { |stack_resource| 
+            if stack_resource['ResourceType'] == 'AWS::EC2::Instance'
+                saveAmi(stack_resource, base_name, fogCompute) 
+            elsif stack_resource['ResourceType'] == 'AWS::CloudFormation::Stack'
+                processStack(stack_resource['PhysicalResourceId'].split('/')[1], base_name, fogCF, fogCompute)
+            end
+        }
+    end
+
+    def saveAmi ec2res, base_name, fogCompute
+        liveInstance = fogCompute.describe_instances({ 'instance-id' => ec2res['PhysicalResourceId'] })[:body]
+        nameTag = liveInstance['reservationSet'][0]['instancesSet'][0]['tagSet']['Name'].split(':')[1].downcase
+
+        ap "Creating AMI #{base_name}-#{nameTag}. Stack resource:"
+        ap ec2res  
+
+        imageSet = fogCompute.describe_images( { 'name' => "#{base_name}-#{nameTag}" } )[:body]
+        imageSet['imagesSet'].each { |image|
+            imageId = image['imageId'] 
+            ap "Deregistering old image #{imageId}..."
+            response = fogCompute.deregister_image(imageId)
+            abort("ERROR: deregistering #{imageId} failed!") unless response[:body]['return'] == 'true'
+        }
+        
+        ap 'Saving new image...'
+        newId = fogCompute.create_image(ec2res['PhysicalResourceId'], "#{base_name}-#{nameTag}", "#{base_name}-#{nameTag} snapshot")[:body]['imageId']
+        ap "Created new AMI #{newId}"
+        ap "Tagging #{newId} with #{base_name}"
+        fogCompute.create_tags(newId, { base_name => base_name })
+    end
+
     def logEvents events
         events.each do |event|
             puts "Timestamp: #{Time.parse(event['Timestamp'].to_s).iso8601}"

data/lib/zergrush_cf/version.rb

@@ -22,5 +22,5 @@
 #++
 
 module ZergrushCF
-  VERSION = "0.0.14"
+  VERSION = "0.0.17"
 end

data/zergrush_cf.gemspec

@@ -17,7 +17,7 @@ Gem::Specification.new do |s|
 
   s.add_development_dependency "bundler", ">= 1.0.0"
   s.add_development_dependency "rake"
-  s.add_development_dependency "zergrush", ">= 0.0.21"
+  s.add_development_dependency "zergrush", ">= 0.0.24"
 
   s.add_dependency "fog", ">=1.20.0"
   s.add_dependency "bunny", ">=1.2.1"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: zergrush_cf
 version: !ruby/object:Gem::Version
-  version: 0.0.14
+  version: 0.0.17
 platform: ruby
 authors:
 - MTN Satellite Communications
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-06-18 00:00:00.000000000 Z
+date: 2014-09-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -44,14 +44,14 @@ dependencies:
     requirements:
     - - ! '>='
       - !ruby/object:Gem::Version
-        version: 0.0.21
+        version: 0.0.24
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ! '>='
       - !ruby/object:Gem::Version
-        version: 0.0.21
+        version: 0.0.24
 - !ruby/object:Gem::Dependency
   name: fog
   requirement: !ruby/object:Gem::Requirement