From b40922610425d3f1a48f37db512b375a2fc85d7f Mon Sep 17 00:00:00 2001
From: dpordomingo <David.Pordomingo.F@gmail.com>
Date: Mon, 25 Nov 2013 15:36:24 +0100
Subject: [PATCH] using Wrapper deffinition

---
 .../Mongator/Processor/Wrapper/DefinitionInjector.php | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/Level3/Mongator/Processor/Wrapper/DefinitionInjector.php b/src/Level3/Mongator/Processor/Wrapper/DefinitionInjector.php
index 8611910..9ead7d5 100644
--- a/src/Level3/Mongator/Processor/Wrapper/DefinitionInjector.php
+++ b/src/Level3/Mongator/Processor/Wrapper/DefinitionInjector.php
@@ -3,10 +3,11 @@
 namespace Level3\Mongator\Processor\Wrapper;
 
 use Mongator\Mongator;
+use Level3\Repository;
 use Level3\Messages\Request;
 use Level3\Messages\Response;
 use Level3\Processor\Wrapper;
-use Level3\Resource;
+use Level3\Resource\Resource;
 use Closure;
 
 class DefinitionInjector extends Wrapper
@@ -18,16 +19,16 @@ public function __construct(Mongator $mongator)
         $this->mongator = $mongator;
     }
 
-    public function error(Closure $execution, Request $request)
+    public function error(Repository $repository, Request $request, Callable $execution)
     {
         return $execution($request);
     }
     
-    protected function processRequest(Closure $execution, Request $request, $method)
+    protected function processRequest(Repository $repository, Request $request, Callable $execution, $method)
     {
-        $response = $execution($request);
+        $response = $execution($repository, $request);
 
-        $key = $request->getKey();
+        $key = $repository->getKey();
         $resource = $response->getResource();
 
         if (!$key || !$resource) {