improvements to JSON Writer

part of the master commit cb95276e6e6bf12a1c06d5c252551e55c788fcb2

Change-Id: Icf18fb4b3ebced375196a8cdbd2a543acf4e43c5
Reviewed-on: https://gerrit.libreoffice.org/c/core/+/100444
Tested-by: Jenkins CollaboraOffice <jenkinscollaboraoffice@gmail.com>
Reviewed-by: Tomaž Vajngerl <quikee@gmail.com>

1 files changed, 98 insertions, 13 deletions

diff --git a/tools/source/misc/json_writer.cxx b/tools/source/misc/json_writer.cxx
index ab109dae669c..251c44c0246f 100644
--- a/tools/source/misc/json_writer.cxx
+++ b/tools/source/misc/json_writer.cxx
@@ -10,6 +10,7 @@
 #include <tools/json_writer.hxx>
 #include <stdio.h>
 #include <cstring>
+#include <rtl/strbuf.hxx>
 
 namespace tools
 {
@@ -22,6 +23,7 @@ JsonWriter::JsonWriter()
     , mpBuffer(static_cast<char*>(malloc(mSpaceAllocated)))
     , mStartNodeCount(0)
     , mPos(mpBuffer)
+    , mbFirstFieldInNode(true)
 {
     *mPos = '{';
     ++mPos;
@@ -38,7 +40,10 @@ JsonWriter::~JsonWriter()
 ScopedJsonWriterNode JsonWriter::startNode(const char* pNodeName)
 {
     auto len = strlen(pNodeName);
-    ensureSpace(len + 4);
+    ensureSpace(len + 6);
+
+    addCommaBeforeField();
+
     *mPos = '"';
     ++mPos;
     memcpy(mPos, pNodeName, len);
@@ -57,15 +62,70 @@ void JsonWriter::endNode()
     ensureSpace(1);
     *mPos = '}';
     ++mPos;
+    mbFirstFieldInNode = false;
 }
 
-void JsonWriter::put(const char* pPropName, const OUString& rPropVal)
+ScopedJsonWriterArray JsonWriter::startArray(const char* pNodeName)
+{
+    auto len = strlen(pNodeName);
+    ensureSpace(len + 6);
+
+    addCommaBeforeField();
+
+    *mPos = '"';
+    ++mPos;
+    memcpy(mPos, pNodeName, len);
+    mPos += len;
+    strncpy(mPos, "\": [ ", 5);
+    mPos += 5;
+    mStartNodeCount++;
+    mbFirstFieldInNode = true;
+    return ScopedJsonWriterArray(*this);
+}
+
+void JsonWriter::endArray()
 {
+    assert(mStartNodeCount && "mismatched StartNode/EndNode somewhere");
+    --mStartNodeCount;
+    ensureSpace(1);
+    *mPos = ']';
+    ++mPos;
+    mbFirstFieldInNode = false;
+}
+
+ScopedJsonWriterStruct JsonWriter::startStruct()
+{
+    ensureSpace(6);
+
     addCommaBeforeField();
 
+    *mPos = '{';
+    ++mPos;
+    *mPos = ' ';
+    ++mPos;
+    mStartNodeCount++;
+    mbFirstFieldInNode = true;
+    return ScopedJsonWriterStruct(*this);
+}
+
+void JsonWriter::endStruct()
+{
+    assert(mStartNodeCount && "mismatched StartNode/EndNode somewhere");
+    --mStartNodeCount;
+    ensureSpace(1);
+    *mPos = '}';
+    ++mPos;
+    mbFirstFieldInNode = false;
+}
+
+void JsonWriter::put(const char* pPropName, const OUString& rPropVal)
+{
     auto nPropNameLength = strlen(pPropName);
     auto nWorstCasePropValLength = rPropVal.getLength() * 2;
-    ensureSpace(nPropNameLength + nWorstCasePropValLength + 6);
+    ensureSpace(nPropNameLength + nWorstCasePropValLength + 8);
+
+    addCommaBeforeField();
+
     *mPos = '"';
     ++mPos;
     memcpy(mPos, pPropName, nPropNameLength);
@@ -120,11 +180,12 @@ void JsonWriter::put(const char* pPropName, const OUString& rPropVal)
 
 void JsonWriter::put(const char* pPropName, const OString& rPropVal)
 {
-    addCommaBeforeField();
-
     auto nPropNameLength = strlen(pPropName);
     auto nWorstCasePropValLength = rPropVal.getLength();
-    ensureSpace(nPropNameLength + nWorstCasePropValLength + 6);
+    ensureSpace(nPropNameLength + nWorstCasePropValLength + 8);
+
+    addCommaBeforeField();
+
     *mPos = '"';
     ++mPos;
     memcpy(mPos, pPropName, nPropNameLength);
@@ -163,12 +224,13 @@ void JsonWriter::put(const char* pPropName, const OString& rPropVal)
 
 void JsonWriter::put(const char* pPropName, const char* pPropVal)
 {
-    addCommaBeforeField();
-
     auto nPropNameLength = strlen(pPropName);
     auto nPropValLength = strlen(pPropVal);
     auto nWorstCasePropValLength = nPropValLength * 2;
-    ensureSpace(nPropNameLength + nWorstCasePropValLength + 6);
+    ensureSpace(nPropNameLength + nWorstCasePropValLength + 8);
+
+    addCommaBeforeField();
+
     *mPos = '"';
     ++mPos;
     memcpy(mPos, pPropName, nPropNameLength);
@@ -210,11 +272,12 @@ void JsonWriter::put(const char* pPropName, const char* pPropVal)
 
 void JsonWriter::put(const char* pPropName, int nPropVal)
 {
-    addCommaBeforeField();
-
     auto nPropNameLength = strlen(pPropName);
     auto nWorstCasePropValLength = 32;
-    ensureSpace(nPropNameLength + nWorstCasePropValLength + 6);
+    ensureSpace(nPropNameLength + nWorstCasePropValLength + 8);
+
+    addCommaBeforeField();
+
     *mPos = '"';
     ++mPos;
     memcpy(mPos, pPropName, nPropNameLength);
@@ -225,6 +288,16 @@ void JsonWriter::put(const char* pPropName, int nPropVal)
     mPos += sprintf(mPos, "%d", nPropVal);
 }
 
+void JsonWriter::putRaw(const rtl::OStringBuffer& rRawBuf)
+{
+    ensureSpace(rRawBuf.getLength() + 2);
+
+    addCommaBeforeField();
+
+    memcpy(mPos, rRawBuf.getStr(), rRawBuf.getLength());
+    mPos += rRawBuf.getLength();
+}
+
 void JsonWriter::addCommaBeforeField()
 {
     if (mbFirstFieldInNode)
@@ -238,12 +311,24 @@ void JsonWriter::addCommaBeforeField()
     }
 }
 
-/** Hands ownership of the the underlying storage buffer to the caller,
+void JsonWriter::reallocBuffer(int noMoreBytesRequired)
+{
+    int currentUsed = mPos - mpBuffer;
+    auto newSize = std::max<int>(mSpaceAllocated * 2, (currentUsed + noMoreBytesRequired) * 2);
+    char* pNew = static_cast<char*>(malloc(newSize));
+    memcpy(pNew, mpBuffer, currentUsed);
+    free(mpBuffer);
+    mpBuffer = pNew;
+    mPos = mpBuffer + currentUsed;
+}
+
+/** Hands ownership of the underlying storage buffer to the caller,
   * after this no more document modifications may be written. */
 char* JsonWriter::extractData()
 {
     assert(mStartNodeCount == 0 && "did not close all nodes");
     assert(mpBuffer && "data already extracted");
+    ensureSpace(2);
     // add closing brace
     *mPos = '}';
     ++mPos;