lib/fog/compute/digitalocean.rb
module Fog
module Compute
class DigitalOcean < Fog::Service
autoload :PagingCollection, File.expand_path('../digitalocean/models/paging_collection', __FILE__)
requires :digitalocean_token
model_path 'fog/compute/digitalocean/models'
model :server
collection :servers
model :image
collection :images
model :region
collection :regions
model :flavor
collection :flavors
model :ssh_key
collection :ssh_keys
model :volume
collection :volumes
request_path 'fog/compute/digitalocean/requests'
request :list_volumes
request :create_volume
request :attach_volume
request :detach_volume
request :destroy_volume
request :list_volume_actions
request :list_servers
request :create_server
request :destroy_server
request :get_server_details
request :change_kernel
request :disable_backups
request :enable_ipv6
request :enable_private_networking
request :power_cycle
request :power_off
request :power_on
request :reboot_server
request :password_reset
request :rebuild
request :rename
request :resize
request :restore
request :shutdown
request :snapshot
request :upgrade
request :list_ssh_keys
request :create_ssh_key
request :get_ssh_key
request :update_ssh_key
request :delete_ssh_key
request :list_images
request :list_flavors
request :get_image_details
request :get_droplet_action
request :list_droplet_actions
request :list_regions
class Mock
def self.data
@data ||= Hash.new do |hash, key|
hash[key] = {
:servers => [],
:ssh_keys => []
}
end
end
def initialize(options={})
@digitalocean_token = options[:digitalocean_token]
end
def data
self.class.data[@digitalocean_token]
end
def reset_data
self.class.data.delete(@digitalocean_token)
end
end
class Real
def initialize(options={})
digitalocean_token = options[:digitalocean_token]
persistent = false
options = {
headers: {
'Authorization' => "Bearer #{digitalocean_token}",
}
}
@connection = Fog::Core::Connection.new 'https://api.digitalocean.com', persistent, options
end
def request(params)
params[:headers] ||= {}
begin
response = @connection.request(params)
rescue Excon::Errors::HTTPStatusError => error
raise case error
when Excon::Errors::NotFound
NotFound.slurp(error)
else
error
end
end
unless response.body.empty?
response.body = Fog::JSON.decode(response.body)
end
response
end
end
end
end
end