From 38061a63977af659ec1e20ad2c23975ae0655c5b Mon Sep 17 00:00:00 2001
From: mathieui <mathieui@mathieui.net>
Date: Sat, 5 Apr 2014 14:19:22 +0200
Subject: Improve /bookmark *

- /bookmark{_local,} * now keeps the order of the tabs, and puts the
  currently non-opened tabs at the end of the list.
- also fix a bug in remote bookmark saving
---
 src/bookmark.py |  4 ++--
 src/core.py     | 25 ++++++++++++++++++-------
 2 files changed, 20 insertions(+), 9 deletions(-)

(limited to 'src')

diff --git a/src/bookmark.py b/src/bookmark.py
index 52235d23..fa013d1e 100644
--- a/src/bookmark.py
+++ b/src/bookmark.py
@@ -133,7 +133,7 @@ def save_privatexml(xmpp):
 
 def save_remote(xmpp, method=preferred):
     """Save the remote bookmarks."""
-    method = "privatexml" if method != 'pep'  else 'pep'
+    method = 'privatexml' if method != 'pep' else 'pep'
 
     try:
         if method is 'privatexml':
@@ -144,7 +144,7 @@ def save_remote(xmpp, method=preferred):
                     method='xep_0223')
     except:
         import traceback
-        log.debug("Could not save the bookmarks:\n%s" % traceback.format_exc())
+        log.error("Could not save the bookmarks:\n%s" % traceback.format_exc())
         return False
     return True
 
diff --git a/src/core.py b/src/core.py
index af5b3992..053c03b0 100644
--- a/src/core.py
+++ b/src/core.py
@@ -1973,21 +1973,26 @@ class Core(object):
         args = common.shell_split(arg)
         nick = None
         password = None
-        if len(args) == 0 and not isinstance(self.current_tab(), tabs.MucTab):
+        if not args and not isinstance(self.current_tab(), tabs.MucTab):
             return
-        if len(args) == 0:
+        if not args:
             tab = self.current_tab()
             roomname = tab.get_name()
             if tab.joined and tab.own_nick != self.own_nick:
                 nick = tab.own_nick
         elif args[0] == '*':
+            new_bookmarks = []
             for tab in self.get_tabs(tabs.MucTab):
                 b = bookmark.get_by_jid(tab.get_name())
                 if not b:
                     b = bookmark.Bookmark(tab.get_name(), autojoin=True, method="local")
-                    bookmark.bookmarks.append(b)
+                    new_bookmarks.append(b)
                 else:
                     b.method = "local"
+                    new_bookmarks.append(b)
+                    bookmark.bookmarks.remove(b)
+            new_bookmarks.extend(bookmark.bookmarks)
+            bookmark.bookmarks = new_bookmarks
             bookmark.save_local()
             bookmark.save_remote(self.xmpp)
             self.information('Bookmarks added and saved.', 'Info')
@@ -2058,9 +2063,9 @@ class Core(object):
             return
         args = common.shell_split(arg)
         nick = None
-        if len(args) == 0 and not isinstance(self.current_tab(), tabs.MucTab):
+        if not args and not isinstance(self.current_tab(), tabs.MucTab):
             return
-        if len(args) == 0:
+        if not args:
             tab = self.current_tab()
             roomname = tab.get_name()
             if tab.joined:
@@ -2072,15 +2077,21 @@ class Core(object):
                 autojoin = False if args[1].lower() != 'true' else True
             else:
                 autojoin = True
+            new_bookmarks = []
             for tab in self.get_tabs(tabs.MucTab):
                 b = bookmark.get_by_jid(tab.get_name())
                 if not b:
                     b = bookmark.Bookmark(tab.get_name(), autojoin=autojoin,
                             method=bookmark.preferred)
-                    bookmark.bookmarks.append(b)
+                    new_bookmarks.append(b)
                 else:
                     b.method = bookmark.preferred
-            if bookmark.save_remote(self.xmpp, self):
+                    bookmark.bookmarks.remove(b)
+                    new_bookmarks.append(b)
+            new_bookmarks.extend(bookmark.bookmarks)
+            bookmark.bookmarks = new_bookmarks
+
+            if bookmark.save_remote(self.xmpp):
                 bookmark.save_local()
                 self.information("Bookmarks added.", "Info")
             else:
-- 
cgit v1.2.3