package com.juick.ws; import com.juick.User; import com.juick.server.MessagesQueries; import com.juick.server.UserQueries; import org.apache.http.NameValuePair; import org.apache.http.client.utils.URLEncodedUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.http.HttpHeaders; import org.springframework.jdbc.core.JdbcTemplate; import org.springframework.util.Assert; import org.springframework.web.socket.CloseStatus; import org.springframework.web.socket.WebSocketSession; import org.springframework.web.socket.handler.TextWebSocketHandler; import java.io.IOException; import java.net.URI; import java.util.ArrayList; import java.util.Collections; import java.util.List; /** * Created by vitalyster on 28.06.2016. */ public class WebsocketComponent extends TextWebSocketHandler { private static final Logger logger = LoggerFactory.getLogger(WebsocketComponent.class); private JdbcTemplate jdbc; private final List clients = Collections.synchronizedList(new ArrayList()); public WebsocketComponent(JdbcTemplate jdbc) { Assert.notNull(jdbc); this.jdbc = jdbc; } @Override public void afterConnectionEstablished(WebSocketSession session) throws Exception { URI hLocation; String hXRealIP = ""; hLocation = session.getUri(); HttpHeaders headers = session.getHandshakeHeaders(); hXRealIP = headers.getOrDefault("X-Real-IP", Collections.singletonList(session.getRemoteAddress().toString())).get(0); // Auth User visitor = new User(); List params = URLEncodedUtils.parse(hLocation, "UTF-8"); for (NameValuePair param : params) { if (param.getName().equals("hash")) { String hash = param.getValue(); if (hash.length() == 16) { visitor = UserQueries.getUserByHash(jdbc, hash); } else { try { logger.info(String.format("wrong hash for %d from %s", visitor.getUid(), hXRealIP)); session.close(new CloseStatus(403, "Forbidden")); } catch (IOException e) { logger.warn("ws error", e); } } break; } } logger.info(String.format("user %d connected to %s from %s", visitor.getUid(), hLocation.getPath(), hXRealIP)); int MID = 0; SocketSubscribed sockSubscr = null; if (hLocation.getPath().equals("/")) { logger.info(String.format("user %d connected", visitor.getUid())); sockSubscr = new SocketSubscribed(session, hXRealIP, visitor, false); } else if (hLocation.getPath().equals("/_all")) { logger.info(String.format("user %d connected to legacy _all (%s)", visitor.getUid(), hLocation.getPath())); sockSubscr = new SocketSubscribed(session, hXRealIP, visitor, true); sockSubscr.allMessages = true; } else if (hLocation.getPath().equals("/_replies")) { logger.info(String.format("user %d connected to legacy _replies (%s)", visitor.getUid(), hLocation.getPath())); sockSubscr = new SocketSubscribed(session, hXRealIP, visitor, true); sockSubscr.allReplies = true; } else if (hLocation.getPath().matches("/\\d+$")) { try { MID = Integer.parseInt(hLocation.getPath().substring(1)); } catch (Exception e) { } if (MID > 0) { if (MessagesQueries.canViewThread(jdbc, MID, visitor.getUid())) { logger.info(String.format("user %d connected to legacy thread (%d) from %s", visitor.getUid(), MID, hXRealIP)); sockSubscr = new SocketSubscribed(session, hXRealIP, visitor, true); sockSubscr.MID = MID; } else { try { session.close(new CloseStatus(403, "Forbidden")); } catch (IOException e) { logger.warn("ws error", e); } } } } if (sockSubscr != null) { synchronized (clients) { clients.add(sockSubscr); logger.info(clients.size() + " clients connected"); } } } @Override public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception { synchronized (clients) { logger.info(String.format("session closed with status %d: %s", status.getCode(), status.getReason())); clients.removeIf(c -> { if (c.session.getId().equals(session.getId())) { return true; } return false; }); logger.info(clients.size() + " clients connected"); } } public List getClients() { return clients; } class SocketSubscribed { WebSocketSession session; String clientName; User visitor; int MID; boolean allMessages; boolean allReplies; long tsConnected; long tsLastData; boolean legacy; public SocketSubscribed(WebSocketSession session, String clientName, User visitor, boolean legacy) { this.session = session; this.clientName = clientName; this.visitor = visitor; tsConnected = tsLastData = System.currentTimeMillis(); this.legacy = legacy; } } }