rudy 0.4.0 → 0.6.0

data/lib/rudy/cli.rb

@@ -0,0 +1,157 @@
+
+require 'drydock'
+
+module Rudy
+    
+  # = CLI
+  # 
+  # These classes provide the functionality for the Command
+  # line interfaces. See the bin/ files if you're interested. 
+  # 
+  module CLI
+    class NoCred < RuntimeError #:nodoc
+    end
+    
+    class Output < Storable
+      # TODO: Use for all CLI responses
+      # Messages and errors should be in @@global.format
+      # Should print messages as they come
+    end
+    
+    class CommandBase < Drydock::Command
+      include Rudy::Huxtable
+      
+      attr_reader :config
+      
+    protected
+      def init
+
+        # The CLI wants output!
+        Rudy::Huxtable.update_logger STDOUT
+        
+        # Send The Huxtables the global values from the command-line
+        Rudy::Huxtable.update_global @global
+        
+        unless @@global.accesskey && @@global.secretkey
+          STDERR.puts "No AWS credentials. Check your configs!"
+          STDERR.puts "Try: rudy init"
+          exit 1
+        end
+        
+        if @@global.environment =~ /^prod/ && Drydock.debug?
+          puts Rudy::Utils.banner("PRODUCTION ACCESS IS DISABLED IN DEBUG MODE")
+          exit 1
+        end
+        
+      end
+      
+      def execute_action(emsg="Failed", &action)
+        begin
+          ret = action.call
+          raise emsg unless ret
+          ret
+        rescue Rudy::AWS::EC2::NoAMI => ex
+          raise Drydock::OptError.new('-a', @alias)
+        end
+      end
+      
+      def execute_check(level=:medium)
+        ret = Annoy.are_you_sure?(level)
+        exit 0 unless ret
+        ret
+      end
+      
+      # Print a default header to the screen for every command.
+      #
+      def print_header
+        
+        # Send The Huxtables the global values again because they could be
+        # updated after initialization but before the command was executed
+        Rudy::Huxtable.update_global @global
+        
+        puts Rudy::CLI.generate_header(@@global, @@config) if @@global.print_header
+        
+        unless @@global.quiet
+          if @@global.environment == "prod"
+            msg = "YOU ARE PLAYING WITH PRODUCTION"
+            puts Rudy::Utils.banner(msg, :huge, :red), $/
+          end
+          puts Rudy::Utils.banner("THIS IS EC2"), $/ if Rudy.in_situ?
+        end
+      end
+      
+    end
+
+    def self.generate_header(global, config)
+      return "" if global.quiet
+      header = StringIO.new
+      title, name = "RUDY v#{Rudy::VERSION}", config.accounts.aws.name
+      now_utc = Time.now.utc.strftime("%Y-%m-%d %H:%M:%S")
+      criteria = []
+      [:region, :zone, :environment, :role, :position].each do |n|
+        key, val = n.to_s.slice(0,1).att, global.send(n) 
+        key = 'R' if n == :region
+        next unless val
+        criteria << "#{key.att}:#{val.to_s.bright}"
+      end
+      if config.accounts && config.accounts.aws
+        if global.verbose > 0
+          header.puts '%s -- %s -- %s UTC' % [title, name, now_utc]
+        end
+        header.puts '[%s]' % [criteria.join("  ")], $/
+      end
+      header.rewind
+      header.read
+    end
+
+
+    # A base for all Drydock executables (bin/rudy etc...). 
+    class Base
+      extend Drydock
+      
+      before do |obj|
+        # Don't print Rudy header unless requested to
+        obj.global.print_header = false  if (obj.global.verbose == 0)
+        @start = Time.now
+      end
+
+      after do |obj|  
+        if obj.global.verbose > 0
+          puts
+          @elapsed = Time.now - @start
+          puts "Elapsed: %.2f seconds" % @elapsed.to_f if @elapsed > 0.1
+        end
+      end
+      
+      
+      # These globals are used by all bin/ executables
+      global :A, :accesskey, String, "AWS Access Key"
+      global :S, :secretkey, String, "AWS Secret Access Key"
+      global :R, :region, String, "Amazon service region (ie: #{Rudy::DEFAULT_REGION})"
+      global :z, :zone, String, "Amazon Availability zone (ie: #{Rudy::DEFAULT_ZONE})"
+      global :u, :user, String, "Provide a username (ie: #{Rudy::DEFAULT_USER})"
+      global :k, :pkey, String, "Path to the private SSH key"
+      global :f, :format, String, "Output format"
+      global :n, :nocolor, "Disable output colors"
+      global :C, :config, String, "Specify another configuration file to read (ie: #{Rudy::CONFIG_FILE})"
+      global :Y, :yes, "Assume a correct answer to confirmation questions"
+      global :q, :quiet, "Run with less output"
+      global :v, :verbose, "Increase verbosity of output (i.e. -v or -vv or -vvv)" do
+        @verbose ||= 0
+        @verbose += 1
+      end
+      global :V, :version, "Display version number" do
+        puts "Rudy version: #{Rudy::VERSION}"
+        exit 0
+      end
+      
+    end
+    
+    
+  end
+
+end
+
+Rudy::Utils.require_glob(RUDY_LIB, 'rudy', 'cli', '**', '*.rb')
+
+

data/lib/rudy/cli/aws/ec2/addresses.rb

@@ -0,0 +1,105 @@
+
+
+module Rudy; module CLI; 
+module AWS; module EC2;
+  
+  class Addresses < Rudy::CLI::CommandBase
+    
+    def addresses_create
+      radd = Rudy::AWS::EC2::Addresses.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      address = radd.create
+      puts @@global.verbose > 0 ? address.inspect : address.dump(@@global.format)
+    end
+    
+    def addresses_destroy_valid?
+      raise Drydock::ArgError.new("IP address", @alias) unless @argv.ipaddress
+      @radd = Rudy::AWS::EC2::Addresses.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      raise "#{@argv.ipaddress} is not allocated to you" unless @radd.exists?(@argv.ipaddress)
+      raise "#{@argv.ipaddress} is associated!" if @radd.associated?(@argv.ipaddress)
+      true
+    end
+    def addresses_destroy
+      address = @radd.get(@argv.ipaddress)
+      raise "Could not fetch #{address.ipaddress}" unless address
+      
+      puts "Destroying address: #{@argv.ipaddress}"
+      puts "NOTE: this IP address will become available to other EC2 customers.".color(:blue)
+      execute_check(:medium)
+      execute_action { @radd.destroy(@argv.ipaddress) }
+      self.addresses
+    end
+
+    def associate_addresses_valid?
+      raise Drydock::ArgError.new('IP address', @alias) if !@argv.ipaddress && !@option.newaddress
+      raise Drydock::OptError.new('instance ID', @alias) if !@option.instance
+      true
+    end
+    def associate_addresses
+      radd = Rudy::AWS::EC2::Addresses.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      rinst = Rudy::AWS::EC2::Instances.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      
+      raise "Instance #{@argv.instid} does not exist!" unless rinst.exists?(@option.instance)
+      
+      if @option.newaddress
+        print "Creating address... "
+        tmp = radd.create
+        puts "#{tmp.ipaddress}"
+        address = tmp.ipaddress
+      else
+        address = @argv.ipaddress
+      end
+      
+      raise "#{address} is not allocated to you" unless radd.exists?(address)
+      raise "#{address} is already associated!" if radd.associated?(address)
+          
+      instance = rinst.get(@option.instance)
+      
+      # If an instance was recently disassoiciated, the dns_public may
+      # not be updated yet
+      instance_name = instance.dns_public
+      instance_name = instance.awsid if !instance_name || instance_name.empty?
+      
+      puts "Associating #{address} to #{instance_name} (#{instance.groups.join(', ')})"
+      execute_check(:low)
+      execute_action { radd.associate(address, instance.awsid) }
+      address = radd.get(address)
+      puts @@global.verbose > 0 ? address.inspect : address.dump(@@global.format)
+    end
+    
+    def disassociate_addresses_valid?
+      raise "You have not supplied an IP addresses" unless @argv.ipaddress
+      true
+    end
+    def disassociate_addresses
+      radd = Rudy::AWS::EC2::Addresses.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      rinst = Rudy::AWS::EC2::Instances.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      raise "#{@argv.ipaddress} is not allocated to you" unless radd.exists?(@argv.ipaddress)
+      raise "#{@argv.ipaddress} is not associated!" unless radd.associated?(@argv.ipaddress)
+      
+      address = radd.get(@argv.ipaddress)
+      instance = rinst.get(address.instid)
+      
+      puts "Disassociating #{address.ipaddress} from #{instance.awsid} (#{instance.groups.join(', ')})"
+      execute_check(:medium)
+      execute_action { radd.disassociate(@argv.ipaddress) }
+      address = radd.get(@argv.ipaddress)
+      puts @@global.verbose > 0 ? address.inspect : address.dump(@@global.format)
+    end
+    
+    def addresses
+      radd = Rudy::AWS::EC2::Addresses.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      addresses = radd.list || []
+      
+      addresses.each do |address|
+        puts @@global.verbose > 0 ? address.inspect : address.dump(@@global.format)
+      end
+      
+      puts "No Addresses" if addresses.empty?
+    end
+    
+    
+  end
+
+end; end
+end; end
+

data/lib/rudy/cli/aws/ec2/candy.rb

@@ -0,0 +1,208 @@
+
+module Rudy; module CLI; 
+module AWS; module EC2;
+  
+  class Candy < Rudy::CLI::CommandBase
+    
+    def status_valid?
+      avail = Rudy::Utils.service_available?('status.aws.amazon.com', 80, 5)
+      raise ServiceUnavailable, 'status.aws.amazon.com' unless avail
+      true
+    end
+    def status
+      url = 'http://status.aws.amazon.com/rss/EC2.rss'
+      # TODO: Move to Rudy::AWS
+      ec2 = Rudy::Utils::RSSReader.run(url) || {}
+      
+      # TODO: Create Storable object
+      if @@global.format == 'yaml'
+        puts ec2.to_yaml
+      elsif @@global.format == 'json'
+        require 'json'
+        puts ec2.to_json
+      else
+        puts "Updated: #{ec2[:pubdate]} (updated every #{ec2[:ttl]} minutes)"
+        ec2[:items].each do |i|
+          puts
+          puts '%s' % i[:title]
+          puts '  %s: %s' % [i[:pubdate], i[:description]]
+          
+        end
+      end
+    end
+    
+    def ssh_valid?
+      if @@global.pkey
+        raise "Cannot find file #{@@global.pkey}" unless File.exists?(@@global.pkey)
+        raise "Insecure permissions for #{@@global.pkey}" unless (File.stat(@@global.pkey).mode & 600) == 0
+      end
+      if @option.group
+        rgroup = Rudy::AWS::EC2::Groups.new(@@global.accesskey, @@global.secretkey, @@global.region)
+        raise "Cannot supply group and instance ID" if @option.instid
+        raise "Group #{@option.group} does not exist" unless rgroup.exists?(@option.group)
+      end
+      if @option.instid && !Rudy::Utils.is_id?(:instance, @option.instid)
+        raise "#{@option.instid} is not an instance ID" 
+      end
+      true
+    end
+    def ssh
+      opts = {}
+      opts[:group] = @option.group if @option.group
+      opts[:group] = :any if @option.all
+      opts[:id] = @option.instid if @option.instid
+      
+      # Options to be sent to Net::SSH
+      ssh_opts = { :user => @global.user || Rudy.sysinfo.user, :debug => nil  }
+      if @@global.pkey 
+        raise "Cannot find file #{@@global.pkey}" unless File.exists?(@@global.pkey)
+        raise InsecureKeyPermissions, @@global.pkey unless File.stat(@@global.pkey).mode == 33152
+        ssh_opts[:keys] = @@global.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::AWS::EC2::Instances.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      lt = rudy.list_group(opts[:group], :running, opts[:id]) do |inst|
+        
+        # Print header
+        if @@global.quiet
+          print "You are #{ssh_opts[:user].bright}. " if !checked # only the 1st
+        else
+          print "Connecting #{ssh_opts[:user].bright}@#{inst.dns_public} "
+          puts "(#{inst.awsid}, groups: #{inst.groups.join(', ')})"
+        end
+        
+        # Make sure we want to run this command on all instances
+        if !checked && command != :interactive_ssh 
+          execute_check(:medium) if ssh_opts[:user] == "root"
+          checked = true
+        end
+        
+        # Open the connection and run the command
+        rbox = Rye::Box.new(inst.dns_public, ssh_opts)
+        ret = rbox.send(command, command_args)
+        puts ret unless command == :interactive_ssh
+      end
+    end
+
+    def copy_valid?
+      raise "You must supply a source and a target. See rudy-ec2 #{@alias} -h" unless @argv.size >= 2
+      raise "You cannot download and upload at the same time" if @option.download && @alias == 'upload'
+      raise "You cannot download and upload at the same time" if @option.upload && @alias == 'download'
+      true
+    end
+    def copy
+      
+      opts = {}
+      opts[:group] = @option.group if @option.group
+      opts[:group] = :any if @option.all
+
+      opts[:id] = @argv.shift if Rudy::Utils.is_id?(:instance, @argv.first)
+      opts[:id] &&= [opts[:id]].flatten
+      
+      # * +:recursive: recursively transfer directories (default: false)
+      # * +:preserve: preserve atimes and ctimes (default: false)
+      # * +:task+ one of: :upload (default), :download.
+      # * +:paths+ an array of paths to copy. The last element is the "to" path.
+      opts[:recursive] = @option.recursive ? true : false
+      opts[:preserve] = @option.preserve ? true : false
+      
+      opts[:paths] = @argv
+      opts[:dest] = opts[:paths].pop
+    
+      opts[:task] = :download if %w(dl download).member?(@alias) || @option.download
+      opts[:task] = :upload if %w(ul upload).member?(@alias)
+      opts[:task] ||= :upload
+      opts[:user] = @global.user || Rudy.sysinfo.user
+    
+      # Options to be sent to Net::SSH
+      ssh_opts = { :user => opts[:user], :debug => nil  }
+      ssh_opts[:keys] = @@global.pkey if @@global.pkey
+
+      if @@global.pkey
+        raise "Cannot find file #{@@global.pkey}" unless File.exists?(@@global.pkey)
+        raise "Insecure permissions for #{@@global.pkey}" unless (File.stat(@@global.pkey).mode & 600) == 0
+      end
+
+      checked = false
+      rudy = Rudy::AWS::EC2::Instances.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      lt = rudy.list_group(opts[:group], :running, opts[:id]) do |inst|
+        
+        if @option.print
+          Rudy::Utils.scp_command inst.dns_public, @@global.pkey, opts[:user], opts[:paths], opts[:dest], (opts[:task] == :download), false, @option.print
+          next
+        end
+        
+        # Print header
+        if @@global.quiet
+          print "You are #{ssh_opts[:user].bright}. " if !checked # only the 1st
+        else
+          print "Connecting #{ssh_opts[:user].bright}@#{inst.dns_public} "
+          puts "(#{inst.awsid}, groups: #{inst.groups.join(', ')})"
+        end
+        
+        # Make sure we want to run this command on all instances
+        if !checked
+          #execute_check(:medium) if opts[:user] == "root"
+          checked = true
+        end
+        
+        scp_opts = {
+          :recursive => opts[:recursive],
+          :preserve => opts[:preserve],
+          :chunk_size => 16384
+        }
+
+        Candy.scp(opts[:task], inst.dns_public, opts[:user], @@global.pkey, opts[:paths], opts[:dest], scp_opts)
+        puts 
+        puts unless @@global.quiet
+      end
+
+    end
+    
+    
+  private
+    
+    def Candy.scp(task, host, user, keypairpath, paths, dest, opts)
+      
+      connect_opts = {}
+      connect_opts[:keys] = [keypairpath] if keypairpath
+      
+      Net::SCP.start(host, user, connect_opts) do |scp|
+        
+        paths.each do |path| 
+          prev_path = nil
+          scp.send("#{task}", path, dest, opts) do |ch, name, sent, total|
+            #print "#{name}: #{sent}/#{total}\r"
+            msg = ((prev_path == name) ? "\r" : "\n") # new line for new file
+            msg << "#{name}: #{sent}/#{total}"  # otherwise, update the same line
+            print msg
+            STDOUT.flush        # update the screen every cycle
+            prev_path = name
+            break if sent == total
+          end
+          puts unless prev_path == path
+        end
+        
+      end
+    end
+    
+    
+    
+  end
+  
+  
+end; end
+end; end