sockjs_test.erl

-module(sockjs_test).
-export([start/0, dispatcher/0]).

start() ->
    Port = 8080,
    application:start(sockjs),
    
    register(sockjs_mqb, self()),

    {ok, _} = misultin:start_link([{loop,        fun misultin_loop/1},
                                   {ws_loop,     fun misultin_ws_loop/1},
                                   {ws_autoexit, false},
                                   {port,        Port}]),

    io:format("~nRunning on port ~p~n~n", [Port]),
    mqb_handle(start),
    receive
        _ -> ok
    end.

%% --------------------------------------------------------------------------

misultin_loop(Req) ->
    try
        handle({misultin, Req})
    catch A:B ->
            io:format("~s ~p ~p~n", [A, B, erlang:get_stacktrace()]),
            Req:respond(500, [], "500")
    end.

misultin_ws_loop(Ws) ->
    {Receive, _} = ws_handle({misultin, Ws}),
    sockjs_http:misultin_ws_loop(Ws, Receive).

%% --------------------------------------------------------------------------

handle(Req) ->
    {Path0, Req1} = sockjs_http:path(Req),
    Path = clean_path(Path0),
    case sockjs_filters:handle_req(
           Req1, Path, sockjs_test:dispatcher()) of
        nomatch -> case Path of
                       "config.js" -> config_js(Req1);
                       _           -> static(Req1, Path)
                   end;
        Req2    -> Req2
    end.

ws_handle(Req) ->
    {Path0, Req1} = sockjs_http:path(Req),
    Path = clean_path(Path0),
    {Receive, _, _, _} = sockjs_filters:dispatch('GET', Path,
                                                 sockjs_test:dispatcher()),
    {Receive, Req1}.

static(Req, Path) ->
    %% TODO unsafe
    LocalPath = filename:join([module_path(), "priv/www", Path]),
    case file:read_file(LocalPath) of
        {ok, Contents} ->
            sockjs_http:reply(200, [], Contents, Req);
        {error, _} ->
            sockjs_http:reply(404, [], "", Req)
    end.

module_path() ->
    {file, Here} = code:is_loaded(?MODULE),
    filename:dirname(filename:dirname(Here)).

config_js(Req) ->
    %% TODO parse the file? Good luck, it's JS not JSON.
    sockjs_http:reply(
      200, [{"content-type", "application/javascript"}],
      "var client_opts = {\"url\":\"http://localhost:8080\",\"disabled_transports\":[],\"sockjs_opts\":{\"devel\":true}};", Req).

clean_path("/")         -> "index.html";
clean_path("/" ++ Path) -> Path.

%% --------------------------------------------------------------------------

dispatcher() ->
     [{mqb, fun mqb_handle/2}].



mqb_handle(start) ->
    ets:new(broadcast_table, [public, named_table]),
    io:format("Initializing~n"),
    ok.
mqb_handle(Conn, init) ->
    true = ets:insert(broadcast_table, {Conn}),
    {_, SessionId} = Conn,
    io:format("~s: New user connected~n", [SessionId]),
    ok;
mqb_handle(Conn, closed) ->
    true = ets:delete_object(broadcast_table, {Conn}),
    {_, SessionId} = Conn,
    io:format("~s: Connection closed~n", [SessionId]),
    ok; 
mqb_handle(_Conn, {recv, Data}) ->
    ets:foldl(fun({Conn}, _Acc) -> Conn:send(Data) end, [], broadcast_table),
    {_, SessionId} = _Conn,
    io:format("~s: Data received / ~s~n", [SessionId, Data]),
    ok.