Handle Nan, Infinity

Author: SinghRajenM

NppJSONViewer/NppJsonViewer/JsonHandler.cpp

@@ -1,33 +1,36 @@
 #include "JsonHandler.h"
 
+namespace rj = rapidjson;
+
+
 JsonHandler::JsonHandler(const ParseOptions &options)
     : m_parseOptions(options)
 {
 }
 
 auto JsonHandler::GetCompressedJson(const std::string &jsonText) -> const Result
 {
-    rapidjson::StringBuffer                    sb;
-    rapidjson::Writer<rapidjson::StringBuffer> handler(sb);
+    rj::StringBuffer                                                                                sb;
+    rj::Writer<rj::StringBuffer, rj::UTF8<>, rj::UTF8<>, rj::CrtAllocator, rj::kWriteNanAndInfFlag> handler(sb);
 
-    return ParseJson(jsonText, sb, handler);
+    return ParseJson<flgBaseWriter>(jsonText, sb, handler);
 }
 
 auto JsonHandler::FormatJson(const std::string &jsonText, LE le, LF lf, char indentChar, unsigned indentLen) -> const Result
 {
-    rapidjson::StringBuffer                          sb;
-    rapidjson::PrettyWriter<rapidjson::StringBuffer> handler(sb);
+    rj::StringBuffer                                                                                      sb;
+    rj::PrettyWriter<rj::StringBuffer, rj::UTF8<>, rj::UTF8<>, rj::CrtAllocator, rj::kWriteNanAndInfFlag> handler(sb);
     handler.SetLineEnding(le);
     handler.SetFormatOptions(lf);
     handler.SetIndent(indentChar, indentLen);
 
-    return ParseJson(jsonText, sb, handler);
+    return ParseJson<flgBaseWriter>(jsonText, sb, handler);
 }
 
 auto JsonHandler::ValidateJson(const std::string &jsonText) -> const Result
 {
-    rapidjson::StringBuffer                    sb;
-    rapidjson::Writer<rapidjson::StringBuffer> handler(sb);
+    rj::StringBuffer                                                                                sb;
+    rj::Writer<rj::StringBuffer, rj::UTF8<>, rj::UTF8<>, rj::CrtAllocator, rj::kWriteNanAndInfFlag> handler(sb);
 
-    return ParseJson(jsonText, sb, handler);
+    return ParseJson<flgBaseWriter>(jsonText, sb, handler);
 }

NppJSONViewer/NppJsonViewer/JsonHandler.h

@@ -9,6 +9,8 @@
 #include <rapidjson/error/en.h>
 #include "Define.h"
 
+namespace rj = rapidjson;
+
 struct Result
 {
     bool        success    = false;
@@ -18,8 +20,11 @@ struct Result
     std::string response;
 };
 
-using LE = rapidjson::LineEndingOption;
-using LF = rapidjson::PrettyFormatOptions;
+using LE = rj::LineEndingOption;
+using LF = rj::PrettyFormatOptions;
+
+constexpr auto flgBaseReader = rj::kParseEscapedApostropheFlag | rj::kParseNanAndInfFlag | rj::kParseNumbersAsStringsFlag;
+constexpr auto flgBaseWriter = rj::kParseEscapedApostropheFlag | rj::kParseNanAndInfFlag | rj::kParseFullPrecisionFlag;
 
 class JsonHandler
 {
@@ -33,23 +38,22 @@ class JsonHandler
     auto FormatJson(const std::string &jsonText, LE le, LF lf, char indentChar, unsigned indentLen) -> const Result;
     auto ValidateJson(const std::string &jsonText) -> const Result;
 
-    template <typename Handler>
-    auto ParseJson(const std::string &jsonText, rapidjson::StringBuffer &sb, Handler &handler) -> const Result;
+    template <unsigned format, typename Handler>
+    auto ParseJson(const std::string &jsonText, rj::StringBuffer &sb, Handler &handler) -> const Result;
 };
 
-template <typename Handler>
-inline auto JsonHandler::ParseJson(const std::string &jsonText, rapidjson::StringBuffer &sb, Handler &handler) -> const Result
+template <unsigned flgBase, typename Handler>
+inline auto JsonHandler::ParseJson(const std::string &jsonText, rj::StringBuffer &sb, Handler &handler) -> const Result
 {
     Result retVal {};
 
-    bool                    success = false;
-    rapidjson::Reader       reader;
-    rapidjson::StringStream ss(jsonText.c_str());
+    bool             success = false;
+    rj::Reader       reader;
+    rj::StringStream ss(jsonText.c_str());
 
     // TODO: Find some better way
-    constexpr auto flgBase         = rapidjson::kParseFullPrecisionFlag | rapidjson::kParseEscapedApostropheFlag | rapidjson::kParseNanAndInfFlag;
-    constexpr auto flgBase_commemt = flgBase | rapidjson::kParseCommentsFlag;
-    constexpr auto flgBase_comma   = flgBase | rapidjson::kParseTrailingCommasFlag;
+    constexpr auto flgBase_commemt = flgBase | rj::kParseCommentsFlag;
+    constexpr auto flgBase_comma   = flgBase | rj::kParseTrailingCommasFlag;
     constexpr auto flgBase_Both    = flgBase_comma | flgBase_commemt;
 
     if (m_parseOptions.bIgnoreComment && m_parseOptions.bIgnoreTraillingComma)
@@ -82,7 +86,7 @@ inline auto JsonHandler::ParseJson(const std::string &jsonText, rapidjson::Strin
     else
     {
         retVal.success    = false;
-        retVal.error_str  = rapidjson::GetParseError_En(reader.GetParseErrorCode());
+        retVal.error_str  = rj::GetParseError_En(reader.GetParseErrorCode());
         retVal.error_pos  = static_cast<int>(reader.GetErrorOffset());
         retVal.error_code = reader.GetParseErrorCode();
         retVal.response.clear();

NppJSONViewer/NppJsonViewer/JsonViewDlg.cpp

@@ -186,7 +186,7 @@ void JsonViewDlg::PopulateTreeUsingSax(HTREEITEM tree_root, const std::string &j
     RapidJsonHandler        handler(this, tree_root);
     rapidjson::StringBuffer sb;
 
-    Result res = JsonHandler(m_pSetting->parseOptions).ParseJson(jsonText, sb, handler);
+    Result res = JsonHandler(m_pSetting->parseOptions).ParseJson<flgBaseReader>(jsonText, sb, handler);
     if (!res.success)
     {
         // Mark the error position