From 6123b27315795ae9b41c8243314322206ff7ced9 Mon Sep 17 00:00:00 2001
From: mailchuck <dev@mailchuck.com>
Date: Sat, 31 Oct 2015 18:42:44 +0100
Subject: [PATCH] removeInboxRowByMsgid fixes

Now undertands messagelists and has better performance.
Haven't tested it yet though because it's triggered through API.
I think it fixes #85
---
 src/bitmessageqt/__init__.py | 31 ++++++++++++++++++++++---------
 1 file changed, 22 insertions(+), 9 deletions(-)

diff --git a/src/bitmessageqt/__init__.py b/src/bitmessageqt/__init__.py
index 740a7bc3..b9d95876 100644
--- a/src/bitmessageqt/__init__.py
+++ b/src/bitmessageqt/__init__.py
@@ -1862,15 +1862,28 @@ class MyForm(QtGui.QMainWindow):
                     self.ui.tableWidgetInbox.item(i, 3).setText(textToDisplay)
 
     def removeInboxRowByMsgid(self, msgid):  # msgid and inventoryHash are the same thing
-        inbox = self.getCurrentMessagelist()
-        for i in range(inbox.rowCount()):
-            if msgid == str(inbox.item(i, 3).data(Qt.UserRole).toPyObject()):
-                self.statusBar().showMessage(_translate(
-                    "MainWindow", "Message trashed"))
-                inbox.removeRow(i)
-                break
-        # this is a callback from core, not initiated by UI. We don't care about performance
-        self.propagateUnreadCount(None, None, None, 0)
+        def widgetConvert (tableWidget):
+            if tableWidget == self.ui.tableWidgetInbox:
+                return self.ui.treeWidgetYourIdentities
+            elif tableWidget == self.ui.tableWidgetInboxSubscriptions:
+                return self.ui.treeWidgetSubscriptions
+            elif tableWidget == self.ui.tableWidgetInboxChans:
+                return self.ui.treeWidgetChans
+            else:
+                return None
+                
+        for inbox in ([
+            self.ui.tableWidgetInbox,
+            self.ui.tableWidgetInboxSubscriptions,
+            self.ui.tableWidgetInboxChans]):
+            for i in range(inbox.rowCount()):
+                if msgid == str(inbox.item(i, 3).data(Qt.UserRole).toPyObject()):
+                    self.statusBar().showMessage(_translate(
+                        "MainWindow", "Message trashed"))
+                    treeWidget = widgetConvert(inbox)
+                    self.propagateUnreadCount(self.getCurrentAccount(treeWidget), self.getCurrentFolder(treeWidget), treeWidget, 0)
+                    inbox.removeRow(i)
+                    break
         
     def newVersionAvailable(self, version):
 #        if (not (self.windowState() & QtCore.Qt.WindowActive)) or (self.windowState() & QtCore.Qt.WindowMinimized):