solutious-rudy 0.5.1 → 0.6.0

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
+
+
+

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

@@ -0,0 +1,53 @@
+
+
+module Rudy; module CLI; 
+module AWS; module EC2;
+  
+  class KeyPairs < Rudy::CLI::CommandBase
+    
+    def create_keypairs_valid?
+      raise Drydock::ArgError.new('name', @alias) unless @argv.name
+      true
+    end
+    def create_keypairs
+      rkey = Rudy::AWS::EC2::KeyPairs.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      kp = execute_action { rkey.create(@argv.name) }
+      if %w[s string].member?(@@global.format)
+        puts "Name: #{kp.name}"
+        puts "Fingerprint: #{kp.fingerprint}", $/
+        puts "Copy the following private key data into a file."
+        puts "Set the permissions to 0600 and keep it safe.", $/
+        puts kp.private_key
+      else
+        puts @@global.verbose > 0 ? kp.inspect : kp.dump(@@global.format)
+      end
+    end
+    
+    def destroy_keypairs_valid?
+      raise Drydock::ArgError.new('name', @alias) unless @argv.name
+      true
+    end
+    def destroy_keypairs
+      rkey = Rudy::AWS::EC2::KeyPairs.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      raise "KeyPair #{@argv.name} does not exist" unless rkey.exists?(@argv.name)
+      kp = rkey.get(@argv.name)
+      puts "Destroying: #{kp.name}"
+      execute_check(:medium)
+      execute_action { rkey.destroy(kp.name) }
+      keypairs
+    end
+    
+    def keypairs
+      rkey = Rudy::AWS::EC2::KeyPairs.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      (rkey.list || []).each do |kp|
+        puts @@global.verbose > 0 ? kp.inspect : kp.dump(@@global.format)
+      end
+      puts "No keypairs" unless rkey.any?
+    end
+    
+    
+  end
+
+
+end; end
+end; end

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

@@ -0,0 +1,49 @@
+
+
+module Rudy; module CLI; 
+module AWS; module EC2;
+  
+  class Snapshots < Rudy::CLI::CommandBase
+    
+    def create_snapshots_valid?
+      raise Drydock::ArgError.new('volume ID', @alias) unless @option.volume
+      @rvol = Rudy::AWS::EC2::Volumes.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      @volume = @rvol.get(@argv.volid)
+      raise "Volume #{@volume.awsid} does not exist" unless @volume
+      true
+    end
+    def create_snapshots
+      rsnap = Rudy::AWS::EC2::Snapshots.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      snap = execute_action { rsnap.create(@volume.awsid) }
+      puts @@global.verbose > 0 ? snap.inspect : snap.dump(@@global.format)
+    end
+    
+    def destroy_snapshots_valid?
+      raise Drydock::ArgError.new('snapshot ID', @alias) unless @argv.snapid
+      @rsnap = Rudy::AWS::EC2::Snapshots.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      @snap = @rsnap.get(@argv.snapid)
+      raise "Snapshot #{@snap.awsid} does not exist" unless @snap
+      true
+    end
+    def destroy_snapshots
+      puts "Destroying: #{@snap.awsid}"
+      execute_check(:medium)
+      execute_action { @rsnap.destroy(@snap.awsid) }
+      snapshots
+    end
+    
+    def snapshots
+      rsnap = Rudy::AWS::EC2::Snapshots.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      snaps = rsnap.list || []
+      snaps.each do |snap|
+        puts @@global.verbose > 0 ? snap.inspect : snap.dump(@@global.format)
+      end
+      puts "No snapshots" if snaps.empty?
+    end
+    
+    
+  end
+
+
+end; end
+end; end

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

@@ -0,0 +1,104 @@
+
+
+module Rudy; module CLI; 
+module AWS; module EC2;
+  
+  class Volumes < Rudy::CLI::CommandBase
+    
+      
+    def volumes_create_valid?
+      raise "You must supply a volume size. See rudy volume -h" unless @option.size
+      raise "You must supply a zone." unless @@global.zone
+      true
+    end
+    def volumes_create
+      puts "Creating #{@option.size}GB volume in #{@@global.zone}"
+      rvol = Rudy::AWS::EC2::Volumes.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      execute_check(:low)
+      vol = execute_action("Create Failed") { 
+        rvol.create(@option.size, @@global.zone, @option.snapshot) 
+      }
+      puts @global.verbose > 1 ? vol.inspect : vol.dump(@@global.format)
+    end
+
+
+    def destroy_volumes_valid?
+      raise "You must supply a volume ID. See rudy volume -h" unless @argv.volid      
+      true
+    end
+    
+    def destroy_volumes
+      @rvol = Rudy::AWS::EC2::Volumes.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      @volume = @rvol.get(@argv.volid)
+      raise "Volume #{@volume.awsid} does not exist" unless @volume
+      raise "Volume #{@volume.awsid} is still in-use" if @volume.in_use?
+      raise "Volume #{@volume.awsid} is still attached" if @volume.attached?
+      raise "Volume #{@volume.awsid} is not available (#{@volume.state})" unless @volume.available?
+      
+      puts "Destroying #{@volume.awsid}"
+      execute_check(:medium)
+      execute_action("Destroy Failed") { @rvol.destroy(@volume.awsid) }
+      
+      vol = @rvol.get(@volume.awsid)
+      puts @global.verbose > 1 ? vol.inspect : vol.dump(@@global.format)
+    end
+    
+    
+    def volumes_attach_valid?
+      raise "You must supply an instance ID." unless @option.instance
+      raise "You must supply a volume ID." unless @argv.volid
+      true
+    end
+    def volumes_attach
+      @option.device ||= "/dev/sdh"
+      
+      rvol = Rudy::AWS::EC2::Volumes.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      rinst = Rudy::AWS::EC2::Instances.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      raise "Volume #{@argv.volid} does not exist" unless rvol.exists?(@argv.volid)
+      raise "Volume #{@argv.volid} is already attached" if rvol.attached?(@argv.volid)
+      raise "Instance #{@option.instance} does not exist" unless rinst.exists?(@option.instance)
+      
+      puts "Attaching #{@argv.volid} to #{@option.instance} on #{@option.device}"
+      execute_check(:low)
+      execute_action("Attach Failed") { 
+        rvol.attach(@argv.volid, @option.instance, @option.device) 
+      }
+      
+      vol = rvol.get(@argv.volid)
+      puts @global.verbose > 1 ? vol.inspect : vol.dump(@@global.format)
+    end
+    
+    def volumes_detach_valid?
+      raise "You must supply a volume ID." unless @argv.volid
+      true
+    end
+    
+    def volumes_detach
+      rvol = Rudy::AWS::EC2::Volumes.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      raise "Volume #{@argv.volid} does not exist" unless rvol.exists?(@argv.volid)
+      
+      vol = rvol.get(@argv.volid)
+      raise "Volume #{vol.awsid} is not attached" unless vol.attached?
+      
+      puts "Detaching #{vol.awsid} from #{vol.instid}"
+      execute_check(:medium)
+      execute_action("Detach Failed") { rvol.detach(vol.awsid) }
+      
+      vol = rvol.get(vol.awsid)
+      puts @global.verbose > 1 ? vol.inspect : vol.dump(@@global.format)
+    end
+    
+    
+    def volumes
+      rvol = Rudy::AWS::EC2::Volumes.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      volumes = rvol.list || []
+      volumes.each do |vol|
+        puts @global.verbose > 1 ? vol.inspect : vol.dump(@@global.format)
+      end
+      puts "No volumes" if volumes.empty?
+    end
+    
+  end
+
+end; end
+end; end

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

@@ -0,0 +1,22 @@
+
+
+module Rudy; module CLI; 
+module AWS; module EC2;
+  
+  class Zones < Rudy::CLI::CommandBase
+    
+    
+    def zones
+      rzone = Rudy::AWS::EC2::Zones.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      rzone.list_as_hash(@argv.name).each_value do |zon|
+        puts zon.dump(@@global.format)
+      end
+      puts "No zones" unless rzone.any?
+    end
+    
+    
+  end
+
+
+end; end
+end; end

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

@@ -0,0 +1,50 @@
+
+
+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
+      execute_check(:low)
+      s3 = Rudy::AWS::S3.new(@@global.accesskey, @@global.secretkey, @@global.region)
+      s3.create_bucket(@argv.name)
+      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/{deploy.rb → candy.rb}

@@ -1,12 +1,8 @@
 
-
-
-
 module Rudy
   module CLI
-    class Deploy < Rudy::CLI::Base
+    class Candy < Rudy::CLI::CommandBase
       
     end
   end
-end
-  
+end

data/lib/rudy/cli/config.rb

@@ -2,12 +2,12 @@
 
 module Rudy
   module CLI
-    class Config < Rudy::CLI::Base
+    class Config < Rudy::CLI::CommandBase
       
       # We force the CLI::Base#print_header to be quiet
       def print_header
-        @global.quiet = true
-        super
+        #@@global.quiet = true
+        #super
       end
       
       
@@ -31,47 +31,52 @@ module Rudy
       #     $ rudy config --all
       #
       def config
-        return if @config.nil? 
+        return if @@config.nil? 
         
         
-        @option.group ||= [@global.environment, @global.role].join(RUDY_DELIM)
+        @option.group ||= [@@global.environment, @@global.role].join(Rudy::DELIM)
         
-        return if @config.empty?
+        return if @@config.empty?
         
         # We need to check whether we're running on a human's computer
         # or within EC2 (we call that running "in-situ"). The userdata
         # available when running in-situ is in a different format.
         if Rudy.in_situ?
-          
-          
+          puts "TODO: do something intelligent with config when running on EC2"
         else
           if @option.all
             puts "# ACCOUNTS: not displayed"
             puts "# MACHINES: "
-            if @config.machines?
-              y @config.machines.to_hash 
+            if @@config.machines?
+              y @@config.machines.to_hash 
             end
-            if @config.routines?
+            if @@config.routines?
               puts "# ROUTINES: "
-              y @config.routines.to_hash 
+              y @@config.routines.to_hash 
             end
           elsif @option.defaults?
-            y @config.defaults.to_hash
+            y @@config.defaults.to_hash
+          elsif @option.script
+            y fetch_script_config.to_hash
+          elsif @option.rudy
+            rf = File.join(RUDY_HOME, 'Rudyfile')
+            raise "Cannot find: #{rf}" unless File.exists?(rf)
+            puts File.read(rf)
           else
-            zon, env, rol = @global.zone, @global.environment, @global.role
-            usr, att = @global.user, @argv.name
-            val = @config.machines.find_deferred(zon, env, rol, usr, att) || ''
+            zon, env, rol = @@global.zone, @@global.environment, @@global.role
+            usr, att = @@global.user, @argv.name
+            val = @@config.machines.find_deferred(zon, env, rol, usr, att) || ''
             puts (val.is_a?(String)) ? val : val.to_hash.to_yaml
           end
           
           #name = @argv.first
-          #if name && @config.userdata.has_key?(which) 
-          #  value = @config.userdata[which][name.to_s]
+          #if name && @@config.userdata.has_key?(which) 
+          #  value = @@config.userdata[which][name.to_s]
           #  puts value if value
           #elsif @option.all
-          #  puts @config.to_yaml
+          #  puts @@config.to_yaml
           #else
-          #  value = @config.userdata[which] 
+          #  value = @@config.userdata[which] 
           #  puts value.to_yaml if value
           #end
         end