config.cc

   1 #include <iostream>
   2 #include <string>
   3 #include <fstream>
   4 #include <streambuf>
   5
   6 // http://open-source-parsers.github.io/jsoncpp-docs/doxygen/namespacemembers.html
   7
   8
   9 #include "config.h"
  10
  11 Config* Config::instance = NULL;
  12
  13 Config::Config()
  14 : jconfigro(jconfig)
  15 {
  16   instance = this;
  17   applyDefaults();
  18 }
  19
  20 Config::~Config()
  21 {
  22   instance = NULL;
  23 }
  24
  25 Config* Config::getInstance()
  26 {
  27   return instance;
  28 }
  29
  30 void Config::applyDefaults()
  31 {
  32   // Insert the value only if it doesn't already exist
  33   auto insertBool   = [&] (const char* s, const char* k, bool v)        { if (jconfig[s][k].isNull()) jconfig[s][k] = v; };
  34   auto insertString = [&] (const char* s, const char* k, const char* v) { if (jconfig[s][k].isNull()) jconfig[s][k] = v; };
  35   auto insertInt    = [&] (const char* s, const char* k, int v)         { if (jconfig[s][k].isNull()) jconfig[s][k] = v; };
  36
  37   insertBool("main", "daemonize", true);
  38
  39   insertBool("log", "enabled", false);
  40   insertString("log", "filename", "stdout");
  41   insertString("log", "level", "debug");
  42
  43   #ifdef VOMP_PLATFORM_RASPBERRY
  44   insertBool("input", "mod_cec_enabled", true);
  45   #else
  46   insertBool("input", "mod_cec_enabled", false);
  47   #endif
  48
  49   insertBool("input", "mod_udp_enabled", true);
  50   insertBool("input", "mod_lirc_enabled", false);
  51
  52   insertInt("server-discovery", "prefer-ipv", 6);
  53 }
  54
  55 bool Config::loadFile()
  56 {
  57   jconfig.clear();
  58
  59   std::ifstream configFile("config.json");
  60   if (configFile.is_open())
  61   {
  62     Json::CharReaderBuilder builder;
  63     builder["collectComments"] = false;
  64     builder["allowTrailingCommas"] = true; // FIXME
  65     std::string errs;
  66
  67     bool ok = Json::parseFromStream(builder, configFile, &jconfig, &errs);
  68     if (!ok)
  69     {
  70       std::cout << errs << std::endl;
  71       return false;
  72     }
  73   }
  74
  75   applyDefaults();
  76   return true;
  77 }
  78
  79 bool Config::getString(const std::string& section, const std::string& key, std::string& out) const
  80 {
  81   Json::Value v = jconfigro[section][key];
  82   if (!v.isString()) return false;
  83   out = v.asString();
  84   return true;
  85 }
  86
  87 bool Config::getInt(const std::string& section, const std::string& key, int& out) const
  88 {
  89   Json::Value v = jconfigro[section][key];
  90   if (!v.isInt64()) return false;
  91   out = v.asInt64();
  92   return true;
  93 }
  94
  95 bool Config::getBool(const std::string& section, const std::string& key, bool& out) const
  96 {
  97   Json::Value v = jconfigro[section][key];
  98   if (!v.isBool()) return false;
  99   out = v.asBool();
 100   return true;
 101 }
 102 /*
 103 bool Config::getBool(const char* section, const char* key, bool& out) const
 104 {
 105   Json::Value v = jconfigro[section][key];
 106   if (!v.isBool()) return false;
 107   out = v.asBool();
 108   return true;
 109 }*/
 110
 111 void Config::dump() const
 112 {
 113   std::cout << jconfigro << std::endl;
 114 }
 115
 116 bool Config::foreachInArray(const std::string& section, const std::string& key, std::function<void(const std::string&)> callback) const
 117 {
 118   const Json::Value& v = jconfigro[section][key];
 119   if (!v.isArray()) return false;
 120
 121   for(Json::Value::const_iterator i = v.begin(); i != v.end(); i++)
 122   {
 123     const Json::Value& jv = *i;
 124     if (jv.isString())
 125       callback(jv.asString());
 126   }
 127
 128   return true;
 129 }
 130
 131 bool Config::foreachPairInObject(const std::string& section, const std::string& key, std::function<void(const std::string&, const std::string&)> callback) const
 132 {
 133   const Json::Value& v = jconfigro[section][key];
 134   if (!v.isObject()) return false;
 135
 136   for(Json::Value::const_iterator i = v.begin(); i != v.end(); i++)
 137   {
 138     const Json::Value& jv = *i;
 139     if (jv.isString())
 140       callback(i.key().asString(), jv.asString());
 141   }
 142
 143   return true;
 144 }
 145
 146 void Config::set(const std::string& section, const std::string& key, const std::string& value)
 147 {
 148   jconfig[section][key] = value;
 149 }
 150
 151 void Config::set(const std::string& section, const std::string& key, bool value)
 152 {
 153   jconfig[section][key] = value;
 154 }
 155
 156 void Config::set(const std::string& section, const std::string& key, const char* value)
 157 {
 158   jconfig[section][key] = value;
 159 }
 160
 161 void Config::set(const std::string& section, const std::string& key, int value)
 162 {
 163   jconfig[section][key] = value;
 164 }