rudy 0.2.4 → 0.3.0

data/lib/console.rb

@@ -0,0 +1,341 @@
+
+# Adapted from: http://github.com/oneup/ruby-console/tree/tput
+
+if __FILE__ == $0
+  puts "This is a module, don't run it from the command line."
+  exit
+end
+
+class String
+  def color(*arg)                         # colorize a string
+    Console.color(arg) +
+    self +
+    Console.color(:default)
+  end
+  def printAt(*arg)
+    row = 0
+    col = 0
+    if not arg.is_a?(Array)
+      arg = [arg]
+    end
+    if arg.length > 0
+      row = arg[0]
+    end
+    if arg.length > 1
+      col = arg[1]
+    end
+    Cursor.save
+    Cursor.position = row, col
+    print self
+    Cursor.restore
+  end
+end
+
+class Console
+  def self.clear
+    system "tput clear"
+  end
+
+  def self.reset
+    system "tput reset"
+  end
+  
+  def self.color(*arg)                         # colorize a string
+    if arg[0].is_a?(Array)
+      arg = arg[0]
+    end
+    if arg.length == 0
+      arg = :default, :red, :bg_default
+    end
+    attribute = {         # mapper for the attributes
+      :normal     => 0,
+      :bright     => 1,
+      :dim        => 2,
+      :underscore => 4,
+      :blink      => 5,
+      :reverse    => 7,
+      :hidden     => 8,
+      :default    => 0
+    }
+    fg_color = {          # mapper for the foreground color
+      :black   => 30,
+      :red     => 31,
+      :green   => 32,
+      :yellow  => 33,
+      :blue    => 34,
+      :magenta => 35,
+      :cyan    => 36,
+      :white   => 37,
+      :default => 39
+    }
+    bg_color = {          # mapper for the background color
+      :bg_black   => 40,
+      :bg_red     => 41,
+      :bg_green   => 42,
+      :bg_yellow  => 43,
+      :bg_blue    => 44,
+      :bg_magenta => 45,
+      :bg_cyan    => 46,
+      :bg_white   => 47,
+      :bg_default => 49
+    }
+    if arg.length > 0                 # turn symbols into numbers
+      arg[0] = attribute[arg[0]]      # attributes
+    end
+    if arg.length > 1
+      arg[1] = fg_color[arg[1]]       # foreground color
+    end
+    if arg.length > 2
+      arg[2] = bg_color[arg[2]]       # background color
+    end
+    "\e[#{arg.join(";")}m"   # magic ansi escape sequence
+  end
+
+  def self.color=(arg)
+    if not arg.is_a?(Array)
+      arg = [arg]
+    end
+    if arg.length == 0
+      arg = :default, :red, :bg_default
+    end
+    print self.color(arg)
+  end
+
+  def self.width
+    `tput cols`.chomp.to_i
+  end
+
+  def self.height
+    `tput lines`.chomp.to_i
+  end
+end
+
+class Cursor
+  def self.position
+    row = ""
+    col = ""
+    c = ""
+
+    termsettings = `stty -g`
+    system("stty raw -echo")
+    print "\e[6n"
+    while (c = STDIN.getc.chr) != ";"
+      if c == "\e" or c == "["
+        next
+      else
+        row += c
+      end
+    end
+    while (c = STDIN.getc.chr) != "R"
+      col += c
+    end
+    system("stty #{termsettings}")
+
+    [row, col]
+  end
+
+  def self.position=(arg)
+    row = 0
+    col = 0
+    if not arg.is_a?(Array)
+      arg = [arg]
+    end
+    if arg.length > 0
+      row = arg[0]
+    end
+    if arg.length > 1
+      col = arg[1]
+    end
+    system "tput cup #{row} #{col}"
+  end
+
+  def self.up(*arg)
+    if arg.length == 0
+      arg[0] == 1
+    end
+    print "\e[#{arg[0]}A"
+  end
+
+  def self.down(*arg)
+    if arg.length == 0
+      arg[0] == 1
+    end
+    print "\e[#{arg[0]}B"
+  end
+
+  def self.right(*arg)
+    if arg.length == 0
+      arg[0] == 1
+    end
+    system "tput cuf #{arg[0]}"
+  end
+
+  def self.left(*arg)
+    if arg.length == 0
+      arg[0] == 1
+    end
+    system "tput cub #{arg[0]}"
+  end
+
+  def self.save
+    system "tput sc"
+  end
+
+  def self.restore
+    system "tput rc"
+  end
+end
+
+class Win
+  def initialize
+    @row = 1
+    @col = 1
+    @width = 10
+    @height = 5
+    @text = ""
+    @border = :single
+    @fg = :default
+    @bg = :bg_default
+    @bordercolor = :default
+  end
+
+  def width=(width)
+    @width = width
+  end
+
+  def width
+    @width
+  end
+
+  def height=(height)
+    @height = height
+  end
+
+  def height
+    @height
+  end
+
+  def text=(text)
+    @text = text
+  end
+
+  def text
+    @text
+  end
+
+  def border=(arg)
+    if not arg.is_a?(Array)
+      arg = [arg]
+    end
+    if arg.length > 0
+      @border = arg[0]
+    end
+    if arg.length > 1
+      @bordercolor = arg[1]
+    end
+  end
+
+  def border
+    [@border, @bordercolor]
+  end
+
+  def position=(arg)
+    if not arg.is_a?(Array)
+      arg = [arg]
+    end
+    if arg.length > 0
+      @row = arg[0]
+    end
+    if arg.length > 1
+      @col = arg[1]
+    end
+  end
+
+  def position
+    [@row, @col]
+  end
+
+  def fg=(fg)
+    @fg = fg
+  end
+
+  def fg
+    @fg
+  end
+
+  def bg=(bg)
+    @bg = bg
+  end
+
+  def bg
+    @bg
+  end
+
+  def drawBorder
+    single = {
+      :upper_left  => "\u250C",
+      :upper_right => "\u2510",
+      :lower_left  => "\u2514",
+      :lower_right => "\u2518",
+      :horizontal  => "\u2500",
+      :vertical    => "\u2502"
+    }
+    bold = {
+      :upper_left  => "\u250D",
+      :upper_right => "\u2511",
+      :lower_left  => "\u2515",
+      :lower_right => "\u2519",
+      :horizontal  => "\u2501",
+      :vertical    => "\u2503"
+    }
+    double = {
+      :upper_left  => "\u2554",
+      :upper_right => "\u2557",
+      :lower_left  => "\u255A",
+      :lower_right => "\u255D",
+      :horizontal  => "\u2550",
+      :vertical    => "\u2551"
+    }
+    round = {
+      :upper_left  => "\u256D",
+      :upper_right => "\u256E",
+      :lower_left  => "\u2570",
+      :lower_right => "\u256F",
+      :horizontal  => "\u2500",
+      :vertical    => "\u2502"
+    }
+    mapper = case @border
+      when :single then single
+      when :bold   then bold
+      when :double then double
+      when :round  then round
+      else single
+    end
+    (mapper[:upper_left] + mapper[:horizontal] * (@width - 2) + mapper[:upper_right]).color(:normal, @bordercolor, @bg).printAt @row, @col
+    (mapper[:lower_left] + mapper[:horizontal] * (@width - 2) + mapper[:lower_right]).color(:normal, @bordercolor, @bg).printAt @row + @height - 1, @col
+    0.upto(@height - 3) do |i|
+      (mapper[:vertical] + " " * (@width - 2) + mapper[:vertical]).color(:normal, @bordercolor, @bg).printAt @row + i + 1, @col
+    end
+  end
+
+  def draw
+    text = @text.split("\n")
+    0.upto(text.length - 1) do |i|
+      if @border != :none
+        text[i] = text[i][0, @width - 2]
+      else
+        text[i] = text[i][0, @width]
+      end
+    end
+    if @border == :none
+      0.upto((text.length > @height ? @height : text.length) - 1) do |i|
+        text[i].color(:normal, @fg, @bg).printAt @row + i , @col
+      end
+    else
+      drawBorder
+      0.upto(@text.length > @height - 3 ? @height - 3 : text.length) do |i|
+        text[i].to_s.color(:normal, @fg, @bg).printAt @row + i + 1, @col + 1
+      end
+    end
+  end
+end

data/lib/rudy/aws/ec2.rb

@@ -14,10 +14,46 @@ module Rudy::AWS
         @aws.describe_images_by_owner('self') || []
       end
       
+      # +id+ AMI ID to deregister (ami-XXXXXXX)
+      # Returns true when successful. Otherwise throws an exception.
+      def deregister(id)
+        @aws.deregister_image(id)
+      end
+      
+      # +path+ the S3 path to the manifest (bucket/file.manifest.xml)
+      # Returns the AMI ID when successful, otherwise throws an exception.
+      def register(path)
+        @aws.register_image(path)
+      end
     end
+    class Snapshots
+      include Rudy::AWS::ObjectBase
+      
+      def list
+        @aws.describe_snapshots || []
+      end
+      
+      def create(vol_id)
+        @aws.create_snapshot(vol_id)
+      end
+      
+      def destroy(snap_id)
+        @aws.delete_snapshot(snap_id)
+      end
+      
+      def exists?(id)
+        list.each do |v|
+          return true if v[:aws_id] === id
+        end
+        false
+      end
+      
+    end
+    
     class Volumes
       include Rudy::AWS::ObjectBase
       
+      
       def list
         list = @aws.describe_volumes() || []
         list.select { |v| v[:aws_status] != "deleting" }
@@ -27,10 +63,18 @@ module Rudy::AWS
         @aws.attach_volume(vol_id, inst_id, device)
       end
       
+      def detach(vol_id)
+        @aws.detach_volume(vol_id)
+      end
+      
       def create(zone, size, snapshot=nil)
         @aws.create_volume(snapshot, size, zone)
       end
       
+      def destroy(vol_id)
+        @aws.delete_volume(vol_id)
+      end
+      
       def exists?(id)
         list.each do |v|
           return true if v[:aws_id] === id
@@ -38,6 +82,7 @@ module Rudy::AWS
         false
       end
       
+      
     end
     
     class Instances

data/lib/rudy/aws/simpledb.rb

@@ -41,6 +41,11 @@ module Rudy::AWS
       items
     end
     
+    def select(query)
+      list = @aws2.select(query) || []
+      list[0]
+    end
+    
     def get_attributes(domain, item, attribute=nil)
       @aws.get_attributes(domain, item, attribute)
     end

data/lib/rudy/aws.rb

@@ -20,6 +20,7 @@ module Rudy
       attr_reader :addresses
       attr_reader :groups
       attr_reader :volumes
+      attr_reader :snapshots
       attr_reader :aws
 
       def initialize(access_key, secret_key)
@@ -28,6 +29,7 @@ module Rudy
         @images = Rudy::AWS::EC2::Images.new(@aws)
         @groups = Rudy::AWS::EC2::Groups.new(@aws)
         @addresses = Rudy::AWS::EC2::Addresses.new(@aws)
+        @snapshots = Rudy::AWS::EC2::Snapshots.new(@aws)
         @volumes = Rudy::AWS::EC2::Volumes.new(@aws)
       end
     
@@ -51,6 +53,7 @@ module Rudy
     
       def initialize(access_key, secret_key)
         @aws = RightAws::SdbInterface.new(access_key, secret_key, {:logger => Logger.new(@@logger)})
+        @aws2 = AwsSdb::Service.new(:access_key_id => access_key, :secret_access_key => secret_key, :logger => Logger.new(@@logger))
         @domains = Rudy::AWS::SimpleDB::Domains.new(@aws)
       end
 

data/lib/rudy/command/base.rb

@@ -28,7 +28,7 @@ module Rudy
       attr_reader :scm
       
       attr_reader :rscripts
-      
+      attr_reader :domains
       attr_reader :machine_images
       
       def init
@@ -81,8 +81,9 @@ module Rudy
           end
         end
         
-        @sdb = Rudy::AWS::SimpleDB.new(@access_key, @secret_key)
         @ec2 = Rudy::AWS::EC2.new(@access_key, @secret_key)
+        # RightAws::SdbInterface is on thin ice. No select support! Or sort!
+        @sdb = Rudy::AWS::SimpleDB.new(@access_key, @secret_key)
         #@s3 = Rudy::AWS::SimpleDB.new(@access_key, @secret_key)
       end
       
@@ -90,7 +91,7 @@ module Rudy
       # not have a valid keypair configured. (See: EC2_KEYPAIR_*)
       def check_keys
         raise "No SSH key provided for #{keypairname}!" unless has_keypair?(keypairname)
-        raise "SSH key provided but cannot be found! (#{keypairpath})" unless File.exists?(keypairpath)
+        raise "SSH key provided but cannot be found! (#{keypairname}: #{keypairpath})" unless File.exists?(keypairpath)
       end  
       
       def has_pem_keys?
@@ -157,10 +158,8 @@ module Rudy
           raise "#{disk.device} is already in use on #{machine[:aws_instance_id]}! (Try umounting it)"
         end
         
-        new_volume = false
         if !disk.awsid || (disk.awsid && !@ec2.volumes.exists?(disk.awsid))
           disk = Rudy::MetaData::Disk.update_volume(@sdb, @ec2, disk, machine)
-          new_volume = true
         end
         
         raise "Unknown error creating volume! #{disk.awsid}" unless disk && disk.awsid
@@ -171,11 +170,16 @@ module Rudy
         
         @user = "root"
         
-        if new_volume
+        if disk.raw_volume
           puts "Creating the filesystem (mkfs.ext3 -F #{disk.device})"
           capture(:stdout) do
             ssh machine[:dns_name], keypairpath, user, "mkfs.ext3 -F #{disk.device}"
           end
+          
+          puts "Saving disk metadata"
+          disk.raw_volume = false
+          Rudy::MetaData::Disk.save(@sdb, disk)
+          
           sleep 2
         end
         
@@ -193,26 +197,25 @@ module Rudy
       # +cmd+ is the name of the command current running. 
       def print_header(cmd=nil)
         title = "RUDY v#{Rudy::VERSION}"
-        title << " -- #{@alias}" if @alias
-        puts title, $/
-
+        now_utc = Time.now.utc
+        criteria = []
+        [:zone, :environment, :role, :position].each do |n|
+          val = instance_variable_get("@#{n}")
+          criteria << "#{n.to_s.color :normal}:#{val.color :bright}"
+        end
+        puts '%s -- %s' % [title, criteria.join(", ")]
+        puts 'UTC: %s' % now_utc.strftime("%Y-%m-%d %H:%M:%S") 
+        puts
+        
         if (@environment == "prod") 
-          puts %q(=======================================================
+          msg = %q(=======================================================
 =======================================================
 !!!!!!!!!   YOU ARE PLAYING WITH PRODUCTION   !!!!!!!!!
 =======================================================
-=======================================================
-
-)
-        end
-
-        criteria = []
-        [:zone, :environment, :role, :position].each do |n|
-          val = instance_variable_get("@#{n}")
-          criteria << "[#{n} = #{val}]"
+=======================================================)
+          puts msg.color(:bright, :red, :bg_white)
+          puts
         end
-        puts criteria.join(" and ")
-        puts 
         
       end
       
@@ -220,6 +223,16 @@ module Rudy
         
       end
       
+      # Returns a hash of info for the requested machine. If the requested machine
+      # is not running, it will raise an exception. 
+      def find_current_machine
+        machine_list = @ec2.instances.list(machine_group)
+        machine = machine_list.values.first  # NOTE: Only one machine per group, for now...
+        raise "There's no machine running in #{machine_group}" unless machine
+        raise "The primary machine in #{machine_group} is not in a running state" unless machine[:aws_state] == 'running'
+
+        machine
+      end
 
       
       def group_metadata(env=@environment, role=@role)
@@ -242,17 +255,28 @@ module Rudy
         
         def print_image(img)
           puts '-'*60
-          puts "Image: #{img[:aws_location]}"
+          puts "Image: #{img[:aws_id]}"
           img.each_pair do |key, value|
              printf(" %22s: %s#{$/}", key, value) if value
            end
           puts
         end
         
-        def print_disk(disk)
+        def print_disk(disk, backups=[])
           puts '-'*60
           puts "Disk: #{disk.name}"
           puts disk.to_s
+          puts "#{backups.size} most recent backups:", backups.collect { |back| "#{back.nice_time} (#{back.awsid})" }
+          puts
+        end
+        
+        
+        def print_volume(vol, disk)
+          puts '-'*60
+          puts "Volume: #{vol[:aws_id]} (disk: #{disk.name})"
+          vol.each_pair do |key, value|
+             printf(" %22s: %s#{$/}", key, value) if value
+           end
           puts
         end