swaymux/tree/SwayTreeModel.h

65 lines
1.7 KiB
C
Raw Normal View History

2024-03-04 00:27:05 +01:00
//
// Created by grimmauld on 03.03.24.
//
#ifndef SWAYMUX_SWAYTREEMODEL_H
#define SWAYMUX_SWAYTREEMODEL_H
#include <QAbstractItemModel>
#include <utility>
#include "AbstractTreeModel.h"
#include "swaytree.h"
#include "../sway_bindings/Sway.h"
#include <nlohmann/json.hpp>
#include <utility>
#include <QTreeView>
2024-03-04 00:27:05 +01:00
using json = nlohmann::json;
class SwayTreeModel : public AbstractTreeModel<SwayTreeNode> {
Q_OBJECT
public:
inline static Sway sway = Sway();
Q_DISABLE_COPY_MOVE(SwayTreeModel)
explicit SwayTreeModel(QObject *parent = nullptr) : AbstractTreeModel<SwayTreeNode>(parent) {
rootItem = nullptr;
}
void generateRoot() {
2024-03-04 00:27:05 +01:00
auto res = sway.sendIPC(SWAY_GET_TREE);
auto rep = json::parse(res.msg);
rootItem = new SwayTreeNode(rep);
}
~SwayTreeModel() override = default;
explicit SwayTreeModel(SwayTreeNode *rootItem, QObject *parent = nullptr) : rootItem(rootItem),
AbstractTreeModel<SwayTreeNode>(
parent) {};
2024-03-04 00:27:05 +01:00
[[nodiscard]] SwayTreeNode *getRoot() const override {
2024-03-04 00:27:05 +01:00
return rootItem;
};
[[nodiscard]] inline QModelIndex findFocusedWindowIndex() const {
return findIndexByNode(getRoot()->findFocused());
}
[[nodiscard]] std::set<const SwayTreeNode *> getSelectedNodes(const QTreeView *treeView) const;
[[nodiscard]] SwayTreeNode *getSelectedNode(const QModelIndex &i) const;
[[nodiscard]] QModelIndex findIndexByNode(const SwayTreeNode *node) const;
2024-03-04 00:27:05 +01:00
private:
SwayTreeNode *rootItem;
2024-03-04 00:27:05 +01:00
};
#endif //SWAYMUX_SWAYTREEMODEL_H