refactor the model to easily get focused, root, output and workspace

mainwindow.cpp

@@ -20,7 +20,7 @@ MainWindow::MainWindow(QWidget *parent)
     ui->treeView->expandAll();
 
     ui->treeView->setSelectionMode(QAbstractItemView::ExtendedSelection);
-    ui->treeView->selectionModel()->select(model->getFocused(), QItemSelectionModel::ClearAndSelect);
+    ui->treeView->selectionModel()->select(model->findFocusedWindowIndex(), QItemSelectionModel::ClearAndSelect);
 
     for (int c = 0; c < model->columnCount(); ++c)
         ui->treeView->resizeColumnToContents(c);

tree/AbstractTreeNode.h

@@ -69,6 +69,16 @@ public:
         return nullptr;
     };
 
+    [[nodiscard]] const T *findParentRecursive(std::function<bool(const T *)> test) const {
+        if (test((const T *) this))
+            return (const T*) this;
+
+        if (parent == nullptr)
+            return nullptr;
+
+        return this->findParentRecursive(test);
+    };
+
     void removeChild(int i) {
         if (i >= 0 && i < childCount()) {
             children.erase(std::next(children.begin(), i));

tree/SwayTreeModel.cpp

@@ -4,15 +4,9 @@
 
 #include "SwayTreeModel.h"
 
-QModelIndex SwayTreeModel::getFocused() const {
-
-    std::function<bool(const SwayTreeNode *)> test = [](const SwayTreeNode *node) {
-        return node->node.focused;
-    };
-    const SwayTreeNode *focused = rootItem->findChildRecursive(test);
-
+QModelIndex SwayTreeModel::findFocusedWindowIndex() const {
     std::vector<int> indices;
-    for (const auto *head = focused; head != nullptr; head = head->parentItem()) {
+    for (const auto *head = getRoot()->findFocused(); head != nullptr; head = head->parentItem()) {
         indices.insert(indices.cbegin(), head->row());
     }
 

tree/SwayTreeModel.h

@@ -42,7 +42,7 @@ public:
     };
 
 
-    [[nodiscard]] QModelIndex getFocused() const;
+    [[nodiscard]] QModelIndex findFocusedWindowIndex() const;
 
 
 private:

tree/swaytree.h

@@ -114,10 +114,6 @@ struct SwayRecord {
     [[nodiscard]] bool operator==(const SwayRecord &other) const {
         return this->id == other.id;
     }
-
-    [[nodiscard]] bool isFocused() const  {
-        return focused;
-    }
 };
 
 class SwayTreeNode : public AbstractTreeNode<SwayTreeNode> {
@@ -153,6 +149,32 @@ public:
     [[nodiscard]] QVariant headerData(int column) const override;
 
     [[nodiscard]] int columnCount() const override { return 1; };
+
+    [[nodiscard]] const SwayTreeNode *findFocused() const {
+        std::function<bool(const SwayTreeNode *)> focused = [](const SwayTreeNode *testNode) {
+            return testNode->node.focused;
+        };
+        return this->findChildRecursive(focused);
+    }
+
+    [[nodiscard]] inline const SwayTreeNode *findRoot() const {
+        return this->findParentRecursive(matchNodeType(NodeType::root));
+    }
+
+    [[nodiscard]] inline const SwayTreeNode *findOutput() const {
+        return this->findParentRecursive(matchNodeType(NodeType::output));
+    }
+
+    [[nodiscard]] inline const SwayTreeNode *findWorkspace() const {
+        return this->findParentRecursive(matchNodeType(NodeType::workspace));
+    }
+
+private:
+    [[nodiscard]] static std::function<bool(const SwayTreeNode *)> matchNodeType(const NodeType::NodeType type) {
+        return [type](const SwayTreeNode *testNode) {
+            return testNode->node.type == type;
+        };
+    }
 };