fix returnUrl

abomfim · May 11, 2015 · ac36f2d · ac36f2d
1 parent 7e57211
commit ac36f2d
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/components/Controller.php b/components/Controller.php
@@ -2,7 +2,8 @@
 namespace yii\easyii\components;
 
 use Yii;
-use \yii\easyii\models;
+use yii\easyii\models;
+use yii\helpers\Url;
 
 class Controller extends \yii\web\Controller
 {
@@ -47,14 +48,12 @@ public function back()
 
     public function setReturnUrl($url = null)
     {
-        Yii::$app->getSession()->set($this->module->id.'_return', $url ? $url : Yii::$app->request->url);
+        Yii::$app->getSession()->set($this->module->id.'_return', $url ? Url::to($url) : Url::current());
     }
 
     public function getReturnUrl($defaultUrl = null)
     {
-        $url = Yii::$app->getSession()->get($this->module->id.'_return', $defaultUrl);
-
-        return $url === null ? Yii::$app->getHomeUrl() : $url;
+        return Yii::$app->getSession()->get($this->module->id.'_return', $defaultUrl ? Url::to($defaultUrl) : Url::to('/admin/'.$this->module->id));
     }
 
     public function formatResponse($success = '', $back = true)
@@ -89,4 +88,4 @@ public function formatResponse($success = '', $back = true)
             return $back ? $this->back() : $this->refresh();
         }
     }
-}
+}