rudy 0.9.1 → 0.9.2

data/examples/solaris.rb

@@ -0,0 +1,35 @@
+# Rudy Solaris Machines
+#
+# This configuration is used to 
+# test solaris instance support.
+
+
+defaults do
+  color true
+  environment :test
+  role :solaris
+end
+
+machines do
+  region :'us-east-1' do
+    ami 'ami-8f30d1e6'               # OpenSolaris 2009.06 32-bit  (US)
+  end
+  region :'eu-west-1' do
+    ami 'ami-2381a957'               # OpenSolaris 2009.06 32-bit  (EU)
+  end
+  env :test do
+    role :solaris do
+      user :root
+    end
+  end
+end
+
+routines do
+  
+  uname do
+    remote :root do
+      uname :a
+    end
+  end
+  
+end

data/examples/windows.rb

@@ -0,0 +1,101 @@
+# Rudy Windows Machines
+#
+# This configuration is used to 
+# test windows instance support.
+
+
+defaults do
+  color true
+  environment :test
+  role :windows
+  zone :'eu-west-1b'
+  bucket 'rudy-ami-eu'
+end
+
+machines do
+  region :'us-east-1' do
+    ami 'ami-de4daab7'               # Amazon Windows Server 2003 (US)
+    size 'm1.small'
+  end
+  region :'eu-west-1' do
+    ami 'ami-8696bef2'               # Rudy Windows 2009-08-24 (EU)
+  end
+  env :test do
+    role :windows do
+      os :windows
+      user :Administrator
+      disks do
+        path "F:" do
+          size 1
+          device 'xvdf'
+          fstype 'ntfs'
+        end
+        path "E:" do
+          size 2
+          device 'xvde'
+        end
+        path 'L:' do
+          size 3
+          device 'xvdl'
+        end
+      end
+    end
+  end
+end
+
+
+commands do
+  allow :rm
+end
+
+routines do
+  
+  create_disks do
+    disks do
+      create "L:"
+      #create "F:"
+      #create "E:"
+    end
+    
+  end
+  
+  shutdown do
+    disks do
+      destroy "L:"
+      #destroy "F:"
+      #destroy "E:"
+    end
+      
+  end
+  
+  
+  upload_config do
+    remote :root do
+      puts "Uploading rudy config"
+      home = guess_user_home
+      mkdir :p, "#{home}/.rudy", "#{home}/.ssh", ".ssh"
+      disable_safe_mode
+      file_upload '~/.rudy/config', "#{home}/.rudy/config"
+      puts "Uploading keypair"
+      file_upload '~/.ssh/id_rsa', '~/.ssh/key-eu-west-1b-test-windows', "#{home}/.ssh/"
+      file_upload '~/.ssh/id_rsa', '~/.ssh/key-eu-west-1b-test-windows', ".ssh/"
+    end
+  end
+  
+end
+
+
+__END__
+
+* diskpart script example
+  * http://social.technet.microsoft.com/Forums/en-US/winserversetup/thread/2cfbaae1-6e33-4197-bb71-63434a34eb3c
+  * http://technet.microsoft.com/en-us/library/cc766465(WS.10).aspx
+  
+* format docs
+  * http://www.computerhope.com/formathl.htm
+
+* Securing Remote Desktop with copSSH
+  * http://www.teamhackaday.com/2008/04/23/securing-windows-remote-desktop-with-copssh/
+  
+* Windows SSH
+  * http://www.windowsnetworking.com/articles_tutorials/install-SSH-Server-Windows-Server-2008.html

data/lib/rudy.rb

@@ -42,7 +42,7 @@ module Rudy
     unless defined?(MAJOR)
       MAJOR = 0.freeze
       MINOR = 9.freeze
-      TINY  = 1.freeze
+      TINY  = 2.freeze
     end
     def self.to_s; [MAJOR, MINOR, TINY].join('.'); end
     def self.to_f; self.to_s.to_f; end
@@ -72,6 +72,12 @@ module Rudy
     DEFAULT_EC2_HOST = "ec2.amazonaws.com"
     DEFAULT_EC2_PORT = 443
     
+    DEFAULT_WINDOWS_FS = 'ntfs'
+    DEFAULT_LINUX_FS = 'ext3'
+    
+    DEFAULT_WINDOWS_DEVICE = 'xvdf'
+    DEFAULT_LINUX_DEVICE = '/dev/sdh'
+    
     MAX_INSTANCES = 5.freeze
     
     ID_MAP = {

data/lib/rudy/aws.rb

@@ -1,6 +1,6 @@
 
 
-require 'EC2'       # amazon-ec2 gem
+require 'AWS'       # amazon-ec2 gem
 
 
 module Rudy
@@ -27,7 +27,7 @@ module Rudy
     
     Rudy::Utils.require_glob(RUDY_LIB, 'rudy', 'aws', '{ec2,s3,sdb}', "*.rb")
     
-    class Error < ::EC2::Error; end
+    class Error < ::AWS::Error; end
   end
   
 end

data/lib/rudy/aws/ec2.rb

@@ -3,6 +3,8 @@ module Rudy; module AWS
   module EC2
 #    include Rudy::Huxtable
     
+    @@mutex = Mutex.new 
+    
     def self.connect(access_key=nil, secret_key=nil, region=nil, logger=nil)
 
       if region
@@ -15,7 +17,7 @@ module Rudy; module AWS
       host ||= DEFAULT_EC2_HOST
       port ||= DEFAULT_EC2_PORT
       
-      @@ec2 = ::EC2::Base.new(:port => port, :server=> host, :access_key_id => access_key, :secret_access_key => secret_key)
+      @@ec2 = ::AWS::EC2::Base.new(:port => port, :server=> host, :access_key_id => access_key, :secret_access_key => secret_key)
       @@logger = logger
     end
     
@@ -28,31 +30,36 @@ module Rudy; module AWS
     # Returns the return value from the request is returned untouched
     # or the default value on error or if the request returned nil. 
     def self.execute_request(default=nil, timeout=nil, &request)
-      timeout ||= 15
+      timeout ||= 30
       raise "No block provided" unless request
       response = nil
-      begin
-        Timeout::timeout(timeout) do
-          response = request.call
-        end
-      # Raise the EC2 exceptions
-      rescue ::EC2::Error, ::EC2::InvalidInstanceIDMalformed => ex  
-        raise Rudy::AWS::Error, ex.message
+      @@mutex.synchronize {
+        begin
+        
+          Timeout::timeout(timeout) do
+            response = request.call
+          end
       
-      # NOTE: The InternalError is returned for non-existent volume IDs. 
-      # It's probably a bug so we're ignoring it -- Dave. 
-      rescue ::EC2::InternalError => ex
-        raise Rudy::AWS::Error, ex.message
+        # Raise the EC2 exceptions
+        rescue ::AWS::Error, ::AWS::InvalidInstanceIDMalformed => ex  
+          raise Rudy::AWS::Error, ex.message
+      
+        # NOTE: The InternalError is returned for non-existent volume IDs. 
+        # It's probably a bug so we're ignoring it -- Dave. 
+        rescue ::AWS::InternalError => ex
+          raise Rudy::AWS::Error, ex.message
         
-      rescue Timeout::Error => ex
-        STDERR.puts "Timeout (#{timeout}): #{ex.message}!"
-      rescue SocketError => ex
-        #STDERR.puts ex.message
-        #STDERR.puts ex.backtrace
-        raise SocketError, "Check your Internets!" unless @@global.offline
-      ensure
-        response ||= default
-      end
+        rescue Timeout::Error => ex
+          STDERR.puts "Timeout (#{timeout}): #{ex.message}!"
+        rescue SocketError => ex
+          #STDERR.puts ex.message
+          #STDERR.puts ex.backtrace
+          raise SocketError, "Check your Internets!" unless @@global.offline
+        ensure
+          response ||= default
+        end
+        sleep 0.1  # defeat race conditions
+      }
       response
     end
     

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

@@ -171,7 +171,7 @@ module Rudy::AWS
       def exists?(name)
         begin
           g = list([name.to_s])
-        rescue ::EC2::InvalidGroupNotFound
+        rescue ::AWS::InvalidGroupNotFound
           return false 
         end
       

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

@@ -13,37 +13,9 @@ module Rudy::AWS
     field :visibility
     field :location
     field :kind
-    
-    def liner_note
-      info = @location
-      # Highlight "debian-5" in /dir/debian-5.manifest.xml
-      #info = info.split(/\//) 
-      #info[-1].gsub!(/(.+?)((\.img)?\.manifest\.xml)/) { |m,n| ($1 || "").bright << $2 }
-      #info = info.join('/')
-      "%s %-6s (%s)" % [self.awsid.bright, self.arch, info] 
-    end
         
     def to_s(with_title=false)
-      lines = []
-      lines << liner_note
-      #if self.available?
-      #  p = public? ? "public" : "private" 
-      #  k, r = @aki || 'aki-unknown', @ari || 'ari-unknown'
-      #  lines << @@sformat % %w{arch owner aki ari visibility} if with_title
-      #  lines << @@sformat % [@arch, @owner, k, r, p]
-      #end
-      lines.join($/)
-    end
-    
-    def inspect
-      lines = []
-      lines << liner_note
-      field_names.each do |key|
-        next unless self.respond_to?(key)
-        val = self.send(key)
-        lines << sprintf(" %22s: %s", key, (val.is_a?(Array) ? val.join(', ') : val))
-      end
-      lines.join($/)
+      [@awsid.bright, @arch, @visibility, @location].join '; '
     end
     
     def available?; @state && @state == "available"; end

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

@@ -23,38 +23,9 @@ module Rudy::AWS
       @groups ||= []
     end
     
-    def liner_note
-      info = self.running? ? self.dns_public : self.state
-      "%s  %-30s" % [self.awsid.bright, info]
-    end
-
-    def to_s(with_title=false)
-      lines = []
-      if @groups
-        gpstr = [@groups].flatten.compact.join(', ')
-        gpstr &&= "(#{gpstr})"
-      else
-        gpstr = ''
-      end
-      
-      lines << "%s  %s" % [liner_note, gpstr]
-      #if self.running?
-      #  k, g = @keyname || 'no-keypair', self.groups.join(', ')
-      #  lines << @@sformat % %w{zone size ami keyname groups} if with_title
-      #  lines << @@sformat % [@zone, @size, @ami, k, g]
-      #end
-      lines.join($/)
-    end
-    
-    def pretty
-      lines = []
-      lines << liner_note
-      field_names.each do |key|
-        next unless self.respond_to?(key)
-        val = self.send(key)
-        lines << sprintf(" %22s: %s", key, (val.is_a?(Array) ? val.join(', ') : val))
-      end
-      lines.join($/)
+    def to_s(*args)
+      groups = [@groups].flatten.compact.join(', ')
+      [self.awsid.bright, self.state, self.dns_public, groups].join '; '
     end
     
     def running?; self.state && self.state == 'running'; end
@@ -307,6 +278,7 @@ module Rudy::AWS
       # +inst_id+ is an instance ID
       # Returns an Instance object
       def get(inst_id)
+        return nil if inst_id.nil?
         inst_id = inst_id.awsid if inst_id.is_a?(Rudy::AWS::EC2::Instance)
         inst = list(:any, inst_id) 
         inst &&= inst.first

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

@@ -8,13 +8,8 @@ module Rudy::AWS
       field :fingerprint
       field :private_key  
       
-      def liner_note
-        "%-20s   %s" % [self.name.bright, self.fingerprint]
-      end
-      
       def to_s(titles=false)
-        str = titles ? "%-20s   %s#{$/}" % ['name', 'fingerprint'] : ""
-        str << liner_note
+        [@name.bright, @fingerprint].join '; '
       end
       
       def public_key

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

@@ -3,7 +3,6 @@
 module Rudy::AWS
   module EC2
     class Snapshot < Storable
-      @@sformat = "%s <- %10s; %s"
       
       field :awsid
       field :progress
@@ -11,25 +10,8 @@ module Rudy::AWS
       field :volid
       field :status
       
-      def liner_note
-        t = Time.parse(@created)
-        info = t.strftime("%Y-%m-%d %H:%M:%S")
-        "%s (%s)" % [(self.awsid || '').bright, info]
-      end
-      
       def to_s(with_title=false)
-        @@sformat % [liner_note,  @volid, @status]
-      end
-      
-      def pretty
-        lines = []
-        lines << liner_note
-        field_names.each do |key|
-          next unless self.respond_to?(key)
-          val = self.send(key)
-          lines << sprintf(" %22s: %s", key, (val.is_a?(Array) ? val.join(', ') : val))
-        end
-        lines.join($/)
+        [@awsid.bright, @volid, @created, @status, @progress].join '; '
       end
       
       def completed?
@@ -98,7 +80,7 @@ module Rudy::AWS
       end
       
       def get(snap_id)
-        list(snap_id).first || nil
+        list(snap_id).first rescue nil
       end
       
       def exists?(snap_id)

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

@@ -20,23 +20,8 @@ module Rudy::AWS
         @size &&= @size.to_i
       end
       
-      def liner_note
-        info = attached? ? "attached to #{@instid}" : @status
-        "%s (%s)" % [(self.awsid || '').bright, info]
-      end
-      
-      def to_s(with_title=false)
-        line = @@sformat % [liner_note, @zone, @size, @device]
-        line << " <- %s" % [@snapid] if @snapid
-        line
-      end
-      
-      def pretty
-        lines = [liner_note]
-        field_names.each do |n|
-           lines << sprintf(" %12s:  %s", n, self.send(n)) if self.send(n)
-         end
-        lines.join($/)
+      def to_s(*args)
+        [@awsid.bright, @zone, @size, @created, @instid, @snapid].join '; '
       end
     
       # Alias for status
@@ -155,7 +140,14 @@ module Rudy::AWS
         volumes = nil if volumes.empty?
         volumes
       end
-    
+      
+      def list_by_instance(instid, &each_vol)
+        instid = instid.awsid if instid.is_a? Rudy::AWS::EC2::Instance
+        volumes = list(:attached, &each_vol)
+        volumes &&= volumes.select { |v| v.instid == instid }
+        volumes
+      end
+      
       def any?(state=nil,vol_id=[])
         vols = list(state, vol_id)
         !vols.nil?
@@ -180,7 +172,7 @@ module Rudy::AWS
           vol_id = Volumes.get_vol_id(vol_id)
           return false unless vol_id
           vol = get(vol_id)
-          (vol && vol.status == state)
+          (vol && vol.status == state) ? true : false
         end
       end