solutious-rudy 0.4.0

data/lib/rudy/command/instances.rb

@@ -0,0 +1,85 @@
+# 
+# 
+# 
+# 
+# 
+# 
+# 
+# 
+# 
+# 
+# 
+
+module Rudy
+  module Command
+    class Instances < Rudy::Command::Base
+      
+      def restart_instances_valid?
+        raise "No instance ID provided" if @argv.filter.nil?
+        raise "No EC2 .pem keys provided" unless has_pem_keys?
+        raise "No SSH key provided for #{@global.user}!" unless has_keypair?
+        raise "No SSH key provided for root!" unless has_keypair?(:root)
+        
+        
+        @list = @ec2.instances.list(machine_group)
+        raise "#{@argv.filter} is not in the current machine group" unless @list.has_key?(@argv.filter)
+        
+        raise "I will not help you destroy production!" if @global.environment == "prod" # TODO: use_caution?, locked?
+        
+        exit unless are_you_sure?(5)
+        true
+      end
+      def restart_instances
+        puts "Restarting #{@argv.filter}!"
+        @ec2.instances.restart @argv.filter
+      end
+      
+      def instances
+        filter = @argv.first
+        filter = machine_group if filter.nil? && !@option.all
+        if instance_id?(filter)
+          inst = @ec2.instances.get(filter)
+          raise "The instance #{filter} does not exist" if inst.empty?
+          list = {inst[:aws_instance_id] => inst}
+        else
+          raise "The security group #{filter} does not exist" if filter && !@ec2.groups.exists?(filter)
+          list = @ec2.instances.list(filter)
+          if list.empty? 
+            msg = "There are no instances running" 
+            msg << " in the group #{filter}" if filter
+            raise msg
+          end
+        end
+        
+        list.each_pair do |id, inst|
+          print_instance inst
+        end
+        
+      end
+      
+      def destroy_instances_valid?
+        filter = argv.first
+        raise "No instance ID provided" if filter.nil?
+        raise "I will not help you destroy production!" if @global.environment == "prod" || filter =~ /^prod/
+        exit unless are_you_sure?
+        true
+      end
+        
+      def destroy_instances
+        filter = argv.first
+
+        if @ec2.groups.exists?(filter)
+          list = @ec2.instances.list(filter)
+          raise "The group #{filter} has no running instances" if list.empty?
+          instance = list.keys.first
+        else 
+          instance = filter
+        end
+        puts "Destroying #{instance}!"
+        @ec2.instances.destroy instance
+      end
+
+    end
+  end
+end
+

data/lib/rudy/command/machines.rb

@@ -0,0 +1,170 @@
+
+
+module Rudy
+  module Command
+    class Machines < Rudy::Command::Base
+      
+
+      def shutdown_valid?
+        raise "No EC2 .pem keys provided" unless has_pem_keys?
+        raise "No SSH key provided for #{@global.user}!" unless has_keypair?
+        raise "No SSH key provided for root!" unless has_keypair?(:root)
+        
+        @list = @ec2.instances.list(machine_group)
+        raise "No machines running in #{machine_group}" unless @list && !@list.empty?
+        
+        raise "I will not help you ruin production!" if @global.environment == "prod" # TODO: use_caution?, locked?
+        
+        true
+      end
+      
+      
+      def shutdown
+        puts "Shutting down #{machine_group}: #{@list.keys.join(', ')}".att(:bright)
+        switch_user("root")
+        puts "This command also affects the volumes attached to the instances! (according to your routines config)"
+        exit unless are_you_sure?(5)
+        
+        execute_routines(@list.values, :shutdown, :before)
+        
+        execute_disk_routines(@list.values, :shutdown)
+        
+        
+        puts $/, "Terminating instances...".att(:bright), $/
+        
+        @ec2.instances.destroy @list.keys
+        sleep 5
+        
+        execute_routines(@list.values, :shutdown, :after)
+        
+        puts "Done!"
+      end
+      
+      def startup_valid?
+        rig = @ec2.instances.list(machine_group)
+        raise "There is already an instance running in #{machine_group}" if rig && !rig.empty?
+        raise "No SSH key provided for #{keypairname}!" unless has_keypair?
+        true
+      end
+      def startup
+        puts "Starting a machine in #{machine_group}".att(:bright)
+        switch_user("root")
+        exit unless are_you_sure?(3)
+        
+        #execute_routines([], :startup, :before_local)
+        
+        @option.image ||= machine_image
+        
+        puts "using AMI: #{@option.image}"
+        
+        instances = @ec2.instances.create(@option.image, machine_group.to_s, File.basename(keypairpath), machine_data.to_yaml, @global.zone)
+        inst = instances.first
+        
+        if @option.address ||= machine_address
+          puts "Associating #{@option.address} to #{inst[:aws_instance_id]}"
+          @ec2.addresses.associate(inst[:aws_instance_id], @option.address)
+        end
+        
+        wait_for_machine(inst[:aws_instance_id])
+        inst = @ec2.instances.get(inst[:aws_instance_id])
+        
+        #inst = @ec2.instances.list(machine_group).values
+        
+        execute_disk_routines(inst, :startup)
+        execute_routines(inst, :startup, :after)
+        
+        puts "Done!"
+      end
+
+      def restart_valid?
+        shutdown_valid?
+      end
+      def restart
+        puts "Restarting #{machine_group}: #{@list.keys.join(', ')}".att(:bright)
+        switch_user("root")
+        exit unless are_you_sure?(5)
+        
+        @list.each do |id, inst|
+          execute_routines(@list.values, :restart, :before)
+        end
+        
+        puts "Restarting instances: #{@list.keys.join(', ')}".att(:bright)
+        @ec2.instances.restart @list.keys
+        sleep 10 # Wait for state to change and SSH to shutdown
+        
+        @list.keys.each do |id|
+          wait_for_machine(id)
+        end
+        
+        execute_disk_routines(@list.values, :restart)
+        
+        @list.each do |id, inst|
+          execute_routines(@list.values, :restart, :after)
+        end
+        
+        puts "Done!"
+      end
+      
+      
+      def status_valid?
+        raise "No EC2 .pem keys provided" unless has_pem_keys?
+        raise "No SSH key provided for #{@global.user}!" unless has_keypair?
+        raise "No SSH key provided for root!" unless has_keypair?(:root)
+        
+        @list = @ec2.instances.list(machine_group)
+        raise "No machines running in #{machine_group}" unless @list
+        true
+      end
+      def status
+        puts "There are no machines running in #{machine_group}" if @list.empty?
+        @list.each_pair do |id, inst|
+          print_instance inst
+        end
+      end
+      
+      
+      def update_valid?
+        raise "No EC2 .pem keys provided" unless has_pem_keys?
+        raise "No SSH key provided for #{@global.user}!" unless has_keypair?
+        raise "No SSH key provided for root!" unless has_keypair?(:root)
+        
+        @scripts = %w[rudy-ec2-startup update-ec2-ami-tools randomize-root-password]
+        @scripts.collect! {|script| File.join(RUDY_HOME, 'support', script) }
+        @scripts.each do |script| 
+          raise "Cannot find #{script}" unless File.exists?(script)
+        end
+        
+        true
+      end
+      
+      
+      def update
+        puts "Updating Rudy "
+        switch_user("root")
+        
+        exit unless are_you_sure?
+        scp do |scp|
+          @scripts.each do |script|
+            puts "Uploading #{File.basename(script)}"
+            scp.upload!(script, "/etc/init.d/")
+          end
+        end
+        
+        ssh do |session|
+          @scripts.each do |script|
+            session.exec!("chmod 700 /etc/init.d/#{File.basename(script)}")
+          end
+          
+          puts "Installing Rudy (#{Rudy::VERSION})"
+          session.exec!("mkdir -p /etc/ec2")
+          session.exec!("gem sources -a http://gems.github.com")
+          puts session.exec!("gem install --no-ri --no-rdoc rudy -v #{Rudy::VERSION}")
+        end
+      end
+      
+      
+    end
+  end
+end
+
+

data/lib/rudy/command/metadata.rb

@@ -0,0 +1,41 @@
+
+
+module Rudy
+  module Command
+    class Metadata < Rudy::Command::Base
+      
+      
+      # Print Rudy's metadata to STDOUT
+      def metadata
+        group_metadata.each_pair do |n,h|
+          puts n.att(:bright)
+          puts h.inspect, ""
+        end
+      end
+      
+      def destroy_metadata_valid?
+        false
+      end
+      
+      def destroy_metadata
+        @sdb.domains.destroy(RUDY_DOMAIN)
+      end
+      
+      
+      
+      def info
+        domains = @sdb.domains.list[:domains]
+        puts "Domains: #{domains.join(", ")}"
+      end
+      
+    private
+      def check_environment
+        raise "No Amazon keys provided!" unless has_keys?
+        raise "No SSH keypairs provided!" unless has_keypair?
+        true
+      end
+      
+    end
+  end
+end
+

data/lib/rudy/command/release.rb

@@ -0,0 +1,174 @@
+
+
+module Rudy
+  module Command
+    class Release < Rudy::Command::Base
+      
+      
+
+      
+      def release_valid?
+        
+        relroutine = @config.routines.find_deferred(@global.environment, @global.role, :release)
+        raise "No release routines defined for #{machine_group}" if relroutine.nil?
+        
+        raise "No EC2 .pem keys provided" unless has_pem_keys?
+        raise "No SSH key provided for #{@global.user} in #{machine_group}!" unless has_keypair?
+        raise "No SSH key provided for root in #{machine_group}!" unless has_keypair?(:root)
+        
+        @list = @ec2.instances.list(machine_group)
+        unless @list.empty?
+          msg = "#{machine_group} is in use, probably with another release. #{$/}"
+          msg << 'Sort it out and run "rudy shutdown" before continuing.' 
+          raise msg
+        end
+        
+        @scm, @scm_params = find_scm(:release)
+        
+        raise "No SCM defined for release routine" unless @scm        
+        raise "#{Dir.pwd} is not a working copy" unless @scm.working_copy?(Dir.pwd)
+        raise "There are local changes. Please revert or check them in." unless @scm.everything_checked_in?
+        raise "Invalid base URI (#{@scm_params[:base]})." unless @scm.valid_uri?(@scm_params[:base])
+        
+        true
+      end
+      
+      
+      
+      def rerelease_valid?
+        relroutine = @config.routines.find_deferred(@global.environment, @global.role, :rerelease)
+        raise "No rerelease routines defined for #{machine_group}" if relroutine.nil?
+        
+        raise "No EC2 .pem keys provided" unless has_pem_keys?
+        raise "No SSH key provided for #{@global.user} in #{machine_group}!" unless has_keypair?
+        raise "No SSH key provided for root in #{machine_group}!" unless has_keypair?(:root)
+        
+        @list = @ec2.instances.list(machine_group)
+        if @list.empty?
+          msg = "There are no machines running in #{machine_group}. #{$/}"
+          msg << 'You must run "rudy release" before you can rerelease.' 
+          raise msg
+        end
+        
+        @scm, @scm_params = find_scm(:rerelease)
+        
+        raise "No SCM defined for release routine" unless @scm        
+        raise "#{Dir.pwd} is not a working copy" unless @scm.working_copy?(Dir.pwd)
+        raise "There are local changes. Please revert or check them in." unless @scm.everything_checked_in?
+        raise "Invalid base URI (#{@scm_params[:base]})." unless @scm.valid_uri?(@scm_params[:base])
+        
+        true
+      end
+      
+      def rerelease
+        puts "Updating release from working copy".att(:bright)
+
+        tag, revision = @scm.local_info
+        puts "tag: #{tag}"
+        puts "rev: #{revision}"
+        
+        execute_disk_routines(@list.values, :rerelease)
+        
+        if @scm
+          
+          puts "Running SCM command".att(:bright)
+          ssh do |session|
+            cmd = "svn #{@scm_params[:command]}"
+            puts "#{cmd}"
+            session.exec!("cd #{@scm_params[:path]}")
+            session.exec!(cmd)
+            puts "#{@scm_params[:command]} complete"
+          end
+          
+        end
+        
+        execute_routines(@list.values, :rerelease, :after)
+        
+      end
+      
+      # <li>Creates a release tag based on the working copy on your machine</li>
+      # <li>Starts a new stage instance</li>
+      # <li>Executes release routines</li>
+      def release
+        # TODO: store metadata about release with local username and hostname
+        puts "Creating release from working copy".att(:bright)
+
+        exit unless are_you_sure?
+
+        tag = @scm.create_release(@global.local_user, @option.msg)        
+        puts "Done! (#{tag})"
+        
+        if @option.switch
+          puts "Switching working copy to new tag".att(:bright)
+          @scm.switch_working_copy(tag)
+        end
+        
+        @option.image ||= machine_image
+        
+        switch_user("root")
+        
+        puts "Starting #{machine_group}".att(:bright)
+        
+        instances = @ec2.instances.create(@option.image, machine_group.to_s, File.basename(keypairpath), machine_data.to_yaml, @global.zone)
+        inst = instances.first
+        
+        if @option.address ||= machine_address
+          puts "Associating #{@option.address} to #{inst[:aws_instance_id]}".att(:bright)
+          @ec2.addresses.associate(inst[:aws_instance_id], @option.address)
+        end
+        
+        wait_for_machine(inst[:aws_instance_id])
+        inst = @ec2.instances.get(inst[:aws_instance_id])
+        
+        #inst = @ec2.instances.list(machine_group).values
+        
+        
+        execute_disk_routines(inst, :release)
+        
+        if @scm
+          
+          puts "Running SCM command".att(:bright)
+          ssh do |session|
+            cmd = "svn #{@scm_params[:command]} #{tag} #{@scm_params[:path]}"
+            puts "#{cmd}"
+            session.exec!(cmd)
+            puts "#{@scm_params[:command]} complete"
+          end
+          
+        end
+        
+        execute_routines(inst, :release, :after)
+        
+        print_instance inst
+        
+        puts "Done!"
+      end
+      
+      
+      def find_scm(routine)
+        env, rol, att = @global.environment, @global.role
+        
+        # Look for the source control engine, checking all known scm values.
+        # The available one will look like [environment][role][release][svn]
+        params = nil
+        scm_name = nil
+        SUPPORTED_SCM_NAMES.each do |v|
+          scm_name = v
+          params = @config.routines.find(env, rol, routine, scm_name)
+          break if params
+        end
+        
+        if params
+          klass = eval "Rudy::SCM::#{scm_name.to_s.upcase}"
+          scm = klass.new(:base => params[:base])
+        end
+        
+        [scm, params]
+        
+      end
+      private :find_scm
+      
+    end
+  end
+end
+