From e432df76d5a0a342fa1b8cc55bbcd1cbd45b40e8 Mon Sep 17 00:00:00 2001
From: Jeffery To <jeffery.to@gmail.com>
Date: Tue, 11 Aug 2009 21:55:24 +0800
Subject: Set returnto when redirected to login of a private install

---
 index.php | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/index.php b/index.php
index 980b9881b..e590e217e 100644
--- a/index.php
+++ b/index.php
@@ -193,6 +193,21 @@ function main()
         }
         if (!in_array($action, $public_actions) &&
             !preg_match('/rss$/', $action)) {
+
+            // set returnto
+            $rargs =& common_copy_args($args);
+            unset($rargs['action']);
+            if (common_config('site', 'fancy')) {
+                unset($rargs['p']);
+            }
+            if (array_key_exists('submit', $rargs)) {
+                unset($rargs['submit']);
+            }
+            foreach (array_keys($_COOKIE) as $cookie) {
+                unset($rargs[$cookie]);
+            }
+            common_set_returnto(common_local_url($action, $rargs));
+
             common_redirect(common_local_url($login_action));
             return;
         }
-- 
cgit v1.2.3-54-g00ecf