src/JsonToArticleConverter.cpp

   1 #include "JsonToArticleConverter.h"
   2 #include "WalkerException.h"
   3
   4 #include <json/json.h>
   5
   6 //! \todo really ugly workaround, passing in the ArticleCollection instance... :/
   7 Article* JsonToArticleConverter::convertToArticle(std::string json, ArticleCollection& articleCache)
   8 {
   9     Json::Reader reader;
  10     Json::Value document;
  11     bool success = reader.parse(json, document, false);
  12
  13     if(!success) {
  14         throw WalkerException("Error parsing JSON");
  15     }
  16
  17     auto allPages = document.get("query", Json::Value::nullSingleton())
  18                             .get("pages", Json::Value::nullSingleton());
  19
  20     // only get first page
  21     auto wantedPage = allPages.get(allPages.getMemberNames()[0],
  22                                    Json::Value::nullSingleton());
  23
  24     if(wantedPage.isMember("missing")) {
  25         throw WalkerException("Article doesn't exist");
  26     } else if(wantedPage.isMember("invalid")) {
  27         throw WalkerException("Invalid article");
  28     }
  29
  30     //! get normalized title not necessary, "title" is already
  31     std::string wantedArticleTitle = wantedPage.get("title", Json::Value::nullSingleton()).asString();
  32     Article* wantedArticle = articleCache.get(wantedArticleTitle);
  33
  34     if(wantedArticle == nullptr) {
  35         wantedArticle = new Article(wantedArticleTitle);
  36     }
  37
  38     articleCache.add(wantedArticle);
  39
  40     // add links
  41     for(const auto &linked : wantedPage.get("links", Json::Value::nullSingleton())) {
  42         auto linkedPageTitle = linked.get("title", Json::Value::nullSingleton()).asString();
  43         auto par = articleCache.get(linkedPageTitle);
  44
  45         if(par == nullptr) {
  46             par = new Article(linkedPageTitle);
  47             articleCache.add(par);
  48         }
  49
  50         wantedArticle->addLink(par);
  51     }
  52
  53     wantedArticle->setAnalyzed(true);
  54
  55     if(!document.isMember("batchcomplete")) {
  56         moreData = true;
  57         continueString =
  58           document.get("continue", Json::Value::nullSingleton())
  59                   .get("plcontinue", Json::Value::nullSingleton())
  60                     .asString();
  61     } else {
  62         moreData = false;
  63         continueString = "";
  64     }
  65
  66     return wantedArticle;
  67 }