1 files changed, 132 insertions, 0 deletions
diff --git a/Swift/QtUI/Roster/QtFilterWidget.cpp b/Swift/QtUI/Roster/QtFilterWidget.cpp
new file mode 100644
index 0000000..5bd4669
--- /dev/null
+++ b/Swift/QtUI/Roster/QtFilterWidget.cpp
@@ -0,0 +1,132 @@
+/*
+ * Copyright (c) 2013 Tobias Markmann
+ * Licensed under the simplified BSD license.
+ * See Documentation/Licenses/BSD-simplified.txt for more information.
+ */
+
+#include <Swift/QtUI/Roster/QtFilterWidget.h>
+
+#include <QLayout>
+#include <QVBoxLayout>
+#include <QKeyEvent>
+#include <QEvent>
+#include <QString>
+#include <QEvent>
+
+#include <Swift/QtUI/QtSwiftUtil.h>
+
+namespace Swift {
+
+QtFilterWidget::QtFilterWidget(QWidget* parent, QtTreeWidget* treeView, QBoxLayout* layout) : QWidget(parent), treeView_(treeView), fuzzyRosterFilter_(0), isModifierSinglePressed_(false) {
+	int targetIndex = layout->indexOf(treeView);
+
+	QVBoxLayout* vboxLayout = new QVBoxLayout(this);
+	vboxLayout->setSpacing(0);
+	vboxLayout->setContentsMargins(0,0,0,0);
+
+	filterLineEdit_ = new QLineEdit(this);
+	filterLineEdit_->hide();
+	vboxLayout->addWidget(filterLineEdit_);
+
+	vboxLayout->addWidget(treeView);
+	setLayout(vboxLayout);
+	layout->insertWidget(targetIndex, this);
+
+	filterLineEdit_->installEventFilter(this);
+	treeView->installEventFilter(this);
+
+	sourceModel_ = treeView_->model();
+}
+
+QtFilterWidget::~QtFilterWidget() {
+
+}
+
+bool QtFilterWidget::eventFilter(QObject*, QEvent* event) {
+	if (event->type() == QEvent::KeyPress ||
+		event->type() == QEvent::KeyRelease ||
+		// InputMethodQuery got introduced in Qt 5.
+#if QT_VERSION >= 0x050000
+		event->type() == QEvent::InputMethodQuery ||
+#endif
+		event->type() == QEvent::InputMethod) {
+		event->ignore();
+		QKeyEvent* keyEvent = dynamic_cast<QKeyEvent*>(event);
+		if (keyEvent) {
+			if (keyEvent->key() == Qt::Key_Up || keyEvent->key() == Qt::Key_Down) {
+				return false;
+			} else if ((keyEvent->key() == Qt::Key_Left || keyEvent->key() == Qt::Key_Right) && filterLineEdit_->text().isEmpty()) {
+				return false;
+			} else if (keyEvent->key() == Qt::Key_Alt && event->type() == QEvent::KeyPress) {
+				isModifierSinglePressed_ = true;
+			} else if (keyEvent->key() == Qt::Key_Alt && event->type() == QEvent::KeyRelease && isModifierSinglePressed_) {
+				QPoint itemOffset(2,2);
+				QPoint contextMenuPosition = treeView_->visualRect(treeView_->currentIndex()).topLeft() + itemOffset;;
+				QApplication::postEvent(treeView_, new QContextMenuEvent(QContextMenuEvent::Keyboard, contextMenuPosition, treeView_->mapToGlobal(contextMenuPosition)));
+				return true;
+			} else if (keyEvent->key() == Qt::Key_Return) {
+				filterLineEdit_->setText("");
+				return false;
+			} else if (keyEvent->key() == Qt::Key_Escape) {
+				filterLineEdit_->setText("");
+			} else {
+				isModifierSinglePressed_ = false;
+			}
+		}
+
+		filterLineEdit_->event(event);
+		filterLineEdit_->setVisible(!filterLineEdit_->text().isEmpty());
+
+		// update roster filters
+		if (event->type() == QEvent::KeyRelease) {
+			if (fuzzyRosterFilter_) {
+				if (filterLineEdit_->text().isEmpty()) {
+					// remove currently installed search filter and put old filters back
+					treeView_->getRoster()->removeFilter(fuzzyRosterFilter_);
+					delete fuzzyRosterFilter_;
+					fuzzyRosterFilter_ = NULL;
+					pushAllFilters();
+				} else {
+					// remove currently intsalled search filter and put new search filter in place
+					updateSearchFilter();
+				}
+			} else {
+				if (!filterLineEdit_->text().isEmpty()) {
+					// remove currently installed filters and put a search filter in place
+					popAllFilters();
+					updateSearchFilter();
+				}
+			}
+		}
+		return true;
+	}
+	return false;
+}
+
+void QtFilterWidget::popAllFilters() {
+	std::vector<RosterFilter*> filters = treeView_->getRoster()->getFilters();
+	foreach(RosterFilter* filter, filters) {
+		filters_.push_back(filter);
+		treeView_->getRoster()->removeFilter(filter);
+	}
+}
+
+void QtFilterWidget::pushAllFilters() {
+	foreach(RosterFilter* filter, filters_) {
+		treeView_->getRoster()->addFilter(filter);
+	}
+	filters_.clear();
+}
+
+void QtFilterWidget::updateSearchFilter() {
+	if (fuzzyRosterFilter_) {
+		treeView_->getRoster()->removeFilter(fuzzyRosterFilter_);
+		delete fuzzyRosterFilter_;
+		fuzzyRosterFilter_ = NULL;
+	}
+	fuzzyRosterFilter_ = new FuzzyRosterFilter(Q2PSTRING(filterLineEdit_->text()));
+	treeView_->getRoster()->addFilter(fuzzyRosterFilter_);
+	treeView_->setCurrentIndex(sourceModel_->index(0, 0, sourceModel_->index(0,0)));
+}
+
+}