WebSocket authentication

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

diff --git a/org.argeo.cms.ee/src/org/argeo/cms/websocket/server/WebSocketHandshakeResponse.java b/org.argeo.cms.ee/src/org/argeo/cms/websocket/server/WebSocketHandshakeResponse.java
new file mode 100644 (file)
index 0000000..3a978c8
--- /dev/null
+++ b/org.argeo.cms.ee/src/org/argeo/cms/websocket/server/WebSocketHandshakeResponse.java
@@ -0,0 +1,22 @@
+package org.argeo.cms.websocket.server;
+
+import java.util.Collections;
+
+import javax.websocket.HandshakeResponse;
+
+import org.argeo.cms.auth.RemoteAuthResponse;
+
+public class WebSocketHandshakeResponse implements RemoteAuthResponse {
+       private final HandshakeResponse handshakeResponse;
+
+       public WebSocketHandshakeResponse(HandshakeResponse handshakeResponse) {
+               this.handshakeResponse = handshakeResponse;
+       }
+
+       @Override
+       public void setHeader(String key, String value) {
+               handshakeResponse.getHeaders().put(key, Collections.singletonList(value));
+
+       }
+
+}