Do not store the list of preview plugins in PreviewGenerator.

1 files changed, 15 insertions, 16 deletions

diff --git a/src/previewgenerator.cpp b/src/previewgenerator.cpp
index c7df5a25..904d679f 100644
--- a/src/previewgenerator.cpp
+++ b/src/previewgenerator.cpp
@@ -27,33 +27,32 @@ along with Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
 

 #include "plugincontainer.h"

 

+using namespace MOBase;

+

 PreviewGenerator::PreviewGenerator(const PluginContainer* pluginContainer) :

   m_PluginContainer(pluginContainer) {

   m_MaxSize = QGuiApplication::primaryScreen()->size() * 0.8;

 }

 

-void PreviewGenerator::registerPlugin(MOBase::IPluginPreview *plugin)

-{

-  for (const QString &extension : plugin->supportedExtensions()) {

-    m_PreviewPlugins.insert(std::make_pair(extension, plugin));

-  }

-}

-

 bool PreviewGenerator::previewSupported(const QString &fileExtension) const

 {

-  auto it = m_PreviewPlugins.find(fileExtension.toLower());

-  if (it == m_PreviewPlugins.end()) {

-    return false;

+  auto& previews = m_PluginContainer->plugins<IPluginPreview>();

+  for (auto* preview : previews) {

+    if (preview->supportedExtensions().contains(fileExtension)) {

+      return true;

+    }

   }

-  return m_PluginContainer->isEnabled(it->second);

+  return false;

 }

 

 QWidget *PreviewGenerator::genPreview(const QString &fileName) const

 {

-  auto iter = m_PreviewPlugins.find(QFileInfo(fileName).suffix().toLower());

-  if (iter != m_PreviewPlugins.end() && m_PluginContainer->isEnabled(iter->second)) {

-    return iter->second->genFilePreview(fileName, m_MaxSize);

-  } else {

-    return nullptr;

+  const QString ext = QFileInfo(fileName).suffix().toLower();

+  auto& previews = m_PluginContainer->plugins<IPluginPreview>();

+  for (auto* preview : previews) {

+    if (m_PluginContainer->isEnabled(preview) && preview->supportedExtensions().contains(ext)) {

+      return preview->genFilePreview(fileName, m_MaxSize);

+    }

   }

+  return nullptr;

 }