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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
%%--------------------------------------------------------------------
%% %CopyrightBegin%
%%
%% Copyright Ericsson AB 2010. All Rights Reserved.
%%
%% The contents of this file are subject to the Erlang Public License,
%% Version 1.1, (the "License"); you may not use this file except in
%% compliance with the License. You should have received a copy of the
%% Erlang Public License along with this software. If not, it can be
%% retrieved online at http://www.erlang.org/.
%%
%% Software distributed under the License is distributed on an "AS IS"
%% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
%% the License for the specific language governing rights and limitations
%% under the License.
%%
%% %CopyrightEnd%
%%----------------------------------------------------------------------
%% File : ct_config_xml.erl
%% Description : CT callback module for reading configs from XML files
%%
%% Created : 16 February 2010
%%----------------------------------------------------------------------
-module(ct_config_xml).
-export([read_config/1, check_parameter/1]).
% read config file
read_config(ConfigFile) ->
case catch do_read_xml_config(ConfigFile) of
{ok, Config}->
{ok, Config};
{error, Error, ErroneousString}->
{error, Error, ErroneousString}
end.
% check against existence of the file
check_parameter(File)->
case filelib:is_file(File) of
true->
{ok, {file, File}};
false->
{nok, {nofile, File}}
end.
% actual reading of the config
do_read_xml_config(ConfigFile)->
{ok, EntityList, _}=
xmerl_sax_parser:file(ConfigFile,
[{event_fun, fun event/3},
{event_state, []}]),
{ok, transform_entity_list(EntityList)}.
% event callback for xmerl_sax_parser
event(Event, _LineNo, State) ->
tag(Event, State).
% document start
tag(startDocument, State) ->
State;
% start of the config
tag({startElement, _Uri, "config", _QName, _Attributes}, []) ->
[{"config", []}];
% start tag
tag({startElement, _Uri, Name, _QName, _Attributes}, Tags) ->
[{Name, []}|Tags];
% value
tag({characters, String}, [{Tag, _Value}|Tags]) ->
[{Tag, String}|Tags];
% end tag
tag({endElement, _Uri, _Name, _QName},
[Entity, {PrevEntityTag, PrevEntityValue}|Tags]) ->
NewHead = {PrevEntityTag, [Entity|PrevEntityValue]},
[NewHead|Tags];
% end of the config
tag({endElement, _Uri, "config", _QName}, [{"config", Config}]) ->
Config;
% end of document, return result
tag(endDocument, {_Tags, Result}) ->
Result;
% default
tag(_El, State) ->
State.
% transform of the ugly deeply nested entity list to the key-value "tree"
transform_entity_list(EntityList)->
lists:map(fun transform_entity/1, EntityList).
% transform entity from {list(), list()} to {atom(), term()}
transform_entity({Tag, [Value|Rest]}) when
is_tuple(Value)->
{list_to_atom(Tag), transform_entity_list([Value|Rest])};
transform_entity({Tag, String})->
case list_to_term(String) of
{ok, Value}->
{list_to_atom(Tag), Value};
Error->
throw(Error)
end.
% transform a string with Erlang terms to the terms
% stolen from trapexit.org :-)
list_to_term(String) ->
{ok, T, _} = erl_scan:string(String++"."),
case erl_parse:parse_term(T) of
{ok, Term} ->
{ok, Term};
Error ->
{error, Error, String}
end.
|
