sabat-rudy 0.8.0

data/lib/rudy/cli/aws/s3/buckets.rb

@@ -0,0 +1,49 @@
+
+
+module Rudy; module CLI; 
+module AWS; module S3;
+  
+  class Buckets < Rudy::CLI::CommandBase
+    
+    
+    def buckets
+      raise "No bucket name supplied" if !@argv.name && @option.list
+      s3 = Rudy::AWS::S3.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      unless @option.list
+        (s3.list_buckets || []).each do |b|
+          puts b.name
+        end
+      else
+        puts "All objects in #{@argv.name}:"
+        (s3.list_bucket_objects(@argv.name) || []).each do |o|
+          puts o
+        end
+      end
+    end
+    
+    def create_buckets_valid?
+      raise "No bucket name supplied" unless @argv.name
+      true
+    end
+    def create_buckets
+      s3 = Rudy::AWS::S3.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      s3.create_bucket(@argv.name, @option.location)
+      buckets
+    end
+    
+    def destroy_buckets_valid?
+      raise "No bucket name supplied" unless @argv.name
+      true
+    end
+    def destroy_buckets
+      execute_check(:medium)
+      s3 = Rudy::AWS::S3.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      s3.destroy_bucket(@argv.name)
+      buckets
+    end
+    
+    
+  end
+  
+end; end
+end; end

data/lib/rudy/cli/aws/s3/store.rb

@@ -0,0 +1,22 @@
+
+
+module Rudy; module CLI; 
+module AWS; module S3;
+  
+  class Store < Rudy::CLI::CommandBase
+    
+    def store_valid?
+      raise "No path specified" unless @argv.path
+      raise "No bucket specified" unless @option.bucket
+      true
+    end
+    def store
+      s3 = Rudy::AWS::S3.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      puts "Success: %s" % s3.store(@argv.path, @option.bucket)
+      
+    end
+    
+  end
+  
+end;end
+end;end

data/lib/rudy/cli/aws/sdb/domains.rb

@@ -0,0 +1,41 @@
+
+
+module Rudy; module CLI; 
+module AWS; module SDB;
+  
+  class Domains < Rudy::CLI::CommandBase
+    
+    
+    def domains
+      sdb = Rudy::AWS::SDB.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      domains = sdb.list_domains
+      puts domains
+      puts "No domains" if domains.nil? || domains.empty?
+    end
+    
+    def domains_create_valid?
+      @sdb = Rudy::AWS::SDB.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      raise "No name specified" unless @argv.name
+      true
+    end
+    def domains_create
+      @sdb.create_domain @argv.name
+      execute_check(:low)
+      domains
+    end
+    
+    def domains_destroy_valid?
+      @sdb = Rudy::AWS::SDB.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      raise "No name specified" unless @argv.name
+      true
+    end
+    def domains_destroy
+      execute_check(:medium)
+      @sdb.destroy_domain @argv.name
+      domains
+    end
+    
+  end
+  
+end; end
+end; end

data/lib/rudy/cli/candy.rb

@@ -0,0 +1,19 @@
+
+module Rudy
+  module CLI
+    class Candy < Rudy::CLI::CommandBase
+      
+      def open
+        rmach = Rudy::Machines.new
+        machines = rmach.list
+        puts 1
+        if machines
+          `open http://#{machines.first.dns_public}`
+        else
+          puts "No machines"
+        end
+      end
+      
+    end
+  end
+end

data/lib/rudy/cli/config.rb

@@ -0,0 +1,81 @@
+
+
+module Rudy
+  module CLI
+    class Config < Rudy::CLI::CommandBase
+      
+      # We force the CLI::Base#print_header to be quiet
+      def print_header
+        #@@global.quiet = true
+        #super
+      end
+      
+      
+      # Display configuration from the local user data file (~/.rudy/config).
+      # This config contains user data which is sent to each EC2 when 
+      # it's created. 
+      #
+      # The primary purpose of this command is to give other apps a way
+      # to check various configuration values. (This is mostly useful for
+      # debugging and checking configuration on an instance itself).
+      #
+      # It will return the most specific configuration available. If the 
+      # attribute isn'e found it will check each parent for the same attribute. 
+      # i.e. if [prod][app][ami] is not available, it will check [prod][ami]
+      # and then [ami]. 
+      #
+      #     # Display the value for a specific machine.
+      #     $ rudy -e prod -r db config param-name
+      #
+      #     # Display all configuration
+      #     $ rudy config --all
+      #
+      def config
+        # if Rudy.in_situ? # TODO: do something intelligent when running on EC2
+        
+        if @@config.nil? || @@config.empty?
+          return if @@global.quiet
+          raise Rudy::NoConfig
+        end
+
+        outform = @@global.format == :json ? :to_json : :to_yaml
+        
+        types = @option.marshal_dump.keys & @@config.keys # Intersections only
+        types = @@config.keys if @option.all
+        types = [:machines] if types.empty?
+          
+        if @option.project
+          rf = File.join(RUDY_HOME, 'Rudyfile')
+          raise "Cannot find: #{rf}" unless File.exists?(rf)
+          puts File.read(rf)
+          
+        elsif @option.script
+          conf = fetch_script_config
+          puts conf.to_hash.send(outform) if conf
+          
+        else
+          puts "# ACCOUNTS: [not displayed]" if types.delete(:accounts)
+          types.each do |conftype|
+            puts "# #{conftype.to_s.upcase}"
+            next unless @@config[conftype]  # Nothing to output
+            puts @@config[conftype].to_hash.send(outform)
+          end
+        end
+        
+      end
+      
+      def print_global
+        # NOTE: This method cannot be called just "global" b/c that conflicts 
+        # with the global accessor for Drydock::Command classes. 
+        if @@global.nil?
+          raise Rudy::NoGlobal
+        end
+        gtmp = @@global.clone
+        gtmp.format = "yaml" if gtmp.format == :s || gtmp.format == :string
+        gtmp.accesskey = gtmp.secretkey = "[not displayed]"
+        puts gtmp.dump(gtmp.format)
+      end
+      
+    end
+  end
+end

data/lib/rudy/cli/disks.rb

@@ -0,0 +1,58 @@
+
+
+module Rudy
+  module CLI
+    class Disks < Rudy::CLI::CommandBase
+
+      
+      def disks
+        rdisk = Rudy::Disks.new
+        rback = Rudy::Backups.new
+        more, less = [], []
+        less = [:environment, :role] if @option.all
+        # We first get the disk metadata
+        disk_list = rdisk.list(more, less) || []
+        # If there are no disks currently, there could be backups
+        # so we grab those to create a list of disks. 
+        if @option.backups
+          backups = rback.list(more, less) || []
+          backups.each_with_index do |b, index|
+            disk_list << b.disk
+          end
+        end
+        # We go through the list of disks but we'll skip ones we've seen
+        seen = []
+        disk_list.each do |d|
+          next if seen.member?(d.name)
+          seen << d.name
+          puts @@global.verbose > 0 ? d.inspect : d.dump(@@global.format)
+          if @option.backups
+            d.list_backups.each_with_index do |b, index|
+              puts '  %s' % b.name
+              break if @option.all.nil? && index >= 2 # display only 3, unless all
+            end
+          end
+        end
+      end
+      
+      def disks_wash
+        rdisk = Rudy::Disks.new
+        dirt = (rdisk.list || [])#.select { |d| d.available? }
+        if dirt.empty?
+          puts "Nothing to wash in #{rdisk.current_machine_group}"
+          return
+        end
+        
+        puts "The following disk metadata will be deleted:"
+        puts dirt.collect {|d| d.name }
+        execute_check(:medium)
+
+        dirt.each do |d|
+          d.destroy(:force)
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/rudy/cli/machines.rb

@@ -0,0 +1,114 @@
+
+
+module Rudy
+  module CLI
+    class Machines < Rudy::CLI::CommandBase
+      
+      
+      def machines
+        # Rudy::Machines.list takes two optional args for adding or 
+        # removing metadata attributes to modify the select query. 
+        # When all is specified we want to find machines in every
+        # environment and role to we remove these attributes from
+        # the select. 
+        more, less = nil, nil
+        less = [:environment, :role] if @option.all
+        
+        rmach = Rudy::Machines.new
+        mlist = rmach.list(more, less) || []
+        if mlist.empty?
+          if @option.all
+            puts "No machines running"
+          else
+            puts "No machines running in #{current_machine_group}" 
+            puts "Try: rudy machines --all"
+          end
+        end
+        mlist.each do |m|
+          puts @@global.verbose > 0 ? m.inspect : m.dump(@@global.format)
+        end
+      end
+      
+      def machines_wash
+        rmach = Rudy::Machines.new
+        dirt = (rmach.list || []).select { |m| !m.running? }
+        if dirt.empty?
+          puts "Nothing to wash in #{rmach.current_machine_group}"
+          return
+        end
+        
+        puts "The following machine metadata will be deleted:".bright
+        puts dirt.collect {|m| m.name.bright }
+        execute_check(:medium)
+        
+        dirt.each do |m|
+          m.destroy
+        end
+        
+      end
+      
+      
+      def ssh
+        # TODO: Give this methos a good look over
+        pkey = user_keypairpath(@@global.user)
+        unless pkey
+          puts "No private key configured for #{@@global.user} in #{current_machine_group}"
+        end
+        
+        # Options to be sent to Net::SSH
+        ssh_opts = { :user => @@global.user || Rudy.sysinfo.user, :debug => nil }
+        if pkey 
+          raise "Cannot find file #{pkey}" unless File.exists?(pkey)
+          raise InsecureKeyPermissions, @pkey unless File.stat(pkey).mode == 33152
+          ssh_opts[:keys] = pkey 
+        end
+
+
+        # The user specified a command to run. We won't create an interactive
+        # session so we need to prepare the command and its arguments
+        if @argv.first
+          command, command_args = @argv.shift, @argv || []
+          puts "#{command} #{command_args.join(' ')}" if @@global.verbose > 1
+
+        # otherwise, we'll open an ssh session or print command
+        else
+          command, command_args = :interactive_ssh, @option.print.nil?
+        end
+
+
+        checked = false
+        rudy = Rudy::Machines.new
+        lt = rudy.list 
+        unless lt
+          puts "No machines running in #{rudy.current_machine_group}"
+          exit
+        end
+        lt.each do |machine|
+          machine.update  # make sure we have the latest DNS info
+          
+          # Print header
+          if @@global.quiet
+            print "You are #{ssh_opts[:user].to_s.bright}. " if !checked # only the 1st
+          else
+            puts machine_separator(machine.name, machine.awsid)
+            puts "Connecting #{ssh_opts[:user].to_s.bright}@#{machine.dns_public} "
+            puts
+          end
+
+          # Make sure we want to run this command on all instances
+          if !checked && command != :interactive_ssh 
+            execute_check(:low) if ssh_opts[:user] == "root"
+            checked = true
+          end
+          
+          # Open the connection and run the command
+          rbox = Rye::Box.new(machine.dns_public, ssh_opts)
+          ret = rbox.send(command, command_args)
+          puts ret unless command == :interactive_ssh
+        end
+      end
+
+
+    end
+  end
+end

data/lib/rudy/cli/routines.rb

@@ -0,0 +1,108 @@
+
+
+module Rudy; module CLI;
+  class Routines < Rudy::CLI::CommandBase
+    
+    def startup_valid?
+      @rr = Rudy::Routines::Startup.new
+      @rr.raise_early_exceptions
+      true
+    end
+    def startup
+      machines = @rr.execute || []
+      puts $/, "The following machines are now available:" unless machines.empty?
+      machines.each do |machine|
+        puts machine.to_s
+      end
+    end
+    
+    def reboot_valid?
+      @rr = Rudy::Routines::Reboot.new
+      @rr.raise_early_exceptions
+      true
+    end
+    def reboot
+      machines = @rr.execute
+      puts $/, "The following machines have been restarted:"
+      machines.each do |machine|
+        puts machine.to_s
+      end
+    end
+    
+    def release_valid?
+      @rr = Rudy::Routines::Release.new(@alias) # Important: could be rerelease
+      @rr.raise_early_exceptions
+      true
+    end
+    def release
+      machines = @rr.execute
+      
+      unless machines.empty?
+        puts $/, "The following machines were processed:"
+        machines.each do |machine|
+          puts machine.to_s
+        end
+      end
+    end
+    
+    def passthrough_valid?
+      @rr = Rudy::Routines::Passthrough.new(@alias)
+      @rr.raise_early_exceptions
+      true
+    end
+    
+    # All unknown commands are sent here (using Drydock's trawler). 
+    # By default, the generic passthrough routine is executed which
+    # does nothing other than execute the routine config block that
+    # matches +@alias+ (the name used on the command-line). Calling
+    #
+    #     $ rudy unknown
+    #
+    # would end up here because it's an unknown command. Passthrough
+    # then looks for a routine config in the current environment and
+    # role called "unknown". If found, it's executed otherwise it'll
+    # raise an exception.
+    #
+    def passthrough
+      machines = @rr.execute
+      
+      unless machines.empty?
+        puts $/, "The following machines were processed:"
+        machines.each do |machine|
+          puts @@global.verbose > 0 ? machine.inspect : machine.dump(@@global.format)
+        end
+      end
+      
+    end
+    
+    def shutdown_valid?
+      @rr = Rudy::Routines::Shutdown.new
+      @rr.raise_early_exceptions
+      true
+    end
+    def shutdown
+      routine = fetch_routine_config(:shutdown)
+      
+      puts "All machines in #{current_machine_group} will be shutdown".bright
+      if routine && routine.disks
+        if routine.disks.destroy
+          puts "The following filesystems will be destroyed:".bright
+          puts routine.disks.destroy.keys.join($/).bright
+        end
+      end
+      
+      execute_check :medium
+      
+      machines = @rr.execute
+      puts $/, "The following instances have been destroyed:"
+      machines.each do |machine|
+        puts '%s %s ' % [machine.name.bright, machine.awsid]
+      end
+      
+      
+    end
+    
+
+  end
+end; end
+