diff options
Diffstat (limited to 'triggers/triggers.py')
| -rw-r--r-- | triggers/triggers.py | 279 |
1 files changed, 126 insertions, 153 deletions
diff --git a/triggers/triggers.py b/triggers/triggers.py index 844f422..aa85386 100644 --- a/triggers/triggers.py +++ b/triggers/triggers.py @@ -35,28 +35,125 @@ class Triggers(GajimPlugin): self.events_handlers = { 'notification': (ged.PREGUI, self._on_notification), - 'decrypted-message-received': (ged.PREGUI2, - self._on_message_received), - 'gc-message-received': (ged.PREGUI2, self._on_gc_message_received), + 'message-received': (ged.PREGUI2, self._on_message_received), + 'gc-message-received': (ged.PREGUI2, self._on_message_received), 'presence-received': (ged.PREGUI, self._on_presence_received), } - def _check_rule_recipients(self, obj, rule): + def _on_notification(self, event): + self._check_all(event, + self._check_rule_apply_notification, + self._apply_rule) + + def _on_message_received(self, event): + self._check_all(event, + self._check_rule_apply_msg_received, + self._apply_rule) + + def _on_presence_received(self, event): + if event.old_show < 2 and event.new_show > 1: + check_func = self._check_rule_apply_connected + elif event.old_show > 1 and event.new_show < 2: + check_func = self._check_rule_apply_disconnected + else: + check_func = self._check_rule_apply_status_changed + self._check_all(event, check_func, self._apply_rule) + + def _check_all(self, event, check_func, apply_func): + # check rules in order + rules_num = [int(item) for item in self.config.keys()] + rules_num.sort() + to_remove = [] + for num in rules_num: + rule = self.config[str(num)] + if check_func(event, rule): + apply_func(event, rule) + if 'one_shot' in rule and rule['one_shot']: + to_remove.append(num) + # Should we stop after first valid rule ? + # break + + decal = 0 + num = 0 + while str(num) in self.config: + if num + decal in to_remove: + num2 = num + while str(num2 + 1) in self.config: + self.config[str(num2)] = self.config[str(num2 + 1)].copy() + num2 += 1 + del self.config[str(num2)] + decal += 1 + else: + num += 1 + + def _check_rule_apply_msg_received(self, event, rule): + return self._check_rule_all('message_received', event, rule) + + def _check_rule_apply_connected(self, event, rule): + return self._check_rule_all('contact_connected', event, rule) + + def _check_rule_apply_disconnected(self, event, rule): + return self._check_rule_all('contact_disconnected', event, rule) + + def _check_rule_apply_status_changed(self, event, rule): + return self._check_rule_all('contact_status_change', event, rule) + + def _check_rule_apply_notification(self, event, rule): + # Check notification type + notif_type = '' + if event.notif_type == 'incoming-message': + notif_type = 'message_received' + if event.notif_type == 'pres': + # TODO: + if (event.base_event.old_show < 2 and + event.base_event.new_show > 1): + notif_type = 'contact_connected' + elif (event.base_event.old_show > 1 and + event.base_event.new_show < 2): + notif_type = 'contact_disconnected' + else: + notif_type = 'contact_status_change' + + return self._check_rule_all(notif_type, event, rule) + + def _check_rule_all(self, notif_type, event, rule): + # Check notification type + if rule['event'] != notif_type: + return False + + # notification type is ok. Now check recipient + if not self._check_rule_recipients(event, rule): + return False + + # recipient is ok. Now check our status + if not self._check_rule_status(event, rule): + return False + + # our_status is ok. Now check opened chat window + if not self._check_rule_tab_opened(event, rule): + return False + + # tab_opened is ok. Now check opened chat window + if not self._check_rule_has_focus(event, rule): + return False + + # All is ok + return True + + def _check_rule_recipients(self, event, rule): rule_recipients = [t.strip() for t in rule['recipients'].split(',')] if rule['recipient_type'] == 'groupchat': - if obj.jid in rule_recipients: + if event.jid in rule_recipients: return True return False - if (rule['recipient_type'] == 'contact' and obj.jid not in + if (rule['recipient_type'] == 'contact' and event.jid not in rule_recipients): return False - contact = app.contacts.get_first_contact_from_jid( - obj.conn.name, obj.jid) - if not contact: # PM? - return False - contact_groups = contact.groups + client = app.get_client(event.account) + contact = client.get_module('Contacts').get_contact(event.jid) + group_found = False - for group in contact_groups: + for group in contact.groups: if group in rule_recipients: group_found = True break @@ -65,19 +162,19 @@ class Triggers(GajimPlugin): return True - def _check_rule_status(self, obj, rule): + def _check_rule_status(self, event, rule): rule_statuses = rule['status'].split() - our_status = app.connections[obj.conn.name].status + our_status = app.connections[event.account].status if rule['status'] != 'all' and our_status not in rule_statuses: return False return True - def _check_rule_tab_opened(self, obj, rule): + def _check_rule_tab_opened(self, event, rule): if rule['tab_opened'] == 'both': return True tab_opened = False - if app.interface.msg_win_mgr.get_control(obj.jid, obj.conn.name): + if app.window.get_control(event.account, event.jid): tab_opened = True if tab_opened and rule['tab_opened'] == 'no': return False @@ -86,17 +183,17 @@ class Triggers(GajimPlugin): return True - def _check_rule_has_focus(self, obj, rule): + def _check_rule_has_focus(self, event, rule): if rule['has_focus'] == 'both': return True if rule['tab_opened'] == 'no': # Does not apply in this case return True - ctrl = app.interface.msg_win_mgr.get_control(obj.jid, obj.conn.name) + ctrl = app.window.get_control(event.account, event.jid) if not ctrl: # Does not apply in this case return True - has_focus = ctrl.parent_win.window.has_focus() + has_focus = ctrl.has_focus() if has_focus and rule['has_focus'] == 'no': return False elif not has_focus and rule['has_focus'] == 'yes': @@ -104,142 +201,18 @@ class Triggers(GajimPlugin): return True - def _check_rule_all(self, event, obj, rule): - # Check notification type - if rule['event'] != event: - return False - - # notification type is ok. Now check recipient - if not self._check_rule_recipients(obj, rule): - return False - - # recipient is ok. Now check our status - if not self._check_rule_status(obj, rule): - return False - - # our_status is ok. Now check opened chat window - if not self._check_rule_tab_opened(obj, rule): - return False - - # tab_opened is ok. Now check opened chat window - if not self._check_rule_has_focus(obj, rule): - return False - - # All is ok - return True - - def _check_rule_apply_notification(self, obj, rule): - # Check notification type - notif_type = '' - if obj.notif_type in ('msg', 'gc-msg'): - notif_type = 'message_received' - elif obj.notif_type == 'pres': - if obj.base_event.old_show < 2 and obj.base_event.new_show > 1: - notif_type = 'contact_connected' - elif obj.base_event.old_show > 1 and obj.base_event.new_show < 2: - notif_type = 'contact_disconnected' - else: - notif_type = 'contact_status_change' - - return self._check_rule_all(notif_type, obj, rule) - - def _check_rule_apply_msg_received(self, obj, rule): - return self._check_rule_all('message_received', obj, rule) - - def _check_rule_apply_connected(self, obj, rule): - return self._check_rule_all('contact_connected', obj, rule) - - def _check_rule_apply_disconnected(self, obj, rule): - return self._check_rule_all('contact_disconnected', obj, rule) - - def _check_rule_apply_status_changed(self, obj, rule): - return self._check_rule_all('contact_status_change', obj, rule) - - def _apply_rule_notification(self, obj, rule): + def _apply_rule(self, event, rule): if rule['sound'] == 'no': - obj.do_sound = False + event.sound = None + event.sound_file = None elif rule['sound'] == 'yes': - obj.do_sound = True - obj.sound_event = '' - obj.sound_file = rule['sound_file'] - - if rule['popup'] == 'no' or obj.control_focused: - obj.do_popup = False - elif rule['popup'] == 'yes': - obj.do_popup = True + event.sound = None + event.sound_file = rule['sound_file'] if rule['run_command']: - obj.do_command = True - obj.command = rule['command'] - else: - obj.do_command = False - - if rule['systray'] == 'no': - obj.show_in_notification_area = False - elif rule['systray'] == 'yes': - obj.show_in_notification_area = True - - if rule['roster'] == 'no': - obj.show_in_roster = False - elif rule['roster'] == 'yes': - obj.show_in_roster = True - - def _apply_rule_message_received(self, obj, rule): - if rule['auto_open'] == 'no': - obj.popup = False - elif rule['auto_open'] == 'yes': - obj.popup = True - - def _apply_rule_presence_received(self, obj, rule): - if rule['auto_open'] == 'no': - obj.popup = False - elif rule['auto_open'] == 'yes': - obj.popup = True - - def _check_all(self, obj, check_func, apply_func): - # check rules in order - rules_num = [int(i) for i in self.config.keys()] - rules_num.sort() - to_remove = [] - for num in rules_num: - rule = self.config[str(num)] - if check_func(obj, rule): - apply_func(obj, rule) - if 'one_shot' in rule and rule['one_shot']: - to_remove.append(num) - # Should we stop after first valid rule ? - # break + event.command = rule['command'] - decal = 0 - num = 0 - while str(num) in self.config: - if num + decal in to_remove: - num2 = num - while str(num2 + 1) in self.config: - self.config[str(num2)] = self.config[str(num2 + 1)].copy() - num2 += 1 - del self.config[str(num2)] - decal += 1 - else: - num += 1 - - def _on_notification(self, obj): - self._check_all(obj, self._check_rule_apply_notification, - self._apply_rule_notification) - - def _on_message_received(self, obj): - self._check_all(obj, self._check_rule_apply_msg_received, - self._apply_rule_message_received) - - def _on_gc_message_received(self, obj): - self._check_all(obj, self._check_rule_apply_msg_received, - self._apply_rule_message_received) - - def _on_presence_received(self, obj): - if obj.old_show < 2 and obj.new_show > 1: - check_func = self._check_rule_apply_connected - elif obj.old_show > 1 and obj.new_show < 2: - check_func = self._check_rule_apply_disconnected - else: - check_func = self._check_rule_apply_status_changed - self._check_all(obj, check_func, self._apply_rule_presence_received) + if rule['popup'] == 'no': + event.show_notification = False + elif rule['popup'] == 'yes': + event.show_notification = True |