/**
 * Copyright (C) 2021-2024 Saturneric <eric@bktus.com>
 *
 * 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 <eric@bktus.com> starting on May 12, 2021.
 *
 * SPDX-License-Identifier: GPL-3.0-or-later
 *
 */

#include "GpgKeyGenerateInfo.h"

#include <cassert>

#include "core/module/ModuleManager.h"
#include "core/utils/CommonUtils.h"
#include "core/utils/GpgUtils.h"

namespace GpgFrontend {

const KeyAlgo KeyGenerateInfo::kNoneAlgo = {
    "none", KeyGenerateInfo::tr("None"), "None", 0, 0, "0.0.0"};

const QContainer<KeyAlgo> KeyGenerateInfo::kPrimaryKeyAlgos = {
    kNoneAlgo,

    /**
     * Algorithm (DSA) as a government standard for digital signatures.
     * Originally, it supported key lengths between 512 and 1024 bits.
     * Recently, NIST has declared 512-bit keys obsolete:
     * now, DSA is available in 1024, 2048 and 3072-bit lengths.
     */
    {"rsa1024", "RSA", "RSA", 1024, kENCRYPT | kSIGN | kAUTH | kCERT, "2.2.0"},
    {"rsa2048", "RSA", "RSA", 2048, kENCRYPT | kSIGN | kAUTH | kCERT, "2.2.0"},
    {"rsa3072", "RSA", "RSA", 3072, kENCRYPT | kSIGN | kAUTH | kCERT, "2.2.0"},
    {"rsa4096", "RSA", "RSA", 4096, kENCRYPT | kSIGN | kAUTH | kCERT, "2.2.0"},
    {"dsa1024", "DSA", "DSA", 1024, kSIGN | kAUTH | kCERT, "2.2.0"},
    {"dsa2048", "DSA", "DSA", 2048, kSIGN | kAUTH | kCERT, "2.2.0"},
    {"dsa3072", "DSA", "DSA", 3072, kSIGN | kAUTH | kCERT, "2.2.0"},
    {"ed25519", "ED25519", "EdDSA", 256, kSIGN | kAUTH | kCERT, "2.2.0"},
    {"nistp256", "NIST", "ECDSA", 256, kSIGN | kAUTH | kCERT, "2.2.0"},
    {"nistp384", "NIST", "ECDSA", 384, kSIGN | kAUTH | kCERT, "2.2.0"},
    {"nistp521", "NIST", "ECDSA", 521, kSIGN | kAUTH | kCERT, "2.2.0"},
    {"brainpoolp256r1", "BrainPooL", "ECDSA", 256, kSIGN | kAUTH | kCERT,
     "2.3.0"},
    {"brainpoolp384r1", "BrainPooL", "ECDSA", 384, kSIGN | kAUTH | kCERT,
     "2.3.0"},
    {"brainpoolp512r1", "BrainPooL", "ECDSA", 512, kSIGN | kAUTH | kCERT,
     "2.3.0"},
    {"ed448", "ED448", "EdDSA", 448, kSIGN | kAUTH | kCERT, "2.3.0"},
    {"secp256k1", "ED448", "EdDSA", 256, kSIGN | kAUTH | kCERT, "2.3.0"},
};

const QContainer<KeyAlgo> KeyGenerateInfo::kSubKeyAlgos = {
    kNoneAlgo,

    {"rsa1024", "RSA", "RSA", 1024, kENCRYPT | kSIGN | kAUTH, "2.2.0"},
    {"rsa2048", "RSA", "RSA", 2048, kENCRYPT | kSIGN | kAUTH, "2.2.0"},
    {"rsa3072", "RSA", "RSA", 3072, kENCRYPT | kSIGN | kAUTH, "2.2.0"},
    {"rsa4096", "RSA", "RSA", 4096, kENCRYPT | kSIGN | kAUTH, "2.2.0"},
    {"dsa1024", "DSA", "DSA", 1024, kSIGN | kAUTH, "2.2.0"},
    {"dsa2048", "DSA", "DSA", 2048, kSIGN | kAUTH, "2.2.0"},
    {"dsa3072", "DSA", "DSA", 3072, kSIGN | kAUTH, "2.2.0"},
    {"ed25519", "ED25519", "EdDSA", 256, kSIGN | kAUTH, "2.2.0"},
    {"cv25519", "CV25519", "ECDH", 256, kENCRYPT, "2.2.0"},
    {"nistp256", "NIST", "ECDH", 256, kENCRYPT, "2.2.0"},
    {"nistp384", "NIST", "ECDH", 384, kENCRYPT, "2.2.0"},
    {"nistp521", "NIST", "ECDH", 521, kENCRYPT, "2.2.0"},
    {"brainpoolp256r1", "BrainPooL", "ECDH", 256, kENCRYPT, "2.3.0"},
    {"brainpoolp384r1", "BrainPooL", "ECDH", 384, kENCRYPT, "2.3.0"},
    {"brainpoolp512r1", "BrainPooL", "ECDH", 512, kENCRYPT, "2.3.0"},
    {"x448", "X448", "ECDH", 448, kENCRYPT, "2.3.0"},
    {"secp256k1", "SECP256K1", "ECDH", 256, kENCRYPT, "2.3.0"},
    /**
     * GnuPG supports the Elgamal asymmetric encryption algorithm in key lengths
     * ranging from 1024 to 4096 bits.
     */
    {"elg1024", "ELG-E", "ELG-E", 1024, kENCRYPT, "2.2.0"},
    {"elg2048", "ELG-E", "ELG-E", 2048, kENCRYPT, "2.2.0"},
    {"elg3072", "ELG-E", "ELG-E", 3072, kENCRYPT, "2.2.0"},
    {"elg4096", "ELG-E", "ELG-E", 4096, kENCRYPT, "2.2.0"},
};

auto KeyGenerateInfo::GetSupportedKeyAlgo(int channel) -> QContainer<KeyAlgo> {
  QContainer<KeyAlgo> algos;

  for (const auto &algo : kPrimaryKeyAlgos) {
    if (!CheckGpgVersion(channel, algo.SupportedVersion())) continue;

    algos.append(algo);
  }

  std::sort(algos.begin(), algos.end(), [](const KeyAlgo &a, const KeyAlgo &b) {
    return a.Name() < b.Name() && a.KeyLength() < b.KeyLength();
  });

  return algos;
}

auto KeyGenerateInfo::GetSupportedSubkeyAlgo(int channel)
    -> QContainer<KeyAlgo> {
  QContainer<KeyAlgo> algos;

  for (const auto &algo : kSubKeyAlgos) {
    if (!CheckGpgVersion(channel, algo.SupportedVersion())) continue;

    algos.append(algo);
  }

  std::sort(algos.begin(), algos.end(), [](const KeyAlgo &a, const KeyAlgo &b) {
    return a.Name() < b.Name() && a.KeyLength() < b.KeyLength();
  });

  return algos;
}

KeyGenerateInfo::KeyGenerateInfo(bool is_subkey)
    : subkey_(is_subkey),
      algo_(kNoneAlgo),
      expired_(QDateTime::currentDateTime().toLocalTime().addYears(2)) {}

auto KeyGenerateInfo::SearchPrimaryKeyAlgo(const QString &algo_id)
    -> std::tuple<bool, KeyAlgo> {
  auto it =
      std::find_if(kPrimaryKeyAlgos.cbegin(), kPrimaryKeyAlgos.cend(),
                   [=](const KeyAlgo &algo) { return algo.Id() == algo_id; });

  if (it != kPrimaryKeyAlgos.cend()) {
    return {true, *it};
  }

  return {false, KeyAlgo{}};
}

auto KeyGenerateInfo::SearchSubKeyAlgo(const QString &algo_id)
    -> std::tuple<bool, KeyAlgo> {
  auto it =
      std::find_if(kSubKeyAlgos.cbegin(), kSubKeyAlgos.cend(),
                   [=](const KeyAlgo &algo) { return algo.Id() == algo_id; });

  if (it != kSubKeyAlgos.cend()) {
    return {true, *it};
  }

  return {false, KeyAlgo{}};
}

void KeyGenerateInfo::SetAlgo(const KeyAlgo &algo) {
  // reset all options
  reset_options();

  this->SetAllowCert(algo.CanCert());
  this->allow_change_certification_ = false;

  SetAllowEncr(algo.CanEncrypt());
  allow_change_encryption_ = algo.CanEncrypt();

  SetAllowSign(algo.CanSign());
  allow_change_signing_ = algo.CanSign();

  SetAllowAuth(algo.CanAuth());
  allow_change_authentication_ = algo.CanAuth();

  this->algo_ = algo;
}

void KeyGenerateInfo::reset_options() {
  allow_change_encryption_ = true;
  SetAllowEncr(true);

  allow_change_certification_ = true;
  SetAllowCert(true);

  allow_change_signing_ = true;
  SetAllowSign(true);

  allow_change_authentication_ = true;
  SetAllowAuth(true);
}

void KeyGenerateInfo::SetExpireTime(const QDateTime &m_expired) {
  KeyGenerateInfo::expired_ = m_expired;
}

void KeyGenerateInfo::SetNonExpired(bool m_non_expired) {
  if (!m_non_expired) this->expired_ = QDateTime::fromSecsSinceEpoch(0);
  KeyGenerateInfo::non_expired_ = m_non_expired;
}

void KeyGenerateInfo::SetAllowEncr(bool m_allow_encryption) {
  if (allow_change_encryption_) {
    KeyGenerateInfo::allow_encryption_ = m_allow_encryption;
  }
}

void KeyGenerateInfo::SetAllowCert(bool m_allow_certification) {
  if (allow_change_certification_) {
    KeyGenerateInfo::allow_certification_ = m_allow_certification;
  }
}

/**
 * @brief
 *
 * @return true
 * @return false
 */
[[nodiscard]] auto KeyGenerateInfo::IsSubKey() const -> bool { return subkey_; }

/**
 * @brief Set the Is Sub Key object
 *
 * @param m_sub_key
 */
void KeyGenerateInfo::SetIsSubKey(bool m_sub_key) {
  KeyGenerateInfo::subkey_ = m_sub_key;
}

/**
 * @brief Get the Userid object
 *
 * @return QString
 */
[[nodiscard]] auto KeyGenerateInfo::GetUserid() const -> QString {
  return QString("%1(%2)<%3>").arg(name_).arg(comment_).arg(email_);
}

/**
 * @brief Set the Name object
 *
 * @param m_name
 */
void KeyGenerateInfo::SetName(const QString &m_name) { this->name_ = m_name; }

/**
 * @brief Set the Email object
 *
 * @param m_email
 */
void KeyGenerateInfo::SetEmail(const QString &m_email) {
  this->email_ = m_email;
}

/**
 * @brief Set the Comment object
 *
 * @param m_comment
 */
void KeyGenerateInfo::SetComment(const QString &m_comment) {
  this->comment_ = m_comment;
}

/**
 * @brief Get the Name object
 *
 * @return QString
 */
[[nodiscard]] auto KeyGenerateInfo::GetName() const -> QString { return name_; }

/**
 * @brief Get the Email object
 *
 * @return QString
 */
[[nodiscard]] auto KeyGenerateInfo::GetEmail() const -> QString {
  return email_;
}

/**
 * @brief Get the Comment object
 *
 * @return QString
 */
[[nodiscard]] auto KeyGenerateInfo::GetComment() const -> QString {
  return comment_;
}

/**
 * @brief Get the Algo object
 *
 * @return const QString&
 */
[[nodiscard]] auto KeyGenerateInfo::GetAlgo() const -> const KeyAlgo & {
  return algo_;
}

/**
 * @brief Get the Key Size object
 *
 * @return int
 */
[[nodiscard]] auto KeyGenerateInfo::GetKeyLength() const -> int {
  return algo_.KeyLength();
}

/**
 * @brief Get the Expired object
 *
 * @return const QDateTime&
 */
[[nodiscard]] auto KeyGenerateInfo::GetExpireTime() const -> const QDateTime & {
  return expired_;
}

/**
 * @brief
 *
 * @return true
 * @return false
 */
[[nodiscard]] auto KeyGenerateInfo::IsNonExpired() const -> bool {
  return non_expired_;
}

/**
 * @brief
 *
 * @return true
 * @return false
 */
[[nodiscard]] auto KeyGenerateInfo::IsNoPassPhrase() const -> bool {
  return this->no_passphrase_;
}

/**
 * @brief Set the Non Pass Phrase object
 *
 * @param m_non_pass_phrase
 */
void KeyGenerateInfo::SetNonPassPhrase(bool m_non_pass_phrase) {
  KeyGenerateInfo::no_passphrase_ = m_non_pass_phrase;
}

/**
 * @brief
 *
 * @return true
 * @return false
 */
[[nodiscard]] auto KeyGenerateInfo::IsAllowSign() const -> bool {
  return allow_signing_;
}

/**
 * @brief
 *
 * @return true
 * @return false
 */
[[nodiscard]] auto KeyGenerateInfo::IsAllowNoPassPhrase() const -> bool {
  return allow_no_pass_phrase_;
}

/**
 * @brief Set the Allow Signing object
 *
 * @param m_allow_signing
 */
void KeyGenerateInfo::SetAllowSign(bool m_allow_signing) {
  if (allow_change_signing_) KeyGenerateInfo::allow_signing_ = m_allow_signing;
}

/**
 * @brief
 *
 * @return true
 * @return false
 */
[[nodiscard]] auto KeyGenerateInfo::IsAllowEncr() const -> bool {
  return allow_encryption_;
}

/**
 * @brief
 *
 * @return true
 * @return false
 */
[[nodiscard]] auto KeyGenerateInfo::IsAllowCert() const -> bool {
  return allow_certification_;
}

/**
 * @brief
 *
 * @return true
 * @return false
 */
[[nodiscard]] auto KeyGenerateInfo::IsAllowAuth() const -> bool {
  return allow_authentication_;
}

/**
 * @brief Set the Allow Authentication object
 *
 * @param m_allow_authentication
 */
void KeyGenerateInfo::SetAllowAuth(bool m_allow_authentication) {
  if (allow_change_authentication_) {
    KeyGenerateInfo::allow_authentication_ = m_allow_authentication;
  }
}

/**
 * @brief
 *
 * @return true
 * @return false
 */
[[nodiscard]] auto KeyGenerateInfo::IsAllowModifySign() const -> bool {
  return allow_change_signing_;
}

/**
 * @brief
 *
 * @return true
 * @return false
 */
[[nodiscard]] auto KeyGenerateInfo::IsAllowModifyEncr() const -> bool {
  return allow_change_encryption_;
}

/**
 * @brief
 *
 * @return true
 * @return false
 */
[[nodiscard]] auto KeyGenerateInfo::IsAllowModifyCert() const -> bool {
  return allow_change_certification_;
}

/**
 * @brief
 *
 * @return true
 * @return false
 */
[[nodiscard]] auto KeyGenerateInfo::IsAllowModifyAuth() const -> bool {
  return allow_change_authentication_;
}

KeyAlgo::KeyAlgo(QString id, QString name, QString type, int length, int opera,
                 QString supported_version)
    : id_(std::move(id)),
      name_(std::move(name)),
      type_(std::move(type)),
      length_(length),
      supported_version_(std::move(supported_version)) {
  encrypt_ = ((opera & kENCRYPT) != 0);
  sign_ = ((opera & kSIGN) != 0);
  auth_ = ((opera & kAUTH) != 0);
  cert_ = ((opera & kCERT) != 0);
};

auto KeyAlgo::Id() const -> QString { return id_; }

auto KeyAlgo::Name() const -> QString { return name_; }

auto KeyAlgo::KeyLength() const -> int { return length_; }

auto KeyAlgo::Type() const -> QString { return type_; }

auto KeyAlgo::CanEncrypt() const -> bool { return encrypt_; }

auto KeyAlgo::CanSign() const -> bool { return sign_; }

auto KeyAlgo::CanAuth() const -> bool { return auth_; }

auto KeyAlgo::CanCert() const -> bool { return cert_; }

auto KeyAlgo::SupportedVersion() const -> QString { return supported_version_; }

auto KeyAlgo::operator==(const KeyAlgo &o) const -> bool {
  return this->id_ == o.id_;
}
}  // namespace GpgFrontend