1 files changed, 194 insertions, 0 deletions
diff --git a/src/script/rapidjson/document.cpp b/src/script/rapidjson/document.cpp
new file mode 100644
index 0000000..2487eb6
--- /dev/null
+++ b/src/script/rapidjson/document.cpp
@@ -0,0 +1,194 @@
+
+#include <cstdio>
+#include <vector>
+#include <fstream>
+
+#include "userdata.hpp"
+#include "values.hpp"
+#include "file.hpp"
+
+#include "lua.hpp"
+#include "rapidjson.h"
+
+using namespace  rapidjson;
+
+template<>
+const char* const Userdata<Document>::metatable = "rapidjson.Document";
+
+template<>
+Document* Userdata<Document>::construct(lua_State * L)
+{
+    auto t = lua_type(L, 1);
+    if (t != LUA_TNONE && t != LUA_TSTRING && t != LUA_TTABLE)
+    {
+        luax::typerror(L, 1, "none, string or table");
+        return nullptr;
+    }
+
+    auto doc = new Document();
+    if (t == LUA_TSTRING)
+    {
+        size_t len;
+        const char* s = luaL_checklstring(L, 1, &len);
+        doc->Parse(s, len);
+    }
+    else if (t == LUA_TTABLE)
+    {
+        values::toDocument(L, 1, doc);
+    }
+    return doc;
+}
+
+static int pushParseResult(lua_State* L, Document* doc)
+{
+    auto err = doc->GetParseError();
+    if (err != kParseErrorNone)
+    {
+        lua_pushnil(L);
+        lua_pushfstring(L, "%s (at Offset %d)", GetParseError_En(err), doc->GetErrorOffset());
+        return 2;
+    }
+
+    lua_pushboolean(L, 1);
+    return 1;
+}
+
+static int Document_parse(lua_State* L)
+{
+    Document* doc = Userdata<Document>::get(L, 1);
+
+    size_t l = 0;
+    auto s = luaL_checklstring(L, 2, &l);
+    doc->Parse(s, l);
+
+    return pushParseResult(L, doc);
+}
+
+static int Document_parseFile(lua_State* L)
+{
+    auto doc = Userdata<Document>::get(L, 1);
+
+    auto s = luaL_checkstring(L, 2);
+    std::ifstream ifs(s);
+    IStreamWrapper isw(ifs);
+
+    doc->ParseStream(isw);
+
+    return pushParseResult(L, doc);;
+}
+
+/**
+ * doc:get('path'[, default])
+ */
+static int Document_get(lua_State* L)
+{
+    auto doc = Userdata<Document>::check(L, 1);
+    auto s = luaL_checkstring(L, 2);
+    Pointer ptr(s);
+    auto v = ptr.Get(*doc);
+
+    if (!v)
+    {
+        if (lua_gettop(L) >= 3)
+        {
+            lua_pushvalue(L, 3);
+        }
+        else
+        {
+            lua_pushnil(L);
+        }
+    }
+    else
+    {
+        values::push(L, *v);
+    }
+    return 1;
+}
+
+static int Document_set(lua_State* L)
+{
+    auto doc = Userdata<Document>::check(L, 1);
+    Pointer ptr(luaL_checkstring(L, 2));
+    auto v = values::toValue(L, 3, doc->GetAllocator());
+
+    ptr.Set(*doc, v, doc->GetAllocator());
+
+    return 0;
+}
+
+/**
+ * local jsonstr = doc:stringify(
+ {pretty=false})
+ */
+static int Document_stringify(lua_State* L)
+{
+    auto doc = Userdata<Document>::check(L, 1);
+
+    auto pretty = luax::optboolfield(L, 2, "pretty", false);
+
+    StringBuffer sb;
+    if (pretty)
+    {
+        PrettyWriter<StringBuffer> writer(sb);
+        doc->Accept(writer);
+    }
+    else
+    {
+        Writer<StringBuffer> writer(sb);
+        doc->Accept(writer);
+    }
+    lua_pushlstring(L, sb.GetString(), sb.GetSize());
+
+    return 1;
+}
+
+/**
+ * doc:save(filename,
+ {pretty=false, sort_keys=false})
+ */
+static int Document_save(lua_State* L)
+{
+    auto doc = Userdata<Document>::check(L, 1);
+    auto filename = luaL_checkstring(L, 2);
+    auto pretty = luax::optboolfield(L, 3, "pretty", false);
+
+    auto fp = file::open(filename, "wb");
+    char buffer[512];
+    FileWriteStream fs(fp, buffer, sizeof(buffer));
+
+    if (pretty)
+    {
+        PrettyWriter<FileWriteStream> writer(fs);
+        doc->Accept(writer);
+    }
+    else
+    {
+        Writer<FileWriteStream> writer(fs);
+        doc->Accept(writer);
+    }
+    fclose(fp);
+
+    return 0;
+}
+
+template <>
+const luaL_Reg* Userdata<Document>::methods()
+{
+    static const luaL_Reg reg[] =
+    {
+        { "parse", Document_parse },
+        { "parseFile", Document_parseFile },
+
+        { "__gc", metamethod_gc },
+        { "__tostring", metamethod_tostring },
+
+        { "get", Document_get },
+        { "set", Document_set },
+
+        { "stringify", Document_stringify },
+        { "save", Document_save },
+
+        { nullptr, nullptr }
+    };
+    return reg;
+}