4 files changed, 76 insertions, 3 deletions

diff --git a/common/domain/applicationdomaintype.h b/common/domain/applicationdomaintype.h
index e0a6de0..b583a6e 100644
--- a/common/domain/applicationdomaintype.h
+++ b/common/domain/applicationdomaintype.h
@@ -97,6 +97,15 @@ private:
     qint64 mRevision;
 };
 
+/*
+ * Should this be specific to the synclistresultset, in other cases we may want to take revision and resource into account.
+ */
+inline bool operator==(const ApplicationDomainType& lhs, const ApplicationDomainType& rhs)
+{
+    return lhs.identifier() == rhs.identifier()
+            && lhs.resourceInstanceIdentifier() == rhs.resourceInstanceIdentifier();
+}
+
 struct Event : public ApplicationDomainType {
     typedef QSharedPointer<Event> Ptr;
     using ApplicationDomainType::ApplicationDomainType;
diff --git a/common/entitystorage.h b/common/entitystorage.h
index 78dbcda..6b09cad 100644
--- a/common/entitystorage.h
+++ b/common/entitystorage.h
@@ -118,10 +118,11 @@ public:
                 break;
             case Akonadi2::Operation_Modification:
                 Trace() << "Got modification";
-                resultProvider->add(copy(*value).template staticCast<DomainType>());
+                resultProvider->modify(copy(*value).template staticCast<DomainType>());
                 break;
             case Akonadi2::Operation_Removal:
                 Trace() << "Got removal";
+                resultProvider->remove(copy(*value).template staticCast<DomainType>());
                 break;
             }
             return true;
diff --git a/common/resultprovider.h b/common/resultprovider.h
index 841fd01..bc03152 100644
--- a/common/resultprovider.h
+++ b/common/resultprovider.h
@@ -81,6 +81,28 @@ public:
         });
     }
 
+    void modify(const T &value)
+    {
+        //Because I don't know how to use bind
+        auto weakEmitter = mResultEmitter;
+        callInMainThreadOnEmitter([weakEmitter, value](){
+            if (auto strongRef = weakEmitter.toStrongRef()) {
+                strongRef->modifyHandler(value);
+            }
+        });
+    }
+
+    void remove(const T &value)
+    {
+        //Because I don't know how to use bind
+        auto weakEmitter = mResultEmitter;
+        callInMainThreadOnEmitter([weakEmitter, value](){
+            if (auto strongRef = weakEmitter.toStrongRef()) {
+                strongRef->removeHandler(value);
+            }
+        });
+    }
+
     void initialResultSetComplete()
     {
         callInMainThreadOnEmitter(&ResultEmitter<T>::initialResultSetComplete);
@@ -176,15 +198,27 @@ public:
     {
         addHandler = handler;
     }
-    // void onRemoved(const std::function<void(const T&)> &handler);
+
+    void onModified(const std::function<void(const DomainType&)> &handler)
+    {
+        modifyHandler = handler;
+    }
+
+    void onRemoved(const std::function<void(const DomainType&)> &handler)
+    {
+        removeHandler = handler;
+    }
+
     void onInitialResultSetComplete(const std::function<void(void)> &handler)
     {
         initialResultSetCompleteHandler = handler;
     }
+
     void onComplete(const std::function<void(void)> &handler)
     {
         completeHandler = handler;
     }
+
     void onClear(const std::function<void(void)> &handler)
     {
         clearHandler = handler;
@@ -195,6 +229,16 @@ public:
         addHandler(value);
     }
 
+    void modify(const DomainType &value)
+    {
+        modifyHandler(value);
+    }
+
+    void remove(const DomainType &value)
+    {
+        removeHandler(value);
+    }
+
     void initialResultSetComplete()
     {
         initialResultSetCompleteHandler();
@@ -214,7 +258,8 @@ private:
     friend class ResultProvider<DomainType>;
 
     std::function<void(const DomainType&)> addHandler;
-    // std::function<void(const T&)> removeHandler;
+    std::function<void(const DomainType&)> modifyHandler;
+    std::function<void(const DomainType&)> removeHandler;
     std::function<void(void)> initialResultSetCompleteHandler;
     std::function<void(void)> completeHandler;
     std::function<void(void)> clearHandler;
diff --git a/common/synclistresult.h b/common/synclistresult.h
index 865d3e0..3ae2bed 100644
--- a/common/synclistresult.h
+++ b/common/synclistresult.h
@@ -4,6 +4,7 @@
 #include <functional>
 #include <QSharedPointer>
 #include <QEventLoop>
+#include <QDebug>
 #include "resultprovider.h"
 
 namespace async {
@@ -25,6 +26,23 @@ public:
         emitter->onAdded([this](const T &value) {
             this->append(value);
         });
+        emitter->onModified([this](const T &value) {
+            for (auto it = this->begin(); it != this->end(); it++) {
+                if (**it == *value) {
+                    it = this->erase(it);
+                    this->insert(it, value);
+                    break;
+                }
+            }
+        });
+        emitter->onRemoved([this](const T &value) {
+            for (auto it = this->begin(); it != this->end(); it++) {
+                if (**it == *value) {
+                    this->erase(it);
+                    break;
+                }
+            }
+        });
         emitter->onInitialResultSetComplete([this]() {
             if (eventLoopAborter) {
                 eventLoopAborter();