112 lines
2.5 KiB
C++
112 lines
2.5 KiB
C++
#include <QString>
|
|
#include <logger/QsLog.h>
|
|
#include <QtXml/QtXml>
|
|
#include "OneSixAssets.h"
|
|
#include "net/NetJob.h"
|
|
#include "net/HttpMetaCache.h"
|
|
#include "net/S3ListBucket.h"
|
|
#include "MultiMC.h"
|
|
|
|
class ThreadedDeleter : public QThread
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
void run()
|
|
{
|
|
QLOG_INFO() << "Cleaning up assets folder...";
|
|
QDirIterator iter(m_base, QDirIterator::Subdirectories);
|
|
int base_length = m_base.length();
|
|
while (iter.hasNext())
|
|
{
|
|
QString filename = iter.next();
|
|
QFileInfo current(filename);
|
|
// we keep the dirs... whatever
|
|
if (current.isDir())
|
|
continue;
|
|
QString trimmedf = filename;
|
|
trimmedf.remove(0, base_length + 1);
|
|
if (m_whitelist.contains(trimmedf))
|
|
{
|
|
QLOG_TRACE() << trimmedf << " gets to live";
|
|
}
|
|
else
|
|
{
|
|
// DO NOT TOLERATE JUNK
|
|
QLOG_TRACE() << trimmedf << " dies";
|
|
QFile f(filename);
|
|
f.remove();
|
|
}
|
|
}
|
|
}
|
|
QString m_base;
|
|
QStringList m_whitelist;
|
|
};
|
|
|
|
void OneSixAssets::downloadFinished()
|
|
{
|
|
deleter = new ThreadedDeleter();
|
|
QDir dir("assets");
|
|
deleter->m_base = dir.absolutePath();
|
|
deleter->m_whitelist = nuke_whitelist;
|
|
connect(deleter, SIGNAL(finished()), SIGNAL(finished()));
|
|
deleter->start();
|
|
}
|
|
|
|
void OneSixAssets::S3BucketFinished()
|
|
{
|
|
QString prefix("http://s3.amazonaws.com/Minecraft.Resources/");
|
|
nuke_whitelist.clear();
|
|
|
|
emit filesStarted();
|
|
|
|
auto firstJob = index_job->first();
|
|
auto objectList = std::dynamic_pointer_cast<S3ListBucket>(firstJob)->objects;
|
|
|
|
NetJob *job = new NetJob("Assets");
|
|
|
|
connect(job, SIGNAL(succeeded()), SLOT(downloadFinished()));
|
|
connect(job, SIGNAL(failed()), SIGNAL(failed()));
|
|
connect(job, SIGNAL(filesProgress(int, int, int)), SIGNAL(filesProgress(int, int, int)));
|
|
|
|
auto metacache = MMC->metacache();
|
|
|
|
for (auto object: objectList)
|
|
{
|
|
// Filter folder keys (zero size)
|
|
if (object.size == 0)
|
|
continue;
|
|
|
|
nuke_whitelist.append(object.Key);
|
|
|
|
auto entry = metacache->resolveEntry("assets", object.Key, object.ETag);
|
|
if (entry->stale)
|
|
{
|
|
job->addNetAction(CacheDownload::make(QUrl(prefix + object.Key), entry));
|
|
}
|
|
}
|
|
if (job->size())
|
|
{
|
|
files_job.reset(job);
|
|
files_job->start();
|
|
}
|
|
else
|
|
{
|
|
delete job;
|
|
emit finished();
|
|
}
|
|
}
|
|
|
|
void OneSixAssets::start()
|
|
{
|
|
auto job = new NetJob("Assets index");
|
|
job->addNetAction(
|
|
S3ListBucket::make(QUrl("http://s3.amazonaws.com/Minecraft.Resources/")));
|
|
connect(job, SIGNAL(succeeded()), SLOT(S3BucketFinished()));
|
|
connect(job, SIGNAL(failed()), SIGNAL(failed()));
|
|
emit indexStarted();
|
|
index_job.reset(job);
|
|
job->start();
|
|
}
|
|
|
|
#include "OneSixAssets.moc"
|