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

---
 .../src/main/java/com/juick/www/controllers/NewMessage.java      | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

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

diff --git a/juick-www/src/main/java/com/juick/www/controllers/NewMessage.java b/juick-www/src/main/java/com/juick/www/controllers/NewMessage.java
index 1993737b..fcd27710 100644
--- a/juick-www/src/main/java/com/juick/www/controllers/NewMessage.java
+++ b/juick-www/src/main/java/com/juick/www/controllers/NewMessage.java
@@ -22,6 +22,7 @@ import com.juick.server.helpers.TagStats;
 import com.juick.server.util.HttpBadRequestException;
 import com.juick.server.util.HttpUtils;
 import com.juick.service.*;
+import com.juick.util.UserUtils;
 import com.juick.www.Utils;
 import com.juick.www.WebApp;
 import org.apache.commons.io.FilenameUtils;
@@ -87,7 +88,7 @@ public class NewMessage {
 
     @RequestMapping(value = "/post", method = RequestMethod.GET)
     protected void doGetNewMessage(HttpServletRequest request, HttpServletResponse response) throws IOException {
-        com.juick.User visitor = webApp.getVisitorUser(request, response);
+        com.juick.User visitor = UserUtils.getCurrentUser();
         if (visitor.getUid() == 0) {
             Utils.sendTemporaryRedirect(response, "/login");
             return;
@@ -184,7 +185,7 @@ public class NewMessage {
     public void doPostMessage(HttpServletRequest request, HttpServletResponse response,
                               @RequestParam(required = false) String img,
                               @RequestParam(required = false) MultipartFile attach) throws IOException {
-        com.juick.User visitor = webApp.getVisitorUser(request, response);
+        com.juick.User visitor = UserUtils.getCurrentUser();
         if (visitor.getUid() == 0) {
             response.sendError(HttpServletResponse.SC_FORBIDDEN);
             return;
@@ -317,7 +318,7 @@ public class NewMessage {
     public void doPostComment(HttpServletRequest request, HttpServletResponse response,
                               @RequestParam(required = false) String img,
                               @RequestParam(required = false) MultipartFile attach) throws IOException {
-        com.juick.User visitor = webApp.getVisitorUser(request, response);
+        com.juick.User visitor = UserUtils.getCurrentUser();
         if (visitor.getUid() == 0) {
             response.sendError(HttpServletResponse.SC_FORBIDDEN);
             return;
@@ -433,7 +434,7 @@ public class NewMessage {
 
     @RequestMapping(value = "/like", method = RequestMethod.POST)
     public void doPostRecomm(HttpServletRequest request, HttpServletResponse response) throws IOException {
-        com.juick.User visitor = webApp.getVisitorUser(request, response);
+        com.juick.User visitor = UserUtils.getCurrentUser();
         if (visitor.getUid() == 0) {
             response.sendError(HttpServletResponse.SC_FORBIDDEN);
             return;
-- 
cgit v1.2.3