1 files changed, 55 insertions, 8 deletions

diff --git a/framework/src/extensionmodel.cpp b/framework/src/extensionmodel.cpp
index f64a0e17..e3fab7d8 100644
--- a/framework/src/extensionmodel.cpp
+++ b/framework/src/extensionmodel.cpp
@@ -22,12 +22,17 @@
 #include <QDir>
 #include <QDebug>
 #include <QTimer>
+#include <QJsonDocument>
+#include <QJsonObject>
 
 using namespace Kube;
 
 ExtensionModel::ExtensionModel(QObject *parent)
     : QSortFilterProxyModel(parent)
 {
+    setDynamicSortFilter(true);
+    sort(0, Qt::DescendingOrder);
+    setFilterCaseSensitivity(Qt::CaseInsensitive);
     QTimer::singleShot(0, this, &ExtensionModel::load);
 }
 
@@ -36,8 +41,7 @@ QHash<int, QByteArray> ExtensionModel::roleNames() const
     return {
         {Name, "name"},
         {Tooltip, "tooltip"},
-        {Icon, "icon"},
-        {Source, "source"}
+        {Icon, "icon"}
     };
 }
 
@@ -48,21 +52,55 @@ void ExtensionModel::load()
     auto engine = qmlEngine(this);
     Q_ASSERT(engine);
     for (const auto &path : engine->importPathList()) {
-        QDir dir{path + "/org/kube/viewextensions"};
+        QDir dir{path + "/org/kube/views"};
         for (const auto &pluginName : dir.entryList(QDir::Dirs | QDir::NoDotAndDotDot)) {
-            auto viewPath = dir.path() + pluginName + "/View.qml";
-            qWarning() << "Plugin path: " << dir.path() + pluginName + "/View.qml";
+            const auto pluginPath = dir.path() + "/" + pluginName;
+            mPaths.insert(pluginName, pluginPath);
             auto item = new QStandardItem;
-            item->setData(viewPath, Source);
             item->setData(pluginName, Name);
             item->setData(pluginName, Tooltip);
-            item->setData("document-decrypt", Icon);
+            item->setData("kdocumentinfo-inverted", Icon);
+
+            if (QFileInfo::exists(pluginPath + "/metadata.json")) {
+                QFile file{pluginPath + "/metadata.json"};
+                file.open(QIODevice::ReadOnly);
+                auto json = QJsonDocument::fromJson(file.readAll());
+                auto map = json.object().toVariantMap();
+                item->setData(map.value("icon").toString(), Icon);
+                item->setData(map.value("tooltip").toString(), Tooltip);
+                if (map.value("hidden", false).toBool()) {
+                    delete item;
+                    continue;
+                }
+            }
+
             model->appendRow(item);
         }
     }
     setSourceModel(model);
 }
 
+QString ExtensionModel::findSource(const QString &extensionName, const QString &sourceName)
+{
+    if (mPaths.isEmpty()) {
+        load();
+    }
+    return mPaths.value(extensionName) + "/" + sourceName;
+}
+
+void ExtensionModel::setSortOrder(const QVariantList &order)
+{
+    mSortOrder.clear();
+    for (const auto &e : order) {
+        mSortOrder << e.toString();
+    }
+}
+
+QVariantList ExtensionModel::sortOrder() const
+{
+    return {};
+}
+
 QVariant ExtensionModel::data(const QModelIndex &idx, int role) const
 {
     return QSortFilterProxyModel::data(idx, role);
@@ -70,5 +108,14 @@ QVariant ExtensionModel::data(const QModelIndex &idx, int role) const
 
 bool ExtensionModel::lessThan(const QModelIndex &left, const QModelIndex &right) const
 {
-    return QSortFilterProxyModel::lessThan(left, right);
+    auto leftIndex = mSortOrder.indexOf(left.data(Name).toString());
+    auto rightIndex = mSortOrder.indexOf(right.data(Name).toString());
+    if (leftIndex >= 0 && rightIndex >= 0) {
+        //Higher index is less than
+        return leftIndex > rightIndex;
+    }
+    if (leftIndex < 0 && rightIndex < 0) {
+        return QSortFilterProxyModel::lessThan(left, right);
+    }
+    return leftIndex < rightIndex;
 }