1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
-module(ephraim).
-compile([debug_info, export_all]).
-record(state, {
conn :: pid(),
convs :: dict(),
uis :: set()
}).
-spec start() -> ok.
start() ->
Pid = spawn(?MODULE, init, []),
case register(ephraim, Pid) of
true ->
ok
end,
open_port({spawn_executable, "ephraim-gtk"}, [stream, use_stdio, out]).
-spec stop() -> ok.
stop() ->
ephraim ! stop,
ok.
-spec init() -> ok.
init() ->
Conn = spawn(ephraim_conn, init, []),
loop(#state{conn=Conn,convs=dict:new(),uis=sets:new()}),
init:stop().
-spec get_conv(#state{},exmpp_jid:jid()) -> {#state{},pid()}.
get_conv(State, JID) ->
Key = exmpp_jid:to_lower(JID),
case dict:find(Key, State#state.convs) of
{ok, Conv} ->
{State, Conv};
error ->
Conv = spawn(ephraim_conv, init, [Key]),
Dict = dict:store(Key, Conv, State#state.convs),
{State#state{convs=Dict}, Conv}
end.
-spec loop(#state{}) -> ok.
loop(State) ->
receive
stop ->
dict:fold(fun(_,Conv,Msg) -> Conv ! Msg end, stop, State#state.convs),
State#state.conn ! stop,
ok;
{register_ui, Pid} ->
io:format("Registered UI ~p~n", [Pid]),
UIs = State#state.uis,
State2 = State#state{uis=sets:add_element(Pid,UIs)},
loop(State2);
{unregister_ui, Pid} ->
io:format("Unregistered UI ~p~n", [Pid]),
UIs = State#state.uis,
State2 = State#state{uis=sets:del_element(Pid,UIs)},
Size = sets:size(State2#state.uis),
if Size =:= 0 ->
self() ! stop;
true ->
ok
end,
loop(State2);
{receive_message, From, Packet} ->
{State2, Conv} = get_conv(State, From),
Conv ! {receive_message, Packet},
loop(State2);
Msg ->
io:format("ephraim: ~p~n", [Msg]),
loop(State)
end.
|
