1 /**
2 * Copyright (c) 2016, Timothy Stack
3 *
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions are met:
8 *
9 * * Redistributions of source code must retain the above copyright notice, this
10 * list of conditions and the following disclaimer.
11 * * Redistributions in binary form must reproduce the above copyright notice,
12 * this list of conditions and the following disclaimer in the documentation
13 * and/or other materials provided with the distribution.
14 * * Neither the name of Timothy Stack nor the names of its contributors
15 * may be used to endorse or promote products derived from this software
16 * without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ''AS IS'' AND ANY
19 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
21 * DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY
22 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
23 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
24 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
25 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
27 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
30 #include "config.h"
31
32 #include <algorithm>
33
34 #include "elem_to_json.hh"
35 #include "yajlpp/yajlpp.hh"
36
37 using namespace std;
38
39 static
element_to_json(yajl_gen gen,data_parser & dp,const data_parser::element & elem)40 void element_to_json(yajl_gen gen, data_parser &dp, const data_parser::element &elem)
41 {
42 size_t value_len;
43 const char *value_str = dp.get_element_string(elem, value_len);
44
45 switch (elem.value_token()) {
46 case DT_NUMBER: {
47 yajl_gen_number(gen, value_str, value_len);
48 break;
49 }
50 case DNT_GROUP: {
51 elements_to_json(gen, dp, elem.get_value_elem().e_sub_elements, false);
52 break;
53 }
54 case DNT_PAIR: {
55 const data_parser::element &pair_elem = elem.get_pair_elem();
56 string key_str = dp.get_element_string(pair_elem.e_sub_elements->front());
57
58 if (!key_str.empty()) {
59 yajlpp_map singleton_map(gen);
60
61 singleton_map.gen(key_str);
62 element_to_json(gen, dp, pair_elem.get_pair_value());
63 }
64 else {
65 element_to_json(gen, dp, pair_elem.get_pair_value());
66 }
67 break;
68 }
69 case DT_CONSTANT: {
70 if (strncasecmp("true", value_str, value_len) == 0) {
71 yajl_gen_bool(gen, true);
72 }
73 else if (strncasecmp("false", value_str, value_len) == 0) {
74 yajl_gen_bool(gen, false);
75 }
76 else {
77 yajl_gen_null(gen);
78 }
79 break;
80 }
81 default:
82 yajl_gen_pstring(gen, value_str, value_len);
83 break;
84 }
85 }
86
87 static
map_elements_to_json2(yajl_gen gen,data_parser & dp,data_parser::element_list_t * el)88 void map_elements_to_json2(yajl_gen gen, data_parser &dp, data_parser::element_list_t *el)
89 {
90 yajlpp_map root_map(gen);
91 int col = 0;
92
93 for (auto &iter : *el) {
94 const data_parser::element &pvalue = iter.get_pair_value();
95
96 if (pvalue.value_token() == DT_INVALID) {
97 log_debug("invalid!!");
98 // continue;
99 }
100
101 std::string key_str = dp.get_element_string(
102 iter.e_sub_elements->front());
103
104 if (key_str.empty()) {
105 char buffer[32];
106
107 snprintf(buffer, sizeof(buffer), "col_%d", col);
108 key_str = buffer;
109 col += 1;
110 }
111 root_map.gen(key_str);
112 element_to_json(gen, dp, pvalue);
113 }
114 }
115
116 static
list_body_elements_to_json(yajl_gen gen,data_parser & dp,data_parser::element_list_t * el)117 void list_body_elements_to_json(yajl_gen gen, data_parser &dp, data_parser::element_list_t *el)
118 {
119 for (auto &iter : *el) {
120 element_to_json(gen, dp, iter);
121 }
122 }
123
124 static
list_elements_to_json(yajl_gen gen,data_parser & dp,data_parser::element_list_t * el)125 void list_elements_to_json(yajl_gen gen, data_parser &dp, data_parser::element_list_t *el)
126 {
127 yajlpp_array root_array(gen);
128
129 list_body_elements_to_json(gen, dp, el);
130 }
131
132 static
map_elements_to_json(yajl_gen gen,data_parser & dp,data_parser::element_list_t * el)133 void map_elements_to_json(yajl_gen gen, data_parser &dp, data_parser::element_list_t *el)
134 {
135 bool unique_names = el->size() > 1;
136 vector<string> names;
137
138 for (auto &iter : *el) {
139 const data_parser::element &pvalue = iter.get_pair_value();
140
141 if (pvalue.value_token() == DT_INVALID) {
142 log_debug("invalid!!");
143 // continue;
144 }
145
146 std::string key_str = dp.get_element_string(
147 iter.e_sub_elements->front());
148 if (key_str.empty()) {
149 continue;
150 }
151 if (find(names.begin(), names.end(), key_str) != names.end()) {
152 unique_names = false;
153 break;
154 }
155 else {
156 names.push_back(key_str);
157 }
158 }
159
160 names.clear();
161
162 if (unique_names) {
163 map_elements_to_json2(gen, dp, el);
164 }
165 else {
166 list_elements_to_json(gen, dp, el);
167 }
168 }
169
elements_to_json(yajl_gen gen,data_parser & dp,data_parser::element_list_t * el,bool root)170 void elements_to_json(yajl_gen gen, data_parser &dp, data_parser::element_list_t *el, bool root)
171 {
172 if (el->empty()) {
173 yajl_gen_null(gen);
174 }
175 else {
176 switch (el->front().e_token) {
177 case DNT_PAIR: {
178 if (root && el->size() == 1) {
179 const data_parser::element &pair_elem = el->front().get_pair_elem();
180 std::string key_str = dp.get_element_string(
181 pair_elem.e_sub_elements->front());
182
183 if (key_str.empty() &&
184 el->front().get_pair_value().value_token() == DNT_GROUP) {
185 element_to_json(gen, dp, el->front().get_pair_value());
186 }
187 else {
188 yajlpp_map singleton_map(gen);
189
190 if (key_str.empty()) {
191 key_str = "col_0";
192 }
193 singleton_map.gen(key_str);
194 element_to_json(gen, dp, pair_elem.get_pair_value());
195 }
196 }
197 else {
198 map_elements_to_json(gen, dp, el);
199 }
200 break;
201 }
202 default:
203 list_elements_to_json(gen, dp, el);
204 break;
205 }
206 }
207 }
208