feat: user can select shown columns at key table

1 files changed, 136 insertions, 0 deletions

diff --git a/src/ui/widgets/KeyTable.cpp b/src/ui/widgets/KeyTable.cpp
new file mode 100644
index 00000000..fc6d083f
--- /dev/null
+++ b/src/ui/widgets/KeyTable.cpp
@@ -0,0 +1,136 @@
+/**
+ * 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 "ui/widgets/KeyTable.h"
+
+#include "ui/UserInterfaceUtils.h"
+
+namespace GpgFrontend::UI {
+
+auto KeyTable::GetChecked() const -> KeyIdArgsListPtr {
+  auto ret = std::make_unique<KeyIdArgsList>();
+  for (size_t i = 0; i < GetRowCount(); i++) {
+    if (IsRowChecked(i)) ret->push_back(GetKeyIdByRow(i));
+  }
+  return ret;
+}
+
+KeyTable::KeyTable(QWidget* parent, QSharedPointer<GpgKeyTableModel> model,
+                   GpgKeyTableDisplayMode select_type,
+                   GpgKeyTableColumn column_filter,
+                   GpgKeyTableProxyModel::KeyFilter filter)
+    : QTableView(parent),
+      model_(std::move(model)),
+      proxy_model_(model_, select_type, column_filter, std::move(filter), this),
+      column_filter_(column_filter) {
+  setModel(&proxy_model_);
+
+  verticalHeader()->hide();
+  horizontalHeader()->setStretchLastSection(false);
+  setSizePolicy(QSizePolicy::Preferred, QSizePolicy::Preferred);
+
+  setShowGrid(false);
+  sortByColumn(2, Qt::AscendingOrder);
+  setSelectionBehavior(QAbstractItemView::SelectRows);
+  setSelectionMode(QAbstractItemView::SingleSelection);
+
+  // table items not editable
+  setEditTriggers(QAbstractItemView::NoEditTriggers);
+
+  setFocusPolicy(Qt::NoFocus);
+  setAlternatingRowColors(true);
+
+  connect(CommonUtils::GetInstance(), &CommonUtils::SignalFavoritesChanged,
+          &proxy_model_, &GpgKeyTableProxyModel::SignalFavoritesChanged);
+  connect(this, &KeyTable::SignalColumnTypeChange, this,
+          [=](GpgKeyTableColumn global_column_filter) {
+            emit(&proxy_model_)
+                ->SignalColumnTypeChange(column_filter_ & global_column_filter);
+          });
+}
+
+void KeyTable::SetFilterKeyword(const QString& keyword) {
+  proxy_model_.SetSearchKeywords(keyword);
+}
+
+void KeyTable::RefreshModel(QSharedPointer<GpgKeyTableModel> model) {
+  model_ = std::move(model);
+  proxy_model_.setSourceModel(model_.get());
+}
+
+auto KeyTable::IsRowChecked(int row) const -> bool {
+  auto index = model()->index(row, 0);
+  return index.data(Qt::CheckStateRole).toInt() == Qt::Checked;
+}
+
+auto KeyTable::GetRowCount() const -> int { return model()->rowCount(); }
+
+auto KeyTable::GetKeyIdByRow(int row) const -> QString {
+  if (row < 0 || row >= model()->rowCount()) return {};
+  auto origin_row = model()->index(row, 0).data().toInt();
+  return model_->GetKeyIDByRow(origin_row);
+}
+
+auto KeyTable::IsPrivateKeyByRow(int row) const -> bool {
+  if (row < 0 || row >= model()->rowCount()) return false;
+  auto origin_row = model()->index(row, 0).data().toInt();
+  return model_->IsPrivateKeyByRow(origin_row);
+}
+
+auto KeyTable::IsPublicKeyByRow(int row) const -> bool {
+  if (row < 0 || row >= model()->rowCount()) return false;
+  auto origin_row = model()->index(row, 0).data().toInt();
+  return !model_->IsPrivateKeyByRow(origin_row);
+}
+
+void KeyTable::SetRowChecked(int row) const {
+  if (row < 0 || row >= model()->rowCount()) return;
+  model()->setData(model()->index(row, 0), Qt::Checked, Qt::CheckStateRole);
+}
+
+void KeyTable::CheckAll() {
+  for (int row = 0; row < model()->rowCount(); ++row) {
+    auto index = model()->index(row, 0);
+    model()->setData(index, Qt::Checked, Qt::CheckStateRole);
+  }
+}
+
+void KeyTable::UncheckAll() {
+  for (int row = 0; row < model()->rowCount(); ++row) {
+    auto index = model()->index(row, 0);
+    model()->setData(index, Qt::Unchecked, Qt::CheckStateRole);
+  }
+}
+
+[[nodiscard]] auto KeyTable::GetRowSelected() const -> int {
+  auto selected_indexes = selectedIndexes();
+  if (selected_indexes.empty()) return -1;
+
+  return selected_indexes.first().row();
+}
+}  // namespace GpgFrontend::UI
\ No newline at end of file