UI Framework Modified.

1 files changed, 199 insertions, 173 deletions

diff --git a/src/ui/main_window/MainWindowServerSlotFunction.cpp b/src/ui/main_window/MainWindowServerSlotFunction.cpp
index dbbf178a..6266efda 100644
--- a/src/ui/main_window/MainWindowServerSlotFunction.cpp
+++ b/src/ui/main_window/MainWindowServerSlotFunction.cpp
@@ -23,216 +23,242 @@
  */
 
 #include "MainWindow.h"
-#include "server/ComUtils.h"
-#include "ui/ShowCopyDialog.h"
 
+#ifdef SERVER_SUPPORT
 #include "rapidjson/document.h"
 #include "rapidjson/prettywriter.h"
+#include "server/ComUtils.h"
+#endif
+#include "ui/ShowCopyDialog.h"
+
+namespace GpgFrontend::UI {
+
+#ifdef SERVER_SUPPORT
 
 /**
  * get full size crypt text from server using short crypto text
  * @param shortenCryptoText short crypto text([GpgFrontend_ShortCrypto]://)
  * @return
  */
-QString MainWindow::getCryptText(const QString &shortenCryptoText) {
-
-    QString ownKeyId = settings.value("general/ownKeyId").toString();
-
-    GpgKey key = mCtx->getKeyRefById(ownKeyId);
-    if (!key.good) {
-        QMessageBox::critical(this, tr("Invalid Own Key"),
-                              tr("Own Key can not be use to do any operation. "
-                                 "Please go to the setting interface to select an OwnKey and get a ServiceToken."));
-        return {};
-    }
-
-    auto utils = new ComUtils(this);
+QString MainWindow::getCryptText(const QString& shortenCryptoText) {
+  QString ownKeyId = settings.value("general/ownKeyId").toString();
+
+  GpgKey key = mCtx->getKeyRefById(ownKeyId);
+  if (!key.good) {
+    QMessageBox::critical(this, tr("Invalid Own Key"),
+                          tr("Own Key can not be use to do any operation. "
+                             "Please go to the setting interface to select an "
+                             "OwnKey and get a ServiceToken."));
+    return {};
+  }
 
-    QString serviceToken = settings.value("general/serviceToken").toString();
-    if (serviceToken.isEmpty() || !utils->checkServiceTokenFormat(serviceToken)) {
-        QMessageBox::critical(this, tr("Error"),
-                              tr("Please obtain a Service Token from the server in the settings."));
-        return {};
-    }
+  auto utils = new ComUtils(this);
 
-    QUrl reqUrl(utils->getUrl(ComUtils::GetFullCryptText));
-    QNetworkRequest request(reqUrl);
-    request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
+  QString serviceToken = settings.value("general/serviceToken").toString();
+  if (serviceToken.isEmpty() || !utils->checkServiceTokenFormat(serviceToken)) {
+    QMessageBox::critical(
+        this, tr("Error"),
+        tr("Please obtain a Service Token from the server in the settings."));
+    return {};
+  }
 
-    // Sign Shorten Text
-    auto outSignTextBase64 = ComUtils::getSignStringBase64(mCtx, shortenCryptoText, key);
+  QUrl reqUrl(utils->getUrl(ComUtils::GetFullCryptText));
+  QNetworkRequest request(reqUrl);
+  request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
 
-    rapidjson::Document doc;
-    doc.SetObject();
+  // Sign Shorten Text
+  auto outSignTextBase64 =
+      ComUtils::getSignStringBase64(mCtx, shortenCryptoText, key);
 
-    rapidjson::Value s, t;
+  rapidjson::Document doc;
+  doc.SetObject();
 
-    // Signature
-    s.SetString(outSignTextBase64.constData(), outSignTextBase64.count());
-    // Service Token
-    const auto t_byte_array = serviceToken.toUtf8();
-    t.SetString(t_byte_array.constData(), t_byte_array.count());
+  rapidjson::Value s, t;
 
-    rapidjson::Document::AllocatorType &allocator = doc.GetAllocator();
+  // Signature
+  s.SetString(outSignTextBase64.constData(), outSignTextBase64.count());
+  // Service Token
+  const auto t_byte_array = serviceToken.toUtf8();
+  t.SetString(t_byte_array.constData(), t_byte_array.count());
 
-    doc.AddMember("signature", s, allocator);
-    doc.AddMember("serviceToken", t, allocator);
+  rapidjson::Document::AllocatorType& allocator = doc.GetAllocator();
 
-    rapidjson::StringBuffer sb;
-    rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(sb);
-    doc.Accept(writer);
+  doc.AddMember("signature", s, allocator);
+  doc.AddMember("serviceToken", t, allocator);
 
-    QByteArray postData(sb.GetString());
-    qDebug() << "postData" << QString::fromUtf8(postData);
+  rapidjson::StringBuffer sb;
+  rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(sb);
+  doc.Accept(writer);
 
-    QNetworkReply *reply = utils->getNetworkManager().post(request, postData);
+  QByteArray postData(sb.GetString());
+  qDebug() << "postData" << QString::fromUtf8(postData);
 
-    auto dialog = new WaitingDialog(tr("Getting Cpt From Server"), this);
-    dialog->show();
+  QNetworkReply* reply = utils->getNetworkManager().post(request, postData);
 
-    while (reply->isRunning()) QApplication::processEvents();
+  auto dialog = new WaitingDialog(tr("Getting Cpt From Server"), this);
+  dialog->show();
 
-    dialog->close();
+  while (reply->isRunning())
+    QApplication::processEvents();
 
-    QByteArray replyData = reply->readAll().constData();
-    if (utils->checkServerReply(replyData)) {
-        /**
-         * {
-         *      "cryptoText" : ...
-         *      "sha": ...
-         *      "serviceToken": ...
-         *      "date": ...
-         * }
-         */
+  dialog->close();
 
-        if (!utils->checkDataValueStr("cryptoText")
-            || !utils->checkDataValueStr("sha")
-            || !utils->checkDataValueStr("serviceToken")) {
-            QMessageBox::critical(this, tr("Error"),
-                                  tr("The communication content with the server does not meet the requirements"));
-            return {};
-        }
+  QByteArray replyData = reply->readAll().constData();
+  if (utils->checkServerReply(replyData)) {
+    /**
+     * {
+     *      "cryptoText" : ...
+     *      "sha": ...
+     *      "serviceToken": ...
+     *      "date": ...
+     * }
+     */
 
-        auto cryptoText = utils->getDataValueStr("cryptoText");
-        auto sha = utils->getDataValueStr("sha");
-        auto serviceTokenFromServer = utils->getDataValueStr("serviceToken");
+    if (!utils->checkDataValueStr("cryptoText") ||
+        !utils->checkDataValueStr("sha") ||
+        !utils->checkDataValueStr("serviceToken")) {
+      QMessageBox::critical(this, tr("Error"),
+                            tr("The communication content with the server does "
+                               "not meet the requirements"));
+      return {};
+    }
 
-        QCryptographicHash sha_generator(QCryptographicHash::Sha256);
-        sha_generator.addData(cryptoText.toUtf8());
+    auto cryptoText = utils->getDataValueStr("cryptoText");
+    auto sha = utils->getDataValueStr("sha");
+    auto serviceTokenFromServer = utils->getDataValueStr("serviceToken");
 
-        if (sha_generator.result().toHex() == sha && serviceToken == serviceTokenFromServer) {
-            return cryptoText;
-        } else QMessageBox::critical(this, tr("Error"), tr("Invalid short ciphertext"));
+    QCryptographicHash sha_generator(QCryptographicHash::Sha256);
+    sha_generator.addData(cryptoText.toUtf8());
 
-        return {};
-    }
+    if (sha_generator.result().toHex() == sha &&
+        serviceToken == serviceTokenFromServer) {
+      return cryptoText;
+    } else
+      QMessageBox::critical(this, tr("Error"), tr("Invalid short ciphertext"));
 
     return {};
-}
+  }
 
-void MainWindow::shortenCryptText() {
+  return {};
+}
 
-    // gather information
-    QString serviceToken = settings.value("general/serviceToken").toString();
-    QString ownKeyId = settings.value("general/ownKeyId").toString();
-    QByteArray cryptoText = edit->curTextPage()->toPlainText().toUtf8();
+#endif
 
-    auto utils = new ComUtils(this);
+#ifdef SERVER_SUPPORT
 
-    if (serviceToken.isEmpty() || !utils->checkServiceTokenFormat(serviceToken)) {
-        QMessageBox::critical(this, tr("Invalid Service Token"),
-                              tr("Please go to the setting interface to get a ServiceToken."));
-        return;
+void MainWindow::shortenCryptText() {
+  // gather information
+  QString serviceToken = settings.value("general/serviceToken").toString();
+  QString ownKeyId = settings.value("general/ownKeyId").toString();
+  QByteArray cryptoText = edit->curTextPage()->toPlainText().toUtf8();
+
+  auto utils = new ComUtils(this);
+
+  if (serviceToken.isEmpty() || !utils->checkServiceTokenFormat(serviceToken)) {
+    QMessageBox::critical(
+        this, tr("Invalid Service Token"),
+        tr("Please go to the setting interface to get a ServiceToken."));
+    return;
+  }
+
+  QUrl reqUrl(utils->getUrl(ComUtils::ShortenCryptText));
+  QNetworkRequest request(reqUrl);
+  request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
+
+  GpgKey key = mCtx->getKeyRefById(ownKeyId);
+  if (!key.good) {
+    QMessageBox::critical(this, tr("Invalid Own Key"),
+                          tr("Own Key can not be use to do any operation."));
+    return;
+  }
+
+  QCryptographicHash ch(QCryptographicHash::Md5);
+  ch.addData(cryptoText);
+  QString md5 = ch.result().toHex();
+
+  qDebug() << "md5" << md5;
+
+  QByteArray signText = QString("[%1][%2]").arg(serviceToken, md5).toUtf8();
+
+  QCryptographicHash sha(QCryptographicHash::Sha256);
+  sha.addData(signText);
+  QString shaText = sha.result().toHex();
+
+  qDebug() << "shaText" << shaText;
+
+  QByteArray outSignTextBase64 =
+      ComUtils::getSignStringBase64(mCtx, signText, key);
+
+  rapidjson::Value c, s, m, t;
+
+  rapidjson::Document doc;
+  doc.SetObject();
+
+  c.SetString(cryptoText.constData(), cryptoText.count());
+  auto m_byte_array = shaText.toUtf8();
+  m.SetString(m_byte_array.constData(), m_byte_array.count());
+  s.SetString(outSignTextBase64.constData(), outSignTextBase64.count());
+  auto t_byte_array = serviceToken.toUtf8();
+  t.SetString(t_byte_array.constData(), t_byte_array.count());
+
+  rapidjson::Document::AllocatorType& allocator = doc.GetAllocator();
+
+  doc.AddMember("cryptoText", c, allocator);
+  doc.AddMember("sha", m, allocator);
+  doc.AddMember("sign", s, allocator);
+  doc.AddMember("serviceToken", t, allocator);
+
+  rapidjson::StringBuffer sb;
+  rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(sb);
+  doc.Accept(writer);
+
+  QByteArray postData(sb.GetString());
+  qDebug() << "postData" << QString::fromUtf8(postData);
+
+  QNetworkReply* reply = networkAccessManager->post(request, postData);
+
+  auto* dialog = new WaitingDialog(tr("Getting Scpt From Server"), this);
+  dialog->show();
+  while (reply->isRunning())
+    QApplication::processEvents();
+  dialog->close();
+
+  if (utils->checkServerReply(reply->readAll().constData())) {
+    /**
+     * {
+     *      "shortenText" : ...
+     *      "md5": ...
+     * }
+     */
+
+    if (!utils->checkDataValueStr("shortenText") ||
+        !utils->checkDataValueStr("md5")) {
+      QMessageBox::critical(this, tr("Error"),
+                            tr("The communication content with the server does "
+                               "not meet the requirements"));
+      return;
     }
 
-    QUrl reqUrl(utils->getUrl(ComUtils::ShortenCryptText));
-    QNetworkRequest request(reqUrl);
-    request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
-
-    GpgKey key = mCtx->getKeyRefById(ownKeyId);
-    if (!key.good) {
-        QMessageBox::critical(this, tr("Invalid Own Key"), tr("Own Key can not be use to do any operation."));
-        return;
+    QString shortenText = utils->getDataValueStr("shortenText");
+
+    QCryptographicHash md5_generator(QCryptographicHash::Md5);
+    md5_generator.addData(shortenText.toUtf8());
+    if (md5_generator.result().toHex() == utils->getDataValueStr("md5")) {
+      auto* dialog =
+          new ShowCopyDialog(shortenText,
+                             tr("Notice: Use Decrypt & Verify operation to "
+                                "decrypt this short crypto text."),
+                             this);
+      dialog->show();
+    } else {
+      QMessageBox::critical(
+          this, tr("Error"),
+          tr("There is a problem with the communication with the server"));
+      return;
     }
-
-    QCryptographicHash ch(QCryptographicHash::Md5);
-    ch.addData(cryptoText);
-    QString md5 = ch.result().toHex();
-
-    qDebug() << "md5" << md5;
-
-    QByteArray signText = QString("[%1][%2]").arg(serviceToken, md5).toUtf8();
-
-    QCryptographicHash sha(QCryptographicHash::Sha256);
-    sha.addData(signText);
-    QString shaText = sha.result().toHex();
-
-    qDebug() << "shaText" << shaText;
-
-    QByteArray outSignTextBase64 = ComUtils::getSignStringBase64(mCtx, signText, key);
-
-    rapidjson::Value c, s, m, t;
-
-    rapidjson::Document doc;
-    doc.SetObject();
-
-    c.SetString(cryptoText.constData(), cryptoText.count());
-    auto m_byte_array = shaText.toUtf8();
-    m.SetString(m_byte_array.constData(), m_byte_array.count());
-    s.SetString(outSignTextBase64.constData(), outSignTextBase64.count());
-    auto t_byte_array = serviceToken.toUtf8();
-    t.SetString(t_byte_array.constData(), t_byte_array.count());
-
-    rapidjson::Document::AllocatorType &allocator = doc.GetAllocator();
-
-    doc.AddMember("cryptoText", c, allocator);
-    doc.AddMember("sha", m, allocator);
-    doc.AddMember("sign", s, allocator);
-    doc.AddMember("serviceToken", t, allocator);
-
-    rapidjson::StringBuffer sb;
-    rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(sb);
-    doc.Accept(writer);
-
-    QByteArray postData(sb.GetString());
-    qDebug() << "postData" << QString::fromUtf8(postData);
-
-    QNetworkReply *reply = networkAccessManager->post(request, postData);
-
-    auto *dialog = new WaitingDialog(tr("Getting Scpt From Server"), this);
-    dialog->show();
-    while (reply->isRunning()) QApplication::processEvents();
-    dialog->close();
-
-    if (utils->checkServerReply(reply->readAll().constData())) {
-
-        /**
-         * {
-         *      "shortenText" : ...
-         *      "md5": ...
-         * }
-         */
-
-        if (!utils->checkDataValueStr("shortenText") || !utils->checkDataValueStr("md5")) {
-            QMessageBox::critical(this, tr("Error"),
-                                  tr("The communication content with the server does not meet the requirements"));
-            return;
-        }
-
-        QString shortenText = utils->getDataValueStr("shortenText");
-
-        QCryptographicHash md5_generator(QCryptographicHash::Md5);
-        md5_generator.addData(shortenText.toUtf8());
-        if (md5_generator.result().toHex() == utils->getDataValueStr("md5")) {
-            auto *dialog = new ShowCopyDialog(shortenText,
-                                              tr("Notice: Use Decrypt & Verify operation to decrypt this short crypto text."),
-                                              this);
-            dialog->show();
-        } else {
-            QMessageBox::critical(this, tr("Error"), tr("There is a problem with the communication with the server"));
-            return;
-        }
-    }
-
+  }
 }
 
+#endif
+
+}  // namespace GpgFrontend::UI