freeShop/src/Config.cpp

#include <cpp3ds/System/FileSystem.hpp>
#include <fstream>
#include <rapidjson/ostreamwrapper.h>
#include <rapidjson/writer.h>
#include <cpp3ds/System/FileInputStream.hpp>
#include "Config.hpp"
#include "Util.hpp"

namespace {
	const char *keyStrings[] = {
		"cache_version",
		"auto-update",
		"trigger_update",
		"last_updated",
		"download_title_keys",
		"key_urls",
		"sleep_mode",
	};
}

namespace FreeShop {

Config::Config()
{
	static_assert(KEY_COUNT == sizeof(keyStrings)/sizeof(*keyStrings), "Key string count much match the enum count.");
	loadDefaults();
}

Config &Config::getInstance()
{
	static Config config;
	return config;
}

bool Config::loadFromFile(const std::string &filename)
{
	rapidjson::Document &json = getInstance().m_json;
	std::string path = cpp3ds::FileSystem::getFilePath(filename);
	std::string jsonString;
	cpp3ds::FileInputStream file;
	if (!file.open(filename))
		return false;
	jsonString.resize(file.getSize());
	file.read(&jsonString[0], jsonString.size());
	json.Parse(jsonString.c_str());
	getInstance().loadDefaults();
	return !json.HasParseError();
}

void Config::saveToFile(const std::string &filename)
{
	std::string path = cpp3ds::FileSystem::getFilePath(filename);
	std::ofstream file(path);
	rapidjson::OStreamWrapper osw(file);
	rapidjson::Writer<rapidjson::OStreamWrapper> writer(osw);
	getInstance().m_json.Accept(writer);
}

bool Config::keyExists(const char *key)
{
	return getInstance().m_json.HasMember(key);
}

const rapidjson::Value &Config::get(Key key)
{
	return getInstance().m_json[keyStrings[key]];
}

#define ADD_DEFAULT(key, val) \
	if (!m_json.HasMember(keyStrings[key])) \
		m_json.AddMember(rapidjson::StringRef(keyStrings[key]), val, m_json.GetAllocator());

void Config::loadDefaults()
{
	if (!m_json.IsObject())
		m_json.SetObject();

	ADD_DEFAULT(CacheVersion, "");

	// Update settings
	ADD_DEFAULT(AutoUpdate, true);
	ADD_DEFAULT(TriggerUpdateFlag, false);
	ADD_DEFAULT(LastUpdatedTime, 0);
	ADD_DEFAULT(DownloadTitleKeys, false);
	ADD_DEFAULT(KeyURLs, rapidjson::kArrayType);

	// Other settings
	ADD_DEFAULT(SleepMode, false);
}

void Config::set(Key key, const char *val)
{
	rapidjson::Value v(val, getInstance().m_json.GetAllocator());
	set(key, v);
}

void Config::set(Key key, rapidjson::Value &val)
{
	const char *keyStr = keyStrings[key];
	if (keyExists(keyStr))
		getInstance().m_json[keyStr] = val;
	else
		getInstance().m_json.AddMember(rapidjson::StringRef(keyStr), val, getInstance().m_json.GetAllocator());
}

rapidjson::Document::AllocatorType &Config::getAllocator()
{
	return getInstance().m_json.GetAllocator();
}


} // namespace FreeShop