juick-www: Spring Security

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;