xbmc/utils/JSONVariantWriter.cpp

   1 /*
   2  *  Copyright (C) 2015-2018 Team Kodi
   3  *  This file is part of Kodi - https://kodi.tv
   4  *
   5  *  SPDX-License-Identifier: GPL-2.0-or-later
   6  *  See LICENSES/README.md for more information.
   7  */
   8
   9 #include "JSONVariantWriter.h"
  10
  11 #include "utils/Variant.h"
  12
  13 #include <rapidjson/prettywriter.h>
  14 #include <rapidjson/stringbuffer.h>
  15 #include <rapidjson/writer.h>
  16
  17 template<class TWriter>
  18 bool InternalWrite(TWriter& writer, const CVariant &value)
  19 {
  20   switch (value.type())
  21   {
  22   case CVariant::VariantTypeInteger:
  23     return writer.Int64(value.asInteger());
  24
  25   case CVariant::VariantTypeUnsignedInteger:
  26     return writer.Uint64(value.asUnsignedInteger());
  27
  28   case CVariant::VariantTypeDouble:
  29     return writer.Double(value.asDouble());
  30
  31   case CVariant::VariantTypeBoolean:
  32     return writer.Bool(value.asBoolean());
  33
  34   case CVariant::VariantTypeString:
  35     return writer.String(value.c_str(), value.size());
  36
  37   case CVariant::VariantTypeArray:
  38     if (!writer.StartArray())
  39       return false;
  40
  41     for (CVariant::const_iterator_array itr = value.begin_array(); itr != value.end_array(); ++itr)
  42     {
  43       if (!InternalWrite(writer, *itr))
  44         return false;
  45     }
  46
  47     return writer.EndArray(value.size());
  48
  49   case CVariant::VariantTypeObject:
  50     if (!writer.StartObject())
  51       return false;
  52
  53     for (CVariant::const_iterator_map itr = value.begin_map(); itr != value.end_map(); ++itr)
  54     {
  55       if (!writer.Key(itr->first.c_str()) ||
  56         !InternalWrite(writer, itr->second))
  57         return false;
  58     }
  59
  60     return writer.EndObject(value.size());
  61
  62   case CVariant::VariantTypeConstNull:
  63   case CVariant::VariantTypeNull:
  64   default:
  65     return writer.Null();
  66   }
  67
  68   return false;
  69 }
  70
  71 bool CJSONVariantWriter::Write(const CVariant &value, std::string& output, bool compact)
  72 {
  73   rapidjson::StringBuffer stringBuffer;
  74   if (compact)
  75   {
  76     rapidjson::Writer<rapidjson::StringBuffer> writer(stringBuffer);
  77
  78     if (!InternalWrite(writer, value) || !writer.IsComplete())
  79       return false;
  80   }
  81   else
  82   {
  83     rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(stringBuffer);
  84     writer.SetIndent('\t', 1);
  85
  86     if (!InternalWrite(writer, value) || !writer.IsComplete())
  87       return false;
  88   }
  89
  90   output = stringBuffer.GetString();
  91   return true;
  92 }