solutious-rudy 0.5.1 → 0.6.0

data/bin/rudy-s3

@@ -0,0 +1,76 @@
+#!/usr/bin/ruby
+
+# = Rudy S3
+# 
+# === A Rudy interface to Amazon Simple Storage Service
+#
+# 
+# 
+# See rudy-s3 -h for usage
+# 
+
+$:.unshift File.join(File.dirname(__FILE__), '..', 'lib') # Put our local lib in first place
+$:.unshift File.join(File.dirname(__FILE__), '..', 'vendor', 'highline-1.5.1', 'lib')
+#%w{amazon-ec2 caesars rye}.each { |dir| $:.unshift File.join(File.dirname(__FILE__), '..', '..', dir, 'lib') }
+#require 'rubygems'
+
+require 'rudy'
+require 'rudy/cli'
+require 'drydock'
+
+# Command-line interface for bin/rudy-s3
+module RudyCLI_S3
+  extend Drydock
+  
+  # ------------------------------------  AMAZON S3 COMMANDS  --------
+  # ------------------------------------------------------------------
+  
+  desc "Amazon S3 Buckets"
+  usage "rudy-s3 buckets"
+  option :l, :list, "List bucket objects"
+  action :C, :create, "Create a bucket"
+  action :D, :destroy, "Destroy a bucket"
+  argv :name
+  command :buckets => Rudy::CLI::AWS::S3::Buckets
+  command_alias :buckets, :b
+
+  #desc "Amazon S3 Storage Interface"
+  #usage "rudy-s3 store file-path"
+  #option :b, :bucket, String, "Bucket name"
+  #argv :path
+  #command :store => Rudy::CLI::AWS::S3::Store
+  #command_alias :store, :s
+
+  
+  
+  # ---------------------------------  RUDY-S3 MISCELLANEOUS  --------
+  # ------------------------------------------------------------------
+  
+  default :buckets
+  debug :off
+  
+  
+end
+
+
+#puts Rudy::Utils.banner("THIS IS RUBY #{RUBY_VERSION}") if Drydock.debug?
+
+# We call Drydock specifically otherwise it will run at_exit. Rye also
+# uses at_exit for shutting down the ssh-agent. Ruby executes at_exit
+# blocks in reverse order so if Drydock is required first, it's block
+# will run after Rye shuts down the ssh-agent. 
+begin
+  Drydock.run!(ARGV, STDIN) if Drydock.run? && !Drydock.has_run?
+rescue Drydock::ArgError, Drydock::OptError=> ex
+  STDERR.puts ex.message
+  STDERR.puts ex.usage
+rescue Rudy::Error => ex
+  STDERR.puts ex.message
+rescue => ex
+  STDERR.puts "ERROR (#{ex.class.to_s}): #{ex.message}"
+  STDERR.puts ex.backtrace if Drydock.debug?
+rescue SystemExit
+  # Don't balk
+end
+
+

data/bin/rudy-sdb

@@ -0,0 +1,67 @@
+#!/usr/bin/ruby
+
+# = Rudy SDB
+# 
+# === A Rudy interface to Amazon Simple Storage Service
+#
+# 
+# 
+# See rudy-sdb -h for usage
+# 
+
+$:.unshift File.join(File.dirname(__FILE__), '..', 'lib') # Put our local lib in first place
+$:.unshift File.join(File.dirname(__FILE__), '..', 'vendor', 'highline-1.5.1', 'lib')
+#%w{amazon-ec2 caesars rye}.each { |dir| $:.unshift File.join(File.dirname(__FILE__), '..', '..', dir, 'lib') }
+#require 'rubygems'
+
+require 'rudy'
+require 'rudy/cli'
+require 'drydock'
+
+# Command-line interface for bin/rudy-sdb
+module RudyCLI_SDB
+  extend Drydock
+  
+  # -----------------------------------  AMAZON SDB COMMANDS  --------
+  # ------------------------------------------------------------------
+  
+  
+  desc "Displays the SimpleDB domains associated to your account"
+  action :C, :create, "Create a domain"
+  action :D, :destroy, "Destroy a domain"
+  argv :name
+  command :domains => Rudy::CLI::AWS::SDB::Domains
+  
+  
+  # --------------------------------  RUDY-SDB MISCELLANEOUS  --------
+  # ------------------------------------------------------------------
+  
+  default :domains
+  debug :off
+  
+  
+end
+
+
+#puts Rudy::Utils.banner("THIS IS RUBY #{RUBY_VERSION}") if Drydock.debug?
+
+# We call Drydock specifically otherwise it will run at_exit. Rye also
+# uses at_exit for shutting down the ssh-agent. Ruby executes at_exit
+# blocks in reverse order so if Drydock is required first, it's block
+# will run after Rye shuts down the ssh-agent. 
+begin
+  Drydock.run!(ARGV, STDIN) if Drydock.run? && !Drydock.has_run?
+rescue Drydock::ArgError, Drydock::OptError=> ex
+  STDERR.puts ex.message
+  STDERR.puts ex.usage
+rescue Rudy::Error => ex
+  STDERR.puts ex.message
+rescue => ex
+  STDERR.puts "ERROR (#{ex.class.to_s}): #{ex.message}"
+  STDERR.puts ex.backtrace if Drydock.debug?
+rescue SystemExit
+  # Don't balk
+end
+
+
+

data/lib/annoy.rb

@@ -1,14 +1,17 @@
+#---
+# TODO: Use Matrix to give a more accurate annoyance factor
+# TODO: Add trivia questions
+#+++
 
 require 'timeout'
 require 'sysinfo'
+require 'highline'
 
-
-# Annoy - your annoying friend that asks you questions all the time.
+# = Annoy 
 #
-# TODO: Use Matrix to give a more accurate annoyance factor
-# TODO: Add trivia questions
+# Like your annoying friend that asks you questions all the time.
 #
-class Annoy #:nodoc:all
+class Annoy
   
   attr_accessor :factor
   attr_accessor :flavor
@@ -18,9 +21,9 @@ class Annoy #:nodoc:all
   attr_accessor :system
   
   @@operators = {
-    :low      => %w(+ - *),
-    :medium   => %w(* % + -),
-    :high     => %w(* % + -),
+    :low      => %w(+ -),
+    :medium   => %w(* -),
+    :high     => %w(& * -),
     :insane   => %w(** << | & *)
   }.freeze
   
@@ -33,15 +36,22 @@ class Annoy #:nodoc:all
   
   @@randsize = {
     :low      => 10,
-    :medium   => 100,
-    :high     => 1000,
+    :medium   => 14,
+    :high     => 50,
     :insane   => 1000
   }.freeze
   
   @@period = 60.freeze    # max seconds to wait
-  
   @@flavors = [:numeric, :string].freeze
+  @@skip = false          # skip questions
   
+  # Calling this method tells Annoy to not prompt for
+  # a response. All questions will return true. 
+  def Annoy.enable_skip; @@skip = true; end
+  # Tells annoy to prompt for a response. 
+  def Annoy.disable_skip; @@skip = false; end
+  # Returns true of Annoy is in skip mode
+  def Annoy.skip?; @@skip; end
   
   # * +factor+ annoyance factor, one of :low (default), :medium, :high, :insane
   # * +flavor+ annoyance flavor, one of :rand (default), :numeric, string
@@ -100,17 +110,36 @@ class Annoy #:nodoc:all
   # Returns true when the answer is correct, otherwise false.
   def Annoy.challenge?(msg="Please confirm.", factor=:medium, flavor=:rand, writer=STDOUT, period=nil)
     return true unless STDIN.tty? # Humans only!
+    return true if Annoy.skip?
     begin
       success = Timeout::timeout(period || @@period) do
         flavor = Annoy.get_flavor(flavor)
         question, answer = Annoy.question(factor, flavor)
-        writer.print "#{msg} To continue, #{Annoy.verb(flavor)} #{question}: "
-        writer.print "(#{answer}) " if ![:high, :insane].member?(factor) && flavor == :numeric
-        writer.flush
-        response = Annoy.get_response(writer)
-        response = response.to_i if flavor == :numeric
-        (response == answer)
+        msg = "#{msg} To continue, #{Annoy.verb(flavor)} #{question}: "
+        #writer.print msg
+        #if ![:medium, :high, :insane].member?(factor) && flavor == :numeric
+        #writer.print "(#{answer}) " 
+        #writer.flush
+        #end
+        #response = Annoy.get_response(writer)
+        
+        trap("SIGINT") { raise Annoy::GiveUp  }
+        
+        highline = HighLine.new 
+        response = highline.ask(msg) { |q| 
+          q.echo = '*'             # Don't display response
+          q.overwrite = true       # Erase the question afterwards
+          q.whitespace = :strip    # Remove whitespace from the response
+          q.answer_type = Integer  if flavor == :numeric
+        }
+
+        ret = (response == answer)
+        writer.puts "Incorrect" unless ret
+        ret
       end
+    rescue Annoy::GiveUp => ex
+      writer.puts $/, "Giving up!"
+      false
     rescue Timeout::Error => ex
       writer.puts $/, "Times up!"
       false
@@ -123,6 +152,12 @@ class Annoy #:nodoc:all
     Annoy.challenge?("Are you sure?", factor, flavor, writer)
   end
   
+  # Runs a challenge with the message, "Proceed?"
+  # See: Annoy.challenge?
+  def Annoy.proceed?(factor=:medium, flavor=:rand, writer=STDOUT)
+    Annoy.challenge?("Proceed?", factor, flavor, writer)
+  end
+  
   # See: Annoy.challenge?
   # Uses the value of @flavor, @factor, and @writer
   def challenge?(msg="Please confirm.")
@@ -143,6 +178,7 @@ class Annoy #:nodoc:all
   # * +regexp+ The regular expression to match the answer. 
   def Annoy.pose_question(msg, regexp, writer=STDOUT, period=nil)
     return true unless STDIN.tty? # Only ask a question if there's a human
+    return true if Annoy.skip?
     begin
       success = Timeout::timeout(period || @@period) do
         regexp &&= Regexp.new regexp
@@ -161,7 +197,9 @@ class Annoy #:nodoc:all
  private 
   def Annoy.get_response(writer=STDOUT)
     return true unless STDIN.tty? # Humans only
-    # TODO: Count the number of keystrokes to prevent copy/paste
+    return true if Annoy.skip?
+    # TODO: Count the number of keystrokes to prevent copy/paste.
+    # We can probably use Highline. 
     # We likely need to be more specific but this will do for now.
     #if ::SystemInfo.new.os == :unix 
     #  begin
@@ -223,5 +261,10 @@ class Annoy #:nodoc:all
   
 end
 
+# = Annoy::GiveUp
+#
+# This is what happens when you don't answer Annoy's questions. 
+class Annoy::GiveUp < RuntimeError
+end
 
 

data/lib/console.rb

@@ -4,7 +4,9 @@
 # See: man terminfo
 #+++
 
-
+#
+#
+#
 class String
   @@print_with_attributes = true
   def String.disable_colour; @@print_with_attributes = false; end
@@ -58,7 +60,8 @@ class String
     Console.print_at(self, args)
   end
   
-  # Returns the string with escape code attributes removed.
+  # Returns the string with ANSI escape codes removed.
+  #
   # NOTE: The non-printable attributes count towards the string size. 
   # You can use this method to get the "visible" size:
   #
@@ -66,8 +69,10 @@ class String
   #     "\e[34;47;5mMONKEY_JUNK\e[39;49;0m".size            # => 31
   #
   def noatt
-    gsub(/\e\[[\d;]*m/, '')
+    gsub(/\e\[?[0-9;]*[mc]?/, '')
   end
+  alias :noansi :noatt
+  
 end
 
 class Object #:nodoc:all
@@ -175,8 +180,8 @@ module Console #:nodoc:all
   end
   def print_at(str, props={})
     print_at_lamb = lambda {
-      #props[:x] ||= 0
-      #props[:y] ||= 0
+      props[:x] ||= 0
+      props[:y] ||= 0
       props[:minus] = false unless props.has_key?(:minus)
       props[:x] = props[:x]-str.noatt.size if props[:x] && props[:minus] # Subtract the str length from the position
       Cursor.save

data/lib/rudy/aws/ec2/address.rb

@@ -2,21 +2,86 @@
 
 module Rudy::AWS
   
-  class EC2
+  module EC2
     
     class Address < Storable
       field :ipaddress
       field :instid
-      def to_s
-        msg = "Address: #{self.ipaddress}"
-        msg << ", instance: #{self.instid}" if self.instid
-        msg
+      
+      def liner_note
+        info = self.associated? ? @instid : "available"
+        "%s (%s)" % [@ipaddress.to_s.bright, info]
+      end
+      
+      def to_s(with_titles=false)
+        liner_note
+      end
+      
+      def associated?
+        !@instid.nil? && !@instid.empty?
       end
     end
   
     class Addresses
       include Rudy::AWS::ObjectBase
+      include Rudy::AWS::EC2::Base
     
+      
+      def create
+        ret = @ec2.allocate_address
+        return false unless ret && ret['publicIp']
+        address = Rudy::AWS::EC2::Address.new
+        address.ipaddress = ret['publicIp']
+        address
+      end
+
+      def destroy(address)
+        address = address.ipaddress if address.is_a?(Rudy::AWS::EC2::Address)
+        raise UnknownAddress unless exists?(address)
+        
+        opts ={
+          :public_ip => address || raise("No public IP address supplied")
+        }
+        ret = @ec2.release_address(opts)
+        (ret && ret['return'] == 'true')
+      end
+
+
+      # Associate an elastic IP to an instance
+      def associate(address, instance)
+        raise NoInstanceID unless instance
+        raise NoAddress unless address
+        
+        address = address.ipaddress if address.is_a?(Rudy::AWS::EC2::Address)
+        instance = instance.awsid if instance.is_a?(Rudy::AWS::EC2::Instance)
+        raise UnknownAddress unless exists?(address)
+        raise AddressAssociated if associated?(address)
+        
+        opts ={
+          :instance_id => instance,
+          :public_ip => address
+        }
+        ret = @ec2.associate_address(opts)
+        (ret && ret['return'] == 'true')
+      end
+
+      # Disssociate an elastic IP from an instance
+      def disassociate(address)
+        raise NoAddress unless address
+        address = address.ipaddress if address.is_a?(Rudy::AWS::EC2::Address)
+        instance = instance.awsid if instance.is_a?(Rudy::AWS::EC2::Instance)
+        raise UnknownAddress unless exists?(address)
+        raise AddressNotAssociated unless associated?(address)
+        
+        opts ={
+          :public_ip => address
+        }
+        ret = @ec2.disassociate_address(opts)
+        (ret && ret['return'] == 'true')
+      end
+
+
+
       # Returns a Array of Rudy::AWS::EC2::Address objects. 
       def list(addresses=[])
         addresses = list_as_hash(addresses)
@@ -28,7 +93,7 @@ module Rudy::AWS
       def list_as_hash(addresses=[])
         addresses ||= []
         addresses = [addresses].flatten.compact
-        alist = @aws.describe_addresses(:addresses=> addresses)
+        alist = @ec2.describe_addresses(:addresses=> addresses)
         
         return nil unless alist['addressesSet'].is_a?(Hash)
       
@@ -43,7 +108,7 @@ module Rudy::AWS
     
     
       def any?
-        !(list_as_hash || {}).empty?
+        !list_as_hash.nil?
       end
       
       def get(address)
@@ -64,38 +129,6 @@ module Rudy::AWS
         address
       end
     
-      # Associate an elastic IP to an instance
-      def associate(instance, address)
-        address = address.ipaddress if address.is_a?(Rudy::AWS::EC2::Address)
-        instance = instance.awsid if instance.is_a?(Rudy::AWS::EC2::Instance)
-        raise "Not a valid address" unless exists?(address)
-        opts ={
-          :instance_id => instance || raise("No instance ID supplied"),
-          :public_ip => address || raise("No public IP address supplied")
-        }
-        ret = @aws.associate_address(opts)
-        (ret && ret['return'] == 'true')
-      end
-    
-    
-      def create
-        ret = @aws.allocate_address
-        return false unless ret && ret['publicIp']
-        address = Rudy::AWS::EC2::Address.new
-        address.ipaddress = ret['publicIp']
-        address
-      end
-    
-      def destroy(address)
-        address = address.ipaddress if address.is_a?(Rudy::AWS::EC2::Address)
-        raise "Not a valid address" unless exists?(address)
-        opts ={
-          :public_ip => address || raise("No public IP address supplied")
-        }
-        ret = @aws.release_address(opts)
-        (ret && ret['return'] == 'true')
-      end
-    
     
       # +address+ is an IP address or Rudy::AWS::EC2::Address object
       # Returns true if the given address is assigned to the current account
@@ -117,5 +150,8 @@ module Rudy::AWS
         false
       end
     end
+    
   end
-end
+end
+
+