juick-www: Spring Security

1 files changed, 5 insertions, 4 deletions

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;