window.cpp

/****************************************************************************
**
** Copyright (C) 2015 The Qt Company Ltd.
** Contact: http://www.qt.io/licensing/
**
** This file is part of the examples of the Qt Toolkit.
**
** $QT_BEGIN_LICENSE:BSD$
** You may use this file under the terms of the BSD license as follows:
**
** "Redistribution and use in source and binary forms, with or without
** modification, are permitted provided that the following conditions are
** met:
**   * Redistributions of source code must retain the above copyright
**     notice, this list of conditions and the following disclaimer.
**   * Redistributions in binary form must reproduce the above copyright
**     notice, this list of conditions and the following disclaimer in
**     the documentation and/or other materials provided with the
**     distribution.
**   * Neither the name of The Qt Company Ltd nor the names of its
**     contributors may be used to endorse or promote products derived
**     from this software without specific prior written permission.
**
**
** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
**
** $QT_END_LICENSE$
**
****************************************************************************/

#include <QtWidgets>
#include <QtNetwork>
#include "uploadslot.h"
#include "window.h"

//! [0]
Window::Window(QWidget *parent)
    : QWidget(parent)
{
    browseButton = createButton(tr("&Browse..."), SLOT(browse()));
    findButton = createButton(tr("&Find"), SLOT(find()));
    tinyAllButton = createButton(tr("&Tiny"), SLOT(tinyAll()));
    outBrowseButton = createButton(tr("&Browse..."), SLOT(browse2()));

    fileComboBox = createComboBox(tr("*"));
    textComboBox = createComboBox();
    directoryComboBox = createComboBox(QDir::currentPath());
    outDirectoryComboBox = createComboBox(QDir::currentPath());
    tokenComboBox = createComboBox(tr("HyDxVk41AiyRezVSYFaycTqq1jwgDmm9"));

    fileLabel = new QLabel(tr("Named:"));
    textLabel = new QLabel(tr("Containing text:"));
    tokenLabel = new QLabel(tr("Token:"));
    directoryLabel = new QLabel(tr("In directory:"));
    outDirecotryLabel = new QLabel(tr("Out Directory"));
    filesFoundLabel = new QLabel;

    createFilesTable();
//! [0]

//! [1]
    QGridLayout *mainLayout = new QGridLayout;
    mainLayout->addWidget(fileLabel, 0, 0);
    mainLayout->addWidget(fileComboBox, 0, 1, 1, 2);
    mainLayout->addWidget(textLabel, 1, 0);
    mainLayout->addWidget(textComboBox, 1, 1, 1, 2);
    mainLayout->addWidget(tokenLabel, 2, 0);
    mainLayout->addWidget(tokenComboBox, 2, 1, 1, 2);
    mainLayout->addWidget(directoryLabel, 3, 0);
    mainLayout->addWidget(directoryComboBox, 3, 1);
    mainLayout->addWidget(browseButton, 3, 2);
    mainLayout->addWidget(outDirecotryLabel, 4, 0);
    mainLayout->addWidget(outDirectoryComboBox, 4, 1);
    mainLayout->addWidget(outBrowseButton, 4, 2);
    mainLayout->addWidget(filesTable, 5, 0, 1, 3);
    mainLayout->addWidget(filesFoundLabel, 6, 1, 1, 2);
    mainLayout->addWidget(tinyAllButton, 6, 2);
    mainLayout->addWidget(findButton, 6, 0);
    setLayout(mainLayout);

    setWindowTitle(tr("*･゜ﾟ･*:.｡..｡.:*･'(*ﾟ▽ﾟ*)'･*:.｡. .｡.:*･゜ﾟ･*"));
    resize(700, 700);
}
//! [1]

//! [2]
void Window::browse()
{
    QString directory = QFileDialog::getExistingDirectory(this,
                               tr("Find Files"), QDir::currentPath());

    QComboBox *comboBox = directoryComboBox;
    if (!directory.isEmpty()) {
        if (comboBox->findText(directory) == -1)
            comboBox->addItem(directory);
        comboBox->setCurrentIndex(comboBox->findText(directory));
    }
}

void Window::browse2()
{
    QString directory = QFileDialog::getExistingDirectory(this,
                               tr("Find Files"), QDir::currentPath());

    QComboBox *comboBox = outDirectoryComboBox;

    if (!directory.isEmpty()) {
        if (comboBox->findText(directory) == -1)
            comboBox->addItem(directory);
        comboBox->setCurrentIndex(comboBox->findText(directory));
    }
}
//! [2]

static void updateComboBox(QComboBox *comboBox)
{
    if (comboBox->findText(comboBox->currentText()) == -1)
        comboBox->addItem(comboBox->currentText());
}

//! [3]
void Window::find()
{
    filesTable->setRowCount(0);

    QString fileName = fileComboBox->currentText();
    QString text = textComboBox->currentText();
    QString path = directoryComboBox->currentText();
//! [3]

    updateComboBox(fileComboBox);
    updateComboBox(textComboBox);
    updateComboBox(directoryComboBox);

//! [4]
    currentDir = QDir(path);
    QStringList files;
    if (fileName.isEmpty())
        fileName = "*";
    files = currentDir.entryList(QStringList(fileName),
                                 QDir::Files | QDir::NoSymLinks);

    if (!text.isEmpty())
        files = findFiles(files, text);
    showFiles(files);
}
//! [4]

void Window::tinyAll()
{
    QModelIndexList selection = filesTable->selectionModel()->selectedRows();
    for (int i = 0; i < selection.count(); i ++) {
        this->openFileOfItem(selection.at(i).row(), 0);
    }
}

//! [5]
QStringList Window::findFiles(const QStringList &files, const QString &text)
{
    QProgressDialog progressDialog(this);
    progressDialog.setCancelButtonText(tr("&Cancel"));
    progressDialog.setRange(0, files.size());
    progressDialog.setWindowTitle(tr("Find Files"));

//! [5] //! [6]
    QStringList foundFiles;

    for (int i = 0; i < files.size(); ++i) {
        progressDialog.setValue(i);
        progressDialog.setLabelText(tr("Searching file number %1 of %2...")
                                    .arg(i).arg(files.size()));
        qApp->processEvents();
//! [6]

        if (progressDialog.wasCanceled())
            break;

//! [7]
        QFile file(currentDir.absoluteFilePath(files[i]));

        if (file.open(QIODevice::ReadOnly)) {
            QString line;
            QTextStream in(&file);
            while (!in.atEnd()) {
                if (progressDialog.wasCanceled())
                    break;
                line = in.readLine();
                if (line.contains(text)) {
                    foundFiles << files[i];
                    break;
                }
            }
        }
    }
    return foundFiles;
}
//! [7]

//! [8]
void Window::showFiles(const QStringList &files)
{
    for (int i = 0; i < files.size(); ++i) {
        QFile file(currentDir.absoluteFilePath(files[i]));
        qint64 size = QFileInfo(file).size();

        QTableWidgetItem *fileNameItem = new QTableWidgetItem(files[i]);
        fileNameItem->setFlags(fileNameItem->flags() ^ Qt::ItemIsEditable);
        QTableWidgetItem *sizeItem = new QTableWidgetItem(tr("%1 KB")
                                             .arg(int((size + 1023) / 1024)));
        sizeItem->setTextAlignment(Qt::AlignRight | Qt::AlignVCenter);
        sizeItem->setFlags(sizeItem->flags() ^ Qt::ItemIsEditable);

        int row = filesTable->rowCount();
        filesTable->insertRow(row);
        filesTable->setItem(row, 0, fileNameItem);
        filesTable->setItem(row, 1, sizeItem);
    }
    filesFoundLabel->setText(tr("%1 file(s) found").arg(files.size()) +
                             (" (Double click on a file to shrink it)"));
    filesFoundLabel->setWordWrap(true);
}
//! [8]

//! [9]
QPushButton *Window::createButton(const QString &text, const char *member)
{
    QPushButton *button = new QPushButton(text);
    connect(button, SIGNAL(clicked()), this, member);
    return button;
}
//! [9]

//! [10]
QComboBox *Window::createComboBox(const QString &text)
{
    QComboBox *comboBox = new QComboBox;
    comboBox->setEditable(true);
    comboBox->addItem(text);
    comboBox->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Preferred);
    return comboBox;
}
//! [10]

//! [11]
void Window::createFilesTable()
{
    filesTable = new QTableWidget(0, 4);
    filesTable->setSelectionBehavior(QAbstractItemView::SelectRows);
    filesTable->setSelectionMode(QAbstractItemView::ExtendedSelection);
    QStringList labels;
    labels << tr("Filename") << tr("Size") << tr("ShrinkedSize") << tr("OutFilename");
    filesTable->setHorizontalHeaderLabels(labels);
    filesTable->horizontalHeader()->setStretchLastSection(true);
    filesTable->horizontalHeader()->setSectionResizeMode(0, QHeaderView::Stretch);
    filesTable->verticalHeader()->hide();
    filesTable->setShowGrid(false);

    connect(filesTable, SIGNAL(cellActivated(int,int)),
            this, SLOT(openFileOfItem(int,int)));
}
//! [11]

//! [12]

void Window::openFileOfItem(int row, int /* column */)
{
    QTableWidgetItem *sizeItem = new QTableWidgetItem(tr("shrinking"));
    sizeItem->setTextAlignment(Qt::AlignRight | Qt::AlignVCenter);
    sizeItem->setFlags(sizeItem->flags() ^ Qt::ItemIsEditable);
    filesTable->setItem(row, 2, sizeItem);

    QTableWidgetItem *item = filesTable->item(row, 0);
    QString path = currentDir.absoluteFilePath(item->text());
    QFileInfo fileInfo(path);
    QString outpath = outDirectoryComboBox->currentText();
    QDir outDir = QDir(outpath);
    QString outPath = outDir.absoluteFilePath(fileInfo.fileName());

    QNetworkRequest *nr = new QNetworkRequest(QUrl("https://api.tinify.com/shrink"));
    QNetworkAccessManager *manager = new QNetworkAccessManager();
    UploadSlot *slot = new UploadSlot(manager, row, outPath);
    QString concatenated = tokenComboBox->currentText();
    QByteArray data = concatenated.toLocal8Bit().toBase64();
    QString headerData = "Basic " + data;
    nr->setRawHeader("Authorization", headerData.toLocal8Bit());
    nr->setRawHeader("Content-Type", "application/x-www-form-urlencoded");
    QByteArray postData("");

    QFile file(path);
            if (!file.open(QIODevice::ReadOnly)){
                qDebug() << "QFile Error: File not found!";
            } else { qDebug() << "File found, proceed as planned"; }
    postData.append(file.readAll());
    file.close();
    connect(manager, SIGNAL(finished(QNetworkReply*)),
            slot, SLOT(replyFinished(QNetworkReply*)));
    connect(slot, SIGNAL(transferDone(int,int,int,QString)), this, SLOT(transferDone(int,int,int,QString)));
    const QNetworkRequest snr = QNetworkRequest(*nr);
    const QByteArray pd = QByteArray(postData);
    manager->post(snr, pd);
}

void Window::transferDone(int row, int status, int size, QString path)
{
    if (status == 0) {
        QTableWidgetItem *sizeItem = new QTableWidgetItem(tr("error"));
        filesTable->setItem(row, 2, sizeItem);
        QTableWidgetItem *errorItem = new QTableWidgetItem(path);
        filesTable->setItem(row, 3, errorItem);
    } else {
        QTableWidgetItem *fileNameItem = new QTableWidgetItem(path);
        fileNameItem->setFlags(fileNameItem->flags() ^ Qt::ItemIsEditable);
        QTableWidgetItem *sizeItem = new QTableWidgetItem(tr("%1 KB")
                                             .arg(int((size + 1023) / 1024)));
        sizeItem->setTextAlignment(Qt::AlignRight | Qt::AlignVCenter);
        sizeItem->setFlags(sizeItem->flags() ^ Qt::ItemIsEditable);

        filesTable->setItem(row, 3, fileNameItem);
        filesTable->setItem(row, 2, sizeItem);
    }

}

//! [12]