From 9f770c26d1e4f392d591bf35886e3dcc7371d64f Mon Sep 17 00:00:00 2001
From: Vitaly Takmazov
Date: Wed, 29 Mar 2017 14:11:46 +0300
Subject: juick-www: Spring Security

---
 juick-www/src/main/java/com/juick/www/controllers/PM.java | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'juick-www/src/main/java/com/juick/www/controllers/PM.java')

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;
-- 
cgit v1.2.3