From c5f98f9756ed9ba5b2241fa11215d99c8bf52461 Mon Sep 17 00:00:00 2001
From: isanae <14251494+isanae@users.noreply.github.com>
Date: Thu, 12 Dec 2019 18:57:36 -0500
Subject: now prunes empty directories handles archives and conflicts
 checkboxes

---
 src/datatab.cpp | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

(limited to 'src/datatab.cpp')

diff --git a/src/datatab.cpp b/src/datatab.cpp
index bc0ff455..b351923c 100644
--- a/src/datatab.cpp
+++ b/src/datatab.cpp
@@ -18,7 +18,7 @@ QString UnmanagedModName();
 DataTab::DataTab(
   OrganizerCore& core, PluginContainer& pc,
   QWidget* parent, Ui::MainWindow* mwui) :
-    m_core(core), m_pluginContainer(pc), m_archives(false), m_parent(parent),
+    m_core(core), m_pluginContainer(pc), m_parent(parent),
     m_model(new FileTreeModel(core)),
     ui{
       mwui->btnRefreshData, mwui->dataTree,
@@ -150,7 +150,6 @@ void DataTab::onRefresh()
 void DataTab::refreshDataTree()
 {
   m_model->refresh();
-  ui.tree->expand(m_model->index(0, 0));
 }
 
 void DataTab::refreshDataTreeKeepExpandedNodes()
@@ -343,12 +342,27 @@ void DataTab::onContextMenu(const QPoint &pos)
 
 void DataTab::onConflicts()
 {
-  refreshDataTreeKeepExpandedNodes();
+  updateOptions();
 }
 
 void DataTab::onArchives()
 {
-  m_archives = (m_core.getArchiveParsing() && ui.archives->isChecked());
+  updateOptions();
+}
+
+void DataTab::updateOptions()
+{
+  FileTreeModel::Flags flags = FileTreeModel::NoFlags;
+
+  if (ui.conflicts->isChecked()) {
+    flags |= FileTreeModel::Conflicts;
+  }
+
+  if (ui.archives->isChecked()) {
+    flags |= FileTreeModel::Archives;
+  }
+
+  m_model->setFlags(flags);
   refreshDataTree();
 }
 
-- 
cgit v1.3.1