fog--fog/lib/fog/compute/models/linode/flavors.rb

require 'fog/core/collection'
require 'fog/compute/models/linode/flavor'

module Fog
  module Linode
    class Compute
      class Flavors < Fog::Collection
        model Fog::Linode::Compute::Flavor

        def all
          load flavors
        end

        def get(id)
          new flavors(id).first
        rescue Fog::Linode::Compute::NotFound
          nil
        end

        private
        def flavors(id=nil)
          connection.avail_linodeplans(id).body['DATA'].tap { |data| map_flavors data }
        end
        
        def map_flavors(flavors)
          flavors.map! { |flavor| flavor.each_with_object({}) { |(k, v), h| h[k.downcase.to_sym] = v  }  }
          flavors.each { |flavor| flavor.merge! :id => flavor[:planid], :name => flavor[:label] }
        end
      end
    end
  end
end