made the checkable columns of the bufferview prettier (code wise that is)

[quassel.git] / src / uisupport / bufferview.cpp

diff --git a/src/uisupport/bufferview.cpp b/src/uisupport/bufferview.cpp
index 8ddd2b3..bc2fe39 100644 (file)
--- a/src/uisupport/bufferview.cpp
+++ b/src/uisupport/bufferview.cpp
@@ -21,7 +21,7 @@
 #include "client.h"
 #include "buffer.h"
 #include "bufferview.h"
-#include "buffertreemodel.h"
+#include "networkmodel.h"
 
 /*****************************************
 * The TreeView showing the Buffers
@@ -33,8 +33,7 @@ BufferView::BufferView(QWidget *parent) : QTreeView(parent) {
 
 void BufferView::init() {
   setIndentation(10);
-  header()->hide();
-  header()->hideSection(1);
+  header()->setContextMenuPolicy(Qt::ActionsContextMenu);
   expandAll();
 
   setAnimated(true);
@@ -60,16 +59,35 @@ void BufferView::setModel(QAbstractItemModel *model) {
   delete selectionModel();
   QTreeView::setModel(model);
   init();
+
+  // remove old Actions
+  QList<QAction *> oldactions = header()->actions();
+  foreach(QAction *action, oldactions) {
+    header()->removeAction(action);
+    action->deleteLater();
+  }
+
+  QString sectionName;
+  QAction *showSection;
+  for(int i = 0; i < model->columnCount(); i++) {
+    sectionName = (model->headerData(i, Qt::Horizontal, Qt::DisplayRole)).toString();
+    showSection = new QAction(sectionName, header());
+    showSection->setCheckable(true);
+    showSection->setChecked(true);
+    showSection->setProperty("column", i);
+    connect(showSection, SIGNAL(toggled(bool)), this, SLOT(toggleHeader(bool)));
+    header()->addAction(showSection);
+  }
   
 }
 
 void BufferView::joinChannel(const QModelIndex &index) {
-  Buffer::Type bufferType = (Buffer::Type)index.data(BufferTreeModel::BufferTypeRole).toInt();
+  Buffer::Type bufferType = (Buffer::Type)index.data(NetworkModel::BufferTypeRole).toInt();
 
   if(bufferType != Buffer::ChannelType)
     return;
   
-  Client::fakeInput(index.data(BufferTreeModel::BufferUidRole).toUInt(), QString("/JOIN %1").arg(index.sibling(index.row(), 0).data().toString()));
+  Client::fakeInput(index.data(NetworkModel::BufferUidRole).toUInt(), QString("/JOIN %1").arg(index.sibling(index.row(), 0).data().toString()));
 }
 
 void BufferView::keyPressEvent(QKeyEvent *event) {
@@ -92,3 +110,8 @@ void BufferView::rowsInserted(const QModelIndex & parent, int start, int end) {
     expand(parent);
   }
 }
+
+void BufferView::toggleHeader(bool checked) {
+  QAction *action = qobject_cast<QAction *>(sender());
+  header()->setSectionHidden((action->property("column")).toInt(), not checked);
+}