jnewland-capsize 0.5.0

data/lib/capsize/meta_tasks.rb

@@ -0,0 +1,156 @@
+Capistrano::Configuration.instance.load do
+
+  namespace :ec2 do
+
+    namespace :setup do
+
+      desc <<-DESC
+      Initialize Capsize config.
+      You can run this command as often as you like.
+      It will not overwrite config files on multiple runs.
+      - Create :capsize_config_dir
+      - Create :capsize_secure_config_dir
+      - Copy capsize.yml.template to :capsize_config_dir/capsize.yml unless it already exists
+      - Automatically generate :capsize_secure_config_dir/secure.yml unless it already exists
+      - Instruct user to test configuration with "cap ec2:images:describe"
+      - Instruct user how to create a new keypair
+      - Instruct user how to setup a new security group.
+      DESC
+      task :default do
+
+        # Make the standard config dir if it doesn't exist already
+        unless File.exists?(fetch(:capsize_config_dir))
+          FileUtils.mkdir fetch(:capsize_config_dir)
+        end
+
+        # Make the secure config dir if it doesn't exist already
+        unless File.exists?(fetch(:capsize_secure_config_dir))
+          FileUtils.mkdir fetch(:capsize_secure_config_dir)
+        end
+
+        # copy the standard config file template, unless the dest file alread exists
+        unless File.exists?("#{fetch(:capsize_secure_config_dir)}/#{fetch(:capsize_secure_config_file_name)}")
+          puts "Please enter your EC2 account information."
+          puts "We'll then write it to a config file at #{fetch(:capsize_secure_config_dir)}/#{fetch(:capsize_secure_config_file_name)}"
+
+          require "yaml"
+          set :aws_access_key_id, proc { Capistrano::CLI.ui.ask("AWS Access Key ID : ") }
+          set :aws_secret_access_key, proc { Capistrano::CLI.ui.ask("AWS Secret Access Key : ") }
+
+          yaml = {}
+
+          # Populate production element
+          yaml['common'] = {}
+          yaml['common']['aws_access_key_id'] = aws_access_key_id
+          yaml['common']['aws_secret_access_key'] = aws_secret_access_key
+
+          yaml = YAML::dump(yaml).split("\n").collect { |line| line == "common: " ? line += "&common" : line }.join("\n")
+
+          env_config =<<EOF
+
+
+development:
+  <<: *common
+
+  # Uncomment and I only apply to the dev environment
+  # or overwrite a common value
+  #foo: 'bar'
+
+test:
+  <<: *common
+
+staging:
+  <<: *common
+
+production:
+  <<: *common
+
+  # Uncomment and I only apply to the production environment
+  # or overwrite a common value
+  #foo: 'baz'
+EOF
+
+          yaml += env_config
+
+          File.open("#{fetch(:capsize_secure_config_dir)}/#{fetch(:capsize_secure_config_file_name)}", 'w') do |file|
+            file.write(yaml)
+          end
+          File.chmod 0664, "#{fetch(:capsize_secure_config_dir)}/#{fetch(:capsize_secure_config_file_name)}"
+        else
+          puts "Warning : The following file was not copied over since it already exists: " + "#{fetch(:capsize_secure_config_dir)}/#{fetch(:capsize_secure_config_file_name)}"
+        end
+
+        unless File.exists?("#{fetch(:capsize_config_dir)}/#{fetch(:capsize_config_file_name)}")
+          FileUtils.cp("#{fetch(:capsize_examples_dir)}/capsize.yml.template", "#{fetch(:capsize_config_dir)}/#{fetch(:capsize_config_file_name)}", :verbose => true)
+        else
+          puts "Warning : The following file was not copied over since it already exists: " + "#{fetch(:capsize_config_dir)}/#{fetch(:capsize_config_file_name)}"
+        end
+
+        check
+
+        ec2.keypairs.create
+
+        ec2.security_groups.create_with_standard_ports
+
+        message = <<-MESSAGE
+
+        Next up: Pick an Amazon Machine Image and Run it!
+
+          Now you need to select an Amazon EC2 image ID that you want to start as a new
+          instance.  The easiest way to do that is to get a list of available images from EC2.
+
+            # Show ALL registered images
+            cap ec2:images:show
+
+            # Show MY registered images
+            cap ec2:images:show OWNER_ID='self'
+
+            # Show the AMAZON registered images
+            cap ec2:images:show OWNER_ID='amazon'
+
+          Select an 'imageId' from the results, and run it:
+
+            cap ec2:instances:run IMAGE_ID='ami-2bb65342'
+
+          You should see some progress information scroll by and finally you should see
+          a description of the key attributes of your running intance (dnsName and instanceId
+          are likely most important).
+
+          Now lets connect to it with SSH (this may take a few tries, sometimes it takes a
+          minute for the new instance to respond to SSH):
+
+            cap ec2:instances:ssh INSTANCE_ID='i-xxxxxx'
+
+          If you want to terminate your instance...
+
+            cap ec2:instances:terminate INSTANCE_ID='i-xxxxxx'
+
+        Enjoy Capsize!
+
+        MESSAGE
+
+        puts message
+
+      end
+
+      desc <<-DESC
+      Test your Capsize config.
+      Run a simple test which will validate that your Capsize config
+      is setup and working properly when querying the Amazon EC2 servers.
+      DESC
+      task :check do
+
+        begin
+          capsize_ec2.describe_images(:owner_id => "amazon")
+          puts "Congratulations!  Your credentials are verified and you are communicating properly with Amazon's EC2 service."
+        rescue Exception => e
+          puts "The test of your Capsize config failed with the following error: " + e
+        end
+
+      end
+
+    end
+
+  end
+
+end

data/lib/capsize/sqs.rb

@@ -0,0 +1,57 @@
+Capistrano::Configuration.instance.load do
+
+  namespace :sqs do
+
+    namespace :messages do
+
+      task :receive do
+        message = capsize_sqs.receive_message
+        unless message.nil?
+          puts message.body
+        else
+          puts "No Message"
+        end
+      end
+
+      task :send_message do
+        raise Exception, "message required" if capsize.get(:message).nil? || capsize.get(:message).empty?
+        begin
+          capsize_sqs.send_message(capsize.get(:message))
+        rescue Exception => e
+          raise e
+        else
+          puts "Message sent to #{capsize.get(:queue_name)} Queue"
+        end
+      end
+
+    end
+
+    namespace :queue do
+
+      task :delete do
+        begin
+          raise Exception unless capsize_sqs.delete
+        rescue Exception => e
+          puts "Queue #{queue_name} not deleted"
+          raise e
+        else
+          puts "Queue #{queue_name} deleted"
+        end
+      end
+
+      task :delete! do
+        begin
+          raise Exception unless capsize_sqs.delete!
+        rescue Exception => e
+          puts "Queue #{queue_name} not deleted"
+          raise e
+        else
+          puts "Queue #{queue_name} deleted"
+        end
+      end
+
+    end
+
+  end
+
+end

data/lib/capsize/sqs_plugin.rb

@@ -0,0 +1,128 @@
+module Capsize
+  module CapsizeSQS
+    include Capsize
+
+    #Creqte the XML for our CapQueue messages
+    def build_xml
+      body = ""
+      xml = Builder::XmlMarkup.new(:target => body, :indent => 2)
+      xml.command do
+        xml.hosts do
+          get(:hosts).split(/,/).each do |host|
+            xml.host host
+          end unless get(:hosts).nil?
+        end
+        xml.environment do
+          get(:environment).split(/,/).each do |keyvar|
+            xml.variable do
+              key, var = keyvar.split(/=/,2)
+              var.gsub!(/("|')/,'')
+              xml.key key
+              xml.var var
+            end
+          end unless get(:environment).nil?
+        end
+        xml.variables do
+          get(:variables).split(/,/).each do |keyvar|
+            xml.variable do
+              key, var = keyvar.split(/=/,2)
+              var.gsub!(/("|')/,'')
+              xml.key key
+              xml.var var
+            end
+          end unless get(:variables).nil?
+        end
+        xml.task get(:task)
+      end
+      body
+    end
+
+    #returns an instance of SQS::Queue
+    def connect
+      set :aws_access_key_id, get(:aws_access_key_id)
+      set :aws_secret_access_key, get(:aws_secret_access_key)
+
+      raise Exception, "You must have an :aws_access_key_id defined in your config." if fetch(:aws_access_key_id).nil? || fetch(:aws_access_key_id).empty?
+      raise Exception, "You must have an :aws_secret_access_key defined in your config." if fetch(:aws_secret_access_key).nil? || fetch(:aws_secret_access_key).empty?
+
+      SQS.access_key_id = fetch(:aws_access_key_id)
+      SQS.secret_access_key = fetch(:aws_secret_access_key)
+
+      queue = nil
+      begin
+        queue = SQS.get_queue queue_name
+      rescue
+        queue = SQS.create_queue queue_name
+      ensure
+        return queue
+      end
+    end
+
+    def delete( options=nil )
+      queue = connect()
+      queue.delete(options)
+    end
+
+    def delete!
+      queue = connect()
+      queue.delete!
+    end
+
+    def send_message( m )
+      queue = connect()
+      queue.send_message(m)
+    end
+
+    def get_queue_attributes( force=false )
+      queue = connect()
+      queue.get_queue_attributes(force)
+    end
+
+    def set_queue_attributes( atts={} )
+      queue = connect()
+      queue.set_queue_attributes(atts)
+    end
+
+    def empty?
+      queue = connect()
+      queue.empty?
+    end
+
+    def peek_message
+      queue = connect()
+      queue.peek_message
+    end
+
+    def peek_messages( options={} )
+      queue = connect()
+      queue.peek_messages(options)
+    end
+
+    def receive_message
+      queue = connect()
+      queue.receive_message
+    end
+
+    def receive_messages( options={} )
+      queue = connect()
+      queue.receive_messages(options)
+    end
+
+    def add_grant( options={} )
+      queue = connect()
+      queue.add_grant(options)
+    end
+
+    def remove_grant( options={} )
+      queue = connect()
+      queue.remove_grant(options)
+    end
+
+    def list_grants
+      queue = connect()
+      queue.list_grants
+    end
+
+  end
+end
+Capistrano.plugin :capsize_sqs, Capsize::CapsizeSQS

data/lib/capsize/version.rb

@@ -0,0 +1,9 @@
+module Capsize #:nodoc:
+  module VERSION #:nodoc:
+    MAJOR = 0
+    MINOR = 5
+    TINY  = 0
+
+    STRING = [MAJOR, MINOR, TINY].join('.')
+  end
+end

data/lib/capsize.rb

@@ -0,0 +1,26 @@
+# Require all necessary libraries
+%w[
+  rubygems
+  ostruct
+  yaml
+  fileutils
+  builder
+  capistrano
+  EC2
+  sqs
+  capsize/version
+  capsize/capsize.rb
+  capsize/meta_tasks
+  capsize/ec2
+  capsize/ec2_plugin
+  capsize/sqs
+  capsize/sqs_plugin
+  capsize/configuration
+].each { |lib|
+  begin
+    require lib
+  rescue Exception => e
+    puts "The loading of '#{lib}' failed in capsize.rb with message : " + e
+    exit
+  end
+}