1
0
Fork 0
mirror of https://github.com/fog/fog.git synced 2022-11-09 13:51:43 -05:00
fog--fog/lib/fog/aws/models/ec2/addresses.rb
2010-03-17 12:26:42 -07:00

70 lines
1.5 KiB
Ruby

require 'fog/collection'
require 'fog/aws/models/ec2/address'
module Fog
module AWS
module EC2
class Mock
def addresses(attributes = {})
Fog::AWS::EC2::Addresses.new({
:connection => self
}.merge!(attributes))
end
end
class Real
def addresses(attributes = {})
Fog::AWS::EC2::Addresses.new({
:connection => self
}.merge!(attributes))
end
end
class Addresses < Fog::Collection
attribute :public_ip
attribute :server
model Fog::AWS::EC2::Address
def initialize(attributes)
@public_ip ||= []
super
end
def all(public_ip = @public_ip)
@public_ip = public_ip
data = connection.describe_addresses(public_ip).body
load(
data['addressesSet'].map do |address|
address.reject {|key, value| value.nil? || value.empty? }
end
)
if server
self.replace(self.select {|address| address.server_id == server.id})
end
self
end
def get(public_ip)
if public_ip
all(public_ip).first
end
rescue Excon::Errors::BadRequest
nil
end
def new(attributes = {})
if server
super({ :server => server }.merge!(attributes))
else
super(attributes)
end
end
end
end
end
end