159 lines
3.6 KiB
Ruby
159 lines
3.6 KiB
Ruby
module Fog
|
|
class Service
|
|
|
|
class Error < Fog::Errors::Error; end
|
|
class NotFound < Fog::Errors::NotFound; end
|
|
|
|
module Collections
|
|
|
|
def collections
|
|
service.collections
|
|
end
|
|
|
|
def requests
|
|
service.requests
|
|
end
|
|
|
|
end
|
|
|
|
class << self
|
|
|
|
def inherited(child)
|
|
child.class_eval <<-EOS, __FILE__, __LINE__
|
|
module Collections
|
|
include Fog::Service::Collections
|
|
|
|
def service
|
|
#{child}
|
|
end
|
|
end
|
|
|
|
def self.service
|
|
#{child}
|
|
end
|
|
EOS
|
|
end
|
|
|
|
def new(options={})
|
|
# attempt to load credentials from config file
|
|
begin
|
|
default_credentials = Fog.credentials.reject {|key, value| !(recognized | requirements).include?(key)}
|
|
options = default_credentials.merge(options)
|
|
rescue LoadError
|
|
# if there are no configured credentials, do nothing
|
|
end
|
|
|
|
validate_options(options)
|
|
setup_requirements
|
|
|
|
if Fog.mocking?
|
|
service::Mock.send(:include, service::Collections)
|
|
service::Mock.new(options)
|
|
else
|
|
service::Real.send(:include, service::Collections)
|
|
service::Real.new(options)
|
|
end
|
|
end
|
|
|
|
def setup_requirements
|
|
if superclass.respond_to?(:setup_requirements)
|
|
superclass.setup_requirements
|
|
end
|
|
|
|
@required ||= false
|
|
unless @required
|
|
for collection in collections
|
|
require [@model_path, collection].join('/')
|
|
constant = collection.to_s.split('_').map {|characters| characters[0...1].upcase << characters[1..-1]}.join('')
|
|
service::Collections.module_eval <<-EOS, __FILE__, __LINE__
|
|
def #{collection}(attributes = {})
|
|
#{service}::#{constant}.new({:connection => self}.merge(attributes))
|
|
end
|
|
EOS
|
|
end
|
|
for model in models
|
|
require [@model_path, model].join('/')
|
|
end
|
|
for request in requests
|
|
require [@request_path, request].join('/')
|
|
end
|
|
@required = true
|
|
end
|
|
end
|
|
|
|
def model_path(new_path)
|
|
@model_path = new_path
|
|
end
|
|
|
|
def collection(new_collection)
|
|
collections << new_collection
|
|
end
|
|
|
|
def collections
|
|
@collections ||= []
|
|
end
|
|
|
|
def model(new_model)
|
|
models << new_model
|
|
end
|
|
|
|
def models
|
|
@models ||= []
|
|
end
|
|
|
|
def request_path(new_path)
|
|
@request_path = new_path
|
|
end
|
|
|
|
def request(new_request)
|
|
requests << new_request
|
|
end
|
|
|
|
def requests
|
|
@requests ||= []
|
|
end
|
|
|
|
def requires(*args)
|
|
requirements.concat(args)
|
|
end
|
|
|
|
def requirements
|
|
@requirements ||= []
|
|
end
|
|
|
|
def recognizes(*args)
|
|
recognized.concat(args)
|
|
end
|
|
|
|
def recognized
|
|
@recognized ||= []
|
|
end
|
|
|
|
def reset_data(keys=Mock.data.keys)
|
|
Mock.reset_data(keys)
|
|
end
|
|
|
|
def reset_data(keys=Mock.data.keys)
|
|
Mock.reset_data(keys)
|
|
end
|
|
|
|
def validate_options(options)
|
|
missing = requirements - options.keys
|
|
unless missing.empty?
|
|
raise ArgumentError, "Missing required arguments: #{missing.join(', ')}"
|
|
end
|
|
|
|
unless recognizes.empty?
|
|
unrecognized = options.keys - requirements - recognized
|
|
unless unrecognized.empty?
|
|
raise ArgumentError, "Unrecognized arguments: #{unrecognized.join(', ')}"
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
end
|
|
|