opendelivery 0.0.1 → 0.0.3

data/lib/opendelivery/artifact.rb

@@ -1,10 +1,12 @@
-class Artifact
-  def initialize
-  end
+module OpenDelivery
+  class Artifact
+    def initialize
+    end
 
-  def add_timestamp(build_identifier, artifact)
-    timestamp = Time.now.strftime("%Y.%m.%d.%H.%M.%S.%L")
-    stamped_artifact = "#{artifact}-#{build_identifier}-#{timestamp}"
-    return stamped_artifact
+    def add_timestamp(build_identifier, artifact)
+      timestamp = Time.now.strftime("%Y.%m.%d.%H.%M.%S.%L")
+      stamped_artifact = "#{artifact}-#{build_identifier}-#{timestamp}"
+      return stamped_artifact
+    end
   end
 end

data/lib/opendelivery/domain.rb

@@ -1,44 +1,48 @@
 require 'aws-sdk'
 
-class Domain
-
-  def initialize
-    @sdb = AWS::SimpleDB.new
-  end
-
-  def create(domain_name)
-    AWS::SimpleDB.consistent_reads do
-      @sdb.domains.create(domain_name)
+module OpenDelivery
+  class Domain
+
+    def initialize(region=nil)
+      if region.nil?
+        @sdb = AWS::SimpleDB.new
+      else
+        @sdb = AWS::SimpleDB.new(:region => region)
+      end
     end
-  end
 
-  def destroy(domain_name)
-    @sdb.domains[domain_name].delete
-  end
+    def create(domain)
+      AWS::SimpleDB.consistent_reads do
+        @sdb.domains.create(domain)
+      end
+    end
 
-  def destroy_item(domain_name, item_name)
-    @sdb.domains[domain_name].items[item_name].delete
-  end
+    def destroy(domain)
+      @sdb.domains[domain].delete
+    end
 
-  def get_property(sdb_domain, item_name, key)
-    AWS::SimpleDB.consistent_reads do
-      item = @sdb.domains[sdb_domain].items[item_name]
+    def destroy_item(domain, item_name)
+      @sdb.domains[domain].items[item_name].delete
+    end
 
-      item.attributes.each_value do |name, value|
-        if name == key
-          @property_value = value.chomp
+    def get_property(domain, item_name, key)
+      AWS::SimpleDB.consistent_reads do
+        item = @sdb.domains[domain].items[item_name]
+        item.attributes.each_value do |name, value|
+          if name == key
+            @property_value = value.chomp
+          end
         end
       end
-    end
 
-    return @property_value
-  end
-
-  def set_property(sdb_domain, item_name, property, value)
-    AWS::SimpleDB.consistent_reads do
-      item = @sdb.domains[sdb_domain].items[item_name]
+      return @property_value
+    end
 
-      item.attributes.set(property => [value])
+    def set_property(domain, item_name, property, value)
+      AWS::SimpleDB.consistent_reads do
+        item = @sdb.domains[domain].items[item_name]
+        item.attributes.set(property => [value])
+      end
     end
   end
 end

data/lib/opendelivery/image.rb

@@ -1,28 +1,42 @@
-class Image
+module OpenDelivery
+  class Image
 
-  def initialize
-    @ec2 = AWS::EC2.new
-    @sdb = AWS::SimpleDB.new
-    @auto_scale = AWS::AutoScaling.new
-  end
-
-  def create(as_group_name, image_name, sdb_domain, type, key)
-    domain = Domain.new
-    image = @ec2.images.create(
-      instance_id: @auto_scale.groups[as_group_name].auto_scaling_instances.first.id,
-      name: image_name
-    )
+    def initialize(region=nil)
+      if region.nil?
+        @ec2 = AWS::EC2.new
+        @sdb = AWS::SimpleDB.new
+        @auto_scale = AWS::AutoScaling.new
+      else
+        @ec2 = AWS::EC2.new(:region => region)
+        @sdb = AWS::SimpleDB.new(:region => region)
+        @auto_scale = AWS::AutoScaling.new(:region => region)
+      end
+    end
 
-    sleep 10
+    protected
 
-    while image.state != :available
+    def wait_for_image(image)
       sleep 10
-      case image.state
-      when :failed
-        image.delete
-        raise RuntimeError, 'Image Creation Failed'
+
+      while image.state != :available
+        sleep 10
+        case image.state
+        when :failed
+          image.delete
+          raise RuntimeError, 'Image Creation Failed'
+        end
+      end
+    end
+
+    def prep_instance(domain stack_name)
+      group_name = @domain.get_property(domain, stack_name, "AWS::AutoScaling::AutoScalingGroup")
+      instance = @auto_scale.groups[group_name].auto_scaling_instances.first.id
+      @ec2.instances[instance].stop
+
+      while @ec2.instances[instance].status != :stopped
+        sleep 10
       end
+      return instance
     end
-    domain.set_property(sdb_domain, key, type, image.id)
   end
 end

data/lib/opendelivery/stack.rb

@@ -1,46 +1,138 @@
-class Stack
+require 'aws-sdk'
 
-  def initialize
-    @cfn = AWS::CloudFormation.new
-  end
+module OpenDelivery
+  class Stack
+
+    def initialize(region=nil, silent=false, sleep_time=30)
+      if region.nil?
+        @cfn = AWS::CloudFormation.new
+      else
+        @cfn = AWS::CloudFormation.new(:region => region)
+      end
+
+      @sleep_time = sleep_time
+      @silent = silent
+    end
+
+
+    SUCCESS_STATUSES =  [ "CREATE_COMPLETE",
+      "UPDATE_COMPLETE" ]
+
+    FAILURE_STATUSES =  [ "CREATED_FAILED",
+      "ROLLBACK_FAILED",
+      "ROLLBACK_COMPLETE",
+      "DELETED_FAILED",
+      "UPDATE_ROLLBACK_FAILED",
+      "UPDATE_ROLLBACK_COMPLETE",
+      "DELETE_COMPLETE" ]
 
-  def create(template, stack_name, parameters = {}, wait=false)
-    stack = @cfn.stacks.create(stack_name,
-                              File.open(template, "r").read,
-                              :parameters => parameters,
-                              :capabilities => ["CAPABILITY_IAM"],
-                              :disable_rollback => true)
+    PROGRESS_STATUSES = [ "CREATE_IN_PROGRESS",
+      "ROLLBACK_IN_PROGRESS",
+      "DELETE_IN_PROGRESS",
+      "UPDATE_IN_PROGRESS",
+      "UPDATE_COMPLETE_CLEANUP_IN_PROGRESS",
+      "UPDATE_ROLLBACK_IN_PROGRESS",
+      "UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS" ]
+
+    attr_accessor :sleep_time, :silent, :cfm
+
+    def watch stack_name
+      success = false
+      begin
+        stack = @cfn.stacks[stack_name]
+        success = watch_loop stack
+      rescue AWS::CloudFormation::Errors::ValidationError => msg
+        print_status "Exception raised: #{msg}"
+        success = false
+      end
+      return success
+    end
+
+
+    def create(template, stack_name, parameters = {}, wait=false)
+      stack = @cfn.stacks.create(stack_name,
+        File.open(template, "r").read,
+        :parameters => parameters,
+        :capabilities => ["CAPABILITY_IAM"],
+        :disable_rollback => true)
+
+      if wait
+        wait_for_stack(stack)
+      end
 
-    if wait
+      stack.resources.each do |resource|
+        @domain.set_property(domain, stack_name, resource.resource_type, resource.physical_resource_id)
+      end
+    end
+
+    def destroy(domain, stack_name)
+      stack = @cfn.stacks[stack_name]
+      stack.delete
+      while stack.exists?
+        sleep 20
+      end
+      @domain.destroy_item(domain, stack_name)
+    end
+
+    def list
+      @cfn.stacks.each do |stack|
+        puts "Stack Name: #{stack.name} | Status: #{stack.status}"
+      end
+    end
+
+    protected
+
+    def wait_for_stack(stack)
       while stack.status != "CREATE_COMPLETE"
-        sleep 30
+        sleep 20
 
         case stack.status
-        when "ROLLBACK_IN_PROGESS"
-          stack.delete
-        when "ROLLBACK_COMPLETE"
+        when "ROLLBACK_IN_PROGESS" || "ROLLBACK_COMPLETE"
           stack.delete
         end
       end
     end
-  end
 
-  def destroy(stack_name)
-    stack = @cfn.stacks[stack_name]
-    stack.delete
-    while stack.exists?
-      sleep 30
+    def print_status status
+      timestamp = Time.now.strftime("%Y.%m.%d %H:%M:%S:%L")
+      if (!@silent)
+        puts "#{timestamp}: #{status}"
+      end
     end
-  end
 
-  def list
-    @cfn.stacks.each do |stack|
-      case stack.status
-      when "CREATE_COMPLETE"
-        puts "Stack Name: #{stack.name} | Status: Available"
-      when "CREATE_IN_PROGRESS"
-        puts "Stack Name: #{stack.name} | Status: In Progress"
+    def watch_loop stack
+      keep_watching = true
+      success = false
+      abort_count = 10
+      while(keep_watching) do
+        begin
+          stack_status = stack.status
+          if (SUCCESS_STATUSES.include? stack_status)
+            print_status "Success: #{stack_status}"
+            success = true
+            keep_watching = false
+          elsif (PROGRESS_STATUSES.include? stack_status)
+            print_status "In Progress: #{stack_status}"
+            success = false
+            keep_watching = true
+          elsif (FAILURE_STATUSES.include? stack_status)
+            print_status "Failed: #{stack_status}"
+            success = false
+            keep_watching = false
+          else
+            print_status "didn't find #{stack_status} in the list of expected statuses"
+            success = false
+            abort_count = abort_count - 1
+            # if we get too many unknown statuses, assume something has gone horribly wrong and quit.
+            keep_watching = (abort_count > 0)
+          end
+        rescue AWS::CloudFormation::Errors::Throttling
+          print_status "Rate limit exceeded, retrying..."
+          sleep (@sleep_time * 0.1)
+        end
+        sleep(@sleep_time)
       end
+      return success
     end
   end
 end

data/lib/opendelivery/storage.rb

@@ -1,26 +1,34 @@
-class Storage
+require 'aws-sdk'
 
-  def initialize
-    @s3 = AWS::S3.new
-  end
+module OpenDelivery
+  class Storage
 
-  def copy(bucket, key, desired_key)
-    @s3.buckets[bucket].objects[key].copy_to(desired_key)
-  end
+    def initialize(region=nil)
+      if region.nil?
+        @s3 = AWS::S3.new
+      else
+        @s3 = AWS::S3.new(:region => region)
+      end
+    end
 
-  def upload(file, bucket, key)
-    @s3.buckets[bucket].objects[key].write(:file => file)
-  end
+    def copy(bucket, key, desired_key)
+      @s3.buckets[bucket].objects[key].copy_to(desired_key)
+    end
+
+    def upload(file, bucket, key)
+      @s3.buckets[bucket].objects[key].write(:file => file)
+    end
 
-  def download(bucket, key, output_directory)
-    obj = @s3.buckets[bucket].objects[key]
+    def download(bucket, key, output_directory)
+      obj = @s3.buckets[bucket].objects[key]
 
-    base = Pathname.new("#{obj.key}").basename
+      base = Pathname.new("#{obj.key}").basename
 
-    Dir.mkdir(output_directory) unless File.exists?(output_directory)
+      Dir.mkdir(output_directory) unless File.exists?(output_directory)
 
-    File.open("#{output_directory}/#{base}", 'w') do |file|
-      file.write(obj.read)
+      File.open("#{output_directory}/#{base}", 'w') do |file|
+        file.write(obj.read)
+      end
     end
   end
 end

data/lib/opendelivery/version.rb

@@ -1,3 +1,3 @@
-module Opendelivery
-  VERSION = '0.0.1'
+module OpenDelivery
+  VERSION = '0.0.3'
 end

data/lib/opendelivery.rb

@@ -1,8 +1,10 @@
+require 'opendelivery/version.rb'
 require 'opendelivery/artifact.rb'
-require 'opendelivery/database.rb'
 require 'opendelivery/domain.rb'
 require 'opendelivery/image.rb'
-require 'opendelivery/instance.rb'
-require 'opendelivery/key_pair.rb'
 require 'opendelivery/stack.rb'
 require 'opendelivery/storage.rb'
+
+
+module OpenDelivery
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: opendelivery
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -11,7 +11,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-08-15 00:00:00.000000000 Z
+date: 2013-08-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk
@@ -45,6 +45,22 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 1.15.0
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: A collection of tools that are used in the Open Delivery platform.
 email: brian.jakovich@stelligent.com
 executables: []
@@ -53,11 +69,8 @@ extra_rdoc_files: []
 files:
 - lib/opendelivery.rb
 - lib/opendelivery/artifact.rb
-- lib/opendelivery/database.rb
 - lib/opendelivery/domain.rb
 - lib/opendelivery/image.rb
-- lib/opendelivery/instance.rb
-- lib/opendelivery/key_pair.rb
 - lib/opendelivery/stack.rb
 - lib/opendelivery/storage.rb
 - lib/opendelivery/version.rb

data/lib/opendelivery/database.rb

@@ -1,10 +0,0 @@
-class Database
-  def initialize
-  end
-
-  def create
-  end
-
-  def destroy
-  end
-end

data/lib/opendelivery/instance.rb

@@ -1,19 +0,0 @@
-class Instance
-  def initialize
-    @ec2 = AWS::EC2.new
-    @auto_scaling= AWS::AutoScaling.new
-  end
-
-  def destroy
-  end
-
-  def find_in_as_group(as_group_name)
-    first_instance = @auto_scaling.groups[as_group_name].auto_scaling_instances.first.id
-
-    return @ec2.instances[first_instance].ip_address
-  end
-
-  def find_sg_name(sg_id)
-    @ec2.security_groups[sgid].name
-  end
-end

data/lib/opendelivery/key_pair.rb

@@ -1,16 +0,0 @@
-class KeyPair
-  def initialize
-    @ec2 = AWS::EC2.new
-  end
-
-  def create(name)
-    key_pair = @ec2.key_pairs.create(name)
-    File.open("/tmp/#{name}", "w") do |f|
-      f.write(key_pair.private_key)
-    end
-  end
-
-  def delete(name)
-    @ec2.key_pairs[name].delete
-  end
-end