solutious-rudy 0.4.0 → 0.5.0

data/lib/rudy/cli/deploy.rb

@@ -0,0 +1,12 @@
+
+
+
+
+module Rudy
+  module CLI
+    class Deploy < Rudy::CLI::Base
+      
+    end
+  end
+end
+  

data/lib/rudy/cli/disks.rb

@@ -0,0 +1,125 @@
+
+
+module Rudy
+  module CLI
+    class Disks < Rudy::CLI::Base
+
+      def disk
+        puts "Disks".bright
+        opts = {}
+        [:all, :path, :device, :size].each do |v| 
+          opts[v] = @option.send(v) if @option.respond_to?(v)
+        end
+        if @argv.diskname
+          key = @argv.diskname =~ /^disk-/ ? :name : :path
+          opts[key] = @argv.diskname
+        end
+
+        rdisks = Rudy::Disks.new(:config => @config, :global => @global)
+        disks = rdisks.list(opts)
+        raise "No disks." unless disks
+        #rbacks = Rudy::Backups.new(:config => @config, :global => @global)
+        disks.each_pair do |diskid, disk|
+          #backups = rbacks.list_by_disk(disk, 2)
+          print_disk(disk)
+        end
+      end
+      
+      def print_disk(disk, backups=[])
+        return unless disk
+        puts '-'*60
+        puts "Disk: #{disk.name.bright}"
+        puts disk.to_s
+        puts "#{backups.size} most recent backups:", backups.collect { |back| "#{back.nice_time} (#{back.awsid})" }
+        puts
+      end
+      
+      def create_disk_valid?
+        raise "No filesystem path specified" unless @option.path
+        raise "No size specified" unless @option.size
+        raise "No device specified" unless @option.device
+        true
+      end
+      
+      def create_disk
+        puts "Creating Disk".bright
+        exit unless Annoy.are_you_sure?(:low)
+        opts = {}
+        [:path, :device, :size, :group].each do |v| 
+          opts[v] = @option.send(v) if @option.respond_to?(v)
+        end
+        opts[:id] = @option.awsid if @option.awsid
+        opts[:id] &&= [opts[:id]].flatten
+        
+        @global.debug = true
+        rmachines = Rudy::Instances.new(:config => @config, :global => @global)
+        rmachines.list(opts).each_pair do |id,machine|
+          rdisks = Rudy::Disks.new(:config => @config, :global => @global)
+          disk = rdisks.create(machine, opts)
+          print_disk(disk) if disk
+        end
+      end
+      
+      
+      def destroy_disk_valid?
+        raise "No disk specified" unless @argv.diskname
+
+        true
+      end
+      
+      def destroy_disk
+        puts "Destroying Disk".bright
+        exit unless Annoy.are_you_sure?(:medium)
+        opts = {}
+        if @argv.diskname
+          key = @argv.diskname =~ /^disk-/ ? :name : :path
+          opts[key] = @argv.diskname
+        end
+        
+        # TODO: This is fucked! Store the machine info with the disk metadata.
+        # Get all disks that match the request and destroy them.
+        rmachines = Rudy::Instances.new(:config => @config, :global => @global)
+        rmachines.list(opts).each_pair do |id,machine|
+          rdisks = Rudy::Disks.new(:config => @config, :global => @global)
+          rdisks.destroy(machine, opts)
+        end
+        
+        puts "Done."
+      end
+
+      def attach_disk_valid?
+        destroy_disk_valid?
+        raise "There are no instances running in #{machine_group}" if !@instances || @instances.empty?
+        true
+      end
+      
+      def attach_disk
+        puts "Attaching #{name}"
+        switch_user("root")
+        exit unless Annoy.are_you_sure?(:medium)
+  
+        machine = @instances.values.first  # AK! Assumes single machine
+        
+        execute_attach_disk(@disk, machine)
+
+        puts
+        ssh_command machine[:dns_name], keypairpath, @global.user, "df -h" # Display current mounts
+        puts 
+
+        puts "Done!"
+      end
+
+
+
+      
+    end
+  end
+end
+
+
+__END__
+
+    
+
+
+

data/lib/rudy/cli/domains.rb

@@ -0,0 +1,17 @@
+
+
+module Rudy
+  module CLI
+    class Domains < Rudy::CLI::Base
+      
+      
+      def domains
+        puts "Domains".bright, $/
+        
+        rdom = Rudy::Domains.new(:config => @config, :global => @global)
+        puts rdom.list
+      end
+      
+    end
+  end
+end

data/lib/rudy/cli/groups.rb

@@ -0,0 +1,77 @@
+
+#
+#
+module Rudy
+  module CLI
+    class Groups < Rudy::CLI::Base
+      
+      def group
+        puts "Machine Groups".bright
+        opts = {}
+        name = @option.all ? nil : @argv.name
+        rudy = Rudy::Groups.new(:config => @config, :global => @global)
+        rudy.list(name).each do |grp|
+          puts '-'*60
+          puts grp.to_s
+        end
+      end
+      
+      def destroy_group_valid?
+        @rgroup = Rudy::Groups.new(:config => @config, :global => @global)
+        raise "Group #{@rgroup.name(@argv.name)} does not exist" unless @rgroup.exists?(@argv.name)
+        true
+      end
+      
+      def destroy_group
+        puts "Destroying Machine Group".bright
+        opts = {}
+        puts "Destroying group: #{@rgroup.name(@argv.name)}"
+        exit unless Annoy.are_you_sure?(:medium)
+        
+        @rgroup.destroy(@argv.name)
+        puts "Done!"
+      end
+      
+      def create_group
+        puts "Creating Machine Group".bright
+        opts = check_options
+        exit unless Annoy.are_you_sure?(:medium)
+        rudy = Rudy::Groups.new(:config => @config, :global => @global)
+        rudy.create(@argv.name, nil, opts)
+        rudy.list(@argv.name)
+      end
+      
+      def revoke_group
+        puts "Revoke Machine Group Rule".bright
+        opts = check_options
+        exit unless Annoy.are_you_sure?(:medium)
+        rudy = Rudy::Groups.new(:config => @config, :global => @global)
+        rudy.revoke(@argv.name, opts)
+        rudy.list(@argv.name)
+      end
+      
+      def authorize_group
+        puts "Authorize Machine Group Rule".bright
+        opts = check_options
+        exit unless Annoy.are_you_sure?(:medium)
+        rudy = Rudy::Groups.new(:config => @config, :global => @global)
+        rudy.authorize(opts)
+        rudy.list(opts)
+      end
+
+      
+    private
+      
+      def check_options
+        opts = {}
+        [:addresses, :protocols, :owner, :group, :ports].each do |opt|
+          opts[opt] = @option.send(opt) if @option.respond_to?(opt)
+        end
+        opts[:ports].collect! { |port| port.split(/:/) } if opts[:ports]
+        opts
+      end
+      
+    end
+  end
+end
+

data/lib/rudy/{command → cli}/images.rb

@@ -1,8 +1,8 @@
 
 
 module Rudy
-  module Command
-    class Images < Rudy::Command::Base
+  module CLI
+    class Images < Rudy::CLI::Base
       
 
       def images
@@ -33,7 +33,7 @@ module Rudy
         
         # ~/.rudy, /etc/motd, history -c, /etc/hosts, /var/log/rudy*
         
-        are_you_sure?(2)
+        exit unless Annoy.are_you_sure?(:medium)
         
         
         machine_list = @ec2.instances.list(machine_group)
@@ -44,7 +44,7 @@ module Rudy
         
         puts "The new image will be based on #{machine_group}_01"
         
-        @option.account ||= @config.awsinfo.account
+        @option.account ||= @config.accounts.aws.accountnum
         
         unless @option.account
           puts "Enter your 12 digit Amazon account number:"
@@ -71,7 +71,7 @@ module Rudy
           session.exec!("touch /root/firstrun")
         end
         
-        puts "Starting bundling process...".att(:bright)
+        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)
 
@@ -83,7 +83,7 @@ module Rudy
         raise "You must supply an AMI ID (ami-XXXXXXX)" unless ami
         puts "Deregistering AMI: #{ami}"
         
-        are_you_sure?
+        exit unless Annoy.are_you_sure?
         
         if @ec2.images.deregister(ami)
           puts "Done!"
@@ -93,6 +93,18 @@ module Rudy
         
       end
       
+      
+      def print_image(img)
+        puts '-'*60
+        puts "Image: #{img[:aws_id].bright}"
+        img.each_pair do |key, value|
+           printf(" %22s: %s#{$/}", key, value) if value
+         end
+        puts
+      end
+      
+      
+      
     end
   end
 end

data/lib/rudy/cli/instances.rb

@@ -0,0 +1,142 @@
+
+
+module Rudy::CLI
+  class Instances < Rudy::CLI::Base
+    
+    def connect
+      puts "Rudy Connect".bright
+
+      if @argv.cmd
+        @argv.cmd = [@argv.cmd].flatten.join(' ')
+        if @global.user.to_s == "root"
+          exit unless Annoy.are_you_sure?(:medium)
+        end
+      end
+      
+      rudy = Rudy::Instances.new(:config => @config, :global => @global)
+      rudy.connect(@option.group, @argv.cmd, @option.awsid, @option.print)
+    end
+
+    def copy_valid?
+      raise "You must supply a source and a target. See rudy #{@alias} -h" unless @argv.size >= 2
+      raise "You cannot download and upload at the same time" if @option.download && @alias == 'upload'
+      true
+    end
+    def copy
+      puts "Rudy Copy".bright
+      opts = {}
+      opts[:group] = @option.group if @option.group
+      opts[:id] = @option.awsid if @option.awsid
+      opts[:id] &&= [opts[:id]].flatten
+      
+      # Is this more clear?
+      @option.recursive && opts[:recursive] = true
+      @option.preserve  && opts[:preserve]  = true
+      @option.print     && opts[:print]     = true
+      
+      
+      opts[:paths] = @argv
+      opts[:dest] = opts[:paths].pop
+      
+      opts[:task] = :download if @alias == 'download' || @option.download
+      opts[:task] = :upload if @alias == 'upload'
+      opts[:task] ||= :upload
+      
+      #exit unless @option.print || Annoy.are_you_sure?(:low)
+      
+      rudy = Rudy::Instances.new(:config => @config, :global => @global)
+      rudy.copy(opts[:group], opts[:id], opts)
+    end
+
+
+    def status
+      puts "Instance Status".bright
+      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.awsid if @argv.awsid
+      opts[:id] &&= [opts[:id]].flatten
+      rudy = Rudy::Instances.new(:config => @config, :global => @global)
+      
+      lt = rudy.list(opts[:state], opts[:group], opts[:id]) do |inst|
+        puts '-'*60
+        puts "Instance: #{inst.awsid.bright} (AMI: #{inst.ami})"
+        puts inst.to_s
+      end
+      puts "No instances running" if !lt || lt.empty?
+    end
+    alias :instance :status
+    
+    def console_valid?
+      
+      @rmach = Rudy::Instances.new(:config => @config, :global => @global)
+    end
+    
+    def console
+      puts "Instance Console".bright
+      opts = {}
+      opts[:group] = @option.group if @option.group
+      opts[:id] = @argv.awsid if @argv.awsid
+      opts[:id] &&= [opts[:id]].flatten
+      
+      unless @rmach.any?
+        puts "No instances running"
+        return
+      end
+      
+      console = @rmach.console(opts[:group], opts[:id])
+      
+      if console
+        puts console
+      else
+        puts "Console output is not available"
+      end
+      
+    end
+    
+    
+    def instance_create
+      puts "Create Instance".bright
+      opts = {}
+      [:group, :ami, :address, :itype, :keypair].each do |n|
+        opts[n] = @option.send(n) if @option.send(n)
+      end
+
+      rmach = Rudy::Instances.new(:config => @config, :global => @global)
+      # TODO: Print number of instances running. If more than 0, use Annoy.are_you_sure?
+      rmach.create(opts) do |inst| # Rudy::AWS::EC2::Instance objects
+        puts '-'*60
+        puts "Instance: #{inst.awsid.bright} (AMI: #{inst.ami})"
+        puts inst.to_s
+      end
+    
+    end
+    
+    
+    def instance_destroy
+      puts "Destroy Instances".bright
+      opts = {}
+      opts[:group] = @option.group if @option.group
+      opts[:id] = @argv.awsid if @argv.awsid
+      opts[:id] &&= [opts[:id]].flatten
+      
+      rmach = Rudy::Instances.new(:config => @config, :global => @global)
+      instances = rmach.list(:running, opts[:group], opts[:id])
+      raise "No instances running" if instances.nil? || instances.empty?
+      puts "Destroying #{instances.size} instances in #{instances.first.groups.first}"
+      exit unless Annoy.are_you_sure?(:low)
+      rmach.destroy(opts[:group], opts[:id])
+      puts "Done!"
+    end
+    
+  end
+end
+
+

data/lib/rudy/cli/keypairs.rb

@@ -0,0 +1,47 @@
+
+
+module Rudy
+  module CLI
+    class KeyPairs < Rudy::CLI::Base
+      
+      def create_keypair
+        puts "Create KeyPairs".bright
+        rkey = Rudy::KeyPairs.new(:config => @config, :global => @global)
+        name = @argv.kpname
+        
+        rkey.create(@argv.kpname, :force => false)
+        rkey.list.each do |kp|
+          puts kp.to_s
+        end
+      end
+      
+      
+      def destroy_keypair
+        puts "Destroy KeyPairs".bright
+        rkey = Rudy::KeyPairs.new(:config => @config, :global => @global)
+        raise "KeyPair #{rkey.name(@argv.kpname)} does not exist" unless rkey.exists?(@argv.kpname)
+        kp = rkey.get(@argv.kpname)
+        puts "Destroying keypair: #{kp.name}"
+        puts "NOTE: the private key file will also be deleted and you will not be able to".color(:blue)
+        puts "connect to instances started with this keypair.".color(:blue)
+        exit unless Annoy.are_you_sure?(:low)
+        ret = rkey.destroy(@argv.kpname)
+        puts ret ? "Success" : "Failed"
+      end
+      
+      def keypair
+        puts "KeyPairs".bright
+        rkey = Rudy::KeyPairs.new(:config => @config, :global => @global)
+        
+        rkey.list.each do |kp|
+          puts kp.to_s
+        end
+        
+        puts "No keypairs" unless rkey.any?
+        
+      end
+      
+      
+    end
+  end
+end