sneedmc/logic/InstanceList.cpp

609 lines
16 KiB
C++
Raw Normal View History

2013-01-14 23:42:38 +00:00
/* 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
2013-10-05 23:13:40 +00:00
*
2013-01-14 23:42:38 +00:00
* 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.
*/
2013-02-19 18:15:22 +00:00
#include <QDir>
#include <QSet>
2013-02-19 18:15:22 +00:00
#include <QFile>
#include <QDirIterator>
2013-03-18 22:00:46 +00:00
#include <QThread>
#include <QTextStream>
#include <QJsonDocument>
#include <QJsonObject>
#include <QJsonArray>
#include <QXmlStreamReader>
#include <QRegularExpression>
#include <pathutils.h>
2013-02-19 18:15:22 +00:00
#include "MultiMC.h"
2014-05-08 19:20:10 +00:00
#include "logic/InstanceList.h"
#include "logic/icons/IconList.h"
2014-05-08 19:20:10 +00:00
#include "logic/minecraft/MinecraftVersionList.h"
#include "logic/BaseInstance.h"
#include "logic/InstanceFactory.h"
#include "logger/QsLog.h"
2014-05-08 19:20:10 +00:00
#include "gui/groupview/GroupView.h"
2013-02-19 18:15:22 +00:00
const static int GROUP_FILE_FORMAT_VERSION = 1;
2013-02-19 18:15:22 +00:00
2013-10-05 23:13:40 +00:00
InstanceList::InstanceList(const QString &instDir, QObject *parent)
: QAbstractListModel(parent), m_instDir(instDir)
2013-02-19 18:15:22 +00:00
{
2013-12-02 10:09:56 +00:00
connect(MMC, &MultiMC::aboutToQuit, this, &InstanceList::saveGroupList);
2013-12-02 11:02:37 +00:00
if (!QDir::current().exists(m_instDir))
{
QDir::current().mkpath(m_instDir);
}
2013-02-19 18:15:22 +00:00
}
InstanceList::~InstanceList()
{
}
2013-10-05 23:13:40 +00:00
int InstanceList::rowCount(const QModelIndex &parent) const
{
2013-10-05 23:13:40 +00:00
Q_UNUSED(parent);
return m_instances.count();
}
2013-10-05 23:13:40 +00:00
QModelIndex InstanceList::index(int row, int column, const QModelIndex &parent) const
{
2013-10-05 23:13:40 +00:00
Q_UNUSED(parent);
if (row < 0 || row >= m_instances.size())
return QModelIndex();
2013-10-05 23:13:40 +00:00
return createIndex(row, column, (void *)m_instances.at(row).get());
}
2013-10-05 23:13:40 +00:00
QVariant InstanceList::data(const QModelIndex &index, int role) const
{
2013-10-05 23:13:40 +00:00
if (!index.isValid())
{
return QVariant();
}
2013-10-05 23:13:40 +00:00
BaseInstance *pdata = static_cast<BaseInstance *>(index.internalPointer());
switch (role)
{
case InstancePointerRole:
{
2013-10-05 23:13:40 +00:00
QVariant v = qVariantFromValue((void *)pdata);
return v;
}
2014-03-30 18:11:05 +00:00
case InstanceIDRole:
{
return pdata->id();
}
case Qt::DisplayRole:
{
return pdata->name();
}
case Qt::ToolTipRole:
{
return pdata->instanceRoot();
}
case Qt::DecorationRole:
{
QString key = pdata->iconKey();
return MMC->icons()->getIcon(key);
}
// for now.
2014-02-04 00:40:51 +00:00
case GroupViewRoles::GroupRole:
{
return pdata->group();
}
default:
break;
}
return QVariant();
}
2013-10-05 23:13:40 +00:00
Qt::ItemFlags InstanceList::flags(const QModelIndex &index) const
{
Qt::ItemFlags f;
2013-10-05 23:13:40 +00:00
if (index.isValid())
{
2013-10-05 23:13:40 +00:00
f |= (Qt::ItemIsEnabled | Qt::ItemIsSelectable);
}
return f;
}
void InstanceList::groupChanged()
{
// save the groups. save all of them.
saveGroupList();
}
QStringList InstanceList::getGroups()
{
return m_groups.toList();
}
void InstanceList::saveGroupList()
{
QString groupFileName = m_instDir + "/instgroups.json";
QFile groupFile(groupFileName);
2013-10-05 23:13:40 +00:00
// if you can't open the file, fail
2013-10-05 23:13:40 +00:00
if (!groupFile.open(QIODevice::WriteOnly | QIODevice::Truncate))
{
// An error occurred. Ignore it.
QLOG_ERROR() << "Failed to save instance group file.";
return;
}
QTextStream out(&groupFile);
2013-10-05 23:13:40 +00:00
QMap<QString, QSet<QString>> groupMap;
for (auto instance : m_instances)
{
QString id = instance->id();
QString group = instance->group();
2013-10-05 23:13:40 +00:00
if (group.isEmpty())
continue;
// keep a list/set of groups for choosing
m_groups.insert(group);
2013-10-05 23:13:40 +00:00
if (!groupMap.count(group))
{
QSet<QString> set;
set.insert(id);
groupMap[group] = set;
}
else
{
QSet<QString> &set = groupMap[group];
set.insert(id);
}
}
QJsonObject toplevel;
2013-10-05 23:13:40 +00:00
toplevel.insert("formatVersion", QJsonValue(QString("1")));
QJsonObject groupsArr;
2013-10-05 23:13:40 +00:00
for (auto iter = groupMap.begin(); iter != groupMap.end(); iter++)
{
auto list = iter.value();
auto name = iter.key();
QJsonObject groupObj;
QJsonArray instanceArr;
2013-10-05 23:13:40 +00:00
groupObj.insert("hidden", QJsonValue(QString("false")));
for (auto item : list)
{
instanceArr.append(QJsonValue(item));
}
2013-10-05 23:13:40 +00:00
groupObj.insert("instances", instanceArr);
groupsArr.insert(name, groupObj);
}
2013-10-05 23:13:40 +00:00
toplevel.insert("groups", groupsArr);
QJsonDocument doc(toplevel);
2013-08-26 02:57:09 +00:00
groupFile.write(doc.toJson());
groupFile.close();
}
2013-10-05 23:13:40 +00:00
void InstanceList::loadGroupList(QMap<QString, QString> &groupMap)
2013-01-14 23:42:38 +00:00
{
QString groupFileName = m_instDir + "/instgroups.json";
2013-10-05 23:13:40 +00:00
// if there's no group file, fail
2013-10-05 23:13:40 +00:00
if (!QFileInfo(groupFileName).exists())
return;
2013-10-05 23:13:40 +00:00
QFile groupFile(groupFileName);
2013-10-05 23:13:40 +00:00
// if you can't open the file, fail
if (!groupFile.open(QIODevice::ReadOnly))
{
// An error occurred. Ignore it.
2013-10-05 23:13:40 +00:00
QLOG_ERROR() << "Failed to read instance group file.";
return;
}
2013-10-05 23:13:40 +00:00
QTextStream in(&groupFile);
QString jsonStr = in.readAll();
groupFile.close();
2013-10-05 23:13:40 +00:00
QJsonParseError error;
QJsonDocument jsonDoc = QJsonDocument::fromJson(jsonStr.toUtf8(), &error);
2013-10-05 23:13:40 +00:00
// if the json was bad, fail
if (error.error != QJsonParseError::NoError)
{
2013-10-05 23:13:40 +00:00
QLOG_ERROR() << QString("Failed to parse instance group file: %1 at offset %2")
.arg(error.errorString(), QString::number(error.offset))
.toUtf8();
return;
}
2013-10-05 23:13:40 +00:00
// if the root of the json wasn't an object, fail
if (!jsonDoc.isObject())
{
QLOG_WARN() << "Invalid group file. Root entry should be an object.";
return;
}
2013-10-05 23:13:40 +00:00
QJsonObject rootObj = jsonDoc.object();
2013-10-05 23:13:40 +00:00
// Make sure the format version matches, otherwise fail.
if (rootObj.value("formatVersion").toVariant().toInt() != GROUP_FILE_FORMAT_VERSION)
return;
2013-10-05 23:13:40 +00:00
// Get the groups. if it's not an object, fail
if (!rootObj.value("groups").isObject())
{
QLOG_WARN() << "Invalid group list JSON: 'groups' should be an object.";
return;
}
2013-10-05 23:13:40 +00:00
// Iterate through all the groups.
QJsonObject groupMapping = rootObj.value("groups").toObject();
for (QJsonObject::iterator iter = groupMapping.begin(); iter != groupMapping.end(); iter++)
{
QString groupName = iter.key();
2013-10-05 23:13:40 +00:00
// If not an object, complain and skip to the next one.
if (!iter.value().isObject())
{
QLOG_WARN() << QString("Group '%1' in the group list should "
"be an object.")
.arg(groupName)
.toUtf8();
continue;
}
2013-10-05 23:13:40 +00:00
QJsonObject groupObj = iter.value().toObject();
if (!groupObj.value("instances").isArray())
{
QLOG_WARN() << QString("Group '%1' in the group list is invalid. "
"It should contain an array "
"called 'instances'.")
.arg(groupName)
.toUtf8();
continue;
}
2013-10-05 23:13:40 +00:00
// keep a list/set of groups for choosing
m_groups.insert(groupName);
// Iterate through the list of instances in the group.
QJsonArray instancesArray = groupObj.value("instances").toArray();
2013-10-05 23:13:40 +00:00
for (QJsonArray::iterator iter2 = instancesArray.begin(); iter2 != instancesArray.end();
iter2++)
{
groupMap[(*iter2).toString()] = groupName;
}
}
}
QList<FTBRecord> InstanceList::discoverFTBInstances()
{
QList<FTBRecord> records;
2014-05-23 16:41:22 +00:00
QDir dir = QDir(MMC->settings()->get("FTBLauncherDataRoot").toString());
QDir dataDir = QDir(MMC->settings()->get("FTBRoot").toString());
2014-05-23 16:41:22 +00:00
if (!dataDir.exists())
{
2014-05-23 16:41:22 +00:00
QLOG_INFO() << "The FTB directory specified does not exist. Please check your settings";
return records;
}
2014-05-23 16:41:22 +00:00
else if (!dir.exists())
{
2014-05-23 16:41:22 +00:00
QLOG_INFO() << "The FTB launcher data directory specified does not exist. Please check your settings";
return records;
}
dir.cd("ModPacks");
auto allFiles = dir.entryList(QDir::Readable | QDir::Files, QDir::Name);
for (auto filename : allFiles)
{
if (!filename.endsWith(".xml"))
continue;
auto fpath = dir.absoluteFilePath(filename);
QFile f(fpath);
QLOG_INFO() << "Discovering FTB instances -- " << fpath;
if (!f.open(QFile::ReadOnly))
continue;
// read the FTB packs XML.
QXmlStreamReader reader(&f);
while (!reader.atEnd())
{
switch (reader.readNext())
{
case QXmlStreamReader::StartElement:
{
if (reader.name() == "modpack")
{
QXmlStreamAttributes attrs = reader.attributes();
FTBRecord record;
record.dirName = attrs.value("dir").toString();
record.instanceDir = dataDir.absoluteFilePath(record.dirName);
record.templateDir = dir.absoluteFilePath(record.dirName);
QDir test(record.instanceDir);
2014-05-23 16:41:22 +00:00
QLOG_DEBUG() << dataDir.absolutePath() << record.instanceDir << record.dirName;
if (!test.exists())
continue;
record.name = attrs.value("name").toString();
record.logo = attrs.value("logo").toString();
record.mcVersion = attrs.value("mcVersion").toString();
record.description = attrs.value("description").toString();
records.append(record);
}
break;
}
case QXmlStreamReader::EndElement:
break;
case QXmlStreamReader::Characters:
break;
default:
break;
}
}
f.close();
}
return records;
}
void InstanceList::loadFTBInstances(QMap<QString, QString> &groupMap,
QList<InstancePtr> &tempList)
{
auto records = discoverFTBInstances();
if (!records.size())
{
QLOG_INFO() << "No FTB instances to load.";
return;
}
QLOG_INFO() << "Loading FTB instances! -- got " << records.size();
// process the records we acquired.
for (auto record : records)
{
QLOG_INFO() << "Loading FTB instance from " << record.instanceDir;
QString iconKey = record.logo;
iconKey.remove(QRegularExpression("\\..*"));
MMC->icons()->addIcon(iconKey, iconKey, PathCombine(record.templateDir, record.logo),
MMCIcon::Transient);
if (!QFileInfo(PathCombine(record.instanceDir, "instance.cfg")).exists())
{
QLOG_INFO() << "Converting " << record.name << " as new.";
2014-03-30 18:11:05 +00:00
InstancePtr instPtr;
auto &factory = InstanceFactory::get();
auto version = MMC->minecraftlist()->findVersion(record.mcVersion);
if (!version)
{
QLOG_ERROR() << "Can't load instance " << record.instanceDir
<< " because minecraft version " << record.mcVersion
<< " can't be resolved.";
continue;
}
auto error = factory.createInstance(instPtr, version, record.instanceDir,
InstanceFactory::FTBInstance);
if (!instPtr || error != InstanceFactory::NoCreateError)
continue;
instPtr->setGroupInitial("FTB");
instPtr->setName(record.name);
instPtr->setIconKey(iconKey);
instPtr->setIntendedVersionId(record.mcVersion);
instPtr->setNotes(record.description);
if(!continueProcessInstance(instPtr, error, record.instanceDir, groupMap))
continue;
tempList.append(InstancePtr(instPtr));
}
else
{
QLOG_INFO() << "Loading existing " << record.name;
2014-03-30 18:11:05 +00:00
InstancePtr instPtr;
auto error = InstanceFactory::get().loadInstance(instPtr, record.instanceDir);
if (!instPtr || error != InstanceFactory::NoLoadError)
continue;
instPtr->setGroupInitial("FTB");
instPtr->setName(record.name);
instPtr->setIconKey(iconKey);
if (instPtr->intendedVersionId() != record.mcVersion)
instPtr->setIntendedVersionId(record.mcVersion);
instPtr->setNotes(record.description);
if(!continueProcessInstance(instPtr, error, record.instanceDir, groupMap))
continue;
tempList.append(InstancePtr(instPtr));
}
}
}
InstanceList::InstListError InstanceList::loadList()
{
// load the instance groups
QMap<QString, QString> groupMap;
loadGroupList(groupMap);
2013-10-05 23:13:40 +00:00
QList<InstancePtr> tempList;
2013-02-19 18:15:22 +00:00
{
QDirIterator iter(m_instDir, QDir::Dirs | QDir::NoDot | QDir::NoDotDot | QDir::Readable,
QDirIterator::FollowSymlinks);
while (iter.hasNext())
{
QString subDir = iter.next();
if (!QFileInfo(PathCombine(subDir, "instance.cfg")).exists())
continue;
QLOG_INFO() << "Loading MultiMC instance from " << subDir;
2014-03-30 18:11:05 +00:00
InstancePtr instPtr;
auto error = InstanceFactory::get().loadInstance(instPtr, subDir);
if(!continueProcessInstance(instPtr, error, subDir, groupMap))
continue;
2014-03-30 18:11:05 +00:00
tempList.append(instPtr);
}
}
2013-10-05 23:13:40 +00:00
if (MMC->settings()->get("TrackFTBInstances").toBool())
{
loadFTBInstances(groupMap, tempList);
}
beginResetModel();
m_instances.clear();
for(auto inst: tempList)
{
inst->setParent(this);
connect(inst.get(), SIGNAL(propertiesChanged(BaseInstance *)), this,
SLOT(propertiesChanged(BaseInstance *)));
connect(inst.get(), SIGNAL(groupChanged()), this, SLOT(groupChanged()));
connect(inst.get(), SIGNAL(nuked(BaseInstance *)), this,
SLOT(instanceNuked(BaseInstance *)));
m_instances.append(inst);
2013-02-19 18:15:22 +00:00
}
endResetModel();
emit dataIsInvalid();
2013-02-19 18:15:22 +00:00
return NoError;
2013-01-14 23:42:38 +00:00
}
2013-03-18 22:00:46 +00:00
/// Clear all instances. Triggers notifications.
void InstanceList::clear()
{
beginResetModel();
saveGroupList();
2013-03-18 22:00:46 +00:00
m_instances.clear();
endResetModel();
emit dataIsInvalid();
2013-10-05 23:13:40 +00:00
}
void InstanceList::on_InstFolderChanged(const Setting &setting, QVariant value)
{
m_instDir = value.toString();
loadList();
}
2013-03-18 22:00:46 +00:00
/// Add an instance. Triggers notifications, returns the new index
int InstanceList::add(InstancePtr t)
{
beginInsertRows(QModelIndex(), m_instances.size(), m_instances.size());
2013-03-18 22:00:46 +00:00
m_instances.append(t);
t->setParent(this);
2013-10-05 23:13:40 +00:00
connect(t.get(), SIGNAL(propertiesChanged(BaseInstance *)), this,
SLOT(propertiesChanged(BaseInstance *)));
connect(t.get(), SIGNAL(groupChanged()), this, SLOT(groupChanged()));
connect(t.get(), SIGNAL(nuked(BaseInstance *)), this, SLOT(instanceNuked(BaseInstance *)));
endInsertRows();
2013-03-18 22:00:46 +00:00
return count() - 1;
}
InstancePtr InstanceList::getInstanceById(QString instId) const
2013-03-18 22:00:46 +00:00
{
if (m_instances.isEmpty())
{
return InstancePtr();
}
2013-03-18 22:00:46 +00:00
QListIterator<InstancePtr> iter(m_instances);
InstancePtr inst;
2013-10-05 23:13:40 +00:00
while (iter.hasNext())
2013-03-18 22:00:46 +00:00
{
inst = iter.next();
if (inst->id() == instId)
break;
}
if (inst->id() != instId)
return InstancePtr();
else
return iter.peekPrevious();
}
QModelIndex InstanceList::getInstanceIndexById(const QString &id) const
{
return index(getInstIndex(getInstanceById(id).get()));
}
int InstanceList::getInstIndex(BaseInstance *inst) const
{
2013-10-05 23:13:40 +00:00
for (int i = 0; i < m_instances.count(); i++)
{
2013-10-05 23:13:40 +00:00
if (inst == m_instances[i].get())
{
return i;
}
}
return -1;
}
2014-03-30 18:11:05 +00:00
bool InstanceList::continueProcessInstance(InstancePtr instPtr, const int error,
const QDir &dir, QMap<QString, QString> &groupMap)
{
if (error != InstanceFactory::NoLoadError && error != InstanceFactory::NotAnInstance)
{
QString errorMsg = QString("Failed to load instance %1: ")
.arg(QFileInfo(dir.absolutePath()).baseName())
.toUtf8();
switch (error)
{
default:
errorMsg += QString("Unknown instance loader error %1").arg(error);
break;
}
QLOG_ERROR() << errorMsg.toUtf8();
return false;
}
else if (!instPtr)
{
QLOG_ERROR() << QString("Error loading instance %1. Instance loader returned null.")
.arg(QFileInfo(dir.absolutePath()).baseName())
.toUtf8();
return false;
}
else
{
auto iter = groupMap.find(instPtr->id());
if (iter != groupMap.end())
{
instPtr->setGroupInitial((*iter));
}
QLOG_INFO() << "Loaded instance " << instPtr->name() << " from " << dir.absolutePath();
return true;
}
}
2013-10-05 23:13:40 +00:00
void InstanceList::instanceNuked(BaseInstance *inst)
{
int i = getInstIndex(inst);
2013-10-05 23:13:40 +00:00
if (i != -1)
{
2013-10-05 23:13:40 +00:00
beginRemoveRows(QModelIndex(), i, i);
m_instances.removeAt(i);
endRemoveRows();
}
}
2013-10-05 23:13:40 +00:00
void InstanceList::propertiesChanged(BaseInstance *inst)
{
int i = getInstIndex(inst);
2013-10-05 23:13:40 +00:00
if (i != -1)
{
emit dataChanged(index(i), index(i));
}
}
InstanceProxyModel::InstanceProxyModel(QObject *parent) : GroupedProxyModel(parent)
{
}
2013-10-05 23:13:40 +00:00
bool InstanceProxyModel::subSortLessThan(const QModelIndex &left,
const QModelIndex &right) const
{
2013-10-05 23:13:40 +00:00
BaseInstance *pdataLeft = static_cast<BaseInstance *>(left.internalPointer());
BaseInstance *pdataRight = static_cast<BaseInstance *>(right.internalPointer());
2013-11-03 23:11:20 +00:00
QString sortMode = MMC->settings()->get("InstSortMode").toString();
if (sortMode == "LastLaunch")
2013-11-03 23:11:20 +00:00
{
return pdataLeft->lastLaunch() > pdataRight->lastLaunch();
}
else
{
return QString::localeAwareCompare(pdataLeft->name(), pdataRight->name()) < 0;
}
}