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
|
|
|
|
2010-09-07 14:30:02 -04:00
|
|
|
identity :id, :aliases => 'instanceId'
|
2009-10-24 01:23:55 -04:00
|
|
|
|
2010-04-30 17:40:29 -04:00
|
|
|
attr_accessor :architecture
|
2010-09-07 14:30:02 -04:00
|
|
|
attribute :ami_launch_index, :aliases => 'amiLaunchIndex'
|
|
|
|
attribute :availability_zone, :aliases => 'availabilityZone'
|
|
|
|
attribute :block_device_mapping, :aliases => 'blockDeviceMapping'
|
2010-10-04 19:16:03 -04:00
|
|
|
attribute :client_token, :aliases => 'clientToken'
|
2010-09-07 14:30:02 -04:00
|
|
|
attribute :dns_name, :aliases => 'dnsName'
|
2010-01-21 23:11:14 -05:00
|
|
|
attribute :groups
|
2010-09-07 14:30:02 -04:00
|
|
|
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
|
2010-09-07 14:30:02 -04:00
|
|
|
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
|
2010-09-07 14:30:02 -04:00
|
|
|
attribute :root_device_name, :aliases => 'rootDeviceName'
|
|
|
|
attribute :root_device_type, :aliases => 'rootDeviceType'
|
|
|
|
attribute :state, :aliases => 'instanceState'
|
2010-10-04 18:46:12 -04:00
|
|
|
attribute :state_reason, :aliases => 'stateReason'
|
2010-09-07 14:30:02 -04:00
|
|
|
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
|
2010-10-20 17:36:41 -04:00
|
|
|
attr_writer :private_key, :private_key_path, :public_key, :public_key_path, :username
|
2010-09-15 20:31:45 -04:00
|
|
|
|
2010-06-19 20:46:21 -04:00
|
|
|
def initialize(attributes={})
|
2010-08-18 11:47:51 -04:00
|
|
|
@groups ||= ["default"] unless attributes[:subnet_id]
|
2010-03-03 19:53:31 -05:00
|
|
|
@flavor_id ||= 'm1.small'
|
2010-01-21 23:11:14 -05:00
|
|
|
super
|
|
|
|
end
|
|
|
|
|
2009-10-20 22:39:57 -04:00
|
|
|
def addresses
|
2009-12-05 17:53:42 -05:00
|
|
|
requires :id
|
2009-11-21 16:56:39 -05:00
|
|
|
|
2010-01-08 14:29:07 -05:00
|
|
|
connection.addresses(:server => self)
|
2009-09-18 03:01:10 -04:00
|
|
|
end
|
|
|
|
|
2010-05-01 19:20:34 -04:00
|
|
|
def console_output
|
|
|
|
requires :id
|
|
|
|
|
|
|
|
connection.get_console_output(@id)
|
|
|
|
end
|
|
|
|
|
2009-09-20 12:21:03 -04:00
|
|
|
def destroy
|
2009-12-05 17:53:42 -05:00
|
|
|
requires :id
|
2009-11-21 16:56:39 -05:00
|
|
|
|
2009-12-05 17:53:42 -05:00
|
|
|
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
|
2010-01-13 00:59:39 -05:00
|
|
|
def flavor_id
|
2010-03-03 19:53:31 -05:00
|
|
|
@flavor && @flavor.id || @flavor_id
|
2010-01-10 16:22:18 -05:00
|
|
|
end
|
|
|
|
|
|
|
|
def flavor=(new_flavor)
|
2010-01-13 00:59:39 -05:00
|
|
|
@flavor = new_flavor
|
|
|
|
end
|
|
|
|
|
|
|
|
def flavor
|
2010-03-03 19:53:31 -05:00
|
|
|
@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
|
2009-11-21 16:56:39 -05:00
|
|
|
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)
|
2009-10-21 21:50:15 -04:00
|
|
|
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
|
|
|
|
|
2010-09-15 20:31:45 -04:00
|
|
|
def private_key_path
|
2010-10-20 17:32:30 -04:00
|
|
|
@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
|
2010-10-20 17:32:30 -04:00
|
|
|
@private_key ||= private_key_path && File.read(private_key_path)
|
2010-09-15 20:31:45 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def public_key_path
|
2010-10-20 17:32:30 -04:00
|
|
|
@public_key_path ||= Fog.credentials[:public_key_path]
|
|
|
|
@public_key_path &&= File.expand_path(@public_key_path)
|
2010-09-15 20:31:45 -04:00
|
|
|
end
|
|
|
|
|
2010-09-23 20:39:25 -04:00
|
|
|
def public_key
|
2010-10-20 17:32:30 -04:00
|
|
|
@public_key ||= public_key_path && File.read(public_key_path)
|
2010-09-23 20:39:25 -04:00
|
|
|
end
|
2010-10-20 17:32:30 -04:00
|
|
|
|
2010-01-13 00:59:39 -05:00
|
|
|
def ready?
|
|
|
|
@state == 'running'
|
|
|
|
end
|
|
|
|
|
2009-11-21 13:42:20 -05:00
|
|
|
def reboot
|
2009-12-05 17:53:42 -05:00
|
|
|
requires :id
|
|
|
|
connection.reboot_instances(@id)
|
2009-11-21 13:42:20 -05:00
|
|
|
true
|
|
|
|
end
|
|
|
|
|
2009-09-16 02:11:06 -04:00
|
|
|
def save
|
2010-10-04 19:08:34 -04:00
|
|
|
raise Fog::Errors::Error.new('Resaving an existing object may create a duplicate') if identity
|
2009-11-21 16:56:39 -05:00
|
|
|
requires :image_id
|
|
|
|
|
2010-05-18 14:22:32 -04:00
|
|
|
options = {
|
2010-09-15 20:31:45 -04:00
|
|
|
'BlockDeviceMapping' => block_device_mapping,
|
2010-10-03 21:07:33 -04:00
|
|
|
'ClientToken' => client_token,
|
2010-05-25 19:37:19 -04:00
|
|
|
'InstanceType' => flavor_id,
|
2010-09-15 20:31:45 -04:00
|
|
|
'KernelId' => kernel_id,
|
|
|
|
'KeyName' => key_name,
|
|
|
|
'Monitoring.Enabled' => monitoring,
|
|
|
|
'Placement.AvailabilityZone' => availability_zone,
|
|
|
|
'RamdiskId' => ramdisk_id,
|
|
|
|
'SecurityGroup' => groups,
|
2010-08-17 06:28:06 -04:00
|
|
|
'SubnetId' => subnet_id,
|
2010-09-15 20:31:45 -04:00
|
|
|
'UserData' => user_data
|
2010-05-18 14:22:32 -04:00
|
|
|
}
|
2010-09-15 20:31:45 -04:00
|
|
|
|
2010-08-17 11:45:18 -04:00
|
|
|
# If subnet is defined we are working on a virtual private cloud.
|
2010-08-17 06:28:06 -04:00
|
|
|
# subnet & security group cannot co-exist. I wish VPC just ignored
|
|
|
|
# the security group parameter instead, it would be much easier!
|
2010-08-18 11:47:51 -04:00
|
|
|
if subnet_id
|
2010-08-17 06:28:06 -04:00
|
|
|
options.delete('SecurityGroup')
|
2010-08-18 11:47:51 -04:00
|
|
|
else
|
|
|
|
options.delete('SubnetId')
|
2010-08-17 06:28:06 -04:00
|
|
|
end
|
2010-05-18 14:22:32 -04:00
|
|
|
|
2010-09-15 20:31:45 -04:00
|
|
|
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
|
|
|
|
|
2010-09-15 20:31:45 -04:00
|
|
|
def setup(credentials = {})
|
2010-10-20 17:32:30 -04:00
|
|
|
requires :identity, :ip_address, :username
|
2010-09-15 20:31:45 -04:00
|
|
|
sleep(10) # takes a bit before EC2 instances will play nice
|
2010-10-20 17:32:30 -04:00
|
|
|
commands = [
|
2010-09-15 20:31:45 -04:00
|
|
|
%{mkdir .ssh},
|
|
|
|
%{passwd -l root},
|
|
|
|
%{echo "#{attributes.to_json}" >> ~/attributes.json}
|
2010-10-20 17:32:30 -04:00
|
|
|
]
|
|
|
|
if public_key
|
|
|
|
commands << %{echo "#{public_key}" >> ~/.ssh/authorized_keys}
|
|
|
|
end
|
|
|
|
Fog::SSH.new(ip_address, username, credentials).run(commands)
|
2010-09-15 20:31:45 -04:00
|
|
|
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])
|
2010-09-15 20:31:45 -04:00
|
|
|
@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
|
|
|
|
2010-09-15 20:31:45 -04:00
|
|
|
def username
|
|
|
|
@username ||= 'root'
|
|
|
|
end
|
|
|
|
|
2009-09-19 15:31:15 -04:00
|
|
|
def volumes
|
2009-12-05 17:53:42 -05:00
|
|
|
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=
|
2009-12-05 17:53:42 -05:00
|
|
|
def state=(new_state)
|
|
|
|
if new_state.is_a?(Hash)
|
|
|
|
@state = new_state['name']
|
2009-10-23 12:30:19 -04:00
|
|
|
else
|
2009-12-05 17:53:42 -05:00
|
|
|
@state = new_state
|
2009-10-23 12:30:19 -04:00
|
|
|
end
|
2009-09-16 02:11:06 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
end
|
|
|
|
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|