mirror of
https://github.com/fog/fog.git
synced 2022-11-09 13:51:43 -05:00
70 lines
1.5 KiB
Ruby
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
|