pollymc/depends/settings/inifile.cpp

/* Copyright 2013 MultiMC Contributors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "inifile.h"

#include <QFile>
#include <QTextStream>
#include <QStringList>

INIFile::INIFile()
{
}

QString INIFile::unescape(QString orig)
{
	orig.replace("\\n", "\n");
	orig.replace("\\t", "\t");
	orig.replace("\\\\", "\\");
	return orig;
}
QString INIFile::escape(QString orig)
{
	orig.replace("\\", "\\\\");
	orig.replace("\n", "\\n");
	orig.replace("\t", "\\t");
	return orig;
}

bool INIFile::saveFile(QString fileName)
{
	// TODO Handle errors.
	QFile file(fileName);
	file.open(QIODevice::WriteOnly);
	QTextStream out(&file);
	out.setCodec("UTF-8");

	for (Iterator iter = begin(); iter != end(); iter++)
	{
		QString value = iter.value().toString();
		value = escape(value);
		out << iter.key() << "=" << value << "\n";
	}

	return true;
}

bool INIFile::loadFile(QString fileName)
{
	QFile file(fileName);
	if (!file.open(QIODevice::ReadOnly))
		return false;
	bool success = loadFile(file.readAll());
	file.close();
	return success;
}
bool INIFile::loadFile(QByteArray file)
{
	QTextStream in(file);
	in.setCodec("UTF-8");

	QStringList lines = in.readAll().split('\n');
	for (int i = 0; i < lines.count(); i++)
	{
		QString &lineRaw = lines[i];
		// Ignore comments.
		QString line = lineRaw.left(lineRaw.indexOf('#')).trimmed();

		int eqPos = line.indexOf('=');
		if (eqPos == -1)
			continue;
		QString key = line.left(eqPos).trimmed();
		QString valueStr = line.right(line.length() - eqPos - 1).trimmed();

		valueStr = unescape(valueStr);

		QVariant value(valueStr);
		this->operator[](key) = value;
	}

	return true;
}

QVariant INIFile::get(QString key, QVariant def) const
{
	if (!this->contains(key))
		return def;
	else
		return this->operator[](key);
}

void INIFile::set(QString key, QVariant val)
{
	this->operator[](key) = val;
}