1 files changed, 162 insertions, 0 deletions
diff --git a/src/gpg/model/GpgKey.h b/src/gpg/model/GpgKey.h
new file mode 100644
index 00000000..53c074e8
--- /dev/null
+++ b/src/gpg/model/GpgKey.h
@@ -0,0 +1,162 @@
+/**
+ * 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.
+ *
+ */
+
+#ifndef GPGFRONTEND_GPGKEY_H
+#define GPGFRONTEND_GPGKEY_H
+
+#include <boost/date_time.hpp>
+#include <boost/date_time/posix_time/conversion.hpp>
+
+#include "GpgSubKey.h"
+#include "GpgUID.h"
+
+namespace GpgFrontend {
+
+class GpgKey {
+ public:
+  [[nodiscard]] bool good() const { return _key_ref != nullptr; }
+
+  [[nodiscard]] std::string id() const { return _key_ref->subkeys->keyid; }
+
+  [[nodiscard]] std::string name() const { return _key_ref->uids->name; };
+
+  [[nodiscard]] std::string email() const { return _key_ref->uids->email; }
+
+  [[nodiscard]] std::string comment() const { return _key_ref->uids->comment; }
+
+  [[nodiscard]] std::string fpr() const { return _key_ref->fpr; }
+
+  [[nodiscard]] std::string protocol() const {
+    return gpgme_get_protocol_name(_key_ref->protocol);
+  }
+
+  [[nodiscard]] std::string owner_trust() const {
+    switch (_key_ref->owner_trust) {
+      case GPGME_VALIDITY_UNKNOWN:
+        return "Unknown";
+      case GPGME_VALIDITY_UNDEFINED:
+        return "Undefined";
+      case GPGME_VALIDITY_NEVER:
+        return "Never";
+      case GPGME_VALIDITY_MARGINAL:
+        return "Marginal";
+      case GPGME_VALIDITY_FULL:
+        return "FULL";
+      case GPGME_VALIDITY_ULTIMATE:
+        return "Ultimate";
+    }
+    return "Invalid";
+  }
+
+  [[nodiscard]] std::string pubkey_algo() const {
+    return gpgme_pubkey_algo_name(_key_ref->subkeys->pubkey_algo);
+  }
+
+  [[nodiscard]] boost::gregorian::date last_update() const {
+    return boost::posix_time::from_time_t(
+               static_cast<time_t>(_key_ref->last_update))
+        .date();
+  }
+
+  [[nodiscard]] boost::gregorian::date expires() const {
+    return boost::posix_time::from_time_t(_key_ref->subkeys->expires).date();
+  };
+
+  [[nodiscard]] boost::gregorian::date create_time() const {
+    return boost::posix_time::from_time_t(_key_ref->subkeys->timestamp).date();
+  };
+
+  [[nodiscard]] unsigned int length() const {
+    return _key_ref->subkeys->length;
+  }
+
+  [[nodiscard]] bool can_encrypt() const { return _key_ref->can_encrypt; }
+
+  [[nodiscard]] bool CanEncrActual() const;
+
+  [[nodiscard]] bool can_sign() const { return _key_ref->can_sign; }
+
+  [[nodiscard]] bool CanSignActual() const;
+
+  [[nodiscard]] bool can_certify() const { return _key_ref->can_certify; }
+
+  [[nodiscard]] bool CanCertActual() const;
+
+  [[nodiscard]] bool can_authenticate() const {
+    return _key_ref->can_authenticate;
+  }
+
+  [[nodiscard]] bool CanAuthActual() const;
+
+  [[nodiscard]] bool is_private_key() const { return _key_ref->secret; }
+
+  [[nodiscard]] bool expired() const { return _key_ref->expired; }
+
+  [[nodiscard]] bool revoked() const { return _key_ref->revoked; }
+
+  [[nodiscard]] bool disabled() const { return _key_ref->disabled; }
+
+  [[nodiscard]] bool has_master_key() const {
+    return _key_ref->subkeys->secret;
+  }
+
+  [[nodiscard]] std::unique_ptr<std::vector<GpgSubKey>> subKeys() const;
+
+  [[nodiscard]] std::unique_ptr<std::vector<GpgUID>> uids() const;
+
+  GpgKey() = default;
+
+  explicit GpgKey(gpgme_key_t&& key);
+
+  ~GpgKey() = default;
+
+  GpgKey(const gpgme_key_t& key) = delete;
+
+  GpgKey(GpgKey&& k) noexcept;
+
+  GpgKey& operator=(GpgKey&& k) noexcept;
+
+  GpgKey& operator=(const gpgme_key_t& key) = delete;
+
+  bool operator==(const GpgKey& o) const { return o.id() == this->id(); }
+
+  bool operator<=(const GpgKey& o) const { return this->id() < o.id(); }
+
+  explicit operator gpgme_key_t() const { return _key_ref.get(); }
+
+ private:
+  struct _key_ref_deletor {
+    void operator()(gpgme_key_t _key) {
+      if (_key != nullptr) gpgme_key_unref(_key);
+    }
+  };
+
+  using KeyRefHandler = std::unique_ptr<struct _gpgme_key, _key_ref_deletor>;
+
+  KeyRefHandler _key_ref = nullptr;
+};
+
+}  // namespace GpgFrontend
+
+#endif  // GPGFRONTEND_GPGKEY_H