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
117
118
|
%%--------------------------------------------------------------------
%% %CopyrightBegin%
%%
%% Copyright Ericsson AB 2010-2011. 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,_} ->
Error
end.
% check file exists
check_parameter(File) ->
case filelib:is_file(File) of
true ->
{ok,{file,File}};
false ->
{error,{nofile,File}}
end.
% actual reading of the config
do_read_xml_config(ConfigFile) ->
case catch xmerl_sax_parser:file(ConfigFile,
[{event_fun,fun event/3},
{event_state,[]}]) of
{ok,EntityList,_} ->
{ok,lists:reverse(transform_entity_list(EntityList))};
Oops ->
{error,{parsing_failed,Oops}}
end.
% 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(lists:reverse([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
list_to_term(String) ->
{ok, T, _} = erl_scan:string(String++"."),
case catch erl_parse:parse_term(T) of
{ok,Term} ->
{ok,Term};
Error ->
{error,{Error,String}}
end.
|
