require 'chef/provisioning'
require 'cheffish'
require 'chef/provisioning/image_spec'

class Chef
module Provisioning
  #
  # Specification for a image. Sufficient information to find and contact it
  # after it has been set up.
  #
  class ChefImageSpec < ImageSpec
    def initialize(node, chef_server)
      super(node)
      @chef_server = chef_server
    end

    #
    # Get a ImageSpec from the chef server.  If the image does not exist on the
    # server, it returns nil.
    #
    def self.get(name, chef_server = Cheffish.default_chef_server)
      chef_api = Cheffish.chef_server_api(chef_server)
      begin
        data = chef_api.get("/data/images/#{name}")
        data['machine_options'] = strings_to_symbols(data['machine_options'])
        ChefImageSpec.new(data, chef_server)
      rescue Net::HTTPServerException => e
        if e.response.code == '404'
          nil
        else
          raise
        end
      end
    end

    # Creates a new empty ImageSpec with the given name.
    def self.empty(id, chef_server = Cheffish.default_chef_server)
      ChefImageSpec.new({ 'id' => id }, chef_server)
    end

    #
    # Globally unique identifier for this image. Does not depend on the image's
    # location or existence.
    #
    def id
      ChefImageSpec.id_from(chef_server, name)
    end

    def self.id_from(chef_server, name)
      "#{chef_server[:chef_server_url]}/data/images/#{name}"
    end

    #
    # Save this image to the server.  If you have significant information that
    # could be lost, you should do this as quickly as possible.  Data will be
    # saved automatically for you after allocate_image and ready_image.
    #
    def save(action_handler)
      # Save the image to the server.
      _self = self
      _chef_server = _self.chef_server
      Chef::Provisioning.inline_resource(action_handler) do
        # Create the data bag if needed
        chef_data_bag 'images'
        # Save the image
        chef_data_bag_item _self.name do
          data_bag 'images'
          chef_server _chef_server
          raw_data _self.image_data
        end
      end
    end

    def delete(action_handler)
      # Delete the image from the server.
      _self = self
      _chef_server = _self.chef_server
      Chef::Provisioning.inline_resource(action_handler) do
        chef_data_bag_item _self.name do
          data_bag 'images'
          chef_server _chef_server
          action :delete
        end
      end
    end

    protected

    attr_reader :chef_server

    #
    # Chef API object for the given Chef server
    #
    def chef_api
      Cheffish.server_api_for(chef_server)
    end

    def self.strings_to_symbols(data)
      if data.is_a?(Hash)
        result = {}
        data.each_pair do |key, value|
          result[key.to_sym] = strings_to_symbols(value)
        end
        result
      else
        data
      end
    end
  end
end
end