src/ToJsonWriter.cpp

   1 //! \file ToJsonWriter.cpp
   2
   3 #include "ToJsonWriter.h"
   4
   5 #include <json/json.h>
   6
   7 #include "Article.h"
   8
   9 /*! Get article links in an array.
  10  * Basically undoing the Wikipedia to article conversion...
  11  * \param article pointer to article which links should be extracted
  12  * \return Json::Value array with titles as string
  13  * \todo should / could be a member function, but then I'd have to expose
  14  * Json::Value, which is ugly and clutters up other classes...
  15  */
  16 static Json::Value getArticleLinks(const Article* article)
  17 {
  18     Json::Value array(Json::ValueType::arrayValue);
  19
  20     for(auto ali = article->linkBegin(); ali != article->linkEnd(); ali++) {
  21         std::string tit = (*ali)->getTitle();
  22         array.append(Json::Value(tit));
  23     }
  24
  25     return array;
  26 }
  27
  28 std::string ToJsonWriter::convertToJson(const Article* a)
  29 {
  30     Json::Value val(Json::ValueType::objectValue);
  31
  32     Json::Value linkObj(Json::ValueType::objectValue);
  33
  34     if(a->isAnalyzed()) {
  35         linkObj["forward_links"] = getArticleLinks(a);
  36     } else {
  37         linkObj["forward_links"] = Json::Value::nullSingleton();
  38     }
  39
  40     val[a->getTitle()] = linkObj;
  41
  42     Json::FastWriter jsw;
  43     jsw.omitEndingLineFeed();
  44
  45     return jsw.write(val);
  46 }
  47
  48 std::string ToJsonWriter::convertToJson(const ArticleCollection& ac)
  49 {
  50     Json::Value val(Json::ValueType::objectValue);
  51
  52     for(auto ar : ac) {
  53         Json::Value linkObj(Json::ValueType::objectValue);
  54
  55         if(ar.second->isAnalyzed()) {
  56             linkObj["forward_links"] = getArticleLinks(ar.second);
  57         } else {
  58             linkObj["forward_links"] = Json::Value::nullSingleton();
  59         }
  60
  61         val[ar.first] = linkObj;
  62     }
  63
  64     Json::FastWriter jsw;
  65     jsw.omitEndingLineFeed();
  66
  67     return jsw.write(val);
  68 }
  69
  70 void ToJsonWriter::output(const Article* article, std::ostream& outstream)
  71 {
  72     outstream << convertToJson(article);
  73 }
  74
  75 void ToJsonWriter::output(const ArticleCollection& collection,
  76                           std::ostream& outstream)
  77 {
  78     outstream << convertToJson(collection);
  79 }