hashviewwidget.cpp

#include "hashviewwidget.h"
#include <QLabel>
#include <QBoxLayout>
#include <QtGui>
#include "crc32.h"
#include "md5.h"
#include "sha1.h"
#include <QHeaderView>
#include <QFileDialog>
#include <QProgressDialog>
#include <QErrorMessage>
#include "enumTypes.h"
#include "algorithminterface.h"

HashViewWidget::HashViewWidget(int type, QFileInfoList filesList, QWidget *parent) :
    QWidget(parent), algorithmType(type), selectedFileList(filesList)
{
    ok = 0;
    corruptData = 0;
    notOpened = 0;
    filesTable = new QTableWidget(0, 3, this);
    QStringList labels;
    labels << "File Name" << "Size (KB)" << "Hash";
    filesTable->setHorizontalHeaderLabels(labels);
    filesTable->horizontalHeader()->setSectionResizeMode(0, QHeaderView::ResizeToContents);
    filesTable->horizontalHeader()->setStretchLastSection(true);
    filesTable->setShowGrid(false);
    filesTable->setSortingEnabled(false);

    QLabel *label = new QLabel("Result for selected files:");
    saveButton = new QPushButton("Save results");
    QVBoxLayout *vbox = new QVBoxLayout();
    vbox->addWidget(label);
    vbox->addWidget(filesTable);
    vbox->addWidget(saveButton);
    setLayout(vbox);
    connect(saveButton, SIGNAL(clicked()), SLOT(saveFilesHash()));
    connect(this, SIGNAL(statusChanged(QString)), this->parent(), SLOT(changeStatus(QString)));
}

void HashViewWidget::saveFilesHash()
{
    QString fileSuffix;
    QString fileFilter;
    switch (algorithmType)
    {
        case AlgorithmType::crc32:
            fileSuffix = "*.crc32";
            break;
        case AlgorithmType::md5:
            fileSuffix = "*.md5";
            break;
        case AlgorithmType::sha1:
            fileSuffix = "*.sha1";
            break;
        default:
            break;
    }
    fileFilter = tr("hash (%1);; text files (*.txt)").arg(fileSuffix);
    QString filePath = QFileDialog::getSaveFileUrl(this, "Save File", QDir::currentPath(), fileFilter).toLocalFile();

    QFile file(filePath);
    if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
    {
        QErrorMessage *mes = new QErrorMessage(this);
        mes->showMessage(tr("File not opened (path: \"%1\")").arg(filePath));
        return;
    }

    QTextStream out(&file);
    int count = filesTable->rowCount();
    for(int i = 0; i < count; i++)
    {
        out << filesTable->item(i, 2)->text() << " *" + filesTable->item(i, 0)->text()<< "\n";
    }
    file.close();
}

void HashViewWidget::addTableItems()
{
    QProgressDialog progress("", "Abort", 0, selectedFileList.count(), this);
    progress.setWindowModality(Qt::WindowModal);
    progress.setWindowTitle("Create files hash...");
    progress.setMinimumDuration(0);

    for (int i = 0; i < selectedFileList.size(); i++)
    {
        progress.setValue(i);
        progress.setLabelText(tr("File number %1 of %2...").arg(i+1).arg(selectedFileList.count()));
        if (progress.wasCanceled())
            break;

        QTableWidgetItem *fileNameItem = new QTableWidgetItem(selectedFileList[i].fileName());
        QTableWidgetItem *sizeItem = new QTableWidgetItem();
        sizeItem->setData(Qt::DisplayRole, (selectedFileList[i].size() + 1023) / 1024);
        sizeItem->setTextAlignment(Qt::AlignCenter);

        QTableWidgetItem *hashItem = new QTableWidgetItem(getHash(selectedFileList[i].absoluteFilePath()));
        hashItem->setTextAlignment(Qt::AlignCenter);

        int row = filesTable->rowCount();
        filesTable->insertRow(row);
        filesTable->setItem(row, 0, fileNameItem);
        filesTable->setItem(row, 1, sizeItem);
        filesTable->setItem(row, 2, hashItem);
    }
    progress.setValue(selectedFileList.count());
    filesTable->setSortingEnabled(true);
    filesTable->sortByColumn(0);
    emit statusChanged(tr("Ok: %1. Errors: %2 (Not opened: %3. Corrupt files: %4)").arg(ok)
                       .arg(notOpened+corruptData).arg(notOpened).arg(corruptData));
}

void HashViewWidget::processStatus(int status)
{
    switch (status)
    {
        case ErrorType::noError:
            ok++;
            break;
        case ErrorType::notOpened:
            notOpened++;
            break;
        case ErrorType::corruptData:
            corruptData++;
            break;
        default:
            break;
    }
}

QString HashViewWidget::getHash(QString path)
{
    QString hash;
    AlgorithmInterface *algorithm;
    switch (algorithmType)
    {
        case AlgorithmType::crc32:
            algorithm = new Crc32();
            break;
        case AlgorithmType::md5:
            algorithm = new MD5();
            break;
        case AlgorithmType::sha1:
            algorithm = new Sha1();
            break;
        default:
            algorithm = new Sha1();
            break;
    }
    int status = algorithm->calculateFile(path);
    processStatus(status);
    hash = algorithm->getHashString();
    delete algorithm;
    return hash;
}