No OneTemporary
Actions

Size

31 KB

Subscribers

None

View Options

	diff --git a/src/widgets/mainwindow.cpp b/src/widgets/mainwindow.cpp
	index 115a89f..0c56ac3 100644
	--- a/src/widgets/mainwindow.cpp
	+++ b/src/widgets/mainwindow.cpp
	@@ -1,822 +1,812 @@
	/*
	SPDX-FileCopyrightText: 2014-2023 Anne Jan Brouwer <brouwer@annejan.com>
	SPDX-FileCopyrightText: 2016-2017 tezeb <tezeb+github@outoftheblue.pl>
	SPDX-FileCopyrightText: 2018 Lukas Vogel <lukedirtwalker@gmail.com>
	SPDX-FileCopyrightText: 2018 Claudio Maradonna <penguyman@stronzi.org>
	SPDX-FileCopyrightText: 2019 Maciej S. Szmigiero <mail@maciej.szmigiero.name>
	SPDX-FileCopyrightText: 2023 g10 Code GmbH
	SPDX-FileContributor: Sune Stolborg Vuorela <sune@vuorela.dk>

	SPDX-License-Identifier: GPL-3.0-or-later
	*/
	#include "mainwindow.h"

	#include <gpgpass_version.h>

	#include "clipboardhelper.h"
	#include "conf/configuredialog.h"
	#include "job/directoryreencryptjob.h"
	#include "job/filedecryptjob.h"
	#include "job/fileencryptjob.h"
	#include "models/addfileinfoproxy.h"
	#include "passentry.h"
	#include "rootfoldersmanager.h"
	#include "setupwidget.h"
	#include "ui_mainwindow.h"
	#include "usersdialog.h"
	#include "util.h"
	#include "widgets/passwordeditorwidget.h"
	#include "widgets/passwordviewerwidget.h"

	#include <KLocalizedString>
	#include <KMessageBox>
	#include <KMessageWidget>
	#include <KStandardGuiItem>
	#include <QCloseEvent>
	#include <QComboBox>
	#include <QDebug>
	#include <QDirIterator>
	#include <QFileInfo>
	#include <QInputDialog>
	#include <QMenu>
	#include <QMessageBox>
	#include <QShortcut>

	enum class StackLayer {
	WelcomePage = 0,
	PasswordViewer = 1,
	ConfigurationPage = 2,
	PasswordEditor = 3,
	};

	static QString directoryName(const QString &dirOrFile)
	{
	QFileInfo fi{dirOrFile};
	if (fi.isDir()) {
	return fi.absoluteFilePath();
	} else {
	return fi.absolutePath();
	}
	}

	/**
	* @brief MainWindow::MainWindow handles all of the main functionality and also
	* the main window.
	* @param searchText for searching from cli
	* @param parent pointer
	*/
	MainWindow::MainWindow(QWidget *parent)
	: QMainWindow(parent)
	, ui(new Ui::MainWindow)
	, m_clipboardHelper(new ClipboardHelper(this))
	, m_passwordViewer(new PasswordViewerWidget(m_clipboardHelper, this))
	, m_passwordEditor(new PasswordEditorWidget(m_clipboardHelper, this))
	, m_rootFoldersManager(new RootFoldersManager(this))
	{
	#ifdef __APPLE__
	// extra treatment for mac os
	// see http://doc.qt.io/qt-5/qkeysequence.html#qt_set_sequence_auto_mnemonic
	qt_set_sequence_auto_mnemonic(true);
	#endif
	ui->setupUi(this);

	ui->stackedWidget->addWidget(m_passwordViewer);
	auto setupWidget = new SetupWidget(this);
	ui->stackedWidget->addWidget(setupWidget);
	ui->stackedWidget->addWidget(m_passwordEditor);
	connect(setupWidget, &SetupWidget::setupComplete, this, &MainWindow::slotSetupFinished);

	ui->actionEdit->setCheckable(true);
	ui->separator->setFixedHeight(1);
	ui->separator->setFrameStyle(QFrame::HLine);
	ui->lineEditWrapper->setContentsMargins(style()->pixelMetric(QStyle::PM_LayoutLeftMargin),
	style()->pixelMetric(QStyle::PM_LayoutTopMargin),
	style()->pixelMetric(QStyle::PM_LayoutRightMargin),
	style()->pixelMetric(QStyle::PM_LayoutBottomMargin));

	m_errorMessage = new KMessageWidget();
	m_errorMessage->setMessageType(KMessageWidget::Error);
	#if QT_VERSION > QT_VERSION_CHECK(6, 0, 0)
	m_errorMessage->setPosition(KMessageWidget::Position::Header);
	#endif
	m_errorMessage->hide();
	ui->messagesArea->addWidget(m_errorMessage);

	m_storeModel.setRootFoldersManager(m_rootFoldersManager);

	connect(&m_storeModel, &StoreModel::errorOccurred, this, [this](auto str) {
	m_errorMessage->setText(str);
	m_errorMessage->animatedShow();
	setUiElementsEnabled(true);
	});
	}

	void MainWindow::setVisible(bool visible)
	{
	// This originated in the ctor, but we want this to happen after the first start wizard has been run
	// so for now, moved to show() on first call
	if (visible && firstShow) {
	firstShow = false;
	// register shortcut ctrl/cmd + Q to close the main window
	new QShortcut(QKeySequence(Qt::CTRL \| Qt::Key_Q), this, SLOT(close()));

	ui->treeView->setModel(&m_storeModel);
	ui->treeView->setHeaderHidden(true);
	ui->treeView->setIndentation(15);
	ui->treeView->setHorizontalScrollBarPolicy(Qt::ScrollBarAsNeeded);
	ui->treeView->setContextMenuPolicy(Qt::CustomContextMenu);
	ui->treeView->header()->setSectionResizeMode(0, QHeaderView::Stretch);
	ui->treeView->sortByColumn(0, Qt::AscendingOrder);
	connect(ui->treeView, &QWidget::customContextMenuRequested, this, &MainWindow::showContextMenu);
	connect(ui->treeView, &DeselectableTreeView::emptyClicked, this, &MainWindow::deselect);

	searchTimer.setInterval(350);
	searchTimer.setSingleShot(true);

	connect(&searchTimer, &QTimer::timeout, this, &MainWindow::onTimeoutSearch);

	initToolBarButtons();

	ui->lineEdit->setClearButtonEnabled(true);

	setUiElementsEnabled(true);

	QTimer::singleShot(10, this, SLOT(focusInput()));
	verifyInitialized();
	}

	QMainWindow::setVisible(visible);
	}

	MainWindow::~MainWindow() = default;

	/**
	* @brief MainWindow::focusInput selects any text (if applicable) in the search
	* box and sets focus to it. Allows for easy searching, called at application
	* start and when receiving empty message in MainWindow::messageAvailable when
	* compiled with SINGLE_APP=1 (default).
	*/
	void MainWindow::focusInput()
	{
	ui->lineEdit->selectAll();
	ui->lineEdit->setFocus();
	}

	/**
	* @brief MainWindow::changeEvent sets focus to the search box
	* @param event
	*/
	void MainWindow::changeEvent(QEvent *event)
	{
	QWidget::changeEvent(event);
	if (event->type() == QEvent::ActivationChange) {
	if (isActiveWindow()) {
	focusInput();
	}
	}
	}

	/**
	* @brief MainWindow::initToolBarButtons init main ToolBar and connect actions
	*/
	void MainWindow::initToolBarButtons()
	{
	connect(ui->actionAddPassword, &QAction::triggered, this, &MainWindow::addPassword);
	connect(ui->actionAddFolder, &QAction::triggered, this, &MainWindow::addFolder);
	connect(ui->actionEdit, &QAction::toggled, this, &MainWindow::onEdit);
	connect(ui->actionDelete, &QAction::triggered, this, &MainWindow::onDelete);
	connect(ui->actionUsers, &QAction::triggered, this, &MainWindow::onUsers);
	connect(ui->actionConfig, &QAction::triggered, this, &MainWindow::onConfig);
	connect(ui->treeView, &QTreeView::clicked, this, &MainWindow::selectTreeItem);
	connect(ui->treeView, &QTreeView::doubleClicked, this, &MainWindow::editTreeItem);
	connect(ui->lineEdit, &QLineEdit::textChanged, this, &MainWindow::filterList);
	connect(ui->lineEdit, &QLineEdit::returnPressed, this, &MainWindow::selectFromSearch);

	ui->actionAddPassword->setIcon(QIcon::fromTheme(QStringLiteral("document-new")));
	ui->actionAddFolder->setIcon(QIcon::fromTheme(QStringLiteral("folder-new")));
	ui->actionEdit->setIcon(QIcon::fromTheme(QStringLiteral("document-properties")));
	ui->actionDelete->setIcon(QIcon::fromTheme(QStringLiteral("edit-delete")));
	ui->actionUsers->setIcon(QIcon::fromTheme(QStringLiteral("x-office-address-book")));
	ui->actionConfig->setIcon(QIcon::fromTheme(QStringLiteral("configure-symbolic")));
	}

	const QModelIndex MainWindow::getCurrentTreeViewIndex()
	{
	return ui->treeView->currentIndex();
	}

	/**
	* @brief MainWindow::config pops up the configuration screen and handles all
	* inter-window communication
	*/
	void MainWindow::config()
	{
	QScopedPointer<ConfigureDialog> d(new ConfigureDialog(m_rootFoldersManager, this));
	d->setModal(true);

	if (d->exec()) {
	if (d->result() == QDialog::Accepted) {
	this->show();

	m_passwordViewer->setPanelTimer();
	m_clipboardHelper->setClipboardTimer();
	}
	}
	}

	/**
	* @brief MainWindow::on_treeView_clicked read the selected password file
	* @param index
	*/
	void MainWindow::selectTreeItem(const QModelIndex &index)
	{
	bool cleared = ui->treeView->currentIndex().flags() == Qt::NoItemFlags;
	// TODO(bezet): "Could not decrypt";
	m_clipboardHelper->clearClippedText();
	QString file = index.data(QFileSystemModel::FilePathRole).toString();
	if (!file.isEmpty() && QFileInfo(file).isFile() && !cleared) {
	m_selectedFile = file;
	auto decryptJob = new FileDecryptJob(file);
	connect(decryptJob, &FileDecryptJob::result, this, &MainWindow::fileDecryptedSlot);
	decryptJob->start();
	ui->stackedWidget->setCurrentIndex((int)StackLayer::PasswordViewer);
	} else {
	m_passwordViewer->clear();
	ui->actionEdit->setEnabled(false);
	ui->actionEdit->setChecked(false);
	ui->actionDelete->setEnabled(true);
	ui->stackedWidget->setCurrentIndex((int)StackLayer::WelcomePage);
	}
	}

	/**
	* @brief MainWindow::on_treeView_doubleClicked when doubleclicked on
	* TreeViewItem, open the edit Window
	* @param index
	*/
	void MainWindow::editTreeItem(const QModelIndex &index)
	{
	QFileInfo fileInfo{index.data(QFileSystemModel::Roles::FilePathRole).toString()};

	if (!fileInfo.isFile()) {
	return;
	}
	auto decryptJob = new FileDecryptJob(fileInfo.absoluteFilePath());
	connect(decryptJob, &FileDecryptJob::finished, this, [this, decryptJob](KJob *) {
	if (decryptJob->error() != KJob::NoError) {
	m_errorMessage->setText(decryptJob->errorText());
	m_errorMessage->animatedShow();
	return;
	}

	openPasswordDialog(decryptJob->filePath(), decryptJob->content());
	});
	decryptJob->start();
	}

	void MainWindow::openPasswordDialog(const QString &filePath, const QString &content)
	{
	PassEntry entry;
	const auto name = QFileInfo(filePath).baseName();
	if (!content.isEmpty()) {
	entry = PassEntry(name, content);
	}

	m_passwordEditor->setPassEntry(entry);
	ui->actionEdit->setChecked(true);
	ui->stackedWidget->setCurrentIndex((int)StackLayer::PasswordEditor);

	connect(m_passwordEditor, &PasswordEditorWidget::save, this, [this, filePath](const QString &content) {
	const QFileInfo fileInfo(filePath);
	const auto recipients = m_storeModel.recipientsForFile(fileInfo);

	auto encryptJob = new FileEncryptJob(fileInfo.absoluteFilePath(), content.toUtf8(), recipients);
	connect(encryptJob, &FileDecryptJob::finished, this, [encryptJob, this](KJob *) {
	if (encryptJob->error() != KJob::NoError) {
	m_errorMessage->setText(encryptJob->errorText());
	m_errorMessage->animatedShow();
	return;
	}

	ui->treeView->setFocus();
	ui->actionEdit->setChecked(false);

	selectTreeItem(getCurrentTreeViewIndex());
	});
	encryptJob->start();
	});

	connect(m_passwordEditor, &PasswordEditorWidget::editorClosed, this, [this]() {
	selectTreeItem(getCurrentTreeViewIndex());
	ui->actionEdit->setChecked(false);
	});
	}

	/**
	* @brief MainWindow::deselect clear the selection, password and copy buffer
	*/
	void MainWindow::deselect()
	{
	m_clipboardHelper->clearClipboard();
	ui->treeView->clearSelection();
	ui->actionEdit->setEnabled(false);
	ui->actionDelete->setEnabled(false);
	m_passwordViewer->clear();
	ui->stackedWidget->setCurrentIndex((int)StackLayer::WelcomePage);
	}

	void MainWindow::fileDecryptedSlot(KJob *job)
	{
	auto decryptJob = qobject_cast<FileDecryptJob *>(job);
	Q_ASSERT(decryptJob);

	if (decryptJob->error() != KJob::NoError) {
	m_errorMessage->setText(decryptJob->errorText());
	m_errorMessage->animatedShow();
	} else {
	m_errorMessage->animatedHide();
	}

	const auto content = decryptJob->content();

	const PassEntry entry(ui->treeView->selectionModel()->currentIndex().data().toString(), content);
	m_passwordViewer->setPassEntry(entry, content);

	setUiElementsEnabled(true);
	}

	/**
	* @brief MainWindow::setUiElementsEnabled enable or disable the relevant UI
	* elements
	* @param state
	*/
	void MainWindow::setUiElementsEnabled(bool state)
	{
	ui->treeView->setEnabled(state);
	ui->lineEdit->setEnabled(state);
	ui->lineEdit->installEventFilter(this);
	ui->actionAddPassword->setEnabled(state);
	ui->actionAddFolder->setEnabled(state);
	ui->actionUsers->setEnabled(state);
	ui->actionConfig->setEnabled(state);
	// is a file selected?
	state &= ui->treeView->currentIndex().isValid();
	ui->actionDelete->setEnabled(state);
	ui->actionEdit->setEnabled(state);
	}

	/**
	* @brief MainWindow::on_configButton_clicked run Mainwindow::config
	*/
	void MainWindow::onConfig()
	{
	config();
	}

	/**
	* @brief Executes when the string in the search box changes, collapses the
	* TreeView
	* @param arg1
	*/
	void MainWindow::filterList(const QString &arg1)
	{
	ui->treeView->expandAll();
	searchTimer.start();
	}

	/**
	* @brief MainWindow::onTimeoutSearch Fired when search is finished or too much
	* time from two keypresses is elapsed
	*/
	void MainWindow::onTimeoutSearch()
	{
	QString query = ui->lineEdit->text();

	if (query.isEmpty()) {
	ui->treeView->collapseAll();
	deselect();
	}

	query.replace(QStringLiteral(" "), QStringLiteral(".*"));
	QRegularExpression regExp(query, QRegularExpression::CaseInsensitiveOption);
	m_storeModel.setFilterRegularExpression(regExp);

	if (m_storeModel.rowCount() > 0 && !query.isEmpty()) {
	selectFirstFile();
	} else {
	ui->actionEdit->setEnabled(false);
	ui->actionDelete->setEnabled(false);
	}
	}

	/**
	* @brief MainWindow::on_lineEdit_returnPressed get searching
	*
	* Select the first possible file in the tree
	*/
	void MainWindow::selectFromSearch()
	{
	if (m_storeModel.rowCount() > 0) {
	selectFirstFile();
	selectTreeItem(ui->treeView->currentIndex());
	}
	}

	/**
	* @brief MainWindow::selectFirstFile select the first possible file in the
	* tree
	*/
	void MainWindow::selectFirstFile()
	{
	auto model = ui->treeView->model();
	auto index = firstFile(model->index(0, 0));
	ui->treeView->selectionModel()->select(index, QItemSelectionModel::ClearAndSelect);
	}

	/**
	* @brief MainWindow::firstFile return location of first possible file
	* @param parentIndex
	* @return QModelIndex
	*/
	QModelIndex MainWindow::firstFile(QModelIndex parentIndex)
	{
	auto model = parentIndex.model();
	int numRows = model->rowCount(parentIndex);
	for (int row = 0; row < numRows; ++row) {
	auto index = model->index(row, 0, parentIndex);
	if (index.data(AddFileInfoProxy::FileInfoRole).value<QFileInfo>().isFile())
	return index;
	if (model->hasChildren(index))
	return firstFile(index);
	}
	return parentIndex;
	}

	QString MainWindow::fallbackStore()
	{
	const auto rootFolders = m_rootFoldersManager->rootFolders();
	if (rootFolders.isEmpty()) {
	QMessageBox::critical(this, i18nc("@title:dialog", "No password store found"), i18nc("@info", "Please add a password store first."));
	return {};
	}
	return rootFolders[0]->path();
	}

	/**
	* @brief MainWindow::addPassword add a new password by showing a
	* number of dialogs.
	*/
	void MainWindow::addPassword()
	{
	bool ok;

	QString dir = directoryName(ui->treeView->currentIndex().data(QFileSystemModel::Roles::FilePathRole).toString());
	if (dir.isEmpty()) {
	dir = fallbackStore();
	if (dir.isEmpty()) {
	return;
	}
	}
	QString file = QInputDialog::getText(this, i18n("New file"), i18n("New password file: \n(Will be placed in %1 )", dir), QLineEdit::Normal, QString{}, &ok);
	if (!ok \|\| file.isEmpty())
	return;
	file = QDir(dir).absoluteFilePath(file + QStringLiteral(".gpg"));
	openPasswordDialog(file);
	}

	/**
	* @brief MainWindow::onDelete remove password, if you are
	* sure.
	*/
	void MainWindow::onDelete()
	{
	QModelIndex currentIndex = ui->treeView->currentIndex();
	if (!currentIndex.isValid()) {
	// If not valid, we might end up passing empty string
	// to delete, and taht might delete unexpected things on disk
	return;
	}

	QFileInfo fileOrFolder{currentIndex.data(QFileSystemModel::FilePathRole).toString()};
	bool isDir = fileOrFolder.isDir();
	QString file = fileOrFolder.absoluteFilePath();

	QString message;
	if (isDir) {
	message = i18nc("deleting a folder; placeholder is folder name", "Are you sure you want to remove %1 and the whole content?", file);
	QDirIterator it(file, QDirIterator::Subdirectories);
	while (it.hasNext()) {
	it.next();
	if (auto fi = it.fileInfo(); fi.isFile()) {
	if (fi.suffix() != QStringLiteral("gpg")) {
	message += QStringLiteral("<br><strong>")
	+ i18nc("extra warning during certain folder deletions",
	"Attention: "
	"there are unexpected files in the given folder, "
	"check them before continue")
	+ QStringLiteral("</strong>");
	break;
	}
	}
	}
	} else {
	message = i18nc("deleting a file; placeholder is file name", "Are you sure you want to remove %1?", file);
	}

	if (KMessageBox::warningTwoActions(this,
	message,
	isDir ? i18n("Remove folder?") : i18n("Remove password?"),
	KStandardGuiItem::remove(),
	KStandardGuiItem::cancel())
	!= KMessageBox::PrimaryAction)
	return;

	if (!isDir) {
	QFile(file).remove();
	} else {
	QDir dir(file);
	dir.removeRecursively();
	}
	}

	/**
	* @brief MainWindow::onEdit try and edit (selected) password.
	*/
	void MainWindow::onEdit(bool edit)
	{
	if (edit) {
	editTreeItem(ui->treeView->currentIndex());
	} else {
	selectTreeItem(getCurrentTreeViewIndex());
	ui->actionEdit->setChecked(false);
	}
	}

	/**
	* @brief MainWindow::userDialog see MainWindow::onUsers()
	* @param dir folder to edit users for.
	*/
	void MainWindow::userDialog(QString dir)
	{
	if (dir.isEmpty()) {
	dir = fallbackStore();
	if (dir.isEmpty()) {
	return;
	}
	}
	QFileInfo fi(dir);
	if (!fi.isDir()) {
	dir = fi.absolutePath();
	}
	const auto recipients = m_storeModel.recipientsForFile(QFileInfo(dir));
	auto usersDialog = new UsersDialog(recipients, this);
	usersDialog->setAttribute(Qt::WA_DeleteOnClose);
	connect(usersDialog, &UsersDialog::save, this, [dir, this, usersDialog](const QList<QByteArray> &recipients) {
	auto reencryptJob = new DirectoryReencryptJob(m_storeModel, recipients, dir);
	connect(reencryptJob, &DirectoryReencryptJob::result, this, [this, usersDialog](KJob *job) {
	if (job->error() != KJob::NoError) {
	usersDialog->showError(job->errorText());
	return;
	}

	usersDialog->close();
	setUiElementsEnabled(true);
	ui->treeView->setFocus();
	verifyInitialized();
	});

	// statusBar()->showMessage(i18n("Re-encrypting folders"), 3000);
	setUiElementsEnabled(false);
	ui->treeView->setDisabled(true);
	reencryptJob->start();
	});
	usersDialog->show();
	}

	/**
	* @brief MainWindow::onUsers edit users for the current
	* folder,
	* gets lists and opens UserDialog.
	*/
	void MainWindow::onUsers()
	{
	QString dir = ui->treeView->currentIndex().data(QFileSystemModel::Roles::FilePathRole).toString();
	if (dir.isEmpty()) {
	dir = fallbackStore();
	if (dir.isEmpty()) {
	return;
	}
	} else {
	QFileInfo fi(dir);
	if (!fi.isDir()) {
	dir = fi.absolutePath();
	}
	dir = Util::normalizeFolderPath(dir);
	}
	userDialog(dir);
	}

	void MainWindow::slotSetupFinished(const QString &location, const QByteArray &keyId)
	{
	const QString gpgIdFile = location + QStringLiteral(".gpg-id");
	QFile gpgId(gpgIdFile);
	if (!gpgId.open(QIODevice::WriteOnly \| QIODevice::Text)) {
	KMessageBox::error(this, i18n("Unable to write user configuration at \"%1\". Error: %2", gpgIdFile, gpgId.errorString()));
	return;
	}
	gpgId.write(keyId);
	gpgId.close();

	m_rootFoldersManager->addRootFolder(i18nc("Default store name", "Local Store"), location);
	m_rootFoldersManager->save();

	ui->lineEdit->clear();
	m_passwordViewer->clear();
	ui->treeView->selectionModel()->clear();

	ui->actionEdit->setEnabled(false);
	ui->actionDelete->setEnabled(false);

	ui->stackedWidget->setCurrentIndex((int)StackLayer::WelcomePage);
	verifyInitialized();
	}

	void MainWindow::verifyInitialized()
	{
	auto alreadyConfigured = !m_rootFoldersManager->rootFolders().isEmpty();
	ui->sidebar->setVisible(alreadyConfigured);
	ui->stackedWidget->setCurrentIndex(alreadyConfigured ? (int)StackLayer::WelcomePage : (int)StackLayer::ConfigurationPage);

	ui->actionAddFolder->setEnabled(alreadyConfigured);
	ui->actionAddPassword->setEnabled(alreadyConfigured);
	ui->actionDelete->setEnabled(ui->actionDelete->isEnabled() && alreadyConfigured);
	ui->actionEdit->setEnabled(ui->actionEdit->isEnabled() && alreadyConfigured);
	}

	/**
	* @brief MainWindow::closeEvent hide or quit
	* @param event
	*/
	void MainWindow::closeEvent(QCloseEvent *event)
	{
	m_clipboardHelper->clearClipboard();
	event->accept();
	}

	/**
	* @brief MainWindow::eventFilter filter out some events and focus the
	* treeview
	* @param obj
	* @param event
	* @return
	*/
	bool MainWindow::eventFilter(QObject obj, QEvent event)
	{
	if (obj == ui->lineEdit && event->type() == QEvent::KeyPress) {
	auto key = dynamic_cast<QKeyEvent >(event);
	if (key != nullptr && key->key() == Qt::Key_Down) {
	ui->treeView->setFocus();
	}
	}
	return QObject::eventFilter(obj, event);
	}

	/**
	* @brief MainWindow::keyPressEvent did anyone press return, enter or escape?
	* @param event
	*/
	void MainWindow::keyPressEvent(QKeyEvent *event)
	{
	switch (event->key()) {
	case Qt::Key_Delete:
	onDelete();
	break;
	case Qt::Key_Return:
	case Qt::Key_Enter:
	if (m_storeModel.rowCount() > 0)
	selectTreeItem(ui->treeView->currentIndex());
	break;
	case Qt::Key_Escape:
	ui->lineEdit->clear();
	break;
	default:
	break;
	}
	}

	/**
	* @brief MainWindow::showContextMenu show us the (file or folder) context
	* menu
	* @param pos
	*/
	void MainWindow::showContextMenu(const QPoint &pos)
	{
	QModelIndex index = ui->treeView->indexAt(pos);
	bool selected = true;
	if (!index.isValid()) {
	ui->treeView->clearSelection();
	ui->actionDelete->setEnabled(false);
	ui->actionEdit->setEnabled(false);
	selected = false;
	}

	ui->treeView->setCurrentIndex(index);

	QPoint globalPos = ui->treeView->viewport()->mapToGlobal(pos);

	QFileInfo fileOrFolder = ui->treeView->currentIndex().data(AddFileInfoProxy::FileInfoRole).value<QFileInfo>();

	QMenu contextMenu;
	if (!selected \|\| fileOrFolder.isDir()) {
	const QAction *addFolderAction = contextMenu.addAction(i18n("Add folder"));
	const QAction *addPasswordAction = contextMenu.addAction(i18n("Add password"));
	const QAction *usersAction = contextMenu.addAction(i18n("Users"));
	connect(addFolderAction, &QAction::triggered, this, &MainWindow::addFolder);
	connect(addPasswordAction, &QAction::triggered, this, &MainWindow::addPassword);
	connect(usersAction, &QAction::triggered, this, &MainWindow::onUsers);
	} else if (fileOrFolder.isFile()) {
	const QAction *edit = contextMenu.addAction(i18n("Edit"));
	connect(edit, &QAction::triggered, this, &MainWindow::onEdit);
	}
	if (selected) {
	contextMenu.addSeparator();
	if (fileOrFolder.isDir()) {
	const QAction *renameFolderAction = contextMenu.addAction(i18n("Rename folder"));
	connect(renameFolderAction, &QAction::triggered, this, &MainWindow::renameFolder);
	} else if (fileOrFolder.isFile()) {
	const QAction *renamePasswordAction = contextMenu.addAction(i18n("Rename password"));
	connect(renamePasswordAction, &QAction::triggered, this, &MainWindow::renamePassword);
	}
	const QAction *deleteItem = contextMenu.addAction(i18n("Delete"));
	connect(deleteItem, &QAction::triggered, this, &MainWindow::onDelete);
	}
	contextMenu.exec(globalPos);
	}

	/**
	* @brief MainWindow::addFolder add a new folder to store passwords in
	*/
	void MainWindow::addFolder()
	{
	bool ok;
	QString dir = directoryName(ui->treeView->currentIndex().data(QFileSystemModel::FilePathRole).toString());

	if (dir.isEmpty()) {
	dir = fallbackStore();
	if (dir.isEmpty()) {
	return;
	}
	}
	QString newdir = QInputDialog::getText(this, i18n("New file"), i18n("New Folder: \n(Will be placed in %1 )", dir), QLineEdit::Normal, QString{}, &ok);
	if (!ok \|\| newdir.isEmpty())
	return;
	QDir(dir).mkdir(newdir);
	}

	/**
	* @brief MainWindow::renameFolder rename an existing folder
	*/
	void MainWindow::renameFolder()
	{
	bool ok;
	QString srcDir = QDir::cleanPath(directoryName(ui->treeView->currentIndex().data(QFileSystemModel::FilePathRole).toString()));
	if (srcDir.isEmpty()) {
	return;
	}
	QString srcDirName = QDir(srcDir).dirName();
	QString newName = QInputDialog::getText(this, i18n("Rename file"), i18n("Rename Folder To: "), QLineEdit::Normal, srcDirName, &ok);
	if (!ok \|\| newName.isEmpty())
	return;
	QString destDir = srcDir;
	destDir.replace(srcDir.lastIndexOf(srcDirName), srcDirName.length(), newName);
	m_storeModel.move(srcDir, destDir);
	}

	/**
	* @brief MainWindow::renamePassword rename an existing password
	*/
	void MainWindow::renamePassword()
	{
	bool ok;
	QString file = ui->treeView->currentIndex().data(QFileSystemModel::FilePathRole).toString();
	QString filePath = QFileInfo(file).path();
	QString fileName = QFileInfo(file).fileName();
	if (fileName.endsWith(QStringLiteral(".gpg"), Qt::CaseInsensitive))
	fileName.chop(4);

	QString newName = QInputDialog::getText(this, i18n("Rename file"), i18n("Rename File To: "), QLineEdit::Normal, fileName, &ok);
	if (!ok \|\| newName.isEmpty())
	return;
	QString newFile = QDir(filePath).filePath(newName);
	m_storeModel.move(file, newFile);
	}
	-
	-/**
	- * @brief MainWindow::critical critical message popup wrapper.
	- * @param title
	- * @param msg
	- */
	-void MainWindow::critical(const QString &title, const QString &msg)
	-{
	- QMessageBox::critical(this, title, msg);
	-}
	diff --git a/src/widgets/mainwindow.h b/src/widgets/mainwindow.h
	index 796e27d..bc019c0 100644
	--- a/src/widgets/mainwindow.h
	+++ b/src/widgets/mainwindow.h
	@@ -1,128 +1,126 @@
	/*
	SPDX-FileCopyrightText: 2014-2023 Anne Jan Brouwer <brouwer@annejan.com>
	SPDX-FileCopyrightText: 2016-2017 tezeb <tezeb+github@outoftheblue.pl>
	SPDX-FileCopyrightText: 2018 Lukas Vogel <lukedirtwalker@gmail.com>
	SPDX-FileCopyrightText: 2018 Claudio Maradonna <penguyman@stronzi.org>
	SPDX-FileCopyrightText: 2019 Maciej S. Szmigiero <mail@maciej.szmigiero.name>
	SPDX-FileCopyrightText: 2023 g10 Code GmbH
	SPDX-FileContributor: Sune Stolborg Vuorela <sune@vuorela.dk>

	SPDX-License-Identifier: GPL-3.0-or-later
	*/
	#ifndef MAINWINDOW_H_
	#define MAINWINDOW_H_

	#include "models/storemodel.h"

	#include <KSelectionProxyModel>
	#include <QFileSystemModel>
	#include <QItemSelectionModel>
	#include <QMainWindow>
	#include <QProcess>
	#include <QTimer>

	#ifdef __APPLE__
	// http://doc.qt.io/qt-5/qkeysequence.html#qt_set_sequence_auto_mnemonic
	void qt_set_sequence_auto_mnemonic(bool b);
	#endif

	namespace Ui
	{
	class MainWindow;
	}

	class QComboBox;
	class ClipboardHelper;
	class KMessageWidget;
	class AddFileInfoProxy;
	class KJob;
	class PasswordViewerWidget;
	class PasswordEditorWidget;
	class RootFoldersManager;

	/*!
	\class MainWindow
	\brief The MainWindow class does way too much, not only is it a switchboard,
	configuration handler and more, it's also the process-manager.

	This class could really do with an overhaul.
	*/
	class MainWindow : public QMainWindow
	{
	Q_OBJECT

	public:
	explicit MainWindow(QWidget *parent = nullptr);
	~MainWindow();

	void restoreWindow();
	void userDialog(QString = {});
	void config();

	void setUiElementsEnabled(bool state);

	const QModelIndex getCurrentTreeViewIndex();

	void setVisible(bool visible) override;

	protected:
	void closeEvent(QCloseEvent *event) override;
	void keyPressEvent(QKeyEvent *event) override;
	void changeEvent(QEvent *event) override;
	bool eventFilter(QObject obj, QEvent event) override;

	public Q_SLOTS:
	void deselect();

	- void critical(const QString &, const QString &);
	-
	void selectTreeItem(const QModelIndex &index);

	private Q_SLOTS:
	void addPassword();
	void addFolder();
	void onEdit(bool edit);
	void onDelete();
	void onUsers();
	void onConfig();
	void editTreeItem(const QModelIndex &index);
	void openPasswordDialog(const QString &filePath, const QString &content = {});
	void filterList(const QString &arg1);
	void selectFromSearch();
	void showContextMenu(const QPoint &pos);
	void renameFolder();
	void renamePassword();
	void focusInput();
	void onTimeoutSearch();
	void verifyInitialized();
	void fileDecryptedSlot(KJob *job);
	void slotSetupFinished(const QString &location, const QByteArray &keyId);

	private:
	QString fallbackStore();

	QScopedPointer<Ui::MainWindow> ui;
	ClipboardHelper *const m_clipboardHelper;
	PasswordViewerWidget *const m_passwordViewer;
	PasswordEditorWidget *const m_passwordEditor;
	RootFoldersManager *const m_rootFoldersManager;
	StoreModel m_storeModel;
	QTimer searchTimer;
	KMessageWidget *m_notInitialized;
	KMessageWidget *m_errorMessage;
	QString m_selectedFile;
	bool firstShow = true;

	void initToolBarButtons();

	void updateText();
	void selectFirstFile();
	QModelIndex firstFile(QModelIndex parentIndex);
	void setPassword(QString, bool isNew = true);

	void initTrayIcon();
	void destroyTrayIcon();
	void reencryptPath(QString dir);
	};

	#endif // MAINWINDOW_H_

File Metadata

Mime Type: text/x-diff
Expires: Fri, Dec 12, 11:48 AM (1 d, 22 h)
Storage Engine: local-disk
Storage Format: Raw Data
Storage Handle: 48/54/7b08a64cb3fbb1f017562b0eb7eb

No OneTemporaryActions

View Options

File Metadata

Event Timeline

No OneTemporary
Actions