rudy 0.4.0 → 0.6.0

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

@@ -0,0 +1,121 @@
+
+module Rudy; module CLI; 
+module AWS; module EC2;
+  
+  class Groups < Rudy::CLI::CommandBase
+    
+
+    
+    def create_groups_valid?
+      @rgroups = Rudy::AWS::EC2::Groups.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      raise Drydock::ArgError.new('group name', @alias) unless @argv.name
+      raise "Group #{@argv.name} alread exists" if @rgroups.exists?(@argv.name)
+      true
+    end
+    def create_groups
+      opts = check_options
+      puts "Creating #{@argv.name}"
+      
+      execute_action { 
+        @rgroups.create(@argv.name, @option.description, opts[:addresses], opts[:ports], opts[:protocols])
+      }
+      
+      @rgroups.list(@argv.name) do |group|
+        puts @@global.verbose > 0 ? group.inspect : group.dump(@@global.format)
+      end
+    end
+    
+    
+    def destroy_groups_valid?
+      @rgroups = Rudy::AWS::EC2::Groups.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      raise Drydock::ArgError.new('group name', @alias) unless @argv.name
+      raise "Group #{@argv.name} does not exist" unless @rgroups.exists?(@argv.name)
+      true
+    end
+    
+    def destroy_groups
+      puts "Destroying group: #{@argv.name}"
+      execute_check(:medium)
+      execute_action { @rgroups.destroy(@argv.name) }
+      @argv.clear # so groups will print all other groups
+      groups
+    end
+    
+    def revoke_groups_valid?; modify_group_valid?; end
+    def revoke_groups; modify_group(:revoke); end
+    
+    def authorize_groups_valid?; modify_group_valid?; end
+    def authorize_groups; modify_group(:authorize); end
+
+    def groups
+      opts = {}
+      name = @option.all ? nil : @argv.name
+      rgroups = Rudy::AWS::EC2::Groups.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      rgroups.list(name).each do |group|
+        puts @@global.verbose > 0 ? group.inspect : group.dump(@@global.format)
+      end
+    end
+    
+  private
+    
+    def modify_group_valid?
+      if @option.owner == 'self'
+        raise "AWS_ACCOUNT_NUMBER not set" unless @@global.accountnum 
+        @option.owner = @@global.accountnum 
+      end
+      
+      if (@option.addresses || @option.ports) && (@option.group || @option.owner)
+        raise Drydock::OptError.new('', @alias, "Cannot mix group and network authorization")
+      end
+      if @option.owner && !@option.group
+        raise Drydock::OptError.new('', @alias, "Must provide -g with -o")
+      end
+      
+      raise Drydock::ArgError.new('group name', @alias) unless @argv.name
+      @groups = Rudy::AWS::EC2::Groups.new(@@global.accesskey, @@global.secretkey, @@global.region)
+    end
+    
+    def modify_group(action)
+      opts = check_options
+      if (@option.group || @option.owner)
+        g = [opts[:owner], opts[:group]].join(':')
+        puts "#{action.to_s.capitalize} access to #{@argv.name.bright} from #{g.bright}"
+      else
+        print "#{action.to_s.capitalize} access to #{@argv.name.bright}"
+         puts " from #{opts[:addresses].join(', ').bright}"
+        print "on #{opts[:protocols].join(', ').bright} "
+         puts "ports: #{opts[:ports].map { |p| "#{p.join(' to ').bright}" }.join(', ')}"
+      end
+      rgroups = Rudy::AWS::EC2::Groups.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      execute_check(:medium)
+      execute_action { 
+        if (@option.group || @option.owner)
+          rgroups.send("#{action.to_s}_group", @argv.name, opts[:group], opts[:owner])
+        else
+          rgroups.send(action, @argv.name, opts[:addresses], opts[:ports], opts[:protocols])
+        end
+      }
+      groups # prints on the modified group b/c of @argv.name
+    end
+    
+    def check_options
+      opts = {}
+      [:addresses, :protocols, :owner, :group, :ports].each do |opt|
+        opts[opt] = @option.send(opt) if @option.respond_to?(opt)
+      end
+      unless @option.group || @option.owner
+        opts[:ports].collect! { |port| port.split(/[:-]/) } if opts[:ports]
+        opts[:ports] ||= [[22,22],[80,80],[443,443]]
+        opts[:addresses] ||= [Rudy::Utils::external_ip_address]
+        opts[:protocols] ||= [:tcp]
+      else
+        opts[:owner] ||= @@global.accountnum
+      end
+      opts
+    end
+    
+  end
+
+
+end; end
+end; end

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

@@ -0,0 +1,196 @@
+
+
+module Rudy; module CLI; 
+module AWS; module EC2;
+  
+  class Images < Rudy::CLI::CommandBase
+    
+    #def print_header
+    #  puts @global.print_header, @@global.print_header
+    #end
+    
+    
+    def images_valid?
+      if @option.owner == 'self'
+        raise "AWS_ACCOUNT_NUMBER not set" unless @@global.accountnum 
+        @option.owner = @@global.accountnum 
+      end
+      
+      true  
+    end
+    def images
+      
+      rimages = Rudy::AWS::EC2::Images.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      unless @option.all
+        @option.owner ||= 'amazon' 
+        puts "Images owned by #{@option.owner.bright}" unless @argv.awsid
+      end
+      
+      images = rimages.list(@option.owner, @argv) || []
+      images.each do |img|
+        puts @@global.verbose > 0 ? img.inspect : img.dump(@@global.format)
+      end
+      puts "No images" if images.empty?
+    end
+
+    def prepare_images_valid?
+      true
+    end
+    def prepare_images
+      opts = {}
+      opts[:group] = @option.group if @option.group
+      opts[:group] = :any if @option.all
+      opts[:id] = @option.instid if @option.instid
+      
+      puts "This will do the following:"
+      puts "- Clear bash history"
+      # NOTE: We can't delete the host keys here. Otherwise we can't create the image. 
+      #puts "- Delete host SSH keys (this is permanent!)"
+      puts "" 
+      
+      # Options to be sent to Net::SSH
+      ssh_opts = { :user => @global.user || Rudy.sysinfo.user, :debug => STDERR  }
+      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
+      
+      execute_check(:medium)
+      
+      rudy = Rudy::AWS::EC2::Instances.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      lt = rudy.list_group(opts[:group], :running, opts[:id]) do |inst|
+        
+        puts "Preparing #{inst.dns_public}..."
+        
+        # Open the connection and run the command
+        rbox = Rye::Box.new(inst.dns_public, ssh_opts)
+        rbox.safe = false
+        # We need to explicitly add the rm command for rbox so we
+        # can delete the SSH host keys. This is will force the instance
+        # to re-create it's SSH keys on first boot.
+        def rbox.rm(*args); cmd('rm', args); end
+        p ret = rbox.history(:c)
+        p ret.exit_code
+        p ret.stderr
+        p ret.stdout
+        
+      end
+      
+      puts "done"
+    end
+    
+    def create_images_valid?
+      raise "No account number" unless @@global.accountnum
+      raise "No Amazon cert-***.pem" unless @@global.cert
+      raise "No Amazon pk-***.pem" unless @@global.privatekey
+      true
+    end
+    
+    def create_images
+      opts = {}
+      opts[:group] = @option.group if @option.group
+      opts[:group] = :any if @option.all
+      opts[:id] = @option.instid if @option.instid
+      
+      puts "You may want to run rudy-ec2 #{@alias} --prepare before this.".color(:blue)
+      puts "This feature is experimental. Make sure you enter the bucket"
+      puts "and image names correctly because if they're wrong the image"
+      puts "won't get created and you'll be annoyed that you waited."
+      
+      # Options to be sent to Net::SSH
+      ssh_opts = { :user => @global.user || Rudy.sysinfo.user, :debug => STDERR  }
+      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
+      
+      unless @option.name
+        print "Enter the image name: "
+        @option.image_name = gets.chomp
+      end
+
+      unless @option.bucket
+        print "Enter the S3 bucket that will store the image: "
+        @option.bucket_name = gets.chomp
+      end
+      
+      execute_check(:medium)
+      
+      rudy = Rudy::AWS::EC2::Instances.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      lt = rudy.list_group(opts[:group], :running, opts[:id]) do |inst|
+        
+        puts inst.dns_public
+        
+        # Open the connection and run the command
+        rbox = Rye::Box.new(inst.dns_public, ssh_opts)
+        
+        # ~/.rudy, /etc/motd, history -c, /etc/hosts, /var/log/rudy*
+        cert = File.read(@@global.cert)
+        pk = File.read(@@global.privatekey)
+        rbox.safe = false
+        rbox.echo("'#{cert}' > /mnt/cert-temporary.pem")
+        rbox.echo("'#{pk}' > /mnt/pk-temporary.pem")
+        rbox.safe = true
+        rbox.touch("/root/firstrun")
+        
+        # TODO: 
+        # We have to delete the host keys just before we run the bundle command. 
+        # The problem is that if we lose the connection we won't be able to connect
+        # to the instance again. A better solution is to ass the keys to the ignore
+        # list for the bundle command. 
+        
+        #ret = rbox.rm('/etc/ssh/ssh_host_*_key*')
+        #puts "Starting bundling process...".bright
+        #puts ssh_command(machine[:dns_name], keypairpath, @global.user, "ec2-bundle-vol -r i386 -p #{@option.image_name} -k /mnt/pk-*pem -c /mnt/cert*pem -u #{@option.account}", @option.print)
+        #puts ssh_command(machine[:dns_name], keypairpath, @global.user, "ec2-upload-bundle -b #{@option.bucket_name} -m /tmp/#{@option.image_name}.manifest.xml -a #{@global.accesskey} -s #{@global.secretkey}", @option.print)
+        #
+        #@ec2.images.register("#{@option.bucket_name}/#{@option.image_name}.manifest.xml") unless @option.print
+        
+        break
+      end
+      
+    end
+    
+   #def create_images_valid?
+   #  puts "Make sure the machine is clean. I don't want archive no crud!"
+   #  switch_user("root")
+   #  
+   #    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)
+   #  true
+   #end
+   #
+   #
+   #def prepare_images
+   #  # TODO: Avail hooks for clean an instance
+   #  # Clean off Rudy specific crap. 
+   #end
+   #
+   #
+   
+   #
+   #def deregister
+   #  ami = @argv.first
+   #  raise "You must supply an AMI ID (ami-XXXXXXX)" unless ami
+   #  puts "Deregistering AMI: #{ami}"
+   #  
+   #  exit unless Annoy.are_you_sure?
+   #  
+   #  if @ec2.images.deregister(ami)
+   #    puts "Done!"
+   #  else
+   #    puts "There was an unknown problem!"
+   #  end
+   #  
+   #end
+
+  end
+
+
+end; end
+end; end
+
+

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

@@ -0,0 +1,194 @@
+
+
+module Rudy; module CLI; 
+module AWS; module EC2;
+  class InstanceAndGroupError < Drydock::ArgError
+    def message; "You cannot provide a group and an instance ID"; end
+  end
+  class NoInstanceError < Drydock::ArgError
+    def message; "You must provide a group or instance ID"; end
+  end
+  
+  class Instances < Rudy::CLI::CommandBase
+    
+    def instances_create_valid?
+      
+      raise "Cannot supply an instance ID" if @option.instid
+      
+      if @option.group
+        rgroup = Rudy::AWS::EC2::Groups.new(@@global.accesskey, @@global.secretkey, @@global.region)
+        raise "Group #{@option.group} does not exist" unless rgroup.exists?(@option.group)
+      end
+      
+      true
+    end
+      
+    def instances_create
+      
+      opts = {                 # Defaults
+        :group => 'default',
+        :size => 'm1.small',
+        :zone => @@global.zone
+      }
+      
+      radd = Rudy::AWS::EC2::Addresses.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      rinst = Rudy::AWS::EC2::Instances.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      
+      if @option.address
+        raise "Cannot specify both -a and -n" if @option.newaddress
+        raise "#{@option.address} is not allocated to you" unless radd.exists?(@option.address)
+        raise "#{@option.address} is already associated!" if radd.associated?(@option.address)
+      end
+      
+      # These can be sent directly to EC2 class
+      [:group, :ami, :size, :keypair, :private].each do |n|
+        opts[n] = @option.send(n) if @option.send(n)
+      end
+      
+      puts "Creating #{opts[:size]} instance in #{@@global.zone}"
+      
+      unless opts[:keypair]
+        puts "You did not specify a keypair. Unless you've prepared a user account".color(:blue)
+        puts "on this image (#{opts[:ami]}) you will not be able to log in to it.".color(:blue)
+        exit unless Annoy.proceed?(:low)
+      end
+      
+      instances = rinst.list_group(opts[:group], :running)
+      
+      if instances && instances.size > 0
+        instance_count = (instances.size == 1) ? 'is 1 instance' : "are #{instances.size} instances"
+        puts "There #{instance_count} running in the #{opts[:group]} group."
+        exit unless Annoy.proceed?(:low)
+      end
+      
+      if @option.newaddress
+        print "Creating address... "
+        address = radd.create
+        puts "#{address.ipaddress}"
+        @option.address = address.ipaddress
+      end
+         
+      execute_action do
+        first_instance = true
+        rinst.create(opts) do |inst| # Rudy::AWS::EC2::Instance objects
+        
+          # Assign IP address to only the first instance
+          if first_instance && @option.address
+            puts "Associating #{@option.address} to #{inst.awsid}"
+            radd.associate(@option.address, inst.awsid)
+            first_instance = false
+          end
+        
+          puts @@global.verbose > 0 ? inst.inspect : inst.dump(@@global.format)
+        end
+      end
+    end
+
+    def instances_restart_valid?
+      raise InstanceAndGroupError.new(nil, @alias) if @option.group && @argv.instid
+      raise NoInstanceError.new(nil, @alias) if !@option.group && !@argv.instid
+      
+      if @option.group
+        rgroup = Rudy::AWS::EC2::Groups.new(@@global.accesskey, @@global.secretkey, @@global.region)
+        raise "Group #{@option.group} does not exist" unless rgroup.exists?(@option.group)
+      end
+      
+      if @option.private
+        raise Drydock::OptsError.new(nil, @alias, "Cannot allocate public IP for private instance") if @option.address || @option.newadress
+      end
+      
+      @rinst = Rudy::AWS::EC2::Instances.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      raise "No instances" unless @rinst.any?
+      true
+    end
+    alias :instances_destroy_valid? :instances_restart_valid?
+    
+    def instances_destroy
+      instances_action :destroy
+    end
+    
+    def instances_restart
+      instances_action :restart
+    end
+    
+    def consoles_valid?
+      @rinst = Rudy::AWS::EC2::Instances.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      raise "No instances" unless @rinst.any?
+      true
+    end
+    def consoles
+      opts = {}
+      opts[:group] = @option.group if @option.group
+      opts[:id] = @argv.instid if @argv.instid
+      opts[:id] &&= [opts[:id]].flatten
+      
+      lt = @rinst.list_group(opts[:group], :any, opts[:id]) do |inst|
+        puts '-'*50
+        puts "Console for: #{inst.liner_note}", $/
+        console = @rinst.console(inst.awsid)
+        output = console ? Base64.decode64(console) : "Unavailable"
+        puts output.noansi # Remove color and clear, etc...
+      end
+      
+    end
+    
+    def status
+      opts = {}
+      
+      opts[:group] = @option.group if @option.group
+      opts[:state] = @option.state if @option.state
+
+      # A nil value forces the @ec2.instances.list to return all instances
+      if @option.all
+        opts[:state] = :any
+        opts[:group] = :any
+      end
+
+      opts[:id] = @argv.instid if @argv.instid
+      opts[:id] &&= [opts[:id]].flatten
+    
+      rudy = Rudy::AWS::EC2::Instances.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      lt = rudy.list_group(opts[:group], opts[:state], opts[:id]) do |inst|
+        puts @@global.verbose > 0 ? inst.inspect : inst.dump(@@global.format)
+      end
+      puts "No instances running" if !lt || lt.empty?
+    end
+    alias :instances :status
+
+
+    private
+    
+    # * +action+ is one of :destroy, :restart
+    def instances_action(action)
+      opts = {}
+      opts[:group] = @option.group if @option.group
+      opts[:id] = @argv.instid if @argv.instid
+      opts[:id] &&= [opts[:id]].flatten
+      
+      instances = @rinst.list_group(opts[:group], :running, opts[:id])
+      raise "No matching instances running" if instances.nil?
+      
+      inst_names = instances.collect { |inst| inst.dns_public || inst.awsid }
+      inst_ids = instances.collect { |inst| inst.awsid }
+      
+      instance_count = (instances.size == 1) ? '1 instance' : "#{instances.size} instances"
+      
+      print "#{action.to_s.capitalize} #{instance_count} (#{inst_names.join(', ')}) "
+      print "in #{opts[:group]}" if opts[:group]
+      puts
+      execute_check(:medium)
+      
+      execute_action("#{action.to_s.capitalize} Failed") { 
+        @rinst.send(action, inst_ids)
+      }
+      status
+    end
+    
+    
+  end
+
+end; end
+end; end
+
+
+