fog-digitalocean 0.1.0

data/lib/fog/digitalocean/models/compute/image.rb

@@ -0,0 +1,25 @@
+module Fog
+  module Compute
+    class DigitalOcean
+      class Image < Fog::Model
+        identity :id
+        attribute :name
+        attribute :type
+        attribute :distribution
+        attribute :slug
+        attribute :public
+        attribute :regions
+        attribute :min_disk_size
+        attribute :created_at
+      end
+
+      def transfer
+        perform_action :transfer_image
+      end
+
+      def convert_to_snapshot
+        perform_action :convert_image_to_snapshot
+      end
+    end
+  end
+end

data/lib/fog/digitalocean/models/compute/images.rb

@@ -0,0 +1,42 @@
+require 'fog/digitalocean/models/paging_collection'
+
+module Fog
+  module Compute
+    class DigitalOcean
+      class Images < Fog::Compute::DigitalOcean::PagingCollection
+        model Fog::Compute::DigitalOcean::Image
+
+        # Retrieves images
+        # @return [Fog::Compute::DigitalOceanV2:Image]
+        # @raise [Fog::Compute::DigitalOceanV2::NotFound] - HTTP 404
+        # @raise [Fog::Compute::DigitalOceanV2::BadRequest] - HTTP 400
+        # @raise [Fog::Compute::DigitalOceanV2::InternalServerError] - HTTP 500
+        # @raise [Fog::Compute::DigitalOceanV2::ServiceError]
+        # @see https://developers.digitalocean.com/documentation/v2/#list-all-images
+        def all(filters = {})
+          data = service.list_images(filters)
+          links = data.body["links"]
+          get_paged_links(links) 
+          images = data.body["images"]
+          load(images)
+        end
+
+        # Retrieves image
+        # @param [String] id for image to be returned
+        # @return [Fog::Compute::DigitalOceanV2:Image]
+        # @raise [Fog::Compute::DigitalOceanV2::NotFound] - HTTP 404
+        # @raise [Fog::Compute::DigitalOceanV2::BadRequest] - HTTP 400
+        # @raise [Fog::Compute::DigitalOceanV2::InternalServerError] - HTTP 500
+        # @raise [Fog::Compute::DigitalOceanV2::ServiceError]
+        # @see https://developers.digitalocean.com/documentation/v2/#retrieve-an-existing-image-by-id
+        def get(id)
+          image = service.get_image_details(id).body['image']
+          new(image) if image
+        rescue Fog::Errors::NotFound
+          nil
+        end
+
+      end
+    end
+  end
+end

data/lib/fog/digitalocean/models/compute/region.rb

@@ -0,0 +1,13 @@
+module Fog
+  module Compute
+    class DigitalOcean
+      class Region < Fog::Model
+        identity :slug
+        attribute :name
+        attribute :sizes
+        attribute :available
+        attribute :features
+      end
+    end
+  end
+end

data/lib/fog/digitalocean/models/compute/regions.rb

@@ -0,0 +1,21 @@
+module Fog
+  module Compute
+    class DigitalOcean
+      class Regions < Fog::Collection
+        model Fog::Compute::DigitalOcean::Region
+
+        # Retrieves regions
+        # @return [Fog::Compute::DigitalOceanV2:Regions]
+        # @raise [Fog::Compute::DigitalOceanV2::NotFound] - HTTP 404
+        # @raise [Fog::Compute::DigitalOceanV2::BadRequest] - HTTP 400
+        # @raise [Fog::Compute::DigitalOceanV2::InternalServerError] - HTTP 500
+        # @raise [Fog::Compute::DigitalOceanV2::ServiceError]
+        # @see https://developers.digitalocean.com/documentation/v2/#list-all-regions
+        def all(filters = {})
+          data = service.list_regions(filters).body["regions"]
+          load(data)
+        end
+      end
+    end
+  end
+end

data/lib/fog/digitalocean/models/compute/server.rb

@@ -0,0 +1,170 @@
+require 'fog/compute/models/server'
+
+module Fog
+  module Compute
+    class DigitalOcean
+      # A DigitalOcean Droplet
+      #
+      class Server < Fog::Compute::Server
+        identity :id
+        attribute :name
+        attribute :memory
+        attribute :vcpus
+        attribute :disk
+        attribute :locked
+        attribute :created_at
+        attribute :status,    :aliases => 'state'
+        attribute :backup_ids
+        attribute :snapshot_ids
+        attribute :features
+        attribute :region
+        attribute :image
+        attribute :size
+        attribute :size_slug
+        attribute :networks
+        attribute :kernel
+        attribute :next_backup_window
+        attribute :private_networking
+        attribute :backups
+        attribute :ipv6
+        attribute :ssh_keys
+
+        def public_ip_address
+          ipv4_address
+        end
+
+        def ipv6_address
+          if (net = networks['v6'].find { |n| n['type'] == 'public' })
+            net['ip_address']
+          end
+        end
+
+        def ipv4_address
+          if (net = networks['v4'].find { |n| n['type'] == 'public' })
+            net['ip_address']
+          end
+        end
+
+        def save
+          raise Fog::Errors::Error.new('Re-saving an existing object may create a duplicate') if persisted?
+          requires :name, :region, :size, :image
+
+          options = {}
+          if attributes[:ssh_keys]
+            options[:ssh_keys] = attributes[:ssh_keys]
+          elsif @ssh_keys
+            options[:ssh_keys] = @ssh_keys.map(&:id)
+          end
+
+          options[:private_networking] = private_networking
+          options[:backups]            = backups
+          options[:ipv6]               = ipv6
+
+          data = service.create_server(name, size, image, region, options)
+
+          merge_attributes(data.body['droplet'])
+          true
+        end
+
+        def delete
+          requires :id
+          response = service.delete_server id
+          response.body
+        end
+
+        def ready?
+          status == 'active'
+        end
+
+        def locked?
+          locked
+        end
+
+        def actions
+          requires :id
+          response = service.list_droplet_actions id
+          response.body
+        end
+
+        def action(action_id)
+          requires :id
+          response = service.get_droplet_action(id, action_id)
+          response.body
+        end
+
+        def reboot
+          perform_action :reboot_server
+        end
+
+        def disable_backups
+          perform_action :disable_backups
+        end
+
+        def power_cycle
+          perform_action :power_cycle
+        end
+
+        def shutdown
+          perform_action :shutdown
+        end
+
+        def power_off
+          perform_action :power_off
+        end
+
+        def power_on
+          perform_action :power_on
+        end
+
+        def restore(image)
+          perform_action :restore, image
+        end
+
+        def password_reset
+          perform_action :password_reset
+        end
+
+        def resize(resize_disk, size)
+          perform_action :resize, resize_disk, size
+        end
+
+        def rebuild(image)
+          perform_action :rebuild, image
+        end
+
+        def rename(name)
+          perform_action :rename, name
+        end
+
+        def change_kernel(kernel)
+          perform_action :change_kernel, kernel
+        end
+
+        def enable_ipv6
+          perform_action :enable_ipv6
+        end
+
+        def enable_private_networking
+          perform_action :enable_private_networking
+        end
+
+        def snapshot(name)
+          perform_action :snapshot, name
+        end
+
+        def upgrade
+          perform_action :upgrade
+        end
+
+        private
+
+        # Performs a droplet action with the given set of arguments.
+        def perform_action(action, *args)
+          requires :id
+          response = service.send(action, id, *args)
+          response.body
+        end
+      end
+    end
+  end
+end

data/lib/fog/digitalocean/models/compute/servers.rb

@@ -0,0 +1,43 @@
+require 'fog/core/collection'
+require 'fog/digitalocean/models/compute/server'
+require 'fog/digitalocean/models/paging_collection'
+
+module Fog
+  module Compute
+    class DigitalOcean
+      class Servers < Fog::Compute::DigitalOcean::PagingCollection
+        model Fog::Compute::DigitalOcean::Server
+
+        # Returns list of servers
+        # @return [Fog::Compute::DigitalOceanV2::Servers]
+        # @raise [Fog::Compute::DigitalOceanV2::NotFound] - HTTP 404
+        # @raise [Fog::Compute::DigitalOceanV2::BadRequest] - HTTP 400
+        # @raise [Fog::Compute::DigitalOceanV2::InternalServerError] - HTTP 500
+        # @raise [Fog::Compute::DigitalOceanV2::ServiceError]
+        # @see https://developers.digitalocean.com/documentation/v2/#droplets
+        def all(filters = {})
+          data = service.list_servers(filters)
+          links = data.body["links"]
+          get_paged_links(links) 
+          droplets = data.body["droplets"]
+          load(droplets)
+        end
+
+        # Retrieves server
+        # @param [String] id for server to be returned
+        # @return [Fog::Compute::DigitalOceanV2:Server]
+        # @raise [Fog::Compute::DigitalOceanV2::NotFound] - HTTP 404
+        # @raise [Fog::Compute::DigitalOceanV2::BadRequest] - HTTP 400
+        # @raise [Fog::Compute::DigitalOceanV2::InternalServerError] - HTTP 500
+        # @raise [Fog::Compute::DigitalOceanV2::ServiceError]
+        # @see https://developers.digitalocean.com/documentation/v2/#retrieve-an-existing-droplet-by-id
+        def get(id)
+          server = service.get_server_details(id).body['droplet']
+          new(server) if server
+        rescue Fog::Errors::NotFound
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/fog/digitalocean/models/compute/ssh_key.rb

@@ -0,0 +1,31 @@
+module Fog
+  module Compute
+    class DigitalOcean
+      class SshKey < Fog::Model
+        identity :id
+        attribute :fingerprint
+        attribute :public_key
+        attribute :name
+
+        def save
+          requires :name, :public_key
+          merge_attributes(service.create_ssh_key(name, public_key).body['ssh_key'])
+          true
+        end
+
+        def destroy
+          requires :id
+          service.delete_ssh_key id
+        end
+
+        def update
+          requires :id, :name
+          data = service.update_server(id, name)
+          merge_attributes(data.body['ssh_key'])
+          true
+        end
+
+      end
+    end
+  end
+end

data/lib/fog/digitalocean/models/compute/ssh_keys.rb

@@ -0,0 +1,40 @@
+require 'fog/digitalocean/models/paging_collection'
+
+module Fog
+  module Compute
+    class DigitalOcean
+      class SshKeys < Fog::Compute::DigitalOcean::PagingCollection
+        model Fog::Compute::DigitalOcean::SshKey
+
+        # Returns list of ssh keys
+        # @return [Fog::Compute::DigitalOceanV2::Sshkeys] Retrieves a list of ssh keys.
+        # @raise [Fog::Compute::DigitalOceanV2::NotFound] - HTTP 404
+        # @raise [Fog::Compute::DigitalOceanV2::BadRequest] - HTTP 400
+        # @raise [Fog::Compute::DigitalOceanV2::InternalServerError] - HTTP 500
+        # @raise [Fog::Compute::DigitalOceanV2::ServiceError]
+        # @see https://developers.digitalocean.com/documentation/v2/#list-all-keys
+        def all(filters = {})
+          data = service.list_ssh_keys(filters)
+          links = data.body["links"]
+          get_paged_links(links)
+          keys = data.body["ssh_keys"]
+          load(keys)
+        end
+
+        # Returns ssh key
+        # @return [Fog::Compute::DigitalOceanV2::Sshkeys] Retrieves a list of ssh keys.
+        # @raise [Fog::Compute::DigitalOceanV2::NotFound] - HTTP 404
+        # @raise [Fog::Compute::DigitalOceanV2::BadRequest] - HTTP 400
+        # @raise [Fog::Compute::DigitalOceanV2::InternalServerError] - HTTP 500
+        # @raise [Fog::Compute::DigitalOceanV2::ServiceError]
+        # @see https://developers.digitalocean.com/documentation/v2/#retrieve-an-existing-key
+        def get(id)
+          key = service.get_ssh_key(id).body['ssh_key']
+          new(key) if key
+        rescue Fog::Errors::NotFound
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/fog/digitalocean/models/paging_collection.rb

@@ -0,0 +1,42 @@
+module Fog
+  module Compute
+    class DigitalOcean
+      class PagingCollection < Fog::Collection
+
+        attribute :next
+        attribute :last
+
+        def next_page
+          all(page: @next) if @next != @last
+        end
+
+        def previous_page
+          if @next.to_i > 2
+            all(page: @next.to_i - 2)
+          end
+        end
+
+        private
+
+        def deep_fetch(hash, *path)
+          path.inject(hash) do |acc, key|
+            acc.respond_to?(:keys) ? acc[key] : nil
+          end
+        end
+
+        def get_page(link)
+          if match = link.match(/page=(?<page>\d+)/)
+            match.captures.last
+          end
+        end
+
+        def get_paged_links(links)
+          next_link = deep_fetch(links, "pages", "next").to_s
+          last_link = deep_fetch(links, "pages", "last").to_s
+          @next = get_page(next_link) || @next
+          @last = get_page(last_link) || @last
+        end
+      end
+    end
+  end
+end

data/lib/fog/digitalocean/requests/compute/change_kernel.rb

@@ -0,0 +1,44 @@
+module Fog
+  module Compute
+    class DigitalOcean
+      class Real
+        def change_kernel(id, kernel)
+          body = { :type => "change_kernel", :kernel => kernel }
+
+          encoded_body = Fog::JSON.encode(body)
+
+          request(
+            :expects => [201],
+            :headers => {
+              'Content-Type' => "application/json; charset=UTF-8",
+            },
+            :method  => 'POST',
+            :path    => "v2/droplets/#{id}/actions",
+            :body    => encoded_body,
+          )
+        end
+      end
+
+      class Mock
+        def change_kernel(id, kernel)
+          response         = Excon::Response.new
+          response.status  = 201
+          response.body    = {
+            'action' => {
+              'id'            => Fog::Mock.random_numbers(1).to_i,
+              'status'        => "in-progress",
+              'type'          => "change_kernel",
+              'started_at'    => "2014-11-14T16:31:00Z",
+              'completed_at'  => null,
+              'resource_id'   => id,
+              'resource_type' => "droplet",
+              'region'        => "nyc3",
+              'region_slug'   => "nyc3"
+            }
+          }
+          response
+        end
+      end
+    end
+  end
+end