From 4319a2c2de6391f3e42993e96675a64baf9f0d78 Mon Sep 17 00:00:00 2001
From: Vitaly Takmazov
Date: Sun, 27 May 2018 23:15:12 +0300
Subject: ws: accept commands from authenticated users

---
 .../src/main/java/com/juick/server/WebsocketManager.java     | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

(limited to 'juick-server/src/main/java/com/juick/server/WebsocketManager.java')

diff --git a/juick-server/src/main/java/com/juick/server/WebsocketManager.java b/juick-server/src/main/java/com/juick/server/WebsocketManager.java
index 85950c2c..5c976416 100644
--- a/juick-server/src/main/java/com/juick/server/WebsocketManager.java
+++ b/juick-server/src/main/java/com/juick/server/WebsocketManager.java
@@ -156,10 +156,14 @@ public class WebsocketManager extends TextWebSocketHandler {
     protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
         User user = clients.stream().filter(c -> c.session.equals(session)).map(c -> c.visitor)
                 .findFirst().orElse(AnonymousUser.INSTANCE);
-        String command = message.getPayload().trim();
-        if (StringUtils.isNotEmpty(command)) {
-            CommandResult result = commandsManager.processCommand(user, command, URI.create(""));
-            session.sendMessage(new TextMessage(result.getText()));
+        if (!user.isAnonymous()) {
+            String command = message.getPayload().trim();
+            if (StringUtils.isNotEmpty(command)) {
+                CommandResult result = commandsManager.processCommand(user, command, URI.create(""));
+                session.sendMessage(new TextMessage(result.getText()));
+            }
+        } else {
+            session.sendMessage(new TextMessage("Authorization required"));
         }
     }
 
-- 
cgit v1.2.3