require 'fog/ecloud/models/compute/task'

module Fog
  module Compute
    class Ecloud
      class Tasks < Fog::Ecloud::Collection
        
        model Fog::Compute::Ecloud::Task

        identity :href
        attribute :other_links, :aliases => :Links
        attribute :total_count, :aliases => :TotalCount

        def all
          data = connection.get_tasks(href).body
          data = data[:Task] ? data[:Task] : data
          load(data)
        end

        def get(uri)
          if data = connection.get_task(uri)
            new(data.body)
          end
        rescue Fog::Errors::NotFound
          nil
        end
      end
    end
  end
end