1 /*

     2  * Hedgewars, a free turn based strategy game

     3  * Copyright (c) 2006-2011 Andrey Korotaev <unC0Rr@gmail.com>

     4  *

     5  * This program is free software; you can redistribute it and/or modify

     6  * it under the terms of the GNU General Public License as published by

     7  * the Free Software Foundation; version 2 of the License

     8  *

     9  * This program is distributed in the hope that it will be useful,

    10  * but WITHOUT ANY WARRANTY; without even the implied warranty of

    11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    12  * GNU General Public License for more details.

    13  *

    14  * You should have received a copy of the GNU General Public License

    15  * along with this program; if not, write to the Free Software

    16  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA

    17  */

    18 

    19 #include <QGridLayout>

    20 #include <QPushButton>

    21 #include <QNetworkAccessManager>

    22 #include <QNetworkRequest>

    23 #include <QNetworkReply>

    24 #include <QFileInfo>

    25 #include <QFileDialog>

    26 #include <QDebug>

    27 #include <QProgressBar>

    28 #include <QBuffer>

    29 

    30 #include "pagedata.h"

    31 #include "databrowser.h"

    32 #include "hwconsts.h"

    33 

    34 #include "quazip.h"

    35 #include "quazipfile.h"

    36 

    37 PageDataDownload::PageDataDownload(QWidget* parent) : AbstractPage(parent)

    38 {

    39     QGridLayout * pageLayout = new QGridLayout(this);

    40     pageLayout->setColumnStretch(0, 1);

    41     pageLayout->setColumnStretch(1, 1);

    42     pageLayout->setColumnStretch(2, 1);

    43 

    44     BtnBack = addButton(":/res/Exit.png", pageLayout, 2, 0, true);

    45 

    46     web = new DataBrowser(this);

    47     connect(web, SIGNAL(anchorClicked(QUrl)), this, SLOT(request(const QUrl&)));

    48     web->setOpenLinks(false);

    49     pageLayout->addWidget(web, 0, 0, 1, 3);

    50 

    51     progressBarsLayout = new QVBoxLayout();

    52     pageLayout->addLayout(progressBarsLayout, 1, 0, 1, 3);

    53 

    54     fetchList();

    55 }

    56 

    57 void PageDataDownload::request(const QUrl &url)

    58 {

    59     QUrl finalUrl;

    60     if(url.host().isEmpty())

    61         finalUrl = QUrl("http://www.hedgewars.org" + url.path());

    62     else

    63         finalUrl = url;

    64 

    65     if(url.path().endsWith(".zip"))

    66     {

    67         qWarning() << "Download Request" << url.toString();

    68         QString fileName = QFileInfo(url.toString()).fileName();

    69 

    70         QNetworkRequest newRequest(finalUrl);

    71         newRequest.setAttribute(QNetworkRequest::User, fileName);

    72 

    73         QNetworkAccessManager *manager = new QNetworkAccessManager(this);

    74         QNetworkReply *reply = manager->get(newRequest);

    75         connect(reply, SIGNAL(finished()), this, SLOT(fileDownloaded()));

    76         connect(reply, SIGNAL(downloadProgress(qint64, qint64)), this, SLOT(downloadProgress(qint64, qint64)));

    77 

    78         QProgressBar *progressBar = new QProgressBar(this);

    79         progressBarsLayout->addWidget(progressBar);

    80         progressBars.insert(reply, progressBar);

    81     } else

    82     {

    83         qWarning() << "Page Request" << url.toString();

    84 

    85         QNetworkRequest newRequest(finalUrl);

    86 

    87         QNetworkAccessManager *manager = new QNetworkAccessManager(this);

    88         QNetworkReply *reply = manager->get(newRequest);

    89         connect(reply, SIGNAL(finished()), this, SLOT(pageDownloaded()));

    90     }

    91 }

    92 

    93 

    94 void PageDataDownload::pageDownloaded()

    95 {

    96     QNetworkReply * reply = qobject_cast<QNetworkReply *>(sender());

    97 

    98     if(reply)

    99     {

   100         QString html = QString::fromUtf8(reply->readAll());

   101         int begin = html.indexOf("<!-- BEGIN -->");

   102         int end = html.indexOf("<!-- END -->");

   103         if(begin != -1 && begin < end)

   104         {

   105             html.truncate(end);

   106             html.remove(0, begin);

   107         }

   108         web->setHtml(html);

   109     }

   110 }

   111 

   112 void PageDataDownload::fileDownloaded()

   113 {

   114     QNetworkReply * reply = qobject_cast<QNetworkReply *>(sender());

   115 

   116     if(reply)

   117     {

   118         QByteArray fileContents = reply->readAll();

   119         QProgressBar *progressBar = progressBars.value(reply, 0);

   120 

   121         if(progressBar)

   122         {

   123             progressBars.remove(reply);

   124             progressBar->deleteLater();

   125         }

   126 

   127         extractDataPack(&fileContents);

   128     }

   129 }

   130 

   131 void PageDataDownload::downloadProgress(qint64 bytesRecieved, qint64 bytesTotal)

   132 {

   133     QNetworkReply * reply = qobject_cast<QNetworkReply *>(sender());

   134 

   135     if(reply)

   136     {

   137         QProgressBar *progressBar = progressBars.value(reply, 0);

   138 

   139         if(progressBar)

   140         {

   141             progressBar->setValue(bytesRecieved);

   142             progressBar->setMaximum(bytesTotal);

   143         }

   144     }

   145 }

   146 

   147 void PageDataDownload::fetchList()

   148 {

   149     request(QUrl("http://hedgewars.org/content.html"));

   150 }

   151 

   152 bool PageDataDownload::extractDataPack(QByteArray * buf)

   153 {

   154     QBuffer buffer;

   155     buffer.setBuffer(buf);

   156 

   157     QuaZip zip;

   158     zip.setIoDevice(&buffer);

   159     if(!zip.open(QuaZip::mdUnzip))

   160     {

   161       qWarning("testRead(): zip.open(): %d", zip.getZipError());

   162       return false;

   163     }

   164 

   165     QuaZipFile file(&zip);

   166 

   167     QDir extractDir(*cfgdir);

   168     extractDir.cd("Data");

   169 

   170     for(bool more = zip.goToFirstFile(); more; more = zip.goToNextFile())

   171     {

   172         if(!file.open(QIODevice::ReadOnly))

   173         {

   174             qWarning("file.open(): %d", file.getZipError());

   175             return false;

   176         }

   177 

   178 

   179         QString fileName = file.getActualFileName();

   180         QString filePath = extractDir.filePath(fileName);

   181         if (fileName.endsWith("/"))

   182         {

   183             QFileInfo fi(filePath);

   184             QDir().mkpath(fi.filePath());

   185         } else

   186         {

   187             qDebug() << "Extracting" << filePath;

   188             QFile out(filePath);

   189             if(!out.open(QFile::WriteOnly))

   190             {

   191                 qWarning() << "out.open():" << out.errorString();

   192                 return false;

   193             }

   194 

   195             out.write(file.readAll());

   196 

   197             out.close();

   198 

   199             if(file.getZipError() != UNZ_OK) {

   200                 qWarning("file.getFileName(): %d", file.getZipError());

   201                 return false;

   202             }

   203 

   204             if(!file.atEnd()) {

   205                 qWarning("read all but not EOF");

   206                 return false;

   207             }

   208         }

   209 

   210         file.close();

   211 

   212         if(file.getZipError()!=UNZ_OK) {

   213             qWarning("file.close(): %d", file.getZipError());

   214             return false;

   215         }

   216     }

   217 

   218     zip.close();

   219 

   220     return true;

   221 }