WebSocket authentication

[lgpl/argeo-commons.git] / org.argeo.cms.ee / src / org / argeo / cms / websocket / server / WebSocketHandshakeRequest.java

diff --git a/org.argeo.cms.ee/src/org/argeo/cms/websocket/server/WebSocketHandshakeRequest.java b/org.argeo.cms.ee/src/org/argeo/cms/websocket/server/WebSocketHandshakeRequest.java
new file mode 100644 (file)
index 0000000..31bcf92
--- /dev/null
+++ b/org.argeo.cms.ee/src/org/argeo/cms/websocket/server/WebSocketHandshakeRequest.java
@@ -0,0 +1,82 @@
+package org.argeo.cms.websocket.server;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Locale;
+import java.util.Map;
+import java.util.Objects;
+
+import javax.servlet.http.HttpSession;
+import javax.websocket.server.HandshakeRequest;
+
+import org.argeo.cms.auth.RemoteAuthRequest;
+import org.argeo.cms.auth.RemoteAuthSession;
+import org.argeo.cms.servlet.ServletHttpSession;
+
+public class WebSocketHandshakeRequest implements RemoteAuthRequest {
+       private final HandshakeRequest handshakeRequest;
+       private final HttpSession httpSession;
+
+       private Map<String, Object> attributes = new HashMap<>();
+
+       public WebSocketHandshakeRequest(HandshakeRequest handshakeRequest) {
+               Objects.requireNonNull(handshakeRequest);
+               this.handshakeRequest = handshakeRequest;
+               this.httpSession = (HttpSession) handshakeRequest.getHttpSession();
+//             Objects.requireNonNull(this.httpSession);
+       }
+
+       @Override
+       public RemoteAuthSession getSession() {
+               if (httpSession == null)
+                       return null;
+               return new ServletHttpSession(httpSession);
+       }
+
+       @Override
+       public RemoteAuthSession createSession() {
+               throw new UnsupportedOperationException();
+       }
+
+       @Override
+       public Locale getLocale() {
+               // TODO check Accept-Language header
+               return Locale.getDefault();
+       }
+
+       @Override
+       public Object getAttribute(String key) {
+               return attributes.get(key);
+       }
+
+       @Override
+       public void setAttribute(String key, Object object) {
+               attributes.put(key, object);
+       }
+
+       @Override
+       public String getHeader(String key) {
+               List<String> values = handshakeRequest.getHeaders().get(key);
+               if (values.size() == 0)
+                       return null;
+               if (values.size() > 1)
+                       throw new IllegalStateException("More that one value for " + key + ": " + values);
+               return values.get(0);
+       }
+
+       @Override
+       public String getRemoteAddr() {
+               throw new UnsupportedOperationException();
+       }
+
+       @Override
+       public int getLocalPort() {
+               throw new UnsupportedOperationException();
+       }
+
+       @Override
+       public int getRemotePort() {
+               throw new UnsupportedOperationException();
+       }
+
+}