Add sorting options to nyaa.si

1 files changed, 11 insertions, 7 deletions

diff --git a/src/plugins/Matrix.cpp b/src/plugins/Matrix.cpp
index 17bc17b..7439bf6 100644
--- a/src/plugins/Matrix.cpp
+++ b/src/plugins/Matrix.cpp
@@ -81,13 +81,14 @@ namespace QuickMedia {
         }
     }
 
-    static void remove_body_item_by_url(BodyItems &body_items, const std::string &url) {
-        for(auto it = body_items.begin(); it != body_items.end();) {
-            if((*it)->url == url)
-                it = body_items.erase(it);
-            else
-                ++it;
+    static bool remove_body_item_by_url(BodyItems &body_items, const std::string &url) {
+        for(auto it = body_items.begin(); it != body_items.end(); ++it) {
+            if((*it)->url == url) {
+                body_items.erase(it);
+                return true;
+            }
         }
+        return false;
     }
 
     static int color_hash_code(const std::string &str) {
@@ -873,7 +874,10 @@ namespace QuickMedia {
     }
 
     void MatrixInvitesPage::remove_body_item_by_room_id(const std::string &room_id) {
-        remove_body_item_by_url(body->items, room_id);
+        if(remove_body_item_by_url(body->items, room_id)) {
+            prev_invite_count = body->items.size();
+            title = "Invites (" + std::to_string(body->items.size()) + ")";
+        }
     }
 
     void MatrixInvitesPage::clear_data() {