First pass through initializers of {mongrel, webrick} REST handlers; hooks into Indirection to look up models from indirected names.

1 files changed, 14 insertions, 0 deletions

diff --git a/lib/puppet/network/http/mongrel/rest.rb b/lib/puppet/network/http/mongrel/rest.rb
index 6e454c7d9..34f1d8f90 100644
--- a/lib/puppet/network/http/mongrel/rest.rb
+++ b/lib/puppet/network/http/mongrel/rest.rb
@@ -1,4 +1,18 @@
 class Puppet::Network::HTTP::MongrelREST
     def initialize(args = {})
+        raise ArgumentError unless args[:server]
+        raise ArgumentError if !args[:handlers] or args[:handlers].empty?
+        
+        @models = {}
+        args[:handlers].each do |handler|
+            @models[handler] = find_model_for_handler(handler)
+        end
+    end
+    
+  private
+    
+    def find_model_for_handler(handler)
+        Puppet::Indirector::Indirection.model(handler) || 
+            raise(ArgumentError, "Cannot locate indirection [#{handler}].")
     end
 end