From ef48d081dfd0e817c4959dbbd49929ae760a310e Mon Sep 17 00:00:00 2001
From: tv <tv@shackspace.de>
Date: Sat, 31 Jan 2015 15:51:57 +0100
Subject: Notmuch: replace {un,}setTag by notmuchTag

---
 test5.hs | 30 ++++++++++++++++++++----------
 1 file changed, 20 insertions(+), 10 deletions(-)

(limited to 'test5.hs')

diff --git a/test5.hs b/test5.hs
index b9ca124..4133e76 100644
--- a/test5.hs
+++ b/test5.hs
@@ -422,18 +422,28 @@ toggleFold q@State{..} =
 
 toggleTagAtCursor :: Tag -> State -> IO State
 toggleTagAtCursor tag q@State{..} = case Z.label cursor of
+
+    TVSearchResult sr -> do
+        let tagOp =
+                if tag `elem` Notmuch.searchTags sr
+                    then DelTag
+                    else AddTag
+            tagOps = [tagOp tag]
+        Notmuch.notmuchTag tagOps sr
+        -- TODO reload or patch whole thread
+        let cursor' = Z.modifyTree (patchRootLabelTags tagOps) cursor
+        return q { cursor = cursor' }
+
     TVMessage m -> do
         -- TODO modify search result tags
-        -- TODO check Notmuch.{set,unset}Tag result
-        if tag `elem` Notmuch.messageTags m
-            then do
-                Notmuch.unsetTag (T.unpack tag) (Notmuch.unMessageID $ Notmuch.messageId m)
-                let cursor' = Z.modifyTree (patchRootLabelTags [DelTag tag]) cursor
-                return q { cursor = cursor' }
-            else do
-                Notmuch.setTag (T.unpack tag) (Notmuch.unMessageID $ Notmuch.messageId m)
-                let cursor' = Z.modifyTree (patchRootLabelTags [AddTag tag]) cursor
-                return q { cursor = cursor' }
+        let tagOp =
+                if tag `elem` Notmuch.messageTags m
+                    then DelTag
+                    else AddTag
+            tagOps = [tagOp tag]
+        Notmuch.notmuchTag tagOps m
+        let cursor' = Z.modifyTree (patchRootLabelTags tagOps) cursor
+        return q { cursor = cursor' }
 
     _ -> return q { flashMessage = "nothing happened" }
 
-- 
cgit v1.2.3