summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authormathieui <mathieui@mathieui.net>2014-03-19 01:04:51 +0100
committermathieui <mathieui@mathieui.net>2014-03-19 01:04:51 +0100
commitbe74128cb78ebd9c8316816f04632acd12396579 (patch)
tree6dbd4d2d484f24c9c5e06f03769771228a11ede1 /src
parent16c57cdae9a2c4dabe3f602ca4e4699c18ebb487 (diff)
downloadpoezio-be74128cb78ebd9c8316816f04632acd12396579.tar.gz
poezio-be74128cb78ebd9c8316816f04632acd12396579.tar.bz2
poezio-be74128cb78ebd9c8316816f04632acd12396579.tar.xz
poezio-be74128cb78ebd9c8316816f04632acd12396579.zip
small refactor
- add a get_tabs(cls) method to avoid manual filtering - move some stuff to avoid cyclic dependency hell between tab classes
Diffstat (limited to 'src')
-rw-r--r--src/core.py111
-rw-r--r--src/tabs.py59
2 files changed, 97 insertions, 73 deletions
diff --git a/src/core.py b/src/core.py
index af0c6348..7a635d48 100644
--- a/src/core.py
+++ b/src/core.py
@@ -732,9 +732,8 @@ class Core(object):
parts of the client (for example, set the MucTabs as not joined, etc)
"""
msg = msg or ''
- for tab in self.tabs:
- if isinstance(tab, tabs.MucTab):
- tab.command_part(msg)
+ for tab in self.get_tabs(tabs.MucTab):
+ tab.command_part(msg)
self.xmpp.disconnect()
if reconnect:
self.xmpp.start()
@@ -781,6 +780,10 @@ class Core(object):
### Tab getters ###
+ def get_tabs(self, cls=tabs.Tab):
+ "Get all the tabs of a type"
+ return filter(lambda tab: isinstance(tab, cls), self.tabs)
+
def current_tab(self):
"""
returns the current room, the one we are viewing
@@ -1037,11 +1040,11 @@ class Core(object):
Open a Private conversation in a MUC and focus if needed.
"""
complete_jid = room_name+'/'+user_nick
- for tab in self.tabs: # if the room exists, focus it and return
- if isinstance(tab, tabs.PrivateTab):
- if tab.get_name() == complete_jid:
- self.command_win('%s' % tab.nb)
- return tab
+ # if the room exists, focus it and return
+ for tab in self.get_tabs(tabs.PrivateTab):
+ if tab.get_name() == complete_jid:
+ self.command_win('%s' % tab.nb)
+ return tab
# create the new tab
tab = self.get_tab_by_name(room_name, tabs.MucTab)
if not tab:
@@ -1105,16 +1108,16 @@ class Core(object):
"""
Disable private tabs when leaving a room
"""
- for tab in self.tabs:
- if isinstance(tab, tabs.PrivateTab) and tab.get_name().startswith(room_name):
+ for tab in self.get_tabs(tabs.PrivateTab):
+ if tab.get_name().startswith(room_name):
tab.deactivate(reason=reason)
def enable_private_tabs(self, room_name, reason='\x195}You joined the chatroom\x193}'):
"""
Enable private tabs when joining a room
"""
- for tab in self.tabs:
- if tab.get_name().startswith(room_name) and isinstance(tab, tabs.PrivateTab):
+ for tab in self.get_tabs(tabs.PrivateTab):
+ if tab.get_name().startswith(room_name):
tab.activate(reason=reason)
def on_user_changed_status_in_private(self, jid, msg):
@@ -1766,9 +1769,8 @@ class Core(object):
def completion_list(self, the_input):
"""Completion for /list"""
muc_serv_list = []
- for tab in self.tabs: # TODO, also from an history
- if isinstance(tab, tabs.MucTab) and\
- tab.get_name() not in muc_serv_list:
+ for tab in self.get_tabs(tabs.MucTab): # TODO, also from an history
+ if tab.get_name() not in muc_serv_list:
muc_serv_list.append(safeJID(tab.get_name()).server)
if muc_serv_list:
return the_input.new_completion(muc_serv_list, 1, quotify=False)
@@ -1925,11 +1927,10 @@ class Core(object):
relevant_rooms = []
relevant_rooms.extend(sorted(self.pending_invites.keys()))
bookmarks = {str(elem.jid): False for elem in bookmark.bookmarks}
- for tab in self.tabs:
- if isinstance(tab, tabs.MucTab):
- name = tab.get_name()
- if name in bookmarks and not tab.joined:
- bookmarks[name] = True
+ for tab in self.get_tabs(tabs.MucTab):
+ name = tab.get_name()
+ if name in bookmarks and not tab.joined:
+ bookmarks[name] = True
relevant_rooms.extend(sorted(room[0] for room in bookmarks.items() if room[1]))
if the_input.last_completion:
@@ -1953,8 +1954,8 @@ class Core(object):
elif jid.user:
# we are writing the server: complete the server
serv_list = []
- for tab in self.tabs:
- if isinstance(tab, tabs.MucTab) and tab.joined:
+ for tab in self.get_tabs(tabs.MucTab):
+ if tab.joined:
serv_list.append('%s@%s'% (jid.user, safeJID(tab.get_name()).host))
serv_list.extend(relevant_rooms)
return the_input.new_completion(serv_list, 1, quotify=True)
@@ -1980,14 +1981,13 @@ class Core(object):
if tab.joined and tab.own_nick != self.own_nick:
nick = tab.own_nick
elif args[0] == '*':
- for tab in self.tabs:
- if isinstance(tab, 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)
- else:
- b.method = "local"
+ 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)
+ else:
+ b.method = "local"
bookmark.save_local()
bookmark.save_remote(self.xmpp)
self.information('Bookmarks added and saved.', 'Info')
@@ -2044,7 +2044,7 @@ class Core(object):
nicks.append(nick)
jids_list = ['%s/%s' % (jid.bare, nick) for nick in nicks]
return the_input.new_completion(jids_list, 1, quotify=True)
- muc_list = [tab.get_name() for tab in self.tabs if isinstance(tab, tabs.MucTab)]
+ muc_list = [tab.get_name() for tab in self.get_tabs(tabs.MucTab)]
muc_list.append('*')
return the_input.new_completion(muc_list, 1, quotify=True)
@@ -2072,15 +2072,14 @@ class Core(object):
autojoin = False if args[1].lower() != 'true' else True
else:
autojoin = True
- for tab in self.tabs:
- if isinstance(tab, 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)
- else:
- b.method = bookmark.preferred
+ 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)
+ else:
+ b.method = bookmark.preferred
if bookmark.save_remote(self.xmpp, self):
bookmark.save_local()
self.information("Bookmarks added.", "Info")
@@ -2146,7 +2145,7 @@ class Core(object):
nicks.append(nick)
jids_list = ['%s/%s' % (jid.bare, nick) for nick in nicks]
return the_input.new_completion(jids_list, 1, quotify=True)
- muc_list = [tab.get_name() for tab in self.tabs if isinstance(tab, tabs.MucTab)]
+ muc_list = [tab.get_name() for tab in self.get_tabs(tabs.MucTab)]
muc_list.sort()
muc_list.append('*')
return the_input.new_completion(muc_list, 1, quotify=True)
@@ -2279,8 +2278,8 @@ class Core(object):
else:
self.information(_("No server specified"), "Error")
return
- for tab in self.tabs:
- if isinstance(tab, tabs.MucTab) and tab.get_name().endswith(domain):
+ for tab in self.get_tabs(tabs.MucTab):
+ if tab.get_name().endswith(domain):
if tab.joined:
muc.leave_groupchat(tab.core.xmpp, tab.get_name(), tab.own_nick, message)
tab.joined = False
@@ -2292,10 +2291,9 @@ class Core(object):
def completion_server_cycle(self, the_input):
"""Completion for /server_cycle"""
serv_list = set()
- for tab in self.tabs:
- if isinstance(tab, tabs.MucTab):
- serv = safeJID(tab.get_name()).server
- serv_list.add(serv)
+ for tab in self.get_tabs(tabs.MucTab):
+ serv = safeJID(tab.get_name()).server
+ serv_list.add(serv)
return the_input.new_completion(sorted(serv_list), 1, ' ')
def command_last_activity(self, arg):
@@ -2421,8 +2419,8 @@ class Core(object):
return the_input.new_completion(sorted(jid for jid in roster.jids()), n, quotify=True)
elif n == 2:
rooms = []
- for tab in self.tabs:
- if isinstance(tab, tabs.MucTab) and tab.joined:
+ for tab in self.get_tabs(tabs.MucTab):
+ if tab.joined:
rooms.append(tab.get_name())
rooms.sort()
return the_input.new_completion(rooms, n, '', quotify=True)
@@ -2924,8 +2922,8 @@ class Core(object):
return
# Differentiate both type of messages, and call the appropriate handler.
jid_from = message['from']
- for tab in self.tabs:
- if tab.get_name() == jid_from.bare and isinstance(tab, tabs.MucTab):
+ for tab in self.get_tabs(tabs.MucTab):
+ if tab.get_name() == jid_from.bare:
if message['type'] == 'error':
return self.room_error(message, jid_from)
else:
@@ -3238,6 +3236,12 @@ class Core(object):
if config.get_by_tabname('disable_beep', 'false', room_from, False).lower() != 'true':
curses.beep()
+ def on_muc_own_nickchange(self, muc):
+ "We changed our nick in a MUC"
+ for tab in self.get_tabs(tabs.PrivateTab):
+ if tab.parent_muc == muc:
+ tab.own_nick = muc.own_nick
+
def on_groupchat_private_message(self, message):
"""
We received a Private Message (from someone in a Muc)
@@ -3552,9 +3556,8 @@ class Core(object):
When we are disconnected from remote server
"""
roster.modified()
- for tab in self.tabs:
- if isinstance(tab, tabs.MucTab):
- tab.disconnect()
+ for tab in self.get_tabs(tabs.MucTab):
+ tab.disconnect()
self.information(_("Disconnected from server."))
def on_failed_auth(self, event):
diff --git a/src/tabs.py b/src/tabs.py
index 979f451f..c81573c6 100644
--- a/src/tabs.py
+++ b/src/tabs.py
@@ -108,10 +108,7 @@ class Tab(object):
def __init__(self):
self.input = None
- if isinstance(self, MucTab) and not self.joined:
- self._state = 'disconnected'
- else:
- self._state = 'normal'
+ self._state = 'normal'
self.need_resize = False
self.need_resize = False
@@ -467,15 +464,19 @@ class ChatTab(Tab):
# Get the logs
log_nb = config.get('load_log', 10)
+ logs = self.load_logs(log_nb)
- if isinstance(self, PrivateTab):
- logs = logger.get_logs(safeJID(self.get_name()).full.replace('/', '\\'), log_nb)
- else:
- logs = logger.get_logs(safeJID(self.get_name()).bare, log_nb)
if logs:
for message in logs:
self._text_buffer.add_message(**message)
+ @property
+ def is_muc(self):
+ return False
+
+ def load_logs(self, log_nb):
+ logs = logger.get_logs(safeJID(self.get_name()).bare, log_nb)
+
def log_message(self, txt, nickname, time=None, typ=1):
"""
Log the messages in the archives.
@@ -535,6 +536,11 @@ class ChatTab(Tab):
""""
/xhtml <custom xhtml>
"""
+ message = self.generate_xhtml_message(arg)
+ if message:
+ message.send()
+
+ def generate_xhtml_message(self, arg):
if not arg:
return
try:
@@ -551,12 +557,7 @@ class ChatTab(Tab):
msg['body'] = body
msg.enable('html')
msg['html']['body'] = arg
- if isinstance(self, MucTab):
- msg['type'] = 'groupchat'
- if isinstance(self, ConversationTab):
- self.core.add_message_to_text_buffer(self._text_buffer, body, None, self.core.own_nick)
- self.refresh()
- msg.send()
+ return msg
def get_dest_jid(self):
return self.get_name()
@@ -573,7 +574,7 @@ class ChatTab(Tab):
"""
Send an empty chatstate message
"""
- if not isinstance(self, MucTab) or self.joined:
+ if not self.is_muc or self.joined:
if state in ('active', 'inactive', 'gone') and self.inactive and not always_send:
return
if config.get_by_tabname('send_chat_states', 'true', self.general_jid, True) and \
@@ -703,6 +704,8 @@ class MucTab(ChatTab):
def __init__(self, jid, nick):
self.joined = False
ChatTab.__init__(self, jid)
+ if self.joined == False:
+ self._state = 'disconnected'
self.own_nick = nick
self.name = jid
self.users = []
@@ -822,6 +825,10 @@ class MucTab(ChatTab):
return self.get_name()
@property
+ def is_muc(self):
+ return True
+
+ @property
def last_connection(self):
last_message = self._text_buffer.last_message
if last_message:
@@ -1288,6 +1295,12 @@ class MucTab(ChatTab):
msg.send()
self.chat_state = needed
+ def command_xhtml(self, arg):
+ message = self.generate_xhtml_message(arg)
+ if message:
+ message['type'] = 'groupchat'
+ message.send()
+
def command_ignore(self, arg):
"""
/ignore <nick>
@@ -1575,10 +1588,8 @@ class MucTab(ChatTab):
new_nick = presence.find('{%s}x/{%s}item' % (NS_MUC_USER, NS_MUC_USER)).attrib['nick']
if user.nick == self.own_nick:
self.own_nick = new_nick
- # also change our nick in all private discussion of this room
- for _tab in self.core.tabs:
- if isinstance(_tab, PrivateTab) and safeJID(_tab.get_name()).bare == self.name:
- _tab.own_nick = new_nick
+ # also change our nick in all private discussions of this room
+ self.core.on_muc_own_nickchange(self)
user.change_nick(new_nick)
color = dump_tuple(user.color) if config.get_by_tabname('display_user_color_in_join_part', '', self.general_jid, True) == 'true' else 3
self.add_message('\x19%(color)s}%(old)s\x19%(info_col)s} is now known as \x19%(color)s}%(new)s' % {
@@ -1918,6 +1929,9 @@ class PrivateTab(ChatTab):
def remove_information_element(plugin_name):
del PrivateTab.additional_informations[plugin_name]
+ def load_logs(self, log_nb):
+ logs = logger.get_logs(safeJID(self.get_name()).full.replace('/', '\\'), log_nb)
+
def log_message(self, txt, nickname, time=None, typ=1):
"""
Log the messages in the archives.
@@ -3228,6 +3242,13 @@ class ConversationTab(ChatTab):
self.text_win.refresh()
self.input.refresh()
+ def command_xhtml(self, arg):
+ message = self.generate_xhtml_message(arg)
+ if message:
+ message.send()
+ self.core.add_message_to_text_buffer(self._text_buffer, body, None, self.core.own_nick)
+ self.refresh()
+
def command_last_activity(self, arg):
"""
/activity [jid]