fog--fog/lib/fog/bluebox.rb

module Fog
  module Bluebox
    require 'fog/bluebox/models/images'
    require 'fog/bluebox/requests/get_templates'
    require 'fog/bluebox/models/flavors'
    require 'fog/bluebox/requests/get_products'
    require 'fog/bluebox/models/flavor'
    require 'fog/bluebox/requests/get_product'
    require 'fog/bluebox/models/servers'
    require 'fog/bluebox/requests/get_blocks'
    require 'fog/bluebox/models/server'
    require 'fog/bluebox/requests/get_block'
    require 'fog/bluebox/requests/create_block'
    require 'fog/bluebox/requests/destroy_block'
    require 'fog/bluebox/requests/reboot_block'

    def self.new(options={})

      unless options[:bluebox_api_key]
        raise ArgumentError.new('bluebox_api_key is required to access Blue Box')
      end
      if Fog.mocking?
        Fog::Bluebox::Mock.new(options)
      else
        Fog::Bluebox::Real.new(options)
      end
    end

    def self.reset_data(keys=Mock.data.keys)
      Mock.reset_data(keys)
    end

    class Mock

      def self.data
        @data ||= Hash.new do |hash, key|
          hash[key] = {}
        end
      end

      def self.reset_data(keys=data.keys)
        for key in [*keys]
          data.delete(key)
        end
      end

      def initialize(options={})
        @bluebox_api_key = options[:bluebox_api_key]
        @data = self.class.data[@bluebox_api_key]
      end

    end

    class Real

      def initialize(options={})
        @bluebox_api_key = options[:bluebox_api_key]
        @host   = options[:bluebox_host]    || "boxpanel.blueboxgrp.com"
        @port   = options[:bluebox_port]    || 443
        @scheme = options[:bluebox_scheme]  || 'https'
      end

      def request(params)
        @connection = Fog::Connection.new("#{@scheme}://#{@host}:#{@port}")
        params[:headers] ||= {}
        params[:headers].merge!({
          'Authorization' => "Basic #{Base64.encode64(@bluebox_api_key).delete("\r\n")}"
        })
        case params[:method]
        when 'DELETE', 'GET', 'HEAD'
          params[:headers]['Accept'] = 'application/xml'
        when 'POST', 'PUT'
          params[:headers]['Content-Type'] = 'application/xml'
        end

        @connection.request({:host => @host}.merge!(params))
      end

    end
  end
end