sabat-rudy 0.8.0

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

@@ -0,0 +1,104 @@
+
+module Rudy::AWS
+  module EC2
+    
+    class KeyPair < Storable
+      
+      field :name
+      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
+      end
+      
+      def public_key
+        return unless @private_key
+        k = Rye::Key.new(@private_key)
+        k.public_key.to_ssh2
+      end
+
+    end
+    
+    class KeyPairs
+      include Rudy::AWS::ObjectBase
+      include Rudy::AWS::EC2::Base
+      
+      def create(name)
+        raise "No name provided" unless name
+        ret = @ec2.create_keypair(:key_name => name)
+        self.class.from_hash(ret)
+      end
+      
+      def destroy(name)
+        name = name.name if name.is_a?(Rudy::AWS::EC2::KeyPair)
+        raise "No name provided" unless name.is_a?(String)
+        ret = @ec2.delete_keypair(:key_name => name)
+        (ret && ret['return'] == 'true') # BUG? Always returns true
+      end
+      
+      def list(*names)
+        keypairs = list_as_hash(names)
+        keypairs &&= keypairs.values
+        keypairs
+      end
+      
+      def list_as_hash(*names)
+        names = names.flatten
+        klist = @ec2.describe_keypairs(:key_name => names)
+        return unless klist['keySet'].is_a?(Hash)
+        keypairs = {}
+        klist['keySet']['item'].each do |oldkp| 
+          kp = self.class.from_hash(oldkp)
+          keypairs[kp.name] = kp
+        end
+        keypairs
+      end
+      
+      def self.from_hash(h)
+        # keyName: test-c5g4v3pe
+        # keyFingerprint: 65:d0:ce:e7:6a:b0:88:4a:9c:c7:2d:b8:33:0c:fd:3b:c8:0f:0a:3c
+        # keyMaterial: |-
+        #   -----BEGIN RSA PRIVATE KEY-----
+        # 
+        keypair = Rudy::AWS::EC2::KeyPair.new
+        keypair.fingerprint = h['keyFingerprint']
+        keypair.name = h['keyName']
+        keypair.private_key = h['keyMaterial']
+        keypair
+      end
+      
+      def any?
+        keypairs = list || []
+        !keypairs.empty?
+      end
+      
+      def get(name)
+        keypairs = list(name) || []
+        return if keypairs.empty?
+        keypairs.first
+      end
+      
+      def exists?(name)
+        return false unless name
+        kp = get(name) rescue nil
+        !kp.nil?
+      end
+      
+    end
+    
+    class Keypairs #:nodoc:
+      def initialize(*args)
+        raise "Oops! The correct class uses a capital 'P': Rudy::AWS::EC2::KeyPairs"
+      end
+    end
+    
+  end
+end
+
+

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

@@ -0,0 +1,109 @@
+
+
+module Rudy::AWS
+  module EC2
+    class Snapshot < Storable
+      @@sformat = "%s <- %10s; %s"
+      
+      field :awsid
+      field :progress
+      field :created
+      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 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($/)
+      end
+      
+      def completed?
+        self.status && self.status == 'completed'
+      end
+      
+    end
+    
+    class Snapshots
+      include Rudy::AWS::ObjectBase
+      include Rudy::AWS::EC2::Base
+
+      def list(snap_id=[])
+        snapshots = list_as_hash(snap_id)
+        if snapshots
+          snapshots = snapshots.values.sort { |a,b| a.created <=> b.created }
+        end
+        snapshots
+      end
+      def list_as_hash(snap_id=[])
+        snap_id = [snap_id].flatten.compact
+        slist = @ec2.describe_snapshots(:snapshot_id => snap_id)
+        return unless slist['snapshotSet'].is_a?(Hash)
+        snapshots = {}
+        slist['snapshotSet']['item'].each do |snap| 
+          kp = self.class.from_hash(snap)
+          snapshots[kp.awsid] = kp
+        end
+        snapshots
+      end
+      
+      def create(vol_id)
+        vol_id = (vol_id.is_a?(Rudy::AWS::EC2::Volume)) ? vol_id.awsid : vol_id
+        shash = @ec2.create_snapshot(:volume_id => vol_id)
+        snap = Snapshots.from_hash(shash)
+        snap
+      end
+      
+      def destroy(snap_id)
+        ret = @ec2.delete_snapshot(:snapshot_id => snap_id)
+        (ret && ret['return'] == 'true') 
+      end
+      
+      
+      def Snapshots.from_hash(h)
+        #snapshotSet: 
+        #  item: 
+        #  - snapshotId: snap-5493653d
+        #    volumeId: vol-0836d761
+        #    status: completed
+        #    startTime: "2009-03-29T20:15:10.000Z"
+        #    progress: 100%
+        vol = Rudy::AWS::EC2::Snapshot.new
+        vol.awsid = h['snapshotId']
+        vol.volid = h['volumeId']
+        vol.created = h['startTime']
+        vol.progress = h['progress']
+        vol.status = h['status']
+        vol 
+      end
+      
+      def any?
+        !(list_as_hash || {}).empty?
+      end
+      
+      def get(snap_id)
+        list(snap_id).first || nil
+      end
+      
+      def exists?(id)
+        !get(snap_id).nil?
+      end
+      
+    end
+    
+  end
+end

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

@@ -0,0 +1,230 @@
+
+
+module Rudy::AWS
+  module EC2
+    class Volume < Storable
+      @@sformat = "%s  %10s;%4sGB;  %s  " # cram the terabyte
+      
+      field :awsid
+      field :status
+      field :size
+      field :snapid
+      field :zone
+      field :create_time
+      field :attach_time
+      field :instid
+      field :device
+    
+      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 inspect
+        lines = [liner_note]
+        field_names.each do |n|
+           lines << sprintf(" %12s:  %s", n, self.send(n)) if self.send(n)
+         end
+        lines.join($/)
+      end
+    
+      # Alias for status
+      def state
+        status
+      end
+    
+      def available?; (status && status == "available"); end
+      def creating?; (status && status == "creating"); end
+      def deleting?; (status && status == "deleting"); end
+      def attached?; (status && status == "attached"); end
+      def in_use?; (status && status == "in-use"); end
+    
+    end
+  
+  
+    class Volumes
+      include Rudy::AWS::ObjectBase
+      include Rudy::AWS::EC2::Base
+    
+      unless defined?(KNOWN_STATES)
+        KNOWN_STATES = [:available, :creating, :deleting, :attached, :detaching].freeze 
+      end
+    
+      # * +size+ the number of GB
+      def create(size, zone, snapid=nil)
+        opts = {
+          :availability_zone => zone.to_s,
+          :size => (size || 1).to_s
+        }
+      
+        opts[:snapshot_id] = snapid if snapid
+      
+        # "status"=>"creating", 
+        # "size"=>"1", 
+        # "snapshotId"=>nil, 
+        # "requestId"=>"d42ff744-48b5-4f47-a3f0-7aba57a13eb9", 
+        # "availabilityZone"=>"us-east-1b", 
+        # "createTime"=>"2009-03-17T20:10:48.000Z", 
+        # "volumeId"=>"vol-48826421"
+        vol = execute_request({}) { @ec2.create_volume(opts) }
+      
+        # TODO: use a waiter?
+        #Rudy.waiter(1, 30) do
+        #  ret = @@ec2.volumes.available?(volume.awsid)
+        #end
+      
+        reqid = vol['requestId']
+        Volumes.from_hash(vol) || nil
+      end
+    
+      def destroy(vol_id)
+        vol_id = Volumes.get_vol_id(vol_id)
+        raise VolumeNotAvailable, vol_id unless available?(vol_id)
+        ret = execute_request({}) { @ec2.delete_volume(:volume_id => vol_id) }
+        (ret['return'] == 'true') 
+      end
+    
+      def attach(vol_id, inst_id, device)
+        vol_id = Volumes.get_vol_id(vol_id)
+        inst_id = inst_id.is_a?(Rudy::AWS::EC2::Instance) ? inst_id.awsid : inst_id
+        raise NoVolumeID unless vol_id
+        raise VolumeAlreadyAttached, vol_id if attached?(vol_id)
+        raise NoInstanceID unless inst_id
+        raise NoDevice unless device
+      
+        opts = {
+          :volume_id => vol_id, 
+          :instance_id => inst_id, 
+          :device => device.to_s    # Solaris devices are numbers
+        }
+        ret = execute_request(false) { @ec2.attach_volume(opts) }
+        (ret['status'] == 'attaching')
+      end
+    
+      def detach(vol_id)
+        vol_id = Volumes.get_vol_id(vol_id)
+        raise NoVolumeID unless vol_id
+        raise VolumeNotAttached, vol_id unless attached?(vol_id)
+        ret = execute_request({}) { @ec2.detach_volume(:volume_id => vol_id) }
+        (ret['status'] == 'detaching') 
+      end
+    
+    
+      def list(state=nil, vol_id=[])
+        list_as_hash(state, vol_id).values
+      end
+    
+      def list_as_hash(state=nil, vol_id=[])
+        state &&= state.to_sym
+        state = nil if state == :any
+        # A nil state is fine, but we don't want an unknown one!
+        raise UnknownState, state if state && !Volumes.known_state?(state)
+      
+        opts = { 
+          :volume_id => vol_id ? [vol_id].flatten : [] 
+        }
+
+        vlist = execute_request({}) { @ec2.describe_volumes(opts) }
+
+        volumes = {}
+        return volumes unless vlist['volumeSet'].is_a?(Hash)
+        vlist['volumeSet']['item'].each do |vol|
+          v = Volumes.from_hash(vol)
+          next if state && v.state != state.to_s
+          volumes[v.awsid] = v
+        end
+        volumes
+      end
+    
+      def any?(state=nil,vol_id=[])
+        !list(state, vol_id).nil?
+      end
+    
+      def exists?(vol_id)
+        vol_id = Volumes.get_vol_id(vol_id)
+        vol = get(vol_id)
+        !vol.nil?
+      end
+    
+      def get(vol_id)
+        vol_id = Volumes.get_vol_id(vol_id)
+        list(:any, vol_id).first rescue nil
+      end
+    
+      # deleting?, available?, etc...
+      %w[deleting available attached in-use].each do |state|
+        define_method("#{state.tr('-', '_')}?") do |vol_id|
+          vol_id = Volumes.get_vol_id(vol_id)
+          return false unless vol_id
+          vol = get(vol_id)
+          (vol && vol.status == state)
+        end
+      end
+    
+      # Creates a Rudy::AWS::EC2::Volume object from:
+      #
+      #     volumeSet: 
+      #       item: 
+      #       - status: available
+      #         size: "1"
+      #         snapshotId: 
+      #         availabilityZone: us-east-1b
+      #         attachmentSet: 
+      #         createTime: "2009-03-17T20:10:48.000Z"
+      #         volumeId: vol-48826421
+      #         attachmentSet: 
+      #           item: 
+      #           - attachTime: "2009-03-17T21:49:54.000Z"
+      #             status: attached
+      #             device: /dev/sdh
+      #             instanceId: i-956af3fc
+      #             volumeId: vol-48826421
+      #         
+      #     requestId: 8fc30e5b-a9c3-4fe0-a979-0f71e639a7c7
+      #
+      def self.from_hash(h)
+        vol = Rudy::AWS::EC2::Volume.new
+        vol.status = h['status']
+        vol.size = h['size']
+        vol.snapid = h['snapshotId']
+        vol.zone = h['availabilityZone']
+        vol.awsid = h['volumeId']
+        vol.create_time = h['createTime']
+        if h['attachmentSet'].is_a?(Hash)
+          item = h['attachmentSet']['item'].first
+          vol.status = item['status']   # Overwrite "available status". Possibly a bad idea. 
+          vol.device = item['device']
+          vol.attach_time = item['attachTime']
+          vol.instid = item['instanceId']
+        end
+        vol
+      end
+
+      # * +vol_id+ is a String or Rudy::AWS::EC2::Volume
+      # Returns the volume ID
+      def self.get_vol_id(vol_id)
+        (vol_id.is_a?(Rudy::AWS::EC2::Volume)) ? vol_id.awsid : vol_id
+      end
+
+      # Is +state+ a known EC2 volume state? See: KNOWN_STATES
+      def self.known_state?(state)
+        return false unless state
+        state &&= state.to_sym
+        KNOWN_STATES.member?(state)
+      end
+    
+    end
+  end
+end
+
+
+class Rudy::AWS::EC2::Volumes
+  
+
+end

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

@@ -0,0 +1,77 @@
+
+module Rudy::AWS
+  module EC2
+    
+    class Zone < Storable
+      
+      field :name
+      field :region
+      field :state  
+      
+      def liner_note
+        "%-10s  %9s  %s" % [self.name, self.region, self.state]
+      end
+      
+      def to_s(titles=false)
+        str = titles ? "%-20s   %s#{$/}" % ['name', 'region', 'state'] : ""
+        str << liner_note
+      end
+
+    end
+    
+    class Zones
+      include Rudy::AWS::ObjectBase
+      include Rudy::AWS::EC2::Base
+      
+      def list(*names)
+        zones = list_as_hash(names)
+        zones &&= zones.values
+        zones
+      end
+      
+      def list_as_hash(*names)
+        names = names.flatten
+        zlist = @ec2.describe_availability_zones(:zone_name => names)
+        return unless zlist['availabilityZoneInfo'].is_a?(Hash)
+        zones = {}
+        zlist['availabilityZoneInfo']['item'].each do |zhash| 
+          zon = Zones.from_hash(zhash)
+          zones[zon.name] = zon
+        end
+        zones
+      end
+
+      def self.from_hash(h)
+        zone = Rudy::AWS::EC2::Zone.new
+        zone.name = h['zoneName']
+        zone.region = h['regionName']
+        zone.state = h['zoneState']
+        zone
+      end
+      
+      def any?
+        zones = list || []
+        !zones.empty?
+      end
+      
+      def get(name)
+        zones = list(name) || []
+        return if zones.empty?
+        zones.first
+      end
+      
+      def zone?(name)
+        begin
+          kp = get(name)
+          kp.is_a?(Rudy::AWS::EC2::Zone)
+        rescue => ex
+          false
+        end
+      end
+      
+    end
+    
+  end
+end
+
+