1 files changed, 37 insertions, 4 deletions
diff --git a/src/filterwidget.h b/src/filterwidget.h
index 762d9b15..5b08c3ae 100644
--- a/src/filterwidget.h
+++ b/src/filterwidget.h
@@ -1,22 +1,55 @@
 #ifndef FILTERWIDGET_H
 #define FILTERWIDGET_H
 
+#include <QObject>
+#include <QLineEdit>
+#include <QToolButton>
+#include <QList>
+#include <QAbstractItemView>
+
 class EventFilter;
+class FilterWidget;
 
-class FilterWidget
+class FilterWidgetProxyModel : public QSortFilterProxyModel
 {
+  Q_OBJECT;
+
 public:
-  std::function<void ()> changed;
+  FilterWidgetProxyModel(FilterWidget& fw, QWidget* parent=nullptr);
+  using QSortFilterProxyModel::invalidateFilter;
+
+protected:
+  bool filterAcceptsRow(int row, const QModelIndex& parent) const override;
+
+private:
+  FilterWidget& m_filter;
+};
+
+
+class FilterWidget : public QObject
+{
+  Q_OBJECT;
+
+public:
+  using predFun = std::function<bool (const QString& what)>;
 
   FilterWidget();
 
-  void set(QLineEdit* edit);
+  void setEdit(QLineEdit* edit);
+  void setList(QAbstractItemView* list);
   void clear();
 
-  bool matches(std::function<bool (const QString& what)> pred) const;
+  QModelIndex map(const QModelIndex& index);
+
+  bool matches(predFun pred) const;
+
+signals:
+  void changed();
 
 private:
   QLineEdit* m_edit;
+  QAbstractItemView* m_list;
+  FilterWidgetProxyModel* m_proxy;
   EventFilter* m_eventFilter;
   QToolButton* m_clear;
   QString m_text;