refactor: improve the code structure of core

1 files changed, 133 insertions, 0 deletions

diff --git a/src/core/utils/GpgUtils.cpp b/src/core/utils/GpgUtils.cpp
new file mode 100644
index 00000000..a8e15e31
--- /dev/null
+++ b/src/core/utils/GpgUtils.cpp
@@ -0,0 +1,133 @@
+/**
+ * Copyright (C) 2021 Saturneric <[email protected]>
+ *
+ * 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.
+ *
+ * GpgFrontend 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 GpgFrontend. If not, see <https://www.gnu.org/licenses/>.
+ *
+ * The initial version of the source code is inherited from
+ * the gpg4usb project, which is under GPL-3.0-or-later.
+ *
+ * All the source code of GpgFrontend was modified and released by
+ * Saturneric <[email protected]> starting on May 12, 2021.
+ *
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ *
+ */
+
+#include "GpgUtils.h"
+
+#include "core/utils/IOUtils.h"
+
+namespace GpgFrontend {
+
+static inline void Ltrim(std::string& s) {
+  s.erase(s.begin(), std::find_if(s.begin(), s.end(), [](unsigned char ch) {
+            return !std::isspace(ch);
+          }));
+}
+
+static inline void Rtrim(std::string& s) {
+  s.erase(std::find_if(s.rbegin(), s.rend(),
+                       [](unsigned char ch) { return !std::isspace(ch); })
+              .base(),
+          s.end());
+}
+
+static inline auto Trim(std::string& s) -> std::string {
+  Ltrim(s);
+  Rtrim(s);
+  return s;
+}
+
+auto CheckGpgError(gpgme_error_t err) -> gpgme_error_t {
+  if (gpg_err_code(err) != GPG_ERR_NO_ERROR) {
+    SPDLOG_ERROR("[error: {}] source: {} description: {}", gpg_err_code(err),
+                 gpgme_strsource(err), gpgme_strerror(err));
+  }
+  return err;
+}
+
+auto CheckGpgError2ErrCode(gpgme_error_t err, gpgme_error_t predict)
+    -> gpg_err_code_t {
+  auto err_code = gpg_err_code(err);
+  if (err_code != gpg_err_code(predict)) {
+    if (err_code == GPG_ERR_NO_ERROR)
+      SPDLOG_WARN("[Warning {}] Source: {} description: {} predict: {}",
+                  gpg_err_code(err), gpgme_strsource(err), gpgme_strerror(err),
+                  gpgme_strerror(err));
+    else
+      SPDLOG_ERROR("[Error {}] Source: {} description: {} predict: {}",
+                   gpg_err_code(err), gpgme_strsource(err), gpgme_strerror(err),
+                   gpgme_strerror(err));
+  }
+  return err_code;
+}
+
+auto CheckGpgError(gpgme_error_t err, const std::string& comment)
+    -> gpgme_error_t {
+  if (gpg_err_code(err) != GPG_ERR_NO_ERROR) {
+    SPDLOG_WARN("[Error {}] Source: {} description: {} predict: {}",
+                gpg_err_code(err), gpgme_strsource(err), gpgme_strerror(err),
+                gpgme_strerror(err));
+  }
+  return err;
+}
+
+auto TextIsSigned(BypeArrayRef text) -> int {
+  using boost::algorithm::ends_with;
+  using boost::algorithm::starts_with;
+
+  auto trim_text = Trim(text);
+  if (starts_with(trim_text, PGP_SIGNED_BEGIN) &&
+      ends_with(trim_text, PGP_SIGNED_END)) {
+    return 2;
+  }
+  if (text.find(PGP_SIGNED_BEGIN) != std::string::npos &&
+      text.find(PGP_SIGNED_END) != std::string::npos) {
+    return 1;
+  }
+  return 0;
+}
+
+auto NewResult(gpgme_encrypt_result_t&& result) -> GpgEncrResult {
+  gpgme_result_ref(result);
+  return {result, ResultRefDeletor()};
+}
+
+auto NewResult(gpgme_decrypt_result_t&& result) -> GpgDecrResult {
+  gpgme_result_ref(result);
+  return {result, ResultRefDeletor()};
+}
+
+auto NewResult(gpgme_sign_result_t&& result) -> GpgSignResult {
+  gpgme_result_ref(result);
+  return {result, ResultRefDeletor()};
+}
+
+auto NewResult(gpgme_verify_result_t&& result) -> GpgVerifyResult {
+  gpgme_result_ref(result);
+  return {result, ResultRefDeletor()};
+}
+
+auto NewResult(gpgme_genkey_result_t&& result) -> GpgGenKeyResult {
+  gpgme_result_ref(result);
+  return {result, ResultRefDeletor()};
+}
+
+void ResultRefDeletor::operator()(void* _result) {
+  SPDLOG_TRACE("gpgme unref {}", _result);
+  if (_result != nullptr) gpgme_result_unref(_result);
+}
+}  // namespace GpgFrontend