/**
 * This file is part of GPGFrontend.
 *
 * GPGFrontend 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 3 of the License, or
 * (at your option) any later version.
 *
 * Foobar 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 Foobar.  If not, see <https://www.gnu.org/licenses/>.
 *
 * The initial version of the source code is inherited from gpg4usb-team.
 * Their source code version also complies with GNU General Public License.
 *
 * The source code version of this software was modified and released
 * by Saturneric<eric@bktus.com> starting on May 12, 2021.
 *
 */


#include "gpg/function/GpgKeyManager.h"
#include "gpg/function/GpgKeyGetter.h"
#include "gpg/function/BasicOperator.h"

bool GpgFrontend::GpgKeyManager::signKey(const GpgFrontend::GpgKey &target, GpgFrontend::KeyArgsList &keys,
                                         const QString &uid, std::unique_ptr<QDateTime> &expires) {
    BasicOperator::getInstance().setSigners(keys);

    unsigned int flags = 0;

    unsigned int expires_time_t = 0;
    if (expires == nullptr) flags |= GPGME_KEYSIGN_NOEXPIRE;
    else expires_time_t = QDateTime::currentDateTime().secsTo(*expires);

    auto err =
            check_gpg_error(
                    gpgme_op_keysign(ctx, gpgme_key_t(target), uid.toUtf8().constData(), expires_time_t, flags));

    if (gpg_err_code(err) == GPG_ERR_NO_ERROR) return true;
    else return false;
}

bool
GpgFrontend::GpgKeyManager::revSign(const GpgFrontend::GpgKey &key, const GpgFrontend::GpgKeySignature &signature) {

    auto &key_getter = GpgKeyGetter::getInstance();
    auto signing_key = key_getter.getKey(signature.keyid());

    auto err = check_gpg_error(gpgme_op_revsig(ctx, gpgme_key_t(key), gpgme_key_t(signing_key),
                                               signature.uid().data(), 0));
    if (gpg_err_code(err) == GPG_ERR_NO_ERROR) return true;
    else return false;

}

bool GpgFrontend::GpgKeyManager::setExpire(const GpgFrontend::GpgKey &key, std::unique_ptr<GpgSubKey> &subkey,
                                           std::unique_ptr<QDateTime> &expires) {
    unsigned long expires_time = 0;
    if (expires != nullptr) {
        qDebug() << "Expire Datetime" << expires->toString();
        expires_time = QDateTime::currentDateTime().secsTo(*expires);
    }

    const char *sub_fprs = nullptr;

    if (subkey != nullptr) sub_fprs = subkey->fpr().toUtf8().constData();

    auto err = check_gpg_error(gpgme_op_setexpire(ctx, gpgme_key_t(key), expires_time, sub_fprs, 0));
    if (gpg_err_code(err) == GPG_ERR_NO_ERROR) return true;
    else return false;
}