1 files changed, 106 insertions, 0 deletions
diff --git a/src/server/api/PubkeyUploader.cpp b/src/server/api/PubkeyUploader.cpp
new file mode 100644
index 00000000..c6101c65
--- /dev/null
+++ b/src/server/api/PubkeyUploader.cpp
@@ -0,0 +1,106 @@
+/**
+ * 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<[email protected]> starting on May 12, 2021.
+ *
+ */
+
+#include "server/api/PubkeyUploader.h"
+
+PubkeyUploader::PubkeyUploader(GpgME::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, tr("Error"),
+                              tr("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;
+        }
+    }
+}