starting chat service
This commit is contained in:
parent
027c020a34
commit
cb600dc0da
100
priv/chat.html
Normal file
100
priv/chat.html
Normal file
@ -0,0 +1,100 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="en">
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8">
|
||||||
|
<title>Chat with Websockets</title>
|
||||||
|
<link rel="stylesheet" href="./default.css">
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div class="content">
|
||||||
|
<h1 class="content-title">Chat with websockets</h1>
|
||||||
|
|
||||||
|
<div class="content-body">
|
||||||
|
<input autofocus label="Nick" id="nick"></input>
|
||||||
|
<textarea hidden disabled id="wfc-output"></textarea>
|
||||||
|
<input hidden id="wfc-input"></input>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
let nelt = document.getElementById('nick');
|
||||||
|
let ielt = document.getElementById('wfc-input');
|
||||||
|
let oelt = document.getElementById('wfc-output');
|
||||||
|
let ws = new WebSocket("/ws/chat");
|
||||||
|
let nick = '';
|
||||||
|
|
||||||
|
// when user hits any key while typing in nick
|
||||||
|
function on_nick(evt) {
|
||||||
|
if (evt.key === 'Enter') {
|
||||||
|
// don't do default thing
|
||||||
|
evt.preventDefault();
|
||||||
|
// grab contents
|
||||||
|
let contents = nelt.value;
|
||||||
|
let trimmed = contents.trim();
|
||||||
|
// if contents are nonempty
|
||||||
|
let nonempty_contents = trimmed.length > 0;
|
||||||
|
if (nonempty_contents) {
|
||||||
|
nick = trimmed;
|
||||||
|
let msg_obj = ['nick', nick];
|
||||||
|
let msg_str = JSON.stringify(msg_obj);
|
||||||
|
console.log('message to server:', contents.trim());
|
||||||
|
// query backend for result
|
||||||
|
ws.send(msg_str);
|
||||||
|
|
||||||
|
// delete element from dom
|
||||||
|
nelt.remove();
|
||||||
|
oelt.hidden = false;
|
||||||
|
ielt.hidden = false;
|
||||||
|
ielt.autofocus = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// when user hits any key while typing in ielt
|
||||||
|
function on_input_key(evt) {
|
||||||
|
if (evt.key === 'Enter') {
|
||||||
|
// don't do default thing
|
||||||
|
evt.preventDefault();
|
||||||
|
// grab contents
|
||||||
|
let contents = ielt.value;
|
||||||
|
let trimmed = contents.trim();
|
||||||
|
// if contents are nonempty
|
||||||
|
let nonempty_contents = trimmed.length > 0;
|
||||||
|
if (nonempty_contents) {
|
||||||
|
let msg_obj = ['chat', trimmed];
|
||||||
|
let msg_str = JSON.stringify(msg_obj);
|
||||||
|
console.log('message to server:', contents.trim());
|
||||||
|
// query backend for result
|
||||||
|
ws.send(msg_str);
|
||||||
|
|
||||||
|
// clear input
|
||||||
|
ielt.value = '';
|
||||||
|
|
||||||
|
// add to output
|
||||||
|
oelt.value += '> ';
|
||||||
|
oelt.value += trimmed;
|
||||||
|
oelt.value += '\n';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function main() {
|
||||||
|
nelt.addEventListener('keydown', on_nick);
|
||||||
|
ielt.addEventListener('keydown', on_input_key);
|
||||||
|
ws.onmessage =
|
||||||
|
function (msg_evt) {
|
||||||
|
console.log('message from server:', msg_evt);
|
||||||
|
let msg_str = msg_evt.data;
|
||||||
|
let msg_obj = JSON.parse(msg_str);
|
||||||
|
|
||||||
|
oelt.value += msg_obj.nick;
|
||||||
|
oelt.value += '> ';
|
||||||
|
oelt.value += msg_obj.msg;
|
||||||
|
oelt.value += '\n';
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
main();
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
@ -10,9 +10,8 @@
|
|||||||
<h1 class="content-title">WFC Demo</h1>
|
<h1 class="content-title">WFC Demo</h1>
|
||||||
|
|
||||||
<ul>
|
<ul>
|
||||||
<li>
|
<li><a href="/chat.html">Websocket Chatroom</a></li>
|
||||||
<a href="/ws-test-echo.html">Websocket Echo Test</a>
|
<li><a href="/ws-test-echo.html">Websocket Echo Test</a></li>
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
|
|
||||||
<div class="content-body">
|
<div class="content-body">
|
||||||
|
|||||||
@ -9,12 +9,15 @@
|
|||||||
|
|
||||||
-export([
|
-export([
|
||||||
join/1,
|
join/1,
|
||||||
|
relay/1,
|
||||||
nick_available/1
|
nick_available/1
|
||||||
]).
|
]).
|
||||||
-export([start_link/0]).
|
-export([start_link/0]).
|
||||||
-export([init/1, handle_call/3, handle_cast/2, handle_info/2,
|
-export([init/1, handle_call/3, handle_cast/2, handle_info/2,
|
||||||
code_change/3, terminate/2]).
|
code_change/3, terminate/2]).
|
||||||
|
|
||||||
|
-include("$zx_include/zx_logger.hrl").
|
||||||
|
|
||||||
|
|
||||||
-record(o, {pid :: pid(),
|
-record(o, {pid :: pid(),
|
||||||
nick :: string()}).
|
nick :: string()}).
|
||||||
@ -46,6 +49,14 @@ nick_available(Nick) ->
|
|||||||
gen_server:call(?MODULE, {nick_available, Nick}).
|
gen_server:call(?MODULE, {nick_available, Nick}).
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
-spec relay(Message) -> ok
|
||||||
|
when Message :: string().
|
||||||
|
|
||||||
|
relay(Message) ->
|
||||||
|
gen_server:cast(?MODULE, {relay, self(), Message}).
|
||||||
|
|
||||||
|
|
||||||
%%% Startup Functions
|
%%% Startup Functions
|
||||||
|
|
||||||
|
|
||||||
@ -65,7 +76,7 @@ start_link() ->
|
|||||||
%% preparatory work necessary for proper function.
|
%% preparatory work necessary for proper function.
|
||||||
|
|
||||||
init(none) ->
|
init(none) ->
|
||||||
ok = io:format("Starting.~n"),
|
ok = tell("~p Starting.", [?MODULE]),
|
||||||
State = #s{},
|
State = #s{},
|
||||||
{ok, State}.
|
{ok, State}.
|
||||||
|
|
||||||
@ -88,12 +99,12 @@ init(none) ->
|
|||||||
|
|
||||||
handle_call({join, Nick}, {Pid, _}, State) ->
|
handle_call({join, Nick}, {Pid, _}, State) ->
|
||||||
{Reply, NewState} = do_join(Pid, Nick, State),
|
{Reply, NewState} = do_join(Pid, Nick, State),
|
||||||
{reply, Reply, NewState}.
|
{reply, Reply, NewState};
|
||||||
handle_call({nick_available, Nick}, _, State = #s{orators = Orators}) ->
|
handle_call({nick_available, Nick}, _, State = #s{orators = Orators}) ->
|
||||||
Reply = is_nick_available(Nick, Orators),
|
Reply = is_nick_available(Nick, Orators),
|
||||||
{reply, Reply, State}.
|
{reply, Reply, State};
|
||||||
handle_call(Unexpected, From, State) ->
|
handle_call(Unexpected, From, State) ->
|
||||||
ok = io:format("~p Unexpected call from ~tp: ~tp~n", [self(), From, Unexpected]),
|
ok = tell("~p Unexpected call from ~tp: ~tp~n", [?MODULE, From, Unexpected]),
|
||||||
{noreply, State}.
|
{noreply, State}.
|
||||||
|
|
||||||
|
|
||||||
@ -105,8 +116,11 @@ handle_call(Unexpected, From, State) ->
|
|||||||
%% The gen_server:handle_cast/2 callback.
|
%% The gen_server:handle_cast/2 callback.
|
||||||
%% See: http://erlang.org/doc/man/gen_server.html#Module:handle_cast-2
|
%% See: http://erlang.org/doc/man/gen_server.html#Module:handle_cast-2
|
||||||
|
|
||||||
|
handle_cast({relay, From, Message}, State = #s{orators = Orators}) ->
|
||||||
|
do_relay(From, Message, Orators),
|
||||||
|
{noreply, State};
|
||||||
handle_cast(Unexpected, State) ->
|
handle_cast(Unexpected, State) ->
|
||||||
ok = io:format("~p Unexpected cast: ~tp~n", [self(), Unexpected]),
|
ok = tell("~p Unexpected cast: ~tp~n", [?MODULE, Unexpected]),
|
||||||
{noreply, State}.
|
{noreply, State}.
|
||||||
|
|
||||||
|
|
||||||
@ -118,11 +132,11 @@ handle_cast(Unexpected, State) ->
|
|||||||
%% The gen_server:handle_info/2 callback.
|
%% The gen_server:handle_info/2 callback.
|
||||||
%% See: http://erlang.org/doc/man/gen_server.html#Module:handle_info-2
|
%% See: http://erlang.org/doc/man/gen_server.html#Module:handle_info-2
|
||||||
|
|
||||||
handle_info(Msg = {'DOWN', Mon, process, Pid, Reason}, State) ->
|
handle_info(Msg = {'DOWN', _Mon, process, _Pid, _Reason}, State) ->
|
||||||
NewState = handle_down(Msg, State),
|
NewState = handle_down(Msg, State),
|
||||||
{noreply, NewState};
|
{noreply, NewState};
|
||||||
handle_info(Unexpected, State) ->
|
handle_info(Unexpected, State) ->
|
||||||
ok = io:format("~p Unexpected info: ~tp~n", [self(), Unexpected]),
|
ok = tell("~p Unexpected info: ~tp~n", [?MODULE, Unexpected]),
|
||||||
{noreply, State}.
|
{noreply, State}.
|
||||||
|
|
||||||
|
|
||||||
@ -177,15 +191,15 @@ do_join(Pid, Nick, State = #s{orators = Orators}) ->
|
|||||||
do_join2(Pid, Nick, State = #s{orators = Orators}) ->
|
do_join2(Pid, Nick, State = #s{orators = Orators}) ->
|
||||||
_Monitor = erlang:monitor(process, Pid),
|
_Monitor = erlang:monitor(process, Pid),
|
||||||
NewOrator = #o{pid = Pid, nick = Nick},
|
NewOrator = #o{pid = Pid, nick = Nick},
|
||||||
NewOrators = [NewOrator | NewOrators],
|
NewOrators = [NewOrator | Orators],
|
||||||
NewState = State#s{orators = NewOrators},
|
NewState = State#s{orators = NewOrators},
|
||||||
{ok, NewState}.
|
{ok, NewState}.
|
||||||
|
|
||||||
|
|
||||||
-spec ensure_can_join(Pid, Nick, Orators) -> Result.
|
-spec ensure_can_join(Pid, Nick, Orators) -> Result
|
||||||
when Pid :: pid(),
|
when Pid :: pid(),
|
||||||
Nick :: string(),
|
Nick :: string(),
|
||||||
Orators :: [orator()]
|
Orators :: [orator()],
|
||||||
Result :: ok
|
Result :: ok
|
||||||
| {error, Reason},
|
| {error, Reason},
|
||||||
Reason :: any().
|
Reason :: any().
|
||||||
@ -195,7 +209,7 @@ do_join2(Pid, Nick, State = #s{orators = Orators}) ->
|
|||||||
ensure_can_join(Pid, _ , [#o{pid = Pid} | _ ]) -> {error, already_joined};
|
ensure_can_join(Pid, _ , [#o{pid = Pid} | _ ]) -> {error, already_joined};
|
||||||
ensure_can_join(_ , Nick, [#o{nick = Nick} | _ ]) -> {error, {nick_taken, Nick}};
|
ensure_can_join(_ , Nick, [#o{nick = Nick} | _ ]) -> {error, {nick_taken, Nick}};
|
||||||
ensure_can_join(Pid, Nick, [_ | Rest]) -> ensure_can_join(Pid, Nick, Rest);
|
ensure_can_join(Pid, Nick, [_ | Rest]) -> ensure_can_join(Pid, Nick, Rest);
|
||||||
ensure_can_join(Pid, _, _, [] ) -> ok.
|
ensure_can_join(_ , _ , [] ) -> ok.
|
||||||
|
|
||||||
|
|
||||||
-spec is_nick_available(Nick, Orators) -> boolean()
|
-spec is_nick_available(Nick, Orators) -> boolean()
|
||||||
@ -204,13 +218,13 @@ ensure_can_join(Pid, _, _, [] ) -> ok.
|
|||||||
|
|
||||||
is_nick_available(Nick, [#o{nick = Nick} | _ ]) -> false;
|
is_nick_available(Nick, [#o{nick = Nick} | _ ]) -> false;
|
||||||
is_nick_available(Nick, [_ | Rest]) -> is_nick_available(Nick, Rest);
|
is_nick_available(Nick, [_ | Rest]) -> is_nick_available(Nick, Rest);
|
||||||
is_nick_available(Nick, [] ) -> true.
|
is_nick_available(_ , [] ) -> true.
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
-spec handle_down(Msg, State) -> NewState
|
-spec handle_down(Msg, State) -> NewState
|
||||||
when Msg :: {'DOWN', Mon, process, Pid, Reason},
|
when Msg :: {'DOWN', Mon, process, Pid, Reason},
|
||||||
Mon :: monitor(),
|
Mon :: erlang:monitor(),
|
||||||
Pid :: pid(),
|
Pid :: pid(),
|
||||||
Reason :: any(),
|
Reason :: any(),
|
||||||
State :: state(),
|
State :: state(),
|
||||||
@ -224,6 +238,25 @@ handle_down(Msg = {'DOWN', _, process, Pid, _}, State = #s{orators = Orators}) -
|
|||||||
% encountered item, removing
|
% encountered item, removing
|
||||||
hdn(_, Pid, [#o{pid = Pid} | Rest], Acc) -> Rest ++ Acc;
|
hdn(_, Pid, [#o{pid = Pid} | Rest], Acc) -> Rest ++ Acc;
|
||||||
hdn(Msg, Pid, [Skip | Rest], Acc) -> hdn(Msg, Pid, Rest, [Skip | Acc]);
|
hdn(Msg, Pid, [Skip | Rest], Acc) -> hdn(Msg, Pid, Rest, [Skip | Acc]);
|
||||||
hdn(Msg, Pid, [] , Acc) ->
|
hdn(Msg, _, [] , Acc) ->
|
||||||
log("~tp: Unexpected message: ~tp", [?MODULE, Msg]),
|
log("~tp: Unexpected message: ~tp", [?MODULE, Msg]),
|
||||||
Acc.
|
Acc.
|
||||||
|
|
||||||
|
|
||||||
|
do_relay(Pid, Message, Orators) ->
|
||||||
|
case lists:keyfind(Pid, #o.pid, Orators) of
|
||||||
|
#o{nick = Nick} ->
|
||||||
|
do_relay2(Nick, Message, Orators);
|
||||||
|
false ->
|
||||||
|
tell("~tp: Message received from outsider ~tp: ~tp", [?MODULE, Pid, Message]),
|
||||||
|
error
|
||||||
|
end.
|
||||||
|
|
||||||
|
% skip
|
||||||
|
do_relay2(Nick, Msg, [#o{nick = Nick} | Rest]) ->
|
||||||
|
do_relay2(Nick, Msg, Rest);
|
||||||
|
do_relay2(Nick, Msg, [#o{pid = Pid} | Rest]) ->
|
||||||
|
Pid ! {chat, {relay, Nick, Msg}},
|
||||||
|
do_relay2(Nick, Msg, Rest);
|
||||||
|
do_relay2(_, _, []) ->
|
||||||
|
ok.
|
||||||
|
|||||||
@ -226,6 +226,7 @@ route(Sock, get, Route, Request) ->
|
|||||||
case Route of
|
case Route of
|
||||||
<<"/">> -> home(Sock);
|
<<"/">> -> home(Sock);
|
||||||
<<"/default.css">> -> default_css(Sock);
|
<<"/default.css">> -> default_css(Sock);
|
||||||
|
<<"/chat.html">> -> chat_html(Sock);
|
||||||
<<"/ws-test-echo.html">> -> ws_test_echo_html(Sock);
|
<<"/ws-test-echo.html">> -> ws_test_echo_html(Sock);
|
||||||
<<"/ws/echo">> -> ws_echo(Sock, Request);
|
<<"/ws/echo">> -> ws_echo(Sock, Request);
|
||||||
_ -> http_err(Sock, 404)
|
_ -> http_err(Sock, 404)
|
||||||
@ -318,6 +319,19 @@ ws_test_echo_html(Sock) ->
|
|||||||
http_err(Sock, 500)
|
http_err(Sock, 500)
|
||||||
end.
|
end.
|
||||||
|
|
||||||
|
chat_html(Sock) ->
|
||||||
|
%% fixme: cache
|
||||||
|
Path_IH = filename:join([zx:get_home(), "priv", "chat.html"]),
|
||||||
|
case file:read_file(Path_IH) of
|
||||||
|
{ok, Body} ->
|
||||||
|
Resp = #response{headers = [{"content-type", "text/html"}],
|
||||||
|
body = Body},
|
||||||
|
respond(Sock, Resp);
|
||||||
|
Error ->
|
||||||
|
io:format("~p error: ~p~n", [self(), Error]),
|
||||||
|
http_err(Sock, 500)
|
||||||
|
end.
|
||||||
|
|
||||||
wfcin(Sock, #request{enctype = json,
|
wfcin(Sock, #request{enctype = json,
|
||||||
cookies = Cookies,
|
cookies = Cookies,
|
||||||
body = #{"wfcin" := Input}}) ->
|
body = #{"wfcin" := Input}}) ->
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user