diff options
Diffstat (limited to 'juick-www/src/main/java/com/juick/www/controllers/PM.java')
-rw-r--r-- | juick-www/src/main/java/com/juick/www/controllers/PM.java | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/juick-www/src/main/java/com/juick/www/controllers/PM.java b/juick-www/src/main/java/com/juick/www/controllers/PM.java index 028feabc..9cc29129 100644 --- a/juick-www/src/main/java/com/juick/www/controllers/PM.java +++ b/juick-www/src/main/java/com/juick/www/controllers/PM.java @@ -22,6 +22,7 @@ import com.juick.service.PMQueriesService; import com.juick.service.TagService; import com.juick.service.UserService; import com.juick.util.MessageUtils; +import com.juick.util.UserUtils; import com.juick.util.WebUtils; import com.juick.www.Utils; import com.juick.www.WebApp; @@ -62,7 +63,7 @@ public class PM { @RequestMapping(value = "/pm/inbox", method = RequestMethod.GET) protected String doGetInbox(HttpServletRequest request, HttpServletResponse response, ModelMap model) { - com.juick.User visitor = webApp.getVisitorUser(request, response); + com.juick.User visitor = UserUtils.getCurrentUser(); if (visitor.getUid() == 0) { Utils.sendTemporaryRedirect(response, "/login"); } @@ -79,7 +80,7 @@ public class PM { @RequestMapping(value = "/pm/sent", method = RequestMethod.GET) protected String doGetSent(HttpServletRequest request, HttpServletResponse response, ModelMap model) { - com.juick.User visitor = webApp.getVisitorUser(request, response); + com.juick.User visitor = UserUtils.getCurrentUser(); if (visitor.getUid() == 0) { Utils.sendTemporaryRedirect(response, "/login"); } @@ -102,7 +103,7 @@ public class PM { @RequestMapping(value = "/pm/send", method = RequestMethod.POST) public void doPostPM(HttpServletRequest request, HttpServletResponse response) throws IOException { - com.juick.User visitor = webApp.getVisitorUser(request, response); + com.juick.User visitor = UserUtils.getCurrentUser(); if (visitor.getUid() == 0 || visitor.isBanned()) { response.sendError(HttpServletResponse.SC_FORBIDDEN); return; |