scummvm/engines/wintermute/Base/BRegistry.cpp

/* ScummVM - Graphic Adventure Engine
 *
 * ScummVM is the legal property of its developers, whose names
 * are too numerous to list here. Please refer to the COPYRIGHT
 * file distributed with this source distribution.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.

 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.

 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 */

/*
 * This file is based on WME Lite.
 * http://dead-code.org/redir.php?target=wmelite
 * Copyright (c) 2011 Jan Nedoma
 */

#include "engines/wintermute/tinyxml/tinyxml.h"
#include "engines/wintermute/Base/BGame.h"
#include "engines/wintermute/Base/BRegistry.h"
#include "engines/wintermute/utils/PathUtil.h"
#include "engines/wintermute/utils/StringUtil.h"
#include "engines/wintermute/utils/utils.h"

#include "common/file.h"

namespace WinterMute {

//////////////////////////////////////////////////////////////////////////
CBRegistry::CBRegistry(CBGame *inGame): CBBase(inGame) {
	_iniName = NULL;

	setIniName("./wme.ini");
	loadValues(true);
}


//////////////////////////////////////////////////////////////////////////
CBRegistry::~CBRegistry() {
	saveValues();
	delete[] _iniName;
	_iniName = NULL;
}



//////////////////////////////////////////////////////////////////////////
AnsiString CBRegistry::readString(const AnsiString &subKey, const AnsiString &key, const AnsiString &init) {
	AnsiString ret = "";

#ifdef __WIN32__
	// check ini file first (so what we can use project files on windows)
	char buffer[32768];
	GetPrivateProfileString(subKey.c_str(), key.c_str(), init.c_str(), buffer, 32768, _iniName);
	ret = AnsiString(buffer);

	if (buffer != init) return ret;
#endif

	bool found = false;
	ret = getValue(_localValues, subKey, key, found);
	if (!found) ret = getValue(_values, subKey, key, found);
	if (!found) ret = init;

	return ret;
}


//////////////////////////////////////////////////////////////////////////
bool CBRegistry::writeString(const AnsiString &subKey, const AnsiString &key, const AnsiString &value) {
	_values[subKey][key] = value;
	return true;
}


//////////////////////////////////////////////////////////////////////////
int CBRegistry::readInt(const AnsiString &subKey, const AnsiString &key, int init) {
#ifdef __WIN32__
	int ret = GetPrivateProfileInt(subKey.c_str(), key.c_str(), init, _iniName);
	if (ret != init) return ret;
#endif

	AnsiString val = readString(subKey, key, "");
	if (val.empty()) return init;
	else return atoi(val.c_str());
}


//////////////////////////////////////////////////////////////////////////
bool CBRegistry::writeInt(const AnsiString &subKey, const AnsiString &key, int value) {
	writeString(subKey, key, StringUtil::toString(value));
	return true;
}


//////////////////////////////////////////////////////////////////////////
bool CBRegistry::readBool(const AnsiString &subKey, const AnsiString &key, bool init) {
	return (readInt(subKey, key, (int)init) != 0);
}


//////////////////////////////////////////////////////////////////////////
bool CBRegistry::writeBool(const AnsiString &subKey, const AnsiString &key, bool value) {
	return writeInt(subKey, key, (int)value);
}


//////////////////////////////////////////////////////////////////////////
void CBRegistry::setIniName(const char *name) {
	delete[] _iniName;
	_iniName = NULL;

	if (strchr(name, '\\') == NULL && strchr(name, '/') == NULL) {
		_iniName = new char [strlen(name) + 3];
		sprintf(_iniName, "./%s", name);
	} else {
		_iniName = new char [strlen(name) + 1];
		strcpy(_iniName, name);
	}
}


//////////////////////////////////////////////////////////////////////////
char *CBRegistry::getIniName() {
	return _iniName;
}

//////////////////////////////////////////////////////////////////////////
void CBRegistry::loadValues(bool local) {
	if (local) loadXml("settings.xml", _localValues);
	else loadXml(PathUtil::combine(Game->getDataDir(), "settings.xml"), _values);
}

//////////////////////////////////////////////////////////////////////////
void CBRegistry::saveValues() {
	saveXml(PathUtil::combine(Game->getDataDir(), "settings.xml"), _values);
}

//////////////////////////////////////////////////////////////////////////
void CBRegistry::setBasePath(const char *basePath) {
	_basePath = PathUtil::getFileNameWithoutExtension(basePath);

	loadValues(false);
}

//////////////////////////////////////////////////////////////////////////
AnsiString CBRegistry::getValue(PathValueMap &values, const AnsiString path, const AnsiString &key, bool &found) {
	found = false;
	PathValueMap::iterator it = values.find(path);
	if (it == values.end()) return "";

	KeyValuePair pairs = (*it)._value;
	KeyValuePair::iterator keyIt = pairs.find(key);
	if (keyIt == pairs.end()) return "";
	else {
		found = true;
		return (*keyIt)._value;
	}
}

//////////////////////////////////////////////////////////////////////////
void CBRegistry::loadXml(const AnsiString fileName, PathValueMap &values) {
	TiXmlDocument doc(fileName.c_str());
	if (!doc.LoadFile()) return;

	TiXmlElement *rootElem = doc.RootElement();
	if (!rootElem || Common::String(rootElem->Value()) != "Settings") // TODO: Avoid this strcmp-use. (Hack for now, since we might drop TinyXML all together)
		return;

	for (TiXmlElement *pathElem = rootElem->FirstChildElement(); pathElem != NULL; pathElem = pathElem->NextSiblingElement()) {
		for (TiXmlElement *keyElem = pathElem->FirstChildElement(); keyElem != NULL; keyElem = keyElem->NextSiblingElement()) {
			values[Common::String(pathElem->Value())][Common::String(keyElem->Value())] = keyElem->GetText();
		}
	}
}


//////////////////////////////////////////////////////////////////////////
void CBRegistry::saveXml(const AnsiString fileName, PathValueMap &values) {
	CBUtils::createPath(fileName.c_str());

	TiXmlDocument doc;
	doc.LinkEndChild(new TiXmlDeclaration("1.0", "utf-8", ""));

	TiXmlElement *root = new TiXmlElement("Settings");
	doc.LinkEndChild(root);

	PathValueMap::iterator pathIt;
	for (pathIt = _values.begin(); pathIt != _values.end(); ++pathIt) {
		TiXmlElement *pathElem = new TiXmlElement((*pathIt)._key.c_str());
		root->LinkEndChild(pathElem);


		KeyValuePair pairs = (*pathIt)._value;
		KeyValuePair::iterator keyIt;
		for (keyIt = pairs.begin(); keyIt != pairs.end(); ++keyIt) {
			TiXmlElement *keyElem = new TiXmlElement((*keyIt)._key.c_str());
			pathElem->LinkEndChild(keyElem);

			keyElem->LinkEndChild(new TiXmlText((*keyIt)._value.c_str()));
		}
	}


	TiXmlPrinter printer;
	doc.Accept(&printer);

	Common::DumpFile stream;
	stream.open(fileName.c_str());

	if (!stream.isOpen()) return;
	else {
		stream.write(printer.CStr(), printer.Size());
		stream.close();
	}
}

} // end of namespace WinterMute