]> git.donarmstrong.com Git - dsa-puppet.git/blob - 3rdparty/modules/aviator/feature/aviator/core/request_builder.rb
add aimonb/aviator to 3rdparty
[dsa-puppet.git] / 3rdparty / modules / aviator / feature / aviator / core / request_builder.rb
1 module Aviator
2
3   class BaseRequestNotFoundError < StandardError
4     attr_reader :base_request_hierarchy
5
6     def initialize(base_hierarchy)
7       @base_request_hierarchy = base_hierarchy
8       super("#{ base_request_hierarchy } could not be found!")
9     end
10   end
11
12
13   class RequestAlreadyDefinedError < StandardError
14     attr_reader :namespace,
15                 :request_name
16
17     def initialize(namespace, request_name)
18       @namespace = namespace
19       @request_name = request_name
20       super("#{ namespace }::#{ request_name } is already defined")
21     end
22   end
23
24
25   class RequestBuilder
26
27     class << self
28
29       def define_request(root_namespace, request_name, options, &block)
30         base_klass = get_request_class(root_namespace, options[:inherit])
31
32         klass = Class.new(base_klass, &block)
33
34         namespace_arr = [
35           klass.provider,
36           klass.service,
37           klass.api_version,
38           klass.endpoint_type
39         ]
40
41         namespace = namespace_arr.inject(root_namespace) do |namespace, sym|
42           const_name = sym.to_s.camelize
43           namespace.const_set(const_name, Module.new) unless namespace.const_defined?(const_name, false)
44           namespace.const_get(const_name, false)
45         end
46
47         klassname = request_name.to_s.camelize
48
49         if namespace.const_defined?(klassname, false)
50           raise RequestAlreadyDefinedError.new(namespace, klassname)
51         end
52
53         namespace.const_set(klassname, klass)
54       end
55
56
57       def get_request_class(root_namespace, request_class_arr)
58         request_class_arr.inject(root_namespace) do |namespace, sym|
59           namespace.const_get(sym.to_s.camelize, false)
60         end
61       rescue NameError => e
62         arr = ['..', '..'] + request_class_arr
63         arr[-1,1] = arr.last.to_s + '.rb'
64         path = Pathname.new(__FILE__).join(*arr.map{|i| i.to_s }).expand_path
65
66         if path.exist?
67           require path
68           request_class_arr.inject(root_namespace) do |namespace, sym|
69             namespace.const_get(sym.to_s.camelize, false)
70           end
71         else
72           raise BaseRequestNotFoundError.new(request_class_arr)
73         end
74       end
75
76     end
77
78   end
79
80
81   class << self
82
83     def define_request(request_name, options={ :inherit => [:request] }, &block)
84       RequestBuilder.define_request self, request_name, options, &block
85     end
86
87   end # class << self
88
89 end