sneedmc/logic/minecraft/AssetsUtils.cpp

185 lines
4.7 KiB
C++
Raw Normal View History

2015-02-02 22:25:30 +00:00
/* Copyright 2013-2015 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 <QDir>
#include <QDirIterator>
#include <QCryptographicHash>
#include <QJsonParseError>
#include <QJsonDocument>
#include <QJsonObject>
#include <QVariant>
#include <QDebug>
#include "AssetsUtils.h"
2015-10-04 23:47:27 +00:00
#include <FileSystem.h>
namespace AssetsUtils
{
/*
* Returns true on success, with index populated
* index is undefined otherwise
*/
bool loadAssetsIndexJson(QString path, AssetsIndex *index)
{
2013-12-10 06:12:52 +00:00
/*
{
"objects": {
"icons/icon_16x16.png": {
"hash": "bdf48ef6b5d0d23bbb02e17d04865216179f510a",
"size": 3665
},
...
}
}
}
2013-12-10 06:12:52 +00:00
*/
QFile file(path);
// Try to open the file and fail if we can't.
// TODO: We should probably report this error to the user.
if (!file.open(QIODevice::ReadOnly))
{
qCritical() << "Failed to read assets index file" << path;
return false;
}
// Read the file and close it.
QByteArray jsonData = file.readAll();
file.close();
QJsonParseError parseError;
QJsonDocument jsonDoc = QJsonDocument::fromJson(jsonData, &parseError);
// Fail if the JSON is invalid.
if (parseError.error != QJsonParseError::NoError)
{
qCritical() << "Failed to parse assets index file:" << parseError.errorString()
2013-12-10 06:12:52 +00:00
<< "at offset " << QString::number(parseError.offset);
return false;
}
// Make sure the root is an object.
if (!jsonDoc.isObject())
{
qCritical() << "Invalid assets index JSON: Root should be an array.";
return false;
}
QJsonObject root = jsonDoc.object();
QJsonValue isVirtual = root.value("virtual");
2013-12-10 06:12:52 +00:00
if (!isVirtual.isUndefined())
{
index->isVirtual = isVirtual.toBool(false);
}
QJsonValue objects = root.value("objects");
QVariantMap map = objects.toVariant().toMap();
2013-12-10 06:12:52 +00:00
for (QVariantMap::const_iterator iter = map.begin(); iter != map.end(); ++iter)
{
// qDebug() << iter.key();
QVariant variant = iter.value();
QVariantMap nested_objects = variant.toMap();
AssetObject object;
2013-12-10 06:12:52 +00:00
for (QVariantMap::const_iterator nested_iter = nested_objects.begin();
nested_iter != nested_objects.end(); ++nested_iter)
{
// qDebug() << nested_iter.key() << nested_iter.value().toString();
QString key = nested_iter.key();
QVariant value = nested_iter.value();
2013-12-10 06:12:52 +00:00
if (key == "hash")
{
object.hash = value.toString();
}
2013-12-10 06:12:52 +00:00
else if (key == "size")
{
object.size = value.toDouble();
}
}
2013-12-10 06:12:52 +00:00
index->objects.insert(iter.key(), object);
}
return true;
}
2015-01-27 21:31:07 +00:00
QDir reconstructAssets(QString assetsId)
{
QDir assetsDir = QDir("assets/");
2015-10-04 23:47:27 +00:00
QDir indexDir = QDir(FS::PathCombine(assetsDir.path(), "indexes"));
QDir objectDir = QDir(FS::PathCombine(assetsDir.path(), "objects"));
QDir virtualDir = QDir(FS::PathCombine(assetsDir.path(), "virtual"));
2015-01-27 21:31:07 +00:00
2015-10-04 23:47:27 +00:00
QString indexPath = FS::PathCombine(indexDir.path(), assetsId + ".json");
2015-01-27 21:31:07 +00:00
QFile indexFile(indexPath);
2015-10-04 23:47:27 +00:00
QDir virtualRoot(FS::PathCombine(virtualDir.path(), assetsId));
2015-01-27 21:31:07 +00:00
if (!indexFile.exists())
{
qCritical() << "No assets index file" << indexPath << "; can't reconstruct assets";
2015-01-27 21:31:07 +00:00
return virtualRoot;
}
qDebug() << "reconstructAssets" << assetsDir.path() << indexDir.path()
2015-01-27 21:31:07 +00:00
<< objectDir.path() << virtualDir.path() << virtualRoot.path();
AssetsIndex index;
bool loadAssetsIndex = AssetsUtils::loadAssetsIndexJson(indexPath, &index);
if (loadAssetsIndex && index.isVirtual)
{
qDebug() << "Reconstructing virtual assets folder at" << virtualRoot.path();
2015-01-27 21:31:07 +00:00
for (QString map : index.objects.keys())
{
AssetObject asset_object = index.objects.value(map);
2015-10-04 23:47:27 +00:00
QString target_path = FS::PathCombine(virtualRoot.path(), map);
2015-01-27 21:31:07 +00:00
QFile target(target_path);
QString tlk = asset_object.hash.left(2);
2015-10-04 23:47:27 +00:00
QString original_path = FS::PathCombine(objectDir.path(), tlk, asset_object.hash);
2015-01-27 21:31:07 +00:00
QFile original(original_path);
if (!original.exists())
continue;
if (!target.exists())
{
QFileInfo info(target_path);
QDir target_dir = info.dir();
// qDebug() << target_dir;
2015-01-27 21:31:07 +00:00
if (!target_dir.exists())
QDir("").mkpath(target_dir.path());
bool couldCopy = original.copy(target_path);
qDebug() << " Copying" << original_path << "to" << target_path
2015-01-27 21:31:07 +00:00
<< QString::number(couldCopy); // << original.errorString();
}
}
// TODO: Write last used time to virtualRoot/.lastused
}
return virtualRoot;
}
}