mirror of
https://github.com/processone/ejabberd.git
synced 2024-12-26 17:38:45 +01:00
77 lines
2.0 KiB
Erlang
77 lines
2.0 KiB
Erlang
|
%%%----------------------------------------------------------------------
|
||
|
|
%%% File : xml_stream.erl
|
||
|
|
%%% Author : Alexey Shchepin <alexey@sevcom.net>
|
||
|
|
%%% Purpose :
|
||
|
|
%%% Created : 17 Nov 2002 by Alexey Shchepin <alexey@sevcom.net>
|
||
|
|
%%% Id : $Id$
|
||
|
|
%%%----------------------------------------------------------------------
|
||
|
|
|
||
|
|
-module(xml_stream).
|
||
|
|
-author('alexey@sevcom.net').
|
||
|
|
-vsn('$Revision$ ').
|
||
|
|
|
||
|
|
-export([start/1, init/1, send_text/2]).
|
||
|
|
|
||
|
|
start(CallbackPid) ->
|
||
|
|
spawn(?MODULE, init, [CallbackPid]).
|
||
|
|
|
||
|
|
init(CallbackPid) ->
|
||
|
|
Port = open_port({spawn, expat_erl}, [binary]),
|
||
|
|
loop(CallbackPid, Port, []).
|
||
|
|
|
||
|
|
loop(CallbackPid, Port, Stack) ->
|
||
|
|
receive
|
||
|
|
{Port, {data, Bin}} ->
|
||
|
|
%CallbackPid ! binary_to_term(Bin),
|
||
|
|
Data = binary_to_term(Bin),
|
||
|
|
loop(CallbackPid, Port, process_data(CallbackPid, Stack, Data));
|
||
|
|
{From, {send, Str}} ->
|
||
|
|
Port ! {self(), {command, Str}},
|
||
|
|
loop(CallbackPid, Port, Stack)
|
||
|
|
end.
|
||
|
|
|
||
|
|
process_data(CallbackPid, Stack, Data) ->
|
||
|
|
case Data of
|
||
|
|
{xmlstart, {Name, Attrs}} ->
|
||
|
|
if Stack == [] ->
|
||
|
|
gen_fsm:send_event(CallbackPid,
|
||
|
|
{xmlstreamstart, Name, Attrs});
|
||
|
|
true -> true
|
||
|
|
end,
|
||
|
|
[{xmlelement, Name, Attrs, []} | Stack];
|
||
|
|
{xmlend, EndName} ->
|
||
|
|
case Stack of
|
||
|
|
[{xmlelement, Name, Attrs, Els} | Tail] ->
|
||
|
|
NewEl = {xmlelement, Name, Attrs, lists:reverse(Els)},
|
||
|
|
Len = length(Tail),
|
||
|
|
if
|
||
|
|
Len > 1 -> add_subelement(NewEl, Tail);
|
||
|
|
Len == 1 ->
|
||
|
|
gen_fsm:send_event(CallbackPid,
|
||
|
|
{xmlstreamelement, NewEl}),
|
||
|
|
Tail;
|
||
|
|
Len == 0 ->
|
||
|
|
gen_fsm:send_event(CallbackPid,
|
||
|
|
{xmlstreamend, EndName}),
|
||
|
|
Tail
|
||
|
|
end
|
||
|
|
end;
|
||
|
|
{xmlcdata, CData} ->
|
||
|
|
add_subelement({xmlcdata, CData}, Stack);
|
||
|
|
{xmlerror, Err} -> gen_fsm:send_event(CallbackPid,
|
||
|
|
{xmlstreamerror, Err})
|
||
|
|
end.
|
||
|
|
|
||
|
|
|
||
|
|
add_subelement(El, Stack) ->
|
||
|
|
case Stack of
|
||
|
|
[{xmlelement, Name, Attrs, Els} | Tail] ->
|
||
|
|
[{xmlelement, Name, Attrs, [El | Els]} | Tail];
|
||
|
|
[] -> []
|
||
|
|
end.
|
||
|
|
|
||
|
|
|
||
|
|
send_text(Pid, Text) ->
|
||
|
|
Pid ! {self(), {send, Text}}.
|
||
|
|
|