/**
 * 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 "server/api/PubkeyUploader.h"

PubkeyUploader::PubkeyUploader(GpgFrontend::GpgContext *ctx,
                               const QVector<GpgKey> &keys)
    : BaseAPI(ComUtils::UploadPubkey), mCtx(ctx), mKeys(keys) {}

void PubkeyUploader::construct_json() {
  document.SetArray();
  QStringList keyIds;
  QCryptographicHash shaGen(QCryptographicHash::Sha256);

  auto &allocator = document.GetAllocator();

  QVector<QByteArray> keysData;
  for (const auto &key : mKeys) {
    QByteArray keyDataBuf;
    keyIds << key.id;

    // The use of multi-threading brings an improvement in UI smoothness
    gpgme_error_t error;
    auto thread = QThread::create(
        [&]() { error = mCtx->exportKeys(&keyIds, &keyDataBuf); });
    thread->start();
    while (thread->isRunning()) QApplication::processEvents();
    thread->deleteLater();
    keysData.push_back(keyDataBuf);
    keyIds.clear();
  }

  int index = 0;
  for (const auto &keyData : keysData) {
    rapidjson::Value publicKeyObj, pubkey, sha, signedFpr;

    shaGen.addData(keyData);
    auto shaStr = shaGen.result().toHex();
    shaGen.reset();

    auto signFprStr =
        ComUtils::getSignStringBase64(mCtx, mKeys[index].fpr, mKeys[index]);
    qDebug() << "signFprStr" << signFprStr;

    pubkey.SetString(keyData.data(), keyData.count(), allocator);
    sha.SetString(shaStr.data(), shaStr.count(), allocator);
    signedFpr.SetString(signFprStr.data(), signFprStr.count(), allocator);

    publicKeyObj.SetObject();
    publicKeyObj.AddMember("publicKey", pubkey, allocator);
    publicKeyObj.AddMember("sha", sha, allocator);
    publicKeyObj.AddMember("signedFpr", signedFpr, allocator);

    document.PushBack(publicKeyObj, allocator);
    index++;
  }
}

void PubkeyUploader::deal_reply() {
  const auto &utils = getUtils();

  /**
   * {
   *      "strings" : [
   *          "...",
   *          "..."
   *      ]
   * }
   */

  if (!utils.checkDataValue("strings")) {
    QMessageBox::critical(nullptr, _("Error"),
                          _("The communication content with the server does "
                            "not meet the requirements"));
  } else {
    auto &strings = utils.getDataValue("strings");
    qDebug() << "Pubkey Uploader" << strings.IsArray()
             << strings.GetArray().Size();
    if (strings.IsArray() && strings.GetArray().Size() == mKeys.size()) {
      good = true;
    } else {
      good = false;
    }
  }
}