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/compute/server.rb

226 lines
6.8 KiB
Ruby
Raw Normal View History

2010-10-04 17:02:08 -04:00
require 'fog/core/model'
2010-03-16 18:46:21 -04:00
2009-09-16 02:11:06 -04:00
module Fog
module AWS
2010-09-08 17:40:02 -04:00
class Compute
2009-09-16 02:11:06 -04:00
2010-01-08 14:29:07 -05:00
class Server < Fog::Model
2009-09-16 02:11:06 -04:00
identity :id, :aliases => 'instanceId'
attr_accessor :architecture
attribute :ami_launch_index, :aliases => 'amiLaunchIndex'
attribute :availability_zone, :aliases => 'availabilityZone'
attribute :block_device_mapping, :aliases => 'blockDeviceMapping'
attribute :client_token, :aliases => 'clientToken'
attribute :dns_name, :aliases => 'dnsName'
2010-01-21 23:11:14 -05:00
attribute :groups
attribute :flavor_id, :aliases => 'instanceType'
attribute :image_id, :aliases => 'imageId'
attribute :ip_address, :aliases => 'ipAddress'
attribute :kernel_id, :aliases => 'kernelId'
attribute :key_name, :aliases => 'keyName'
attribute :created_at, :aliases => 'launchTime'
2009-09-16 02:11:06 -04:00
attribute :monitoring
attribute :product_codes, :aliases => 'productCodes'
attribute :private_dns_name, :aliases => 'privateDnsName'
attribute :private_ip_address, :aliases => 'privateIpAddress'
attribute :ramdisk_id, :aliases => 'ramdiskId'
2009-09-16 02:11:06 -04:00
attribute :reason
attribute :root_device_name, :aliases => 'rootDeviceName'
attribute :root_device_type, :aliases => 'rootDeviceType'
attribute :state, :aliases => 'instanceState'
attribute :state_reason, :aliases => 'stateReason'
attribute :subnet_id, :aliases => 'subnetId'
2010-10-12 14:00:59 -04:00
attribute :tags, :aliases => 'tagSet'
2009-09-16 02:11:06 -04:00
attribute :user_data
2010-10-12 19:22:12 -04:00
attr_accessor :password
attr_writer :private_key, :private_key_path, :public_key, :public_key_path, :username
def initialize(attributes={})
@groups ||= ["default"] unless attributes[:subnet_id]
@flavor_id ||= 'm1.small'
2010-01-21 23:11:14 -05:00
super
end
def addresses
requires :id
2010-01-08 14:29:07 -05:00
connection.addresses(:server => self)
end
def console_output
requires :id
connection.get_console_output(@id)
end
2009-09-20 12:21:03 -04:00
def destroy
requires :id
connection.terminate_instances(@id)
2009-09-16 02:11:06 -04:00
true
end
2010-10-12 19:22:12 -04:00
remove_method :flavor_id
def flavor_id
@flavor && @flavor.id || @flavor_id
2010-01-10 16:22:18 -05:00
end
def flavor=(new_flavor)
@flavor = new_flavor
end
def flavor
@flavor ||= connection.flavors.all.detect {|flavor| flavor.id == @flavor_id}
2010-01-10 16:22:18 -05:00
end
2009-09-18 11:42:47 -04:00
def key_pair
requires :key_name
2009-09-18 11:42:47 -04:00
connection.keypairs.all(@key_name).first
end
def key_pair=(new_keypair)
2010-09-23 20:39:25 -04:00
@key_name = new_keypair && new_keypair.name
2009-09-18 11:42:47 -04:00
end
2010-10-12 19:22:12 -04:00
remove_method :monitoring=
2009-09-16 02:11:06 -04:00
def monitoring=(new_monitoring)
if new_monitoring.is_a?(Hash)
@monitoring = new_monitoring['state']
else
@monitoring = new_monitoring
end
end
def placement=(new_placement)
if new_placement.is_a?(Hash)
2009-09-16 02:11:06 -04:00
@availability_zone = new_placement['availabilityZone']
else
@availability_zone = new_placement
end
end
def private_key_path
@private_key_path ||= Fog.credentials[:private_key_path]
@private_key_path &&= File.expand_path(@private_key_path)
2010-09-23 20:39:25 -04:00
end
def private_key
@private_key ||= private_key_path && File.read(private_key_path)
end
def public_key_path
@public_key_path ||= Fog.credentials[:public_key_path]
@public_key_path &&= File.expand_path(@public_key_path)
end
2010-09-23 20:39:25 -04:00
def public_key
@public_key ||= public_key_path && File.read(public_key_path)
2010-09-23 20:39:25 -04:00
end
def ready?
@state == 'running'
end
def reboot
requires :id
connection.reboot_instances(@id)
true
end
2009-09-16 02:11:06 -04:00
def save
raise Fog::Errors::Error.new('Resaving an existing object may create a duplicate') if identity
requires :image_id
options = {
'BlockDeviceMapping' => block_device_mapping,
'ClientToken' => client_token,
'InstanceType' => flavor_id,
'KernelId' => kernel_id,
'KeyName' => key_name,
'Monitoring.Enabled' => monitoring,
'Placement.AvailabilityZone' => availability_zone,
'RamdiskId' => ramdisk_id,
'SecurityGroup' => groups,
'SubnetId' => subnet_id,
'UserData' => user_data
}
2010-08-17 11:45:18 -04:00
# If subnet is defined we are working on a virtual private cloud.
# subnet & security group cannot co-exist. I wish VPC just ignored
# the security group parameter instead, it would be much easier!
if subnet_id
options.delete('SecurityGroup')
else
options.delete('SubnetId')
end
data = connection.run_instances(image_id, 1, 1, options)
2009-09-16 02:11:06 -04:00
merge_attributes(data.body['instancesSet'].first)
true
end
def setup(credentials = {})
requires :identity, :ip_address, :username
sleep(10) # takes a bit before EC2 instances will play nice
commands = [
%{mkdir .ssh},
%{passwd -l root},
%{echo "#{attributes.to_json}" >> ~/attributes.json}
]
if public_key
commands << %{echo "#{public_key}" >> ~/.ssh/authorized_keys}
end
Fog::SSH.new(ip_address, username, credentials).run(commands)
rescue Errno::ECONNREFUSED => e
sleep(1)
retry
end
def ssh(commands)
2010-09-23 20:39:25 -04:00
requires :identity, :ip_address, :private_key, :username
@ssh ||= Fog::SSH.new(ip_address, username, :key_data => [private_key])
@ssh.run(commands)
end
2010-07-23 00:52:27 -04:00
def start
requires :id
2010-07-23 15:16:19 -04:00
connection.start_instances(@id)
2010-07-23 00:52:27 -04:00
true
end
def stop
requires :id
connection.stop_instances(@id)
true
end
2010-10-11 16:40:34 -04:00
def username
@username ||= 'root'
end
2009-09-19 15:31:15 -04:00
def volumes
requires :id
2010-01-08 14:29:07 -05:00
connection.volumes(:server => self)
2009-09-19 15:31:15 -04:00
end
2009-09-16 02:11:06 -04:00
private
2010-10-12 19:22:12 -04:00
remove_method :state=
def state=(new_state)
if new_state.is_a?(Hash)
@state = new_state['name']
else
@state = new_state
end
2009-09-16 02:11:06 -04:00
end
end
end
end
end