class NextStage(Page):
def __init__(self):
Page.__init__(self)
self.set_valign(Gtk.Align.FILL)
self.complete = False
self.title = _('Change Password')
self._current_form = None
self.show_all()
def _on_is_valid(self, _widget, is_valid):
self.complete = is_valid
self.get_toplevel().update_page_complete()
def set_form(self, form):
if self._current_form is not None:
self.remove(self._current_form)
self._current_form.destroy()
self._current_form = DataFormWidget(form)
self._current_form.connect('is-valid', self._on_is_valid)
self._current_form.validate()
self.pack_start(self._current_form, True, True, 0)
self._current_form.show_all()
def get_submit_form(self):
return self._current_form.get_submit_form()
def _build_dataform(self, form, is_form):
if not is_form:
return FakeDataFormWidget(form)
dataform = dataforms.extend_form(node=form)
form_widget = DataFormWidget(dataform)
form_widget.connect('is-valid', self._on_is_valid)
form_widget.validate()
return form_widget
class Form(Gtk.Box):
type_ = Gtk.AssistantPageType.CUSTOM
title = _('Search')
complete = True
def __init__(self):
super().__init__(orientation=Gtk.Orientation.VERTICAL)
self.set_spacing(18)
self._dataform_widget = None
self.show_all()
@property
def search_form(self):
return self._dataform_widget.get_submit_form()
def clear(self):
self._show_form(None)
def process_search_form(self, form):
self._show_form(form)
def _show_form(self, form):
if self._dataform_widget is not None:
self.remove(self._dataform_widget)
self._dataform_widget.destroy()
if form is None:
return
options = {'form-width': 350}
form = dataforms.extend_form(node=form)
self._dataform_widget = DataFormWidget(form, options=options)
self._dataform_widget.connect('is-valid', self._on_is_valid)
self._dataform_widget.validate()
self._dataform_widget.show_all()
self.add(self._dataform_widget)
def _on_is_valid(self, _widget, is_valid):
self.get_toplevel().set_stage_complete(is_valid)
def get_submit_form(self):
return self._dataform_widget.get_submit_form()
class GroupchatConfig(Gtk.ApplicationWindow):
def __init__(self, account, jid, own_affiliation, form=None):
Gtk.ApplicationWindow.__init__(self)
self.set_application(app.app)
self.set_position(Gtk.WindowPosition.CENTER)
self.set_show_menubar(False)
self.set_title(_('Group Chat Configuration'))
self._destroyed = False
self.account = account
self.jid = jid
self._own_affiliation = own_affiliation
self._ui = get_builder('groupchat_config.ui')
self.add(self._ui.grid)
# Activate Add button only for Admins and Owners
if self._own_affiliation in ('admin', 'owner'):
self._ui.add_button.set_sensitive(True)
self._ui.add_button.set_tooltip_text('')
visible = muc_caps_cache.supports(jid, nbxmpp.NS_REGISTER)
self._ui.reserved_name_column.set_visible(visible)
self._ui.info_button.set_sensitive(False)
self._form = form
self._affiliations = {}
self._new_affiliations = {}
con = app.connections[self.account]
for affiliation in ('owner', 'admin', 'member', 'outcast'):
con.get_module('MUC').get_affiliation(
self.jid,
affiliation,
callback=self._on_affiliations_received,
user_data=affiliation)
if form is not None:
self._ui.stack.set_visible_child_name('config')
self._data_form_widget = DataFormWidget(form)
self._data_form_widget.connect('is-valid', self._on_is_valid)
self._data_form_widget.validate()
self._ui.config_grid.add(self._data_form_widget)
else:
self._ui.stack.get_child_by_name('config').hide()
self._ui.stack.get_child_by_name('config').set_no_show_all(True)
self._ui.stack.set_visible_child_name('affiliation')
self._ui.connect_signals(self)
self.connect('delete-event', self._cancel)
self.connect('destroy', self._on_destroy)
self.connect('key-press-event', self._on_key_press)
self.show_all()
self._ui.stack.notify('visible-child-name')
def _on_is_valid(self, _widget, is_valid):
self._ui.ok_button.set_sensitive(is_valid)
def _get_current_treeview(self):
page_name = self._ui.stack.get_visible_child_name()
return getattr(self._ui, '%s_treeview' % page_name)
def _on_add(self, *args):
page_name = self._ui.stack.get_visible_child_name()
if page_name == 'outcast':
affiliation_edit, jid_edit = self._allowed_to_edit('outcast')
text = None
affiliation = 'outcast'
else:
affiliation_edit, jid_edit = self._allowed_to_edit('member')
text = _('Member')
affiliation = 'member'
treeview = self._get_current_treeview()
iter_ = treeview.get_model().append([None,
None,
None,
affiliation,
text,
affiliation_edit,
jid_edit])
# Scroll to added row
path = treeview.get_model().get_path(iter_)
treeview.scroll_to_cell(path, None, False, 0, 0)
treeview.get_selection().unselect_all()
treeview.get_selection().select_path(path)
def _on_remove(self, *args):
treeview = self._get_current_treeview()
model, paths = treeview.get_selection().get_selected_rows()
owner_count = self._get_owner_count()
references = []
for path in paths:
if model[path][MUCUser.AFFILIATION] == 'owner':
if owner_count < 2:
# There must be at least one owner
ErrorDialog(_('Error'),
_('A Group Chat needs at least one Owner'))
return
owner_count -= 1
references.append(Gtk.TreeRowReference.new(model, path))
for ref in references:
iter_ = model.get_iter(ref.get_path())
model.remove(iter_)
def _on_jid_edited(self, _renderer, path, new_text):
old_text = self._ui.affiliation_store[path][MUCUser.JID]
if new_text == old_text:
return
if self._jid_exists(new_text):
self._raise_error()
return
self._ui.affiliation_store[path][MUCUser.JID] = new_text
def _on_outcast_jid_edited(self, _renderer, path, new_text):
old_text = self._ui.outcast_store[path][MUCUser.JID]
if new_text == old_text:
return
if self._jid_exists(new_text):
self._raise_error()
return
self._ui.outcast_store[path][MUCUser.JID] = new_text
self._ui.outcast_store[path][MUCUser.AFFILIATION] = 'outcast'
def _on_nick_edited(self, _renderer, path, new_text):
self._ui.affiliation_store[path][MUCUser.NICK] = new_text
def _on_reason_edited(self, _renderer, path, new_text):
self._ui.outcast_store[path][MUCUser.REASON] = new_text
def _on_affiliation_changed(self, cell_renderer_combo,
path_string, new_iter):
combo_store = cell_renderer_combo.get_property('model')
affiliation_text = combo_store.get_value(new_iter, 0)
affiliation = combo_store.get_value(new_iter, 1)
store = self._ui.affiliation_treeview.get_model()
store[path_string][MUCUser.AFFILIATION] = affiliation
store[path_string][MUCUser.AFFILIATION_TEXT] = affiliation_text
def _on_selection_changed(self, tree_selection):
sensitive = bool(tree_selection.count_selected_rows())
selected_affiliations = self._get_selected_affiliations(tree_selection)
self._set_remove_button_state(sensitive, selected_affiliations)
def _jid_exists(self, jid):
stores = [self._ui.affiliation_store, self._ui.outcast_store]
for store in stores:
for row in store:
if row[MUCUser.JID] == jid:
return True
return False
@staticmethod
def _get_selected_affiliations(tree_selection):
model, paths = tree_selection.get_selected_rows()
selected_affiliations = set()
for path in paths:
selected_affiliations.add(model[path][MUCUser.AFFILIATION])
return selected_affiliations
def _on_switch_page(self, stack, _pspec):
page_name = stack.get_visible_child_name()
self._set_button_box_state(page_name)
if page_name == 'config':
return
treeview = getattr(self._ui, '%s_treeview' % page_name)
sensitive = bool(treeview.get_selection().count_selected_rows())
selected_affiliations = self._get_selected_affiliations(
treeview.get_selection())
self._set_remove_button_state(sensitive, selected_affiliations)
def _set_button_box_state(self, page_name):
affiliation = self._own_affiliation in ('admin', 'owner')
page = page_name != 'config'
self._ui.treeview_buttonbox.set_visible(affiliation and page)
self._ui.info_button.set_sensitive(page_name == 'outcast')
def _set_remove_button_state(self, sensitive, selected_affiliations):
if self._own_affiliation not in ('admin', 'owner'):
self._ui.remove_button.set_sensitive(False)
return
self._ui.remove_button.set_tooltip_text('')
if not sensitive:
self._ui.remove_button.set_sensitive(False)
return
if self._own_affiliation == 'owner':
self._ui.remove_button.set_sensitive(True)
return
if set(['owner', 'admin']).intersection(selected_affiliations):
self._ui.remove_button.set_sensitive(False)
self._ui.remove_button.set_tooltip_text(
_('You are not allowed to modify the affiliation '
'of Admins and Owners'))
return
self._ui.remove_button.set_sensitive(True)
def _get_owner_count(self):
owner_count = 0
for row in self._ui.affiliation_store:
if row[MUCUser.AFFILIATION] == 'owner':
owner_count += 1
return owner_count
def _allowed_to_edit(self, affiliation):
if self._own_affiliation == 'owner':
return True, True
if self._own_affiliation == 'admin':
if affiliation in ('admin', 'owner'):
return False, False
return False, True
return False, False
def _on_ok(self, *args):
if self._own_affiliation in ('admin', 'owner'):
self._set_affiliations()
if self._form is not None and self._own_affiliation == 'owner':
form = self._data_form_widget.get_submit_form()
con = app.connections[self.account]
con.get_module('MUC').set_config(self.jid, form)
self.destroy()
def _get_diff(self):
stores = [self._ui.affiliation_store, self._ui.outcast_store]
self._new_affiliations = {}
for store in stores:
for row in store:
if not row[MUCUser.JID]:
# Ignore empty JID field
continue
attr = 'nick'
if row[MUCUser.AFFILIATION] == 'outcast':
attr = 'reason'
self._new_affiliations[row[MUCUser.JID]] = {
'affiliation': row[MUCUser.AFFILIATION],
attr: row[MUCUser.NICK_OR_REASON]}
old_jids = set(self._affiliations.keys())
new_jids = set(self._new_affiliations.keys())
remove = old_jids - new_jids
add = new_jids - old_jids
modified = new_jids - remove - add
return add, remove, modified
def _on_key_press(self, _widget, event):
if event.keyval == Gdk.KEY_Escape:
self._on_cancel()
def _on_cancel(self, *args):
self._cancel()
self.destroy()
def _cancel(self, *args):
if self._form and self._own_affiliation == 'owner':
con = app.connections[self.account]
con.get_module('MUC').cancel_config(self.jid)
def _on_destroy(self, *args):
self._destroyed = True
def _set_affiliations(self):
add, remove, modified = self._get_diff()
diff_dict = {}
for jid in add:
diff_dict[jid] = self._new_affiliations[jid]
for jid in remove:
diff_dict[jid] = {'affiliation': 'none'}
for jid in modified:
if self._new_affiliations[jid] == self._affiliations[jid]:
# Not modified
continue
diff_dict[jid] = self._new_affiliations[jid]
if self._new_affiliations[jid]['affiliation'] == 'outcast':
# New affiliation is outcast, check if the reason changed.
# In case the affiliation was 'admin', 'owner' or 'member'
# before, there is no reason.
new_reason = self._new_affiliations[jid]['reason']
old_reason = self._affiliations[jid].get('reason')
if new_reason == old_reason:
diff_dict[jid].pop('reason', None)
else:
# New affiliation is not outcast, check if the nick has changed.
# In case the affiliation was 'outcast' there is no nick.
new_nick = self._new_affiliations[jid]['nick']
old_nick = self._affiliations[jid].get('nick')
if new_nick == old_nick:
diff_dict[jid].pop('nick', None)
if not diff_dict:
# No changes were made
return
con = app.connections[self.account]
con.get_module('MUC').set_affiliation(self.jid, diff_dict)
@ensure_not_destroyed
def _on_affiliations_received(self, result, affiliation):
if is_error_result(result):
log.info('Error while requesting %s affiliations: %s',
affiliation, result)
return
if affiliation == 'outcast':
self._ui.stack.get_child_by_name('outcast').show()
for jid, attrs in result.users.items():
affiliation_edit, jid_edit = self._allowed_to_edit(affiliation)
if affiliation == 'outcast':
reason = attrs.get('reason')
self._ui.outcast_store.append(
[str(jid),
reason,
None,
affiliation,
None,
affiliation_edit,
jid_edit])
self._affiliations[jid] = {'affiliation': affiliation,
'reason': reason}
else:
nick = attrs.get('nick')
role = attrs.get('role')
self._ui.affiliation_store.append(
[str(jid),
nick,
role,
affiliation,
_(affiliation.capitalize()),
affiliation_edit,
jid_edit])
self._affiliations[jid] = {'affiliation': affiliation,
'nick': nick}
if role is not None:
self._ui.role_column.set_visible(True)
@staticmethod
def _raise_error():
ErrorDialog(_('Error'),
_('An entry with this XMPP Address already exists'))
class Stage(Gtk.Box):
type_ = Gtk.AssistantPageType.CUSTOM
title = _('Settings')
complete = True
def __init__(self):
super().__init__(orientation=Gtk.Orientation.VERTICAL)
self.set_spacing(18)
self._dataform_widget = None
self._notes = []
self._last_stage_data = None
self.show_all()
@property
def stage_data(self):
return self._last_stage_data, self._dataform_widget.get_submit_form()
@property
def actions(self):
return self._last_stage_data.actions
def clear(self):
self._show_form(None)
self._show_notes(None)
self._last_stage_data = None
def process_stage(self, stage_data):
self._last_stage_data = stage_data
self._show_notes(stage_data.notes)
self._show_form(stage_data.data)
def _show_form(self, form):
if self._dataform_widget is not None:
self.remove(self._dataform_widget)
self._dataform_widget.destroy()
if form is None:
return
form = dataforms.extend_form(node=form)
self._dataform_widget = DataFormWidget(form)
self._dataform_widget.connect('is-valid', self._on_is_valid)
self._dataform_widget.validate()
self._dataform_widget.show_all()
self.add(self._dataform_widget)
def _show_notes(self, notes):
for note in self._notes:
self.remove(note)
self._notes = []
if notes is None:
return
for note in notes:
label = Gtk.Label(label=note.text)
label.show()
self._notes.append(label)
self.add(label)
def _on_is_valid(self, _widget, is_valid):
self.get_toplevel().set_stage_complete(is_valid)
