1 files changed, 7 insertions, 0 deletions
diff --git a/beetview.h b/beetview.h
index 763dc8e..f4ae39f 100644
--- a/beetview.h
+++ b/beetview.h
@@ -6,10 +6,17 @@
 class BeetView : public QTreeView {
     Q_OBJECT
     public:
+        enum ExpandOrCollapseMode { Expand, Collapse };
         explicit BeetView(QWidget *parent = 0);
 
+    public slots:
+        void expandOrCollapse(int mode);
+
     protected:
         virtual void contextMenuEvent(QContextMenuEvent *e);
+
+    private:
+        void expandOrCollapseRecursive(const QModelIndex &idx, int mode);
 };
 
 #endif // BEETVIEW_H