From 5ef4a15b48bfc1b5ca867b893b7cbd1b535175f7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= <essen@ninenines.eu>
Date: Mon, 3 Dec 2012 14:13:46 +0100
Subject: Allow passing the Req and an updated Opts when upgrading protocols

---
 test/ws_upgrade_with_opts_handler.erl | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 test/ws_upgrade_with_opts_handler.erl

(limited to 'test/ws_upgrade_with_opts_handler.erl')

diff --git a/test/ws_upgrade_with_opts_handler.erl b/test/ws_upgrade_with_opts_handler.erl
new file mode 100644
index 0000000..02d755e
--- /dev/null
+++ b/test/ws_upgrade_with_opts_handler.erl
@@ -0,0 +1,28 @@
+%% Feel free to use, reuse and abuse the code in this file.
+
+-module(ws_upgrade_with_opts_handler).
+-behaviour(cowboy_websocket_handler).
+
+-export([init/3]).
+-export([websocket_init/3]).
+-export([websocket_handle/3]).
+-export([websocket_info/3]).
+-export([websocket_terminate/3]).
+
+init(_Any, Req, _Opts) ->
+	{upgrade, protocol, cowboy_websocket, Req, <<"success">>}.
+
+websocket_init(_TransportName, Req, Response) ->
+	Req2 = cowboy_req:compact(Req),
+	erlang:send_after(10, self(), send_response),
+	{ok, Req2, Response}.
+
+websocket_handle(_Frame, Req, State) ->
+	{ok, Req, State}.
+
+websocket_info(send_response, Req, State = Response)
+		when is_binary(Response) ->
+	{reply, {text, Response}, Req, State}.
+
+websocket_terminate(_Reason, _Req, _State) ->
+	ok.
-- 
cgit v1.2.3