screengrab-packaging/src/modules/uploader/uploader.cpp

/***************************************************************************
 *   Copyright (C) 2009 - 2013 by Artem 'DOOMer' Galichkin                 *
 *   doomer3d@gmail.com                                                    *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>. *
 ***************************************************************************/

#include "uploader.h"
#include "core/core.h"
#include "uploaderconfig.h"

#include <QDir>
#include <QUuid>
#include <QRegExp>

#include <QDebug>

Uploader::Uploader(QObject *parent) :
    QObject(parent), _multipartData(0)
{
    qsrand(126);
    _strBoundary = "uploadbound" + QByteArray::number(qrand());
    _net = new QNetworkAccessManager(this);
    _serverReply = 0;
    initUploadedStrList();

    UploaderConfig config;
    if (config.checkExistsConfigFile() == false)
    {
        config.defaultSettings();
    }
}

Uploader::~Uploader()
{

}

/*!
 * Create boundary string
 * \param cleared -detect for str boundary line in httpRequest
 */
QByteArray Uploader::boundary(bool cleared)
{
    QByteArray retBoundary = _strBoundary;

    if (cleared == false)
    {
        retBoundary.append("\r\n");
        retBoundary.prepend("--");
    }

    return retBoundary;
}

void Uploader::replyProgress(qint64 bytesSent, qint64 bytesTotal)
{
    Q_EMIT uploadProgress(bytesSent, bytesTotal);
}

/*!
 *  Get user selected params from uploader widget (not from config file)
 */
void Uploader::getUserSettings(const QVariantMap& settings)
{
    _userSettings = settings;
}

/*!
 *  Start uploading base class
 */
void Uploader::startUploading()
{
    connect(_net, &QNetworkAccessManager::finished, this, &Uploader::replyFinished);

    if (!_multipartData && !imageData.isEmpty())
    {
        _serverReply = _net->post(_request, imageData);
    }
    if (_multipartData && imageData.isEmpty())
    {
        _serverReply = _net->post(_request, _multipartData);
    }

    connect(_serverReply, &QNetworkReply::uploadProgress, this, &Uploader::replyProgress);
}

QMap< QByteArray, ResultString_t > Uploader::parsedLinks()
{
    return _uploadedStrings;
}

QList<ResultString_t> Uploader::parsedLinksToGui()
{
    QList<ResultString_t> list;
    ResultString_t delete_url;
    ResultString_t direct_link;

    for (int i =  0; i < _uploadedStrings.count(); ++i)
    {
        QByteArray key = _uploadedStrings.keys().at(i);

        if (key == "delete_url")
        {
            delete_url = _uploadedStrings[key];
        }
        else if(key == "direct_link")
        {
            direct_link = _uploadedStrings[key];
        }
        else
        {
            ResultString_t val = _uploadedStrings[key];
            list.append(val);
        }
    }

    list.prepend(direct_link);

    if (delete_url.first.isEmpty() == false)
    {
        list.append(delete_url);
    }

    return list;
}


/*!
 * Return url for uploaded image
 */
QUrl Uploader::apiUrl()
{
    return QUrl();
}

/*!
 * Prepare image data for uploading
 */
void Uploader::createData(bool inBase64)
{
    Core *core = Core::instance();
    _formatString = core->config()->getSaveFormat();
    _uploadFilename = core->getTempFilename(_formatString);
    core->writeScreen(_uploadFilename, _formatString , true);

    if (inBase64 == false)
    {
        imageData = core->getScreenData();
    }
    else
    {
        imageData = core->getScreenData().toBase64();
    }
    core->killTempFile();
}

/*!
 *  Create request for send to server.
 *  this method called from subclasses.
 */
void Uploader::createRequest(const QByteArray& requestData, const QUrl url)
{
    Q_UNUSED(requestData);
    _request.setUrl(url);
}

/*!
 *  Parsing server reply and get map with server returned links
 *  \param keytags List of tags for parsing
 *  \param result String wuth server reply
 */
QMap<QByteArray, QByteArray> Uploader::parseResultStrings(const QVector< QByteArray >& keytags, const QByteArray& result)
{
    QMap<QByteArray, QByteArray> replyMap;

    QRegExp re;
    QRegExp re2;

    int inStart = 0;
    int outStart = 0;
    int len = 0;

    // parsing xml
    for (int i = 0; i < keytags.count(); ++i)
    {
        // set patterns  for full lenght item
        re.setPattern("<"+keytags[i]+">"); // open tag
        re2.setPattern("</"+keytags[i]+">"); //close tag

        // get start pos and lenght ite in xml
        inStart = re.indexIn(result); // ops open tag start
        outStart = re2.indexIn(result); // pos close tag start
        len = outStart - inStart + re2.matchedLength(); // length of full string

        // extract item and replace spec html symbols
        QByteArray extractedText = result.mid(inStart, len);
        extractedText = extractedText.replace("&quot;","'");
        extractedText = extractedText.replace("&lt;","<");
        extractedText = extractedText.replace("&gt;",">");
        extractedText = extractedText.replace("<"+keytags[i]+">","");
        extractedText = extractedText.replace("</"+keytags[i]+">","");

        // to map
        replyMap.insert(keytags[i], extractedText);

    }

    return replyMap;
}

void Uploader::initUploadedStrList()
{
    ResultString_t strPair = qMakePair(QByteArray(), tr("Direct link"));
    _uploadedStrings.insert(UL_DIRECT_LINK, strPair);

    strPair = qMakePair(QByteArray(), tr("HTML code"));
    _uploadedStrings.insert(UL_HTML_CODE ,strPair);

    strPair = qMakePair(QByteArray(), tr("BB code"));
    _uploadedStrings.insert(UL_BB_CODE, strPair);

    strPair = qMakePair(QByteArray(), tr("HTML code with thumb image"));
    _uploadedStrings.insert(UL_HTML_CODE_THUMB ,strPair);

    strPair = qMakePair(QByteArray("bb_code_thumb"), tr("BB code with thumb image"));
    _uploadedStrings.insert(UL_BB_CODE_THUMB, strPair);

    strPair = qMakePair(QByteArray(), tr("URl to delete image"));
    _uploadedStrings.insert(UL_DELETE_URL ,strPair);
}