0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-06-02 18:18:56 +02:00
construct/ircd/json.cc

1109 lines
29 KiB
C++
Raw Normal View History

2016-11-29 16:23:38 +01:00
/*
* charybdis: standing on the shoulders of giant build times
*
* Copyright (C) 2017 Charybdis Development Team
* Copyright (C) 2017 Jason Volk <jason@zemos.net>
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice is present in all copies.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
* IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
#include <ircd/spirit.h>
2016-11-29 16:23:38 +01:00
BOOST_FUSION_ADAPT_STRUCT
(
ircd::json::doc::member,
( decltype(ircd::json::doc::member::first), first )
( decltype(ircd::json::doc::member::second), second )
)
2017-03-11 04:31:20 +01:00
2016-11-29 16:23:38 +01:00
namespace ircd {
namespace json {
namespace spirit = boost::spirit;
namespace ascii = spirit::ascii;
namespace karma = spirit::karma;
namespace qi = spirit::qi;
using spirit::unused_type;
using qi::lit;
using qi::char_;
2017-03-20 23:59:14 +01:00
using qi::long_;
2016-11-29 16:23:38 +01:00
using qi::double_;
using qi::raw;
using qi::omit;
using qi::matches;
using qi::hold;
using qi::eoi;
using qi::eps;
using qi::attr;
using karma::lit;
using karma::char_;
2017-03-20 23:59:14 +01:00
using karma::long_;
using karma::double_;
using karma::bool_;
2016-11-29 16:23:38 +01:00
using karma::maxwidth;
using karma::buffer;
using karma::eps;
2017-03-11 04:31:20 +01:00
using karma::attr_cast;
2016-11-29 16:23:38 +01:00
template<class it>
struct input
:qi::grammar<it, unused_type>
2016-11-29 16:23:38 +01:00
{
template<class T = unused_type> using rule = qi::rule<it, T>;
2017-03-20 23:59:14 +01:00
rule<> NUL { lit('\0') ,"nul" };
2016-11-29 16:23:38 +01:00
// insignificant whitespaces
2017-03-20 23:59:14 +01:00
rule<> SP { lit('\x20') ,"space" };
rule<> HT { lit('\x09') ,"horizontal tab" };
rule<> CR { lit('\x0D') ,"carriage return" };
rule<> LF { lit('\x0A') ,"line feed" };
2016-11-29 16:23:38 +01:00
// whitespace skipping
2017-03-20 23:59:14 +01:00
rule<> WS { SP | HT | CR | LF ,"whitespace" };
rule<> ws { *(WS) ,"whitespace monoid" };
rule<> wsp { +(WS) ,"whitespace semigroup" };
2016-11-29 16:23:38 +01:00
// structural
2017-03-20 23:59:14 +01:00
rule<> object_begin { lit('{') ,"object begin" };
rule<> object_end { lit('}') ,"object end" };
rule<> array_begin { lit('[') ,"array begin" };
rule<> array_end { lit(']') ,"array end" };
rule<> name_sep { lit(':') ,"name sep" };
rule<> value_sep { lit(',') ,"value sep" };
2016-11-29 16:23:38 +01:00
// literal
2017-03-20 23:59:14 +01:00
rule<string_view> lit_true { lit("true") ,"literal true" };
rule<string_view> lit_false { lit("false") ,"literal false" };
rule<string_view> lit_null { lit("null") ,"literal null" };
2016-11-29 16:23:38 +01:00
2017-03-20 23:59:14 +01:00
rule<> quote { lit('"') ,"quote" };
rule<string_view> chars { raw[*(char_ - quote)] ,"characters" };
rule<string_view> string { quote >> chars >> quote ,"string" };
rule<string_view> name { string ,"name" };
2016-11-29 16:23:38 +01:00
2017-03-20 23:59:14 +01:00
rule<string_view> boolean { lit_true | lit_false ,"boolean" };
rule<string_view> literal { lit_true | lit_false | lit_null ,"literal" };
rule<string_view> number { raw[double_] ,"number" };
2016-11-29 16:23:38 +01:00
rule<string_view> array
{
array_begin >> -(omit[ws >> value >> ws] % value_sep) >> ws >> array_end
,"array"
};
2016-11-29 16:23:38 +01:00
rule<string_view> object
{
object_begin >> -(omit[ws >> member >> ws] % value_sep) >> ws >> object_end
,"object"
};
2016-11-29 16:23:38 +01:00
rule<string_view> value
{
lit_false | lit_true | lit_null | object | array | number | string
,"value"
};
2016-11-29 16:23:38 +01:00
rule<doc::member> member
{
name >> ws >> name_sep >> ws >> value
,"member"
};
rule<int> type
{
(omit[object_begin] >> attr(json::OBJECT)) |
(omit[array_begin] >> attr(json::ARRAY)) |
(omit[quote] >> attr(json::STRING)) |
(omit[number] >> attr(json::NUMBER)) |
(omit[literal] >> attr(json::LITERAL))
,"type"
};
input()
:input::base_type{rule<>{}}
{
array %= array_begin >> -(omit[ws >> value >> ws] % value_sep) >> ws >> array_end;
object %= object_begin >> -(omit[ws >> member >> ws] % value_sep) >> ws >> object_end;
}
};
2016-11-29 16:23:38 +01:00
template<class it>
struct output
:karma::grammar<it, unused_type>
{
template<class T = unused_type> using rule = karma::rule<it, T>;
rule<> NUL { lit('\0') ,"nul" };
// insignificant whitespaces
rule<> SP { lit('\x20') ,"space" };
rule<> HT { lit('\x09') ,"horizontal tab" };
rule<> CR { lit('\x0D') ,"carriage return" };
rule<> LF { lit('\x0A') ,"line feed" };
// whitespace skipping
rule<> WS { SP | HT | CR | LF ,"whitespace" };
rule<> ws { *(WS) ,"whitespace monoid" };
rule<> wsp { +(WS) ,"whitespace semigroup" };
// structural
rule<> object_begin { lit('{') ,"object begin" };
rule<> object_end { lit('}') ,"object end" };
rule<> array_begin { lit('[') ,"array begin" };
rule<> array_end { lit(']') ,"array end" };
rule<> name_sep { lit(':') ,"name separator" };
rule<> value_sep { lit(',') ,"value separator" };
rule<> quote { lit('"') ,"quote" };
2016-11-29 16:23:38 +01:00
rule<string_view> lit_true { karma::string("true") ,"literal true" };
rule<string_view> lit_false { karma::string("false") ,"literal false" };
rule<string_view> lit_null { karma::string("null") ,"literal null" };
rule<string_view> boolean { lit_true | lit_false ,"boolean" };
rule<string_view> literal { lit_true | lit_false | lit_null ,"literal" };
2017-03-20 23:59:14 +01:00
rule<string_view> chars { *(~char_('"')) ,"characters" };
2016-11-29 16:23:38 +01:00
rule<string_view> string { quote << chars << quote ,"string" };
rule<string_view> number { double_ ,"number" };
2016-11-29 16:23:38 +01:00
rule<string_view> name { string ,"name" };
2017-03-20 23:59:14 +01:00
rule<string_view> value { rule<string_view>{} /* subclass implemented */ ,"value" };
rule<const doc::member &> member { name << name_sep << value ,"member" };
2016-11-29 16:23:38 +01:00
2017-03-20 23:59:14 +01:00
rule<const string_view &> elem { value ,"element" };
rule<const json::arr &> elems { -(value % value_sep) ,"elements" };
rule<const json::arr &> array { array_begin << elems << array_end ,"array" };
2016-11-29 16:23:38 +01:00
2017-03-20 23:59:14 +01:00
rule<const json::doc &> members { -(member % value_sep) ,"members" };
rule<const json::doc &> document { object_begin << members << object_end ,"document" };
2016-11-29 16:23:38 +01:00
output()
:output::base_type{rule<>{}}
2017-03-11 04:31:20 +01:00
{}
2016-11-29 16:23:38 +01:00
};
} // namespace json
} // namespace ircd
namespace ircd {
namespace json {
struct parser
:input<const char *>
{
using input<const char *>::input;
}
const parser;
struct printer
:output<char *>
{
2017-03-11 04:31:20 +01:00
printer();
2016-11-29 16:23:38 +01:00
}
const printer;
struct ostreamer
:output<karma::ostream_iterator<char>>
{
2017-03-11 04:31:20 +01:00
ostreamer();
2016-11-29 16:23:38 +01:00
}
const ostreamer;
size_t print(char *const &buf, const size_t &max, const arr &);
2016-11-29 16:23:38 +01:00
size_t print(char *const &buf, const size_t &max, const doc &);
size_t print(char *const &buf, const size_t &max, const obj &);
2017-03-20 23:59:14 +01:00
arr serialize(const arr &, char *&start, char *const &stop);
doc serialize(const doc &, char *&start, char *const &stop);
doc serialize(const obj &, char *&start, char *const &stop);
2016-11-29 16:23:38 +01:00
std::ostream &operator<<(std::ostream &, const arr &);
2016-11-29 16:23:38 +01:00
std::ostream &operator<<(std::ostream &, const doc::member &);
std::ostream &operator<<(std::ostream &, const doc &);
std::ostream &operator<<(std::ostream &, const obj &);
} // namespace json
} // namespace ircd
2017-03-11 04:31:20 +01:00
ircd::json::printer::printer()
{
const auto recursor([this](auto &a, auto &b, auto &c)
{
const auto recurse_array([&]
{
char *out(const_cast<char *>(a.data()));
2017-03-20 23:59:14 +01:00
const arr r(serialize(json::arr(a), out, out + a.size()));
a.resize(size_t(out - r.data()));
});
2017-03-11 04:31:20 +01:00
const auto recurse_document([&]
{
char *out(const_cast<char *>(a.data()));
2017-03-20 23:59:14 +01:00
const doc d(serialize(json::doc(a), out, out + a.size()));
a.resize(size_t(out - d.data()));
2017-03-11 04:31:20 +01:00
});
const auto quote_string([&]
{
a.insert(a.end(), '"');
a.insert(a.begin(), '"');
});
2017-03-11 04:31:20 +01:00
if(likely(!a.empty())) switch(a.front())
{
case '{': recurse_document(); break;
case '[': recurse_array(); break;
case '"': break;
case '0': break;
case '1': break;
case '2': break;
case '3': break;
case '4': break;
case '5': break;
case '6': break;
case '7': break;
case '8': break;
case '9': break;
case 't':
case 'f':
case 'n':
if(a == "true" || a == "false" || a == "null")
break;
default:
quote_string();
break;
2017-03-11 04:31:20 +01:00
}
});
value %= karma::string[recursor];
}
ircd::json::ostreamer::ostreamer()
{
const auto recursor([this](auto &a, auto &b, auto &c)
{
const auto recurse_array([&]
{
char *out(const_cast<char *>(a.data()));
2017-03-20 23:59:14 +01:00
const arr r(serialize(json::arr(a), out, out + a.size()));
a.resize(size_t(out - r.data()));
});
2017-03-11 04:31:20 +01:00
const auto recurse_document([&]
{
char *out(const_cast<char *>(a.data()));
2017-03-20 23:59:14 +01:00
const doc d(serialize(json::doc(a), out, out + a.size()));
a.resize(size_t(out - d.data()));
2017-03-11 04:31:20 +01:00
});
const auto quote_string([&]
{
a.insert(a.end(), '"');
a.insert(a.begin(), '"');
});
2017-03-11 04:31:20 +01:00
if(likely(!a.empty())) switch(a.front())
{
case '{': recurse_document(); break;
case '[': recurse_array(); break;
case '"': break;
case '0': break;
case '1': break;
case '2': break;
case '3': break;
case '4': break;
case '5': break;
case '6': break;
case '7': break;
case '8': break;
case '9': break;
case 't':
case 'f':
case 'n':
if(a == "true" || a == "false" || a == "null")
break;
default:
quote_string();
break;
2017-03-11 04:31:20 +01:00
}
});
value %= karma::string[recursor];
}
2016-11-29 16:23:38 +01:00
size_t
ircd::json::print(char *const &buf,
const size_t &max,
const obj &obj)
{
if(unlikely(!max))
return 0;
char *out(buf);
serialize(obj, out, out + (max - 1));
*out = '\0';
return out - buf;
2016-11-29 16:23:38 +01:00
}
ircd::json::doc
ircd::json::serialize(const obj &obj,
char *&out,
2016-11-29 16:23:38 +01:00
char *const &stop)
{
static const auto throws([]
{
throw print_error("The JSON generator failed to serialize object");
});
const auto print_string([&stop, &out](const val &val)
2016-11-29 16:23:38 +01:00
{
2017-03-20 23:59:14 +01:00
karma::generate(out, maxwidth(stop - out)[printer.string] | eps[throws], string_view{val});
});
const auto print_literal([&stop, &out](const val &val)
{
karma::generate(out, maxwidth(stop - out)[karma::string] | eps[throws], string_view{val});
});
const auto print_object([&stop, &out](const val &val)
{
if(val.serial)
{
2017-03-20 23:59:14 +01:00
karma::generate(out, maxwidth(stop - out)[printer.document] | eps[throws], string_view{val});
return;
}
assert(val.object);
serialize(*val.object, out, stop);
});
2016-11-29 16:23:38 +01:00
const auto print_array([&stop, &out](const val &val)
{
if(val.serial)
{
2017-03-20 23:59:14 +01:00
karma::generate(out, maxwidth(stop - out)[printer.array] | eps[throws], string_view{val});
return;
}
2017-03-20 23:59:14 +01:00
assert(0);
//assert(val.object);
//serialize(*val.object, out, stop);
});
2017-03-20 23:59:14 +01:00
const auto print_number([&stop, &out](const val &val)
{
if(val.serial)
{
if(val.floats)
karma::generate(out, maxwidth(stop - out)[double_] | eps[throws], string_view{val});
else
karma::generate(out, maxwidth(stop - out)[long_] | eps[throws], string_view{val});
return;
}
if(val.floats)
karma::generate(out, maxwidth(stop - out)[double_] | eps[throws], val.floating);
else
karma::generate(out, maxwidth(stop - out)[long_] | eps[throws], val.integer);
});
const auto print_member([&](const obj::member &member)
2016-11-29 16:23:38 +01:00
{
const auto generate_name
2016-11-29 16:23:38 +01:00
{
maxwidth(stop - out)[printer.name << printer.name_sep] | eps[throws]
2016-11-29 16:23:38 +01:00
};
karma::generate(out, generate_name, member.first);
switch(member.second.type)
{
case STRING: print_string(member.second); break;
case OBJECT: print_object(member.second); break;
case ARRAY: print_array(member.second); break;
2017-03-20 23:59:14 +01:00
case NUMBER: print_number(member.second); break;
case LITERAL: print_literal(member.second); break;
}
2016-11-29 16:23:38 +01:00
});
char *const start(out);
2016-11-29 16:23:38 +01:00
karma::generate(out, maxwidth(stop - out)[printer.object_begin] | eps[throws]);
auto it(begin(obj));
if(it != end(obj))
2016-11-29 16:23:38 +01:00
{
print_member(*it);
for(++it; it != end(obj); ++it)
{
karma::generate(out, maxwidth(stop - out)[printer.value_sep] | eps[throws]);
print_member(*it);
}
2016-11-29 16:23:38 +01:00
}
karma::generate(out, maxwidth(stop - out)[printer.object_end] | eps[throws]);
return string_view{start, out};
}
std::ostream &
ircd::json::operator<<(std::ostream &s, const obj &obj)
{
karma::ostream_iterator<char> osi(s);
2016-11-29 16:23:38 +01:00
static const auto throws([]
{
throw print_error("The JSON generator failed to output object to stream");
});
const auto stream_string([&osi](const val &val)
{
2017-03-20 23:59:14 +01:00
karma::generate(osi, ostreamer.string, string_view{val});
});
const auto stream_literal([&osi](const val &val)
{
karma::generate(osi, karma::string, string_view{val});
});
2016-11-29 16:23:38 +01:00
const auto stream_object([&osi, &s](const val &val)
{
if(val.serial)
{
2017-03-20 23:59:14 +01:00
karma::generate(osi, ostreamer.document, string_view{val});
return;
}
2017-03-11 04:31:20 +01:00
assert(val.object);
s << *val.object;
});
2016-11-29 16:23:38 +01:00
2017-03-20 23:59:14 +01:00
const auto stream_array([&osi](const val &val)
{
if(val.serial)
{
2017-03-20 23:59:14 +01:00
karma::generate(osi, ostreamer.array, string_view{val});
return;
}
assert(0);
//assert(val.object);
//s << *val.object;
});
2017-03-20 23:59:14 +01:00
const auto stream_number([&osi](const val &val)
{
if(val.serial)
{
if(val.floats)
karma::generate(osi, double_, string_view{val});
else
karma::generate(osi, long_, string_view{val});
return;
}
if(val.floats)
karma::generate(osi, double_, val.floating);
else
karma::generate(osi, long_, val.integer);
});
const auto stream_member([&](const obj::member &member)
{
karma::generate(osi, ostreamer.name << ostreamer.name_sep, string_view(member.first));
2016-11-29 16:23:38 +01:00
switch(member.second.type)
{
2017-03-20 23:59:14 +01:00
case STRING: stream_string(member.second); break;
case OBJECT: stream_object(member.second); break;
case ARRAY: stream_array(member.second); break;
case NUMBER: stream_number(member.second); break;
case LITERAL: stream_literal(member.second); break;
}
});
2016-11-29 16:23:38 +01:00
karma::generate(osi, ostreamer.object_begin);
auto it(begin(obj));
if(it != end(obj))
2016-11-29 16:23:38 +01:00
{
stream_member(*it);
for(++it; it != end(obj); ++it)
{
karma::generate(osi, ostreamer.value_sep);
stream_member(*it);
}
2016-11-29 16:23:38 +01:00
}
karma::generate(osi, ostreamer.object_end);
return s;
2016-11-29 16:23:38 +01:00
}
2017-03-20 23:59:14 +01:00
ircd::json::obj::obj(const doc &doc,
const bool &recurse)
:idx{doc.count()}
2016-11-29 16:23:38 +01:00
{
2017-03-20 23:59:14 +01:00
const auto non_recursive_transform([]
(const doc::member &m) -> obj::member
2016-11-29 16:23:38 +01:00
{
2017-03-20 23:59:14 +01:00
return obj::member{m};
});
const auto recursive_transform([&non_recursive_transform]
(const doc::member &m) -> obj::member
{
switch(type(m.second))
{
case OBJECT:
return obj::member{m.first, std::make_unique<obj>(m.second)};
default:
return non_recursive_transform(m);
};
2016-11-29 16:23:38 +01:00
});
2017-03-20 23:59:14 +01:00
if(recurse)
std::transform(std::begin(doc), std::end(doc), std::begin(idx), recursive_transform);
else
std::transform(std::begin(doc), std::end(doc), std::begin(idx), non_recursive_transform);
2016-11-29 16:23:38 +01:00
}
2017-03-20 23:59:14 +01:00
ircd::json::obj::obj(std::initializer_list<member> builder)
:idx{builder.size()}
2016-11-29 16:23:38 +01:00
{
2017-03-20 23:59:14 +01:00
std::transform(std::begin(builder), std::end(builder), std::begin(idx), []
(auto&& m)
2016-11-29 16:23:38 +01:00
{
2017-03-20 23:59:14 +01:00
return std::move(const_cast<member &>(m));
});
2016-11-29 16:23:38 +01:00
}
bool
ircd::json::obj::erase(const string_view &name)
2016-11-29 16:23:38 +01:00
{
const auto it(find(name));
if(it == end())
return false;
2016-11-29 16:23:38 +01:00
erase(it);
return true;
2016-11-29 16:23:38 +01:00
}
void
ircd::json::obj::erase(const const_iterator &it)
2016-11-29 16:23:38 +01:00
{
idx.erase(it);
2016-11-29 16:23:38 +01:00
}
ircd::json::obj::const_iterator
ircd::json::obj::erase(const const_iterator &start,
const const_iterator &stop)
2016-11-29 16:23:38 +01:00
{
return { idx.erase(start, stop) };
2016-11-29 16:23:38 +01:00
}
2017-03-20 23:59:14 +01:00
const ircd::json::val &
ircd::json::obj::at(const string_view &path)
const
{
const auto elem(split(path, '.'));
const auto ident(split(elem.first, '['));
const auto &name(ident.first);
const auto &indice(ident.second);
const auto it(find(name));
if(unlikely(it == end()))
throw not_found("'%s'", name);
const auto &val(it->second);
if(!indice.empty())
{
const auto idx(lex_cast<size_t>(chomp(indice, "]")));
if(type(val) != ARRAY)
throw not_found("cannot recurse through non-array \"%s\" for indice [%zu]", name, idx);
if(val.serial)
throw not_found("cannot recurse through unparsed array \"%s\" for indice [%zu]", name, idx);
assert(0);
//return val.array->at(idx);
}
else if(!elem.second.empty())
{
if(type(val) != OBJECT)
throw not_found("cannot recurse through non-object \"%s\" for \"%s\"", name, elem.second);
if(val.serial)
throw not_found("cannot recurse through unparsed object \"%s\" for \"%s\"", name, elem.second);
return val.object->at(elem.second);
}
return it->second;
}
size_t
ircd::json::obj::size()
2016-11-29 16:23:38 +01:00
const
{
const size_t ret(1 + idx.empty());
return std::accumulate(std::begin(idx), std::end(idx), ret, [this]
(auto ret, const auto &member)
{
return ret += member.first.size() + 1 + 1 + member.second.size() + 1;
});
2016-11-29 16:23:38 +01:00
}
ircd::json::obj::operator std::string()
2016-11-29 16:23:38 +01:00
const
{
std::string ret(size(), char());
ret.resize(print(const_cast<char *>(ret.data()), ret.size() + 1, *this));
return ret;
2016-11-29 16:23:38 +01:00
}
inline
ircd::json::val::~val()
noexcept
2016-11-29 16:23:38 +01:00
{
switch(type)
{
case STRING: if(alloc) delete[] string; break;
case OBJECT: if(alloc) delete object; break;
//case ARRAY: if(alloc) delete array; break;
default: break;
}
2016-11-29 16:23:38 +01:00
}
ircd::json::val::operator std::string()
const
2016-11-29 16:23:38 +01:00
{
switch(type)
{
case STRING:
2017-03-20 23:59:14 +01:00
case LITERAL:
return std::string(unquote(string_view(*this)));
case OBJECT:
if(serial)
return std::string(string_view(*this));
else
return std::string(*object);
case ARRAY:
if(serial)
return std::string(string_view(*this));
else
break;
2017-03-20 23:59:14 +01:00
case NUMBER:
if(serial)
return std::string(string_view(*this));
else if(floats)
return std::string(lex_cast(floating));
else
return std::string(lex_cast(integer));
}
throw type_error("cannot stringify type[%d]", int(type));
2016-11-29 16:23:38 +01:00
}
ircd::json::val::operator string_view()
const
2016-11-29 16:23:38 +01:00
{
switch(type)
{
case STRING:
return unquote(string_view{string, len});
case ARRAY:
case OBJECT:
2017-03-20 23:59:14 +01:00
case NUMBER:
case LITERAL:
if(serial)
return string_view{string, len};
else
break;
}
throw type_error("value type[%d] is not a string", int(type));
2016-11-29 16:23:38 +01:00
}
size_t
ircd::json::val::size()
const
2016-11-29 16:23:38 +01:00
{
switch(type)
2016-11-29 16:23:38 +01:00
{
2017-03-20 23:59:14 +01:00
case NUMBER: return lex_cast(integer).size();
case STRING: return 1 + len + 1;
case OBJECT: return serial? len : object->size();
case ARRAY: return serial? len : 2;
case LITERAL: return len;
};
throw type_error("cannot size type[%u]", int(type));
2016-11-29 16:23:38 +01:00
}
2017-03-20 23:59:14 +01:00
std::ostream &
ircd::json::operator<<(std::ostream &s, const val &v)
{
switch(v.type)
{
case STRING:
case LITERAL:
s << string_view(v);
break;
case OBJECT:
if(v.serial)
s << string_view(v);
else
s << *v.object;
break;
case ARRAY:
if(v.serial)
s << string_view(v);
else
assert(0);
break;
case NUMBER:
if(v.serial)
s << string_view(v);
else if(v.floats)
s << v.floating;
else
s << v.integer;
break;
}
return s;
}
inline bool
ircd::json::operator>(const val &a, const val &b)
{
if(unlikely(type(a) != STRING || type(b) != STRING))
throw type_error("cannot compare values");
return static_cast<string_view>(a) > static_cast<string_view>(b);
}
inline bool
ircd::json::operator<(const val &a, const val &b)
{
if(unlikely(type(a) != STRING || type(b) != STRING))
throw type_error("cannot compare values");
return static_cast<string_view>(a) < static_cast<string_view>(b);
}
inline bool
ircd::json::operator>=(const val &a, const val &b)
{
if(unlikely(type(a) != STRING || type(b) != STRING))
throw type_error("cannot compare values");
return static_cast<string_view>(a) >= static_cast<string_view>(b);
}
inline bool
ircd::json::operator<=(const val &a, const val &b)
{
if(unlikely(type(a) != STRING || type(b) != STRING))
throw type_error("cannot compare values");
return static_cast<string_view>(a) <= static_cast<string_view>(b);
}
inline bool
ircd::json::operator!=(const val &a, const val &b)
{
if(unlikely(type(a) != STRING || type(b) != STRING))
throw type_error("cannot compare values");
return static_cast<string_view>(a) != static_cast<string_view>(b);
}
inline bool
ircd::json::operator==(const val &a, const val &b)
{
if(unlikely(type(a) != STRING || type(b) != STRING))
throw type_error("cannot compare values");
return static_cast<string_view>(a) == static_cast<string_view>(b);
}
2016-11-29 16:23:38 +01:00
size_t
ircd::json::print(char *const &buf,
const size_t &max,
const doc &doc)
{
if(unlikely(!max))
return 0;
2016-11-29 16:23:38 +01:00
char *out(buf);
serialize(doc, out, out + (max - 1));
*out = '\0';
2016-11-29 16:23:38 +01:00
return std::distance(buf, out);
}
ircd::json::doc
ircd::json::serialize(const doc &doc,
char *&out,
char *const &stop)
{
static const auto throws([]
{
throw print_error("The JSON generator failed to print document");
});
char *const start(out);
karma::generate(out, maxwidth(stop - start)[printer.document] | eps[throws], doc);
return string_view{start, out};
}
2016-11-29 16:23:38 +01:00
std::ostream &
ircd::json::operator<<(std::ostream &s, const doc &doc)
{
2017-03-11 04:31:20 +01:00
const auto &os(ostreamer);
2016-11-29 16:23:38 +01:00
static const auto throws([]
{
throw print_error("The JSON generator failed to output document to stream");
});
karma::ostream_iterator<char> osi(s);
2017-03-11 04:31:20 +01:00
karma::generate(osi, os.document | eps[throws], doc);
2016-11-29 16:23:38 +01:00
return s;
}
std::ostream &
ircd::json::operator<<(std::ostream &s, const doc::member &member)
{
static const auto throws([]
{
throw print_error("The JSON generator failed to output document member to stream");
});
karma::ostream_iterator<char> osi(s);
karma::generate(osi, ostreamer.member | eps[throws], member);
2016-11-29 16:23:38 +01:00
return s;
}
2017-03-11 04:31:20 +01:00
ircd::json::doc::const_iterator &
ircd::json::doc::const_iterator::operator++()
2016-11-29 16:23:38 +01:00
{
static const qi::rule<const char *, json::doc::member> member
{
parser.name >> parser.ws >> parser.name_sep >> parser.ws >> raw[parser.value]
};
static const qi::rule<const char *, json::doc::member> parse_next
{
parser.object_end | (parser.value_sep >> parser.ws >> member)
};
state.first = string_view{};
state.second = string_view{};
if(!qi::phrase_parse(start, stop, parse_next, parser.WS, state))
start = stop;
return *this;
}
2017-03-20 23:59:14 +01:00
ircd::json::doc::operator std::string()
const
{
//TODO: tmp
std::stringstream ret;
ret << (*this);
return ret.str();
}
2017-03-11 04:31:20 +01:00
ircd::json::doc::const_iterator
2016-11-29 16:23:38 +01:00
ircd::json::doc::begin()
const
{
static const qi::rule<const char *, json::doc::member> member
{
parser.name >> parser.ws >> parser.name_sep >> parser.ws >> raw[parser.value]
};
static const qi::rule<const char *, json::doc::member> parse_begin
{
parser.object_begin >> parser.ws >> (parser.object_end | member)
};
const_iterator ret(string_view::begin(), string_view::end());
2016-11-29 16:23:38 +01:00
if(!qi::phrase_parse(ret.start, ret.stop, parse_begin, parser.WS, ret.state))
ret.start = ret.stop;
return ret;
}
2017-03-11 04:31:20 +01:00
ircd::json::doc::const_iterator
2016-11-29 16:23:38 +01:00
ircd::json::doc::end()
const
{
return { string_view::end(), string_view::end() };
}
size_t
ircd::json::print(char *const &buf,
const size_t &max,
const arr &arr)
{
if(unlikely(!max))
return 0;
char *out(buf);
serialize(arr, out, out + (max - 1));
*out = '\0';
return std::distance(buf, out);
}
ircd::json::arr
ircd::json::serialize(const arr &a,
char *&out,
char *const &stop)
{
static const auto throws([]
{
throw print_error("The JSON generator failed to print array");
});
2017-03-20 23:59:14 +01:00
char *const start(out);
karma::generate(out, maxwidth(stop - out)[printer.array_begin] | eps[throws]);
auto it(begin(a));
if(it != end(a))
{
2017-03-20 23:59:14 +01:00
karma::generate(out, maxwidth(stop - out)[printer.elem] | eps[throws], *it);
for(++it; it != end(a); ++it)
karma::generate(out, maxwidth(stop - out)[printer.value_sep << printer.elem] | eps[throws], *it);
}
2017-03-20 23:59:14 +01:00
karma::generate(out, maxwidth(stop - out)[printer.array_end] | eps[throws]);
return string_view{start, out};
}
std::ostream &
2017-03-20 23:59:14 +01:00
ircd::json::operator<<(std::ostream &s, const arr &a)
{
const auto &os(ostreamer);
static const auto throws([]
{
throw print_error("The JSON generator failed to output array to stream");
});
karma::ostream_iterator<char> osi(s);
2017-03-20 23:59:14 +01:00
karma::generate(osi, ostreamer.array_begin | eps[throws]);
auto it(begin(a));
if(it != end(a))
{
karma::generate(osi, ostreamer.elem | eps[throws], *it);
for(++it; it != end(a); ++it)
karma::generate(osi, (ostreamer.value_sep << ostreamer.elem) | eps[throws], *it);
}
karma::generate(osi, ostreamer.array_end | eps[throws]);
return s;
}
ircd::json::arr::const_iterator &
ircd::json::arr::const_iterator::operator++()
2016-11-29 16:23:38 +01:00
{
static const qi::rule<const char *, string_view> parse_next
{
parser.array_end | (parser.value_sep >> parser.ws >> raw[parser.value])
2016-11-29 16:23:38 +01:00
};
state = string_view{};
if(!qi::phrase_parse(start, stop, parse_next, parser.WS, state))
start = stop;
return *this;
}
2017-03-20 23:59:14 +01:00
ircd::json::arr::operator std::string()
const
{
//TODO: tmp
std::stringstream ret;
ret << (*this);
return ret.str();
}
ircd::json::arr::const_iterator
ircd::json::arr::begin()
2016-11-29 16:23:38 +01:00
const
{
static const qi::rule<const char *, string_view> parse_begin
{
parser.array_begin >> parser.ws >> (parser.array_end | raw[parser.value])
};
const_iterator ret(string_view::begin(), string_view::end());
2016-11-29 16:23:38 +01:00
if(!qi::phrase_parse(ret.start, ret.stop, parse_begin, parser.WS, ret.state))
ret.start = ret.stop;
return ret;
}
ircd::json::arr::const_iterator
ircd::json::arr::end()
2016-11-29 16:23:38 +01:00
const
{
return { string_view::end(), string_view::end() };
2016-11-29 16:23:38 +01:00
}
enum ircd::json::type
ircd::json::type(const string_view &buf)
{
static const auto flag(qi::skip_flag::dont_postskip);
enum type ret;
if(!qi::phrase_parse(begin(buf), end(buf), parser.type, parser.WS, flag, ret))
throw type_error("Failed to get type from buffer");
return ret;
}
2017-03-20 23:59:14 +01:00
enum ircd::json::type
ircd::json::type(const string_view &buf,
std::nothrow_t)
{
static const auto flag(qi::skip_flag::dont_postskip);
enum type ret;
if(!qi::phrase_parse(begin(buf), end(buf), parser.type, parser.WS, flag, ret))
return STRING;
return ret;
}