1 files changed, 163 insertions, 0 deletions
diff --git a/src/QMenuModel/qmenumodel.cpp b/src/QMenuModel/qmenumodel.cpp
new file mode 100644
index 0000000..173dbbe
--- /dev/null
+++ b/src/QMenuModel/qmenumodel.cpp
@@ -0,0 +1,163 @@
+#include "qmenumodel.h"
+#include <QDebug>
+
+QMenuModel::QMenuModel(QObject *parent, GMenuModel *other)
+    : QAbstractListModel(parent),
+      m_menuModel(0),
+      m_signalChangedId(0)
+{
+    static QHash<int, QByteArray> rolesNames;
+    if (rolesNames.empty()) {
+        rolesNames[Action] = "action";
+        rolesNames[Label] = "label";
+        rolesNames[LinkSection] = "linkSection";
+        rolesNames[LinkSubMenu] = "linkSubMenu";
+    }
+    setRoleNames(rolesNames);
+    setMenuModel(other);
+}
+
+QMenuModel::~QMenuModel()
+{
+   setMenuModel(NULL);
+}
+
+void QMenuModel::setMenuModel(GMenuModel *other)
+{
+    if (m_menuModel == other) {
+        return;
+    }
+
+    beginResetModel();
+
+    if (m_menuModel) {
+        g_signal_handler_disconnect(m_menuModel, m_signalChangedId);
+        m_signalChangedId = 0;
+        g_object_unref(m_menuModel);
+    }
+
+    m_menuModel = other;
+
+    endResetModel();
+
+    if (m_menuModel) {
+        qDebug() << "Menu size:" << g_menu_model_get_n_items(m_menuModel);
+        m_signalChangedId = g_signal_connect(m_menuModel,
+                                             "items-changed",
+                                             G_CALLBACK(QMenuModel::onItemsChanged),
+                                             this);
+    }
+}
+
+GMenuModel *QMenuModel::menuModel() const
+{
+    return m_menuModel;
+}
+
+/* QAbstractItemModel */
+int QMenuModel::columnCount(const QModelIndex &) const
+{
+    return 1;
+}
+
+QVariant QMenuModel::data(const QModelIndex &index, int role) const
+{
+    QVariant attribute;
+    int rowCountValue = rowCount();
+
+    if ((rowCountValue > 0) && (index.row() >= 0) && (index.row() < rowCountValue)) {
+        qDebug() << "GetData: " << index.row() << role;
+        if (m_menuModel) {
+            switch (role)
+            {
+            case Action:
+                attribute = getStringAttribute(index, G_MENU_ATTRIBUTE_ACTION);
+                break;
+            case Label:
+                attribute = getStringAttribute(index, G_MENU_ATTRIBUTE_LABEL);
+                break;
+            case LinkSection:
+                attribute = getLink(index, G_MENU_LINK_SECTION);
+                break;
+            case LinkSubMenu:
+                attribute = getLink(index, G_MENU_LINK_SUBMENU);
+                break;
+            default:
+                break;
+            }
+        }
+    }
+    qDebug() << "GetData done" << attribute;
+    return attribute;
+}
+
+QModelIndex QMenuModel::parent(const QModelIndex &index) const
+{
+    return QModelIndex();
+}
+
+int QMenuModel::rowCount(const QModelIndex &) const
+{
+    if (m_menuModel) {
+        return g_menu_model_get_n_items(m_menuModel);
+    }
+    return 0;
+}
+
+QVariant QMenuModel::getStringAttribute(const QModelIndex &index,
+                                        const QString &attribute) const
+{
+    QVariant result;
+    gchar* value = NULL;
+    g_menu_model_get_item_attribute(m_menuModel,
+                                    index.row(),
+                                    attribute.toLatin1(),
+                                    "s", &value);
+    if (value) {
+        result = QVariant(QString::fromLatin1(value));
+        g_free(value);
+    }
+    return result;
+}
+
+QVariant QMenuModel::getLink(const QModelIndex &index,
+                             const QString &linkName) const
+{
+    GMenuModel *link;
+
+    link = g_menu_model_get_item_link(m_menuModel,
+                                      index.row(),
+                                      linkName.toLatin1());
+
+    if (link) {
+        qDebug() << "link: " << (void*)link;
+        QMenuModel *other = new QMenuModel(const_cast<QMenuModel*>(this), link);
+        qDebug() << "link created: " << (void*)link;
+        return QVariant::fromValue<QObject*>(other);
+    }
+
+    return QVariant();
+}
+
+void QMenuModel::onItemsChanged(GMenuModel *,
+                                gint position,
+                                gint removed,
+                                gint added,
+                                gpointer data)
+{
+    QMenuModel *self = reinterpret_cast<QMenuModel*>(data);
+
+    qDebug() << "model changed" << position << removed << added;
+
+    if (removed > 0) {
+        self->beginRemoveRows(QModelIndex(), position, position + removed - 1);
+        self->endRemoveRows();
+    }
+
+    if (added > 0) {
+        self->beginInsertRows(QModelIndex(), position, position + added - 1);
+        self->endInsertRows();
+    }
+    qDebug() << "model size: " << self->rowCount();
+}
+