CactiChameleon9/PrismLauncher
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add a job that removes any files that don't match a whitelist from a folder.

Browse Source
This commit is contained in:
Petr Mrázek 2013-07-06 02:41:23 +02:00
parent 5c07e1b042
commit eb16d977c8
1 changed files with 53 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
asset_test.cpp
View File

@ -1,5 +1,7 @@
#include <QString>
#include <QDebug> #include <QDebug>
#include <QtXml/QtXml> #include <QtXml/QtXml>
#include "dlqueue.h" #include "dlqueue.h"
inline QDomElement getDomElementByTagName(QDomElement parent, QString tagname) inline QDomElement getDomElementByTagName(QDomElement parent, QString tagname)
@ -11,6 +13,51 @@ inline QDomElement getDomElementByTagName(QDomElement parent, QString tagname)
return QDomElement(); return QDomElement();
} }
// a job that removes all files from the base folder that don't match the whitelist
// runs in whatever thread owns the queue. it is fast though.
class NukeAndPaveJob: public Job
{
public:
explicit NukeAndPaveJob(QString base, QStringList whitelist)
:Job()
{
QDir dir(base);
m_base = dir.absolutePath();
m_whitelist = whitelist;
};
virtual void start()
{
QDirIterator iter(m_base, QDirIterator::Subdirectories);
QStringList nuke_list;
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))
{
//qDebug() << trimmedf << " gets to live";
}
else
{
// DO NOT TOLERATE JUNK
//qDebug() << trimmedf << " dies";
QFile f (filename);
f.remove();
}
}
emit finish();
};
private:
QString m_base;
QStringList m_whitelist;
};
class DlMachine : public QObject class DlMachine : public QObject
{ {
Q_OBJECT Q_OBJECT
@ -22,6 +69,10 @@ public slots:
void fetchFinished() void fetchFinished()
{ {
QString prefix("http://s3.amazonaws.com/Minecraft.Resources/");
QString fprefix("assets/");
QStringList nuke_whitelist;
JobPtr firstJob = index_job->getFirstJob(); JobPtr firstJob = index_job->getFirstJob();
auto DlJob = firstJob.dynamicCast<DownloadJob>(); auto DlJob = firstJob.dynamicCast<DownloadJob>();
QByteArray ba = DlJob->m_data; QByteArray ba = DlJob->m_data;
@ -64,10 +115,10 @@ public slots:
continue; continue;
QString trimmedEtag = etagStr.remove('"'); QString trimmedEtag = etagStr.remove('"');
QString prefix("http://s3.amazonaws.com/Minecraft.Resources/");
QString fprefix("assets/");
job->add(DownloadJob::create(QUrl(prefix + keyStr),fprefix + keyStr, trimmedEtag)); job->add(DownloadJob::create(QUrl(prefix + keyStr),fprefix + keyStr, trimmedEtag));
nuke_whitelist.append(keyStr);
} }
job->add(JobPtr(new NukeAndPaveJob(fprefix, nuke_whitelist)));
files_job.reset(job); files_job.reset(job);
dl.enqueue(files_job); dl.enqueue(files_job);
} }