Added di support

samsoncms · Aug 31, 2016 · fd295c0 · fd295c0
1 parent 1c0cfce
commit fd295c0
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/Controller.php b/src/Controller.php
@@ -15,6 +15,7 @@
 use samsonframework\core\ResourcesInterface;
 use samsonframework\core\SystemInterface;
 use samsonframework\i18n\i18nInterface;
+use samsonframework\orm\QueryInterface;
 
 /**
  * SamsonCMS security controller
@@ -52,7 +53,7 @@ class Controller extends \samsoncms\Application
     protected $query;
 
     /**
-     * @var \samsonframework\i18n\I18nInterface
+     * @var \samsonframework\i18n\i18nInterface
      * @Injectable
      */
     protected $i18n;
@@ -87,12 +88,13 @@ public function prepare()
      * @param ResourcesInterface $resources
      * @param SystemInterface    $system
      *
+     * @InjectArgument(query="\samsonframework\orm\QueryInterface")
      * @InjectArgument(resources="\samsonframework\core\ResourcesInterface")
      * @InjectArgument(system="\samsonframework\core\SystemInterface")
      */
-    public function __construct($path, ResourcesInterface $resources, SystemInterface $system)
+    public function __construct(QueryInterface $query, $path, ResourcesInterface $resources, SystemInterface $system)
     {
-        $this->query = new dbQuery();
+        $this->query = $query;
 
         parent::__construct($path, $resources, $system);
     }