UTPass/plugins/Pass/pass.cpp

#include <QUrl>
#include <QtCore/QStandardPaths>
#include <QtCore/QDir>

#include "pass.h"
#include "gpg.h"
#include "passkeymodel.h"



Pass::Pass():
    m_password_store (QStandardPaths::writableLocation(
                          QStandardPaths::AppDataLocation).append("/.password-store")),
    m_sem(std::unique_ptr<QSemaphore>(new QSemaphore(1))),
    m_show_filename(QString())
{}

void Pass::initialize(QObject *window)
{
    if (!window) {
        qFatal("window is invalid. Abording.");
    }

    this->m_gpg = std::unique_ptr<Gpg>(new Gpg(window));

    QObject::connect(this, &Pass::responsePassphraseDialogPropagate, this->m_gpg->passphrase_provider(),
                     &UTPassphraseProvider::handleResponse);
    QObject::connect(this->m_gpg.get(), &Gpg::importKeysFromFileResult, this, &Pass::importGPGKeyResult);
    QObject::connect(this->m_gpg.get(), &Gpg::getKeysResult, this, &Pass::getAllGPGKeysResult);
    QObject::connect(this->m_gpg.get(), &Gpg::deleteKeyResult, this, &Pass::deleteGPGKeyResult);
    QObject::connect(this->m_gpg.get(), &Gpg::decryptResult, this, &Pass::showResult);

    QDir dir(m_password_store);
    if (!dir.exists()) {
        dir.mkpath(".");
    }
    qInfo() << "Password Store is :" << m_password_store;
}

bool Pass::show(QUrl url)
{
    if (!this->m_sem->tryAcquire(1, 500)) {
        return false;
    }
    auto path = url.toLocalFile();
    qInfo() << "Pass show  " << path;
    QFileInfo file_info(path);
    this->m_show_filename = file_info.completeBaseName();
    return this->m_gpg->decryptFromFile(path);
}

void Pass::showResult(Error err, QString plain_text)
{
    qDebug() << "Pass show Result";
    if (err) {
        qInfo() << "Pass show Failed";
        emit showFailed(err.asString());

    }  else if (err.isCanceled()) {
        qInfo() << "Pass show Cancelled";
        emit showCancelled();
    } else {
        qInfo() << "Pass show Succeed";
        emit showSucceed(this->m_show_filename, plain_text);
    }
    this->m_show_filename = QString();
    this->m_sem->release(1);
}

bool Pass::deleteGPGKey(PassKeyModel* key)
{
    if (!this->m_sem->tryAcquire(1, 500)) {
        return false;
    }
    qInfo() << "Delete Key " << key->uid();
    return this->m_gpg->deleteKey(key->key());
}

void Pass::deleteGPGKeyResult(Error err)
{
    qDebug() << "Delete Ke yResult";
    if (err) {
        qInfo() << "Delete Key Failed";
        emit deleteGPGKeyFailed(err.asString());
    } else {
        qInfo() << "Delete Key Succeed";
        emit deleteGPGKeySucceed();
    }
    this->m_sem->release(1);
}

bool Pass::importGPGKey(QUrl url)
{
    if (!this->m_sem->tryAcquire(1, 500)) {
        return false;
    }
    qInfo() << "Import GPG Key from " << url;
    return this->m_gpg->importKeysFromFile(url.toLocalFile());
}

void Pass::importGPGKeyResult(Error err)
{
    qDebug() << "Import GPG Key Result";
    if (err) {
        qInfo() << "Delete Key Failed";
        emit importGPGKeyFailed(err.asString());
    } else {
        qInfo() << "Delete Key Succeed";
        emit importGPGKeySucceed();
    }
    this->m_sem->release(1);
}

bool Pass::getAllGPGKeys()
{
    if (!this->m_sem->tryAcquire(1, 500)) {
        return false;
    }
    qInfo() << "Get GPG keys";
    return this->m_gpg->getAllKeys();
}

void Pass::getAllGPGKeysResult(Error err,  std::vector<GpgME::Key> keys_info)
{
    qDebug() << "Get GPG keys Result";
    if (err) {
        qInfo() << "Get GPG Failed";
        emit getAllGPGKeysFailed(err.asString());
    } else {
        qInfo() << "Get GPG Succeed";
        emit getAllGPGKeysSucceed(QVariant::fromValue(PassKeyModel::keysToPassKey(keys_info)));
    }
    this->m_sem->release(1);
}

void Pass::responsePassphraseDialog(bool cancel, QString passphrase)
{
    qDebug() << "Propagate responsePassphraseDialog";
    emit responsePassphraseDialogPropagate(cancel, passphrase);
}