solutious-rudy 0.4.0 → 0.5.0

data/lib/rudy/cli.rb

@@ -0,0 +1,142 @@
+
+require 'drydock'
+
+module Rudy
+  class UnknownInstance < RuntimeError #:nodoc
+  end
+end
+
+module Rudy
+  
+  # 
+  module CLI
+    class NoCred < RuntimeError #:nodoc
+    end
+    
+    class Base < Drydock::Command
+
+      attr_reader :config
+      
+    protected
+      def init
+        
+        # This is duplicated in Huxtable :[
+        String.disable_color if @global.nocolor
+        Rudy.enable_quiet if @global.quiet
+        
+        @config = Rudy::Config.new
+        @config.look_and_load(@global.config)
+        
+        unless !@config.empty? && @config.accounts? && @config.accounts.aws && @config.accounts.aws.accesskey
+          STDERR.puts "No AWS credentials. Check your configs!"
+          STDERR.puts "Try: #{$0} init"
+          exit 1
+        end
+                
+        # EVERYTHING BELOW THIS LINE IS ATROCIOUS 
+        
+        # These are here so we can print the machine group shit in the header. 
+        # The dupilcation annoys me (see Rudy::Huxtable#init_globals) and I'd
+        # like to find a cleaner solution. 
+        if @config.defaults
+          @global.region ||= @config.defaults.region 
+          @global.zone ||= @config.defaults.zone 
+          @global.environment ||= @config.defaults.environment
+          @global.role ||= @config.defaults.role 
+          @global.position ||= @config.defaults.position 
+          @global.user ||= @config.defaults.user
+        end
+        
+        @global.region ||= DEFAULT_REGION
+        @global.zone ||= DEFAULT_ZONE
+        @global.environment ||= DEFAULT_ENVIRONMENT
+        @global.role ||= DEFAULT_ROLE
+        @global.position ||= DEFAULT_POSITION
+        @global.user ||= DEFAULT_USER
+        
+        
+        if @global.verbose > 1
+          puts "CONFIGS: ", @config.paths, $/
+          
+          puts "GLOBALS:"
+          @global.marshal_dump.each_pair do |n,v|
+            puts "#{n}: #{v}"
+          end
+          ["machines", "routines"].each do |type|
+            puts "#{$/*2}#{type.upcase}:"
+            val = @config.send(type).find_deferred(@global.environment, @global.role)
+            puts val.to_hash.to_yaml if val
+          end
+          puts
+        end
+        
+        if @global.environment =~ /^prod/ && Drydock.debug?
+          puts Rudy.banner("PRODUCTION ACCESS IS DISABLED IN DEBUG MODE")
+          exit 1
+        end
+
+        
+        # TODO: enforce home directory permissions
+        #if File.exists?(RUDY_CONFIG_DIR)
+        #  puts "Checking #{check_environment} permissions..."
+        #end
+        
+      end
+      
+      
+      # Print a default header to the screen for every command.
+      #
+      def print_header
+        puts Rudy::CLI.generate_header(@global, @config)
+        unless @global.quiet
+          puts # a new line
+          
+          if @global.environment == "prod"
+            msg = "YOU ARE PLAYING WITH PRODUCTION"
+            puts Rudy.banner(msg, :huge, :red), $/
+          end
+        
+          puts Rudy.banner("THIS IS EC2"), $/ if Rudy.in_situ?
+        end
+      end
+      
+
+    end
+
+
+
+    def self.generate_header(global, config)
+      header = StringIO.new
+      title = "RUDY v#{Rudy::VERSION}" unless global.quiet
+      now_utc = Time.now.utc.strftime("%Y-%m-%d %H:%M:%S")
+      criteria = []
+      [:zone, :environment, :role, :position].each do |n|
+        val = global.send(n)
+        next unless val
+        criteria << "#{n.to_s.slice(0,1).att :normal}:#{val.to_s.bright}"
+      end
+      if config.accounts && config.accounts.aws
+        header.puts '%s -- %s -- %s UTC' % [title, config.accounts.aws.name, now_utc] unless global.quiet
+        header.puts '[%s]' % criteria.join("  ") unless global.quiet
+      end
+      
+      header.rewind
+      header.read
+    end
+
+
+  end
+end
+
+# Require EC2, S3, Simple DB class
+begin
+  # TODO: Use autoload
+  Dir.glob(File.join(RUDY_LIB, 'rudy', 'cli', "*.rb")).each do |path|
+    require path
+  end
+rescue LoadError => ex
+  puts "Error: #{ex.message}"
+  exit 1
+end
+
+

data/lib/rudy/cli/addresses.rb

@@ -0,0 +1,85 @@
+
+
+module Rudy
+  module CLI
+    class Addresses < Rudy::CLI::Base
+
+
+      def associate_addresses_valid?
+        raise "You have not supplied an IP addresses" unless @argv.ipaddress
+        raise "You did not supply an instance ID" unless @argv.instanceid
+        
+        @rmach = Rudy::Instances.new(:config => @config, :global => @global)
+        @radd = Rudy::Addresses.new(:config => @config, :global => @global)
+        
+        raise "Instance #{@argv.instid} does not exist!" unless @rmach.exists?(@argv.instid)
+        
+        raise "#{@argv.ipaddress} is not allocated to you" unless @radd.exists?(@argv.ipaddress)
+        raise "#{@argv.ipaddress} is already associated!" if @radd.associated?(@argv.ipaddress)
+        
+        true
+      end
+      
+      def associate_addresses
+        puts "Associating #{@argv.address} to #{@inst[:aws_groups]}: #{@inst[:dns_name]}"
+        
+        address = @radd.get(@argv.ipaddress)
+        puts address.to_s
+        
+        puts "Done!"
+        puts
+      end
+      
+      def addresses
+        puts "Addresses".bright, $/
+        
+        radd = Rudy::Addresses.new(:config => @config, :global => @global)
+        addresses = radd.list || []
+        
+        addresses.each do |address|
+          puts address.to_s
+        end
+        
+        puts "No Addresses" if addresses.empty?
+      end
+      
+      
+      def addresses_create
+        puts "Create Address".bright, $/
+        radd = Rudy::Addresses.new(:config => @config, :global => @global)
+        address = radd.create
+        puts address.to_s
+      end
+      
+      def addresses_destroy_valid?
+        raise "You have not supplied an IP addresses" unless @argv.ipaddress
+        
+        @radd = Rudy::Addresses.new(:config => @config, :global => @global)
+        @rmach = Rudy::Instances.new(:config => @config, :global => @global)
+        
+        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
+        puts "Destroy Address".bright, $/
+        
+        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)
+        exit unless Annoy.are_you_sure?(:low)
+        
+        ret = @radd.destroy(@argv.ipaddress)
+        raise "Destroy failed" unless ret
+        
+        puts
+        puts "Done"
+      end
+        
+    end
+  end
+end
+

data/lib/rudy/cli/backups.rb

@@ -0,0 +1,175 @@
+
+
+
+
+module Rudy
+  module CLI
+    class Backups < Rudy::CLI::Base
+      
+      
+      def backup
+        criteria = [@global.zone]
+        criteria += [@global.environment, @global.role] unless @option.all
+        
+        Rudy::MetaData::Backup.list(@sdb, *criteria).each do |backup|
+          puts "%s (%s)" % [backup.name, backup.awsid]
+        end
+      end
+      
+      # Check for backups pointing to snapshots that don't exist. 
+      def sync_backup
+        unless argv.empty?
+          puts "The disk you specified will be ignored."
+          argv.clear
+        end
+        
+        criteria = [@global.zone]
+        criteria += [@global.environment, @global.role] unless @option.all
+        
+        puts "Looking for backup metadata with delinquent snapshots..."
+        to_be_deleted = {} # snap-id => backup
+        Rudy::MetaData::Backup.list(@sdb, *criteria).each do |backup|
+          to_be_deleted[backup.awsid] = backup unless @ec2.snapshots.exists?(backup.awsid)
+        end
+        
+        if to_be_deleted.empty?
+          puts "All backups are in-sync with snapshots. Nothing to do."
+          return
+        end
+        
+        puts
+        puts "These backup metadata will be deleted:"
+        to_be_deleted.each do |snap_id, backup|
+          puts "%s: %s" % [snap_id, backup.name]
+        end
+        
+        puts
+        exit unless Annoy.are_you_sure?
+        
+        puts 
+        puts "Deleting..."
+        to_be_deleted.each do |snap_id, backup|
+          print " -> #{backup.name}... "
+          @sdb.destroy(RUDY_DOMAIN, backup.name)
+          puts "done"
+        end
+        
+        puts "Done!"
+      end
+      
+      def destroy_backup_valid?
+        raise "No backup specified" if argv.empty?
+        exit unless Annoy.are_you_sure?(:high)
+        true
+      end
+      
+      def destroy_backup
+        name = @argv.first
+        puts "Destroying #{name}"
+        begin
+          backup = Rudy::MetaData::Backup.get(@sdb, name)
+        rescue => ex
+          puts "Error deleteing backup: #{ex.message}"
+        end
+        
+        return unless backup
+        
+        begin
+          puts " -> deleting snapshot..."
+          @ec2.snapshots.destroy(backup.awsid)
+        rescue => ex
+          puts "Error deleting snapshot: #{ex.message}."
+          puts "Continuing..."
+        ensure
+          puts " -> deleting metadata..."
+          @sdb.destroy(RUDY_DOMAIN, name)
+        end
+        puts "Done."
+      end
+      
+      def create_backup
+        diskname = @argv.first
+        
+        machine = find_current_machine
+        
+        disks = Rudy::MetaData::Disk.list(@sdb, machine[:aws_availability_zone], @global.environment, @global.role, @global.position)
+        raise "The machine #{machine_name} does not have any disk metadata" if disks.empty?
+        
+        puts "Machine: #{machine_name}"
+        
+        if @option.snapshot
+          raise "You must supply a diskname when using an existing snapshot" unless diskname
+          raise "The snapshot #{@option.snapshot} does not exist" unless @ec2.snapshots.exists?(@option.snapshot)
+          disk = Rudy::MetaData::Disk.get(@sdb, diskname)
+          
+          raise "The disk #{diskname} does not exist" unless disk
+          backup = Rudy::MetaData::Backup.new
+          backup.awsid = @option.snapshot
+          backup.time_stamp
+            
+          # Populate machine infos
+          [:zone, :environment, :role, :position].each do |n|
+            backup.send("#{n}=", @global.send(n)) if @global.send(n)
+          end
+        
+          # Populate disk infos
+          [:path, :size].each do |n|
+            backup.send("#{n}=", disk.send(n)) if disk.send(n)
+          end
+          
+          
+          Rudy::MetaData::Backup.save(@sdb, backup)
+          
+          puts backup.name
+          
+        else
+          volumes = @ec2.instances.volumes(machine[:aws_instance_id])
+          raise "The machine #{machine_name} does not have any volumes attached." if volumes.empty?
+          
+          puts "#{disks.size} Disk(s) defined with #{volumes.size} Volume(s) running"
+        
+          volumes.each do |volume|
+            print "Volume #{volume[:aws_id]}... "
+            disk = Rudy::MetaData::Disk.find_from_volume(@sdb, volume[:aws_id])
+            backup = Rudy::MetaData::Backup.new
+          
+            # TODO: Look for the disk based on the machine
+            raise "No disk associated to volume #{volume[:aws_id]}" unless disk 
+          
+            backup.volume = volume[:aws_id]
+
+            # Populate machine infos
+            [:zone, :environment, :role, :position].each do |n|
+              backup.send("#{n}=", @global.send(n)) if @global.send(n)
+            end
+
+            # Populate disk infos
+            [:path, :size].each do |n|
+              backup.send("#{n}=", disk.send(n)) if disk.send(n)
+            end
+            
+            backup.time_stamp
+            
+            raise "There was a problem creating the backup metadata" unless backup.valid?
+            
+            snap = @ec2.snapshots.create(volume[:aws_id])
+          
+            if !snap || !snap.is_a?(Hash)
+              puts "There was an unknown problem creating #{backup.name}. Continuing with the next volume..."
+              next
+            end
+            
+            backup.awsid = snap[:aws_id]
+
+            Rudy::MetaData::Backup.save(@sdb, backup)
+
+            puts backup.name
+            
+          end
+        end          
+      end
+      
+      
+    end
+  end
+end

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

@@ -1,10 +1,10 @@
 
 
 module Rudy
-  module Command
-    class Config < Rudy::Command::Base
+  module CLI
+    class Config < Rudy::CLI::Base
       
-      # We force the Command::Base#print_header to be quiet
+      # We force the CLI::Base#print_header to be quiet
       def print_header
         @global.quiet = true
         super
@@ -32,11 +32,9 @@ module Rudy
       #
       def config
         return if @config.nil? 
-        puts "Config: #{@config.paths}" if @global.verbose > 0
         
-        which = @option.defaults ? @global.user : machine_name
-        puts "Machine: #{which}" if @global.verbose > 0
-        puts "User: #{@global.user}" if @global.verbose > 0
+        
+        @option.group ||= [@global.environment, @global.role].join(RUDY_DELIM)
         
         return if @config.empty?
         
@@ -47,15 +45,22 @@ module Rudy
           
           
         else
-          
           if @option.all
-            y @config.machines.to_hash
-            y @config.routines.to_hash
-          elsif @option.defaults
+            puts "# ACCOUNTS: not displayed"
+            puts "# MACHINES: "
+            if @config.machines?
+              y @config.machines.to_hash 
+            end
+            if @config.routines?
+              puts "# ROUTINES: "
+              y @config.routines.to_hash 
+            end
+          elsif @option.defaults?
             y @config.defaults.to_hash
           else
-            env, rol, usr, att = @global.environment, @global.role, @global.user, @argv.name
-            val = @config.machines.find_deferred(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