pollymc/logic/OneSixAssets.cpp

146 lines
3.8 KiB
C++
Raw Normal View History

2013-08-04 03:28:39 +05:30
#include <QString>
#include <QDebug>
#include <QtXml/QtXml>
#include "OneSixAssets.h"
#include "net/DownloadJob.h"
2013-08-04 03:28:39 +05:30
inline QDomElement getDomElementByTagName(QDomElement parent, QString tagname)
{
QDomNodeList elementList = parent.elementsByTagName(tagname);
if (elementList.count())
return elementList.at(0).toElement();
else
return QDomElement();
}
class ThreadedDeleter : public QThread
{
Q_OBJECT
public:
void run()
{
2013-09-02 03:55:40 +05:30
QDirIterator iter ( m_base, QDirIterator::Subdirectories );
2013-08-04 03:28:39 +05:30
int base_length = m_base.length();
2013-09-02 03:55:40 +05:30
while ( iter.hasNext() )
2013-08-04 03:28:39 +05:30
{
QString filename = iter.next();
2013-09-02 03:55:40 +05:30
QFileInfo current ( filename );
2013-08-04 03:28:39 +05:30
// we keep the dirs... whatever
2013-09-02 03:55:40 +05:30
if ( current.isDir() )
2013-08-04 03:28:39 +05:30
continue;
QString trimmedf = filename;
2013-09-02 03:55:40 +05:30
trimmedf.remove ( 0, base_length + 1 );
if ( m_whitelist.contains ( trimmedf ) )
2013-08-04 03:28:39 +05:30
{
// qDebug() << trimmedf << " gets to live";
2013-08-04 03:28:39 +05:30
}
else
{
// DO NOT TOLERATE JUNK
// qDebug() << trimmedf << " dies";
2013-09-02 03:55:40 +05:30
QFile f ( filename );
2013-08-04 03:28:39 +05:30
f.remove();
}
}
2013-09-02 03:55:40 +05:30
}
2013-08-04 03:28:39 +05:30
QString m_base;
QStringList m_whitelist;
};
2013-09-02 03:55:40 +05:30
void OneSixAssets::downloadFinished()
2013-08-04 03:28:39 +05:30
{
2013-09-02 03:55:40 +05:30
deleter = new ThreadedDeleter();
QDir dir("assets");
deleter->m_base = dir.absolutePath();
deleter->m_whitelist = nuke_whitelist;
connect(deleter, SIGNAL(finished()), SIGNAL(finished()));
deleter->start();
}
2013-08-04 03:28:39 +05:30
2013-09-02 03:55:40 +05:30
void OneSixAssets::fetchXMLFinished()
2013-08-04 03:28:39 +05:30
{
QString prefix ( "http://s3.amazonaws.com/Minecraft.Resources/" );
QString fprefix ( "assets/" );
2013-09-02 03:55:40 +05:30
nuke_whitelist.clear();
2013-08-04 03:28:39 +05:30
2013-09-02 03:55:40 +05:30
auto firstJob = index_job->first();
QByteArray ba = firstJob->m_data;
2013-08-04 03:28:39 +05:30
QString xmlErrorMsg;
QDomDocument doc;
if ( !doc.setContent ( ba, false, &xmlErrorMsg ) )
{
2013-09-02 03:55:40 +05:30
qDebug() << "Failed to process s3.amazonaws.com/Minecraft.Resources. XML error:" << xmlErrorMsg << ba;
2013-08-04 03:28:39 +05:30
}
//QRegExp etag_match(".*([a-f0-9]{32}).*");
QDomNodeList contents = doc.elementsByTagName ( "Contents" );
2013-09-02 03:55:40 +05:30
DownloadJob *job = new DownloadJob();
connect ( job, SIGNAL(succeeded()), SLOT(downloadFinished()) );
connect ( job, SIGNAL(failed()), SIGNAL(failed()) );
2013-08-04 03:28:39 +05:30
for ( int i = 0; i < contents.length(); i++ )
{
QDomElement element = contents.at ( i ).toElement();
if ( element.isNull() )
continue;
QDomElement keyElement = getDomElementByTagName ( element, "Key" );
QDomElement lastmodElement = getDomElementByTagName ( element, "LastModified" );
QDomElement etagElement = getDomElementByTagName ( element, "ETag" );
QDomElement sizeElement = getDomElementByTagName ( element, "Size" );
if ( keyElement.isNull() || lastmodElement.isNull() || etagElement.isNull() || sizeElement.isNull() )
continue;
QString keyStr = keyElement.text();
QString lastModStr = lastmodElement.text();
QString etagStr = etagElement.text();
QString sizeStr = sizeElement.text();
//Filter folder keys
if ( sizeStr == "0" )
continue;
QString filename = fprefix + keyStr;
QFile check_file ( filename );
QString client_etag = "nonsense";
// if there already is a file and md5 checking is in effect and it can be opened
if ( check_file.exists() && check_file.open ( QIODevice::ReadOnly ) )
{
// check the md5 against the expected one
client_etag = QCryptographicHash::hash ( check_file.readAll(), QCryptographicHash::Md5 ).toHex().constData();
check_file.close();
}
2013-08-04 03:28:39 +05:30
QString trimmedEtag = etagStr.remove ( '"' );
nuke_whitelist.append ( keyStr );
if(trimmedEtag != client_etag)
2013-09-02 03:55:40 +05:30
{
job->add ( QUrl ( prefix + keyStr ), filename );
}
}
if(job->size())
{
files_job.reset ( job );
files_job->start();
}
else
{
delete job;
emit finished();
2013-08-04 03:28:39 +05:30
}
}
void OneSixAssets::start()
{
2013-09-02 03:55:40 +05:30
DownloadJob * job = new DownloadJob(QUrl ( "http://s3.amazonaws.com/Minecraft.Resources/" ));
connect ( job, SIGNAL(succeeded()), SLOT ( fetchXMLFinished() ) );
index_job.reset ( job );
2013-09-02 03:55:40 +05:30
job->start();
2013-08-04 03:28:39 +05:30
}
2013-09-02 03:55:40 +05:30
#include "OneSixAssets.moc"