1 files changed, 83 insertions, 0 deletions
diff --git a/libs/installer_bsplugins/src/BSPluginInfo/AuxTreeModel.cpp b/libs/installer_bsplugins/src/BSPluginInfo/AuxTreeModel.cpp
new file mode 100644
index 0000000..b5df2ab
--- /dev/null
+++ b/libs/installer_bsplugins/src/BSPluginInfo/AuxTreeModel.cpp
@@ -0,0 +1,83 @@
+#include "AuxTreeModel.h"
+
+namespace BSPluginInfo
+{
+
+AuxTreeModel::AuxTreeModel(const TESData::AssociatedEntry* entry, QObject* parent)
+    : QAbstractItemModel(parent), m_Entry{entry}
+{}
+
+QModelIndex AuxTreeModel::index(int row, int column, const QModelIndex& parent) const
+{
+  if (row < 0 || column != 0) {
+    return QModelIndex();
+  }
+
+  const auto parentItem = parent.isValid()
+                              ? static_cast<const Item*>(parent.internalPointer())
+                              : m_Entry->root().get();
+
+  const auto item = parentItem->getByIndex(row).get();
+  return item ? createIndex(row, column, item) : QModelIndex();
+}
+
+QModelIndex AuxTreeModel::parent(const QModelIndex& index) const
+{
+  if (!index.isValid())
+    return QModelIndex();
+
+  const auto item   = static_cast<const Item*>(index.internalPointer());
+  const auto parent = item ? item->parent() : nullptr;
+  if (!parent || !parent->parent()) {
+    return QModelIndex();
+  }
+
+  const int row = parent->parent()->indexOf(parent);
+
+  return createIndex(row, 0, parent);
+}
+
+int AuxTreeModel::rowCount(const QModelIndex& parent) const
+{
+  const auto parentItem = parent.isValid()
+                              ? static_cast<const Item*>(parent.internalPointer())
+                              : m_Entry->root().get();
+
+  return parentItem ? parentItem->numChildren() : 0;
+}
+
+int AuxTreeModel::columnCount([[maybe_unused]] const QModelIndex& parent) const
+{
+  return 1;
+}
+
+QVariant AuxTreeModel::data(const QModelIndex& index, int role) const
+{
+  if (!index.isValid()) {
+    return QVariant();
+  }
+
+  const auto item = static_cast<const Item*>(index.internalPointer());
+  switch (role) {
+  case Qt::DisplayRole:
+    return QString::fromStdString(item->name());
+  }
+
+  return QVariant();
+}
+
+QVariant AuxTreeModel::headerData(int section, Qt::Orientation orientation,
+                                  int role) const
+{
+  if (orientation != Qt::Horizontal || role != Qt::DisplayRole) {
+    return QVariant();
+  }
+
+  if (section == 0) {
+    return tr("File Name");
+  } else {
+    return QVariant();
+  }
+}
+
+}  // namespace BSPluginInfo