template model return type

mvc5 · Apr 9, 2018 · 9c5258c · 9c5258c
1 parent 6000092
commit 9c5258c
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/Plugins/View.php b/src/Plugins/View.php
@@ -6,16 +6,18 @@
 namespace Mvc5\Plugins;
 
 use Mvc5\Arg;
+use Mvc5\Template\TemplateLayout;
+use Mvc5\Template\TemplateModel;
 
 trait View
 {
     /**
      * @param array $vars
      * @param string|null $template
      * @param string $model
-     * @return \Mvc5\Template\TemplateLayout|mixed
+     * @return TemplateLayout|mixed
      */
-    protected function layout(array $vars = [], string $template = null, string $model = Arg::LAYOUT)
+    protected function layout(array $vars = [], string $template = null, string $model = Arg::LAYOUT) : TemplateLayout
     {
         return $this->model($vars, $template, $model);
     }
@@ -24,9 +26,9 @@ protected function layout(array $vars = [], string $template = null, string $mod
      * @param array $vars
      * @param string|null $template
      * @param string|null $model
-     * @return \Mvc5\Template\TemplateModel|\Mvc5\Template\TemplateLayout|mixed
+     * @return TemplateModel|mixed
      */
-    protected function model(array $vars = [], string $template = null, string $model = null)
+    protected function model(array $vars = [], string $template = null, string $model = null) : TemplateModel
     {
         !$model && $model = defined('static::VIEW_MODEL') ? constant('static::VIEW_MODEL') : Arg::VIEW_MODEL;
 
@@ -48,9 +50,9 @@ protected abstract function plugin($plugin, array $args = [], callable $callback
     /**
      * @param string|null $template
      * @param array $vars
-     * @return \Mvc5\Template\TemplateModel|mixed
+     * @return TemplateModel|mixed
      */
-    protected function view(string $template = null, array $vars = [])
+    protected function view(string $template = null, array $vars = []) : TemplateModel
     {
         return $this->model($vars, $template);
     }