Mercurial > libervia-backend
view sat_frontends/primitivus/xmlui.py @ 3696:52ee22d78e18
plugin XEP-0384: fix encryption for user own devices:
our own jid is added automatically by OMEMO library, but during the generation of the
stanza, our list of jids was used to get the keys (and thus our own jid and devices keys
was missing).
fix 402
author | Goffi <goffi@goffi.org> |
---|---|
date | Fri, 22 Oct 2021 18:46:51 +0200 |
parents | be6d91572633 |
children | 524856bd7b19 |
line wrap: on
line source
#!/usr/bin/env python3 # Primitivus: a SAT frontend # Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org) # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Affero General Public License for more details. # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. from sat.core.i18n import _ import urwid import copy from sat.core import exceptions from urwid_satext import sat_widgets from urwid_satext import files_management from sat.core.log import getLogger log = getLogger(__name__) from sat_frontends.primitivus.constants import Const as C from sat_frontends.primitivus.widget import PrimitivusWidget from sat_frontends.tools import xmlui class PrimitivusEvents(object): """ Used to manage change event of Primitivus widgets """ def _event_callback(self, ctrl, *args, **kwargs): """" Call xmlui callback and ignore any extra argument """ args[-1](ctrl) def _xmluiOnChange(self, callback): """ Call callback with widget as only argument """ urwid.connect_signal(self, "change", self._event_callback, callback) class PrimitivusEmptyWidget(xmlui.EmptyWidget, urwid.Text): def __init__(self, _xmlui_parent): urwid.Text.__init__(self, "") class PrimitivusTextWidget(xmlui.TextWidget, urwid.Text): def __init__(self, _xmlui_parent, value, read_only=False): urwid.Text.__init__(self, value) class PrimitivusLabelWidget(xmlui.LabelWidget, PrimitivusTextWidget): def __init__(self, _xmlui_parent, value): super(PrimitivusLabelWidget, self).__init__(_xmlui_parent, value + ": ") class PrimitivusJidWidget(xmlui.JidWidget, PrimitivusTextWidget): pass class PrimitivusDividerWidget(xmlui.DividerWidget, urwid.Divider): def __init__(self, _xmlui_parent, style="line"): if style == "line": div_char = "─" elif style == "dot": div_char = "·" elif style == "dash": div_char = "-" elif style == "plain": div_char = "█" elif style == "blank": div_char = " " else: log.warning(_("Unknown div_char")) div_char = "─" urwid.Divider.__init__(self, div_char) class PrimitivusStringWidget( xmlui.StringWidget, sat_widgets.AdvancedEdit, PrimitivusEvents ): def __init__(self, _xmlui_parent, value, read_only=False): sat_widgets.AdvancedEdit.__init__(self, edit_text=value) self.read_only = read_only def selectable(self): if self.read_only: return False return super(PrimitivusStringWidget, self).selectable() def _xmluiSetValue(self, value): self.set_edit_text(value) def _xmluiGetValue(self): return self.get_edit_text() class PrimitivusJidInputWidget(xmlui.JidInputWidget, PrimitivusStringWidget): pass class PrimitivusPasswordWidget( xmlui.PasswordWidget, sat_widgets.Password, PrimitivusEvents ): def __init__(self, _xmlui_parent, value, read_only=False): sat_widgets.Password.__init__(self, edit_text=value) self.read_only = read_only def selectable(self): if self.read_only: return False return super(PrimitivusPasswordWidget, self).selectable() def _xmluiSetValue(self, value): self.set_edit_text(value) def _xmluiGetValue(self): return self.get_edit_text() class PrimitivusTextBoxWidget( xmlui.TextBoxWidget, sat_widgets.AdvancedEdit, PrimitivusEvents ): def __init__(self, _xmlui_parent, value, read_only=False): sat_widgets.AdvancedEdit.__init__(self, edit_text=value, multiline=True) self.read_only = read_only def selectable(self): if self.read_only: return False return super(PrimitivusTextBoxWidget, self).selectable() def _xmluiSetValue(self, value): self.set_edit_text(value) def _xmluiGetValue(self): return self.get_edit_text() class PrimitivusBoolWidget(xmlui.BoolWidget, urwid.CheckBox, PrimitivusEvents): def __init__(self, _xmlui_parent, state, read_only=False): urwid.CheckBox.__init__(self, "", state=state) self.read_only = read_only def selectable(self): if self.read_only: return False return super(PrimitivusBoolWidget, self).selectable() def _xmluiSetValue(self, value): self.set_state(value == "true") def _xmluiGetValue(self): return C.BOOL_TRUE if self.get_state() else C.BOOL_FALSE class PrimitivusIntWidget(xmlui.IntWidget, sat_widgets.AdvancedEdit, PrimitivusEvents): def __init__(self, _xmlui_parent, value, read_only=False): sat_widgets.AdvancedEdit.__init__(self, edit_text=value) self.read_only = read_only def selectable(self): if self.read_only: return False return super(PrimitivusIntWidget, self).selectable() def _xmluiSetValue(self, value): self.set_edit_text(value) def _xmluiGetValue(self): return self.get_edit_text() class PrimitivusButtonWidget( xmlui.ButtonWidget, sat_widgets.CustomButton, PrimitivusEvents ): def __init__(self, _xmlui_parent, value, click_callback): sat_widgets.CustomButton.__init__(self, value, on_press=click_callback) def _xmluiOnClick(self, callback): urwid.connect_signal(self, "click", callback) class PrimitivusListWidget(xmlui.ListWidget, sat_widgets.List, PrimitivusEvents): def __init__(self, _xmlui_parent, options, selected, flags): sat_widgets.List.__init__(self, options=options, style=flags) self._xmluiSelectValues(selected) def _xmluiSelectValue(self, value): return self.selectValue(value) def _xmluiSelectValues(self, values): return self.selectValues(values) def _xmluiGetSelectedValues(self): return [option.value for option in self.getSelectedValues()] def _xmluiAddValues(self, values, select=True): current_values = self.getAllValues() new_values = copy.deepcopy(current_values) for value in values: if value not in current_values: new_values.append(value) if select: selected = self._xmluiGetSelectedValues() self.changeValues(new_values) if select: for value in values: if value not in selected: selected.append(value) self._xmluiSelectValues(selected) class PrimitivusJidsListWidget(xmlui.ListWidget, sat_widgets.List, PrimitivusEvents): def __init__(self, _xmlui_parent, jids, styles): sat_widgets.List.__init__( self, options=jids + [""], # the empty field is here to add new jids if needed option_type=lambda txt, align: sat_widgets.AdvancedEdit( edit_text=txt, align=align ), on_change=self._onChange, ) self.delete = 0 def _onChange(self, list_widget, jid_widget=None, text=None): if jid_widget is not None: if jid_widget != list_widget.contents[-1] and not text: # if a field is empty, we delete the line (except for the last line) list_widget.contents.remove(jid_widget) elif jid_widget == list_widget.contents[-1] and text: # we always want an empty field as last value to be able to add jids list_widget.contents.append(sat_widgets.AdvancedEdit()) def _xmluiGetSelectedValues(self): # XXX: there is not selection in this list, so we return all non empty values return [jid_ for jid_ in self.getAllValues() if jid_] class PrimitivusAdvancedListContainer( xmlui.AdvancedListContainer, sat_widgets.TableContainer, PrimitivusEvents ): def __init__(self, _xmlui_parent, columns, selectable="no"): options = {"ADAPT": ()} if selectable != "no": options["HIGHLIGHT"] = () sat_widgets.TableContainer.__init__( self, columns=columns, options=options, row_selectable=selectable != "no" ) def _xmluiAppend(self, widget): self.addWidget(widget) def _xmluiAddRow(self, idx): self.setRowIndex(idx) def _xmluiGetSelectedWidgets(self): return self.getSelectedWidgets() def _xmluiGetSelectedIndex(self): return self.getSelectedIndex() def _xmluiOnSelect(self, callback): """ Call callback with widget as only argument """ urwid.connect_signal(self, "click", self._event_callback, callback) class PrimitivusPairsContainer(xmlui.PairsContainer, sat_widgets.TableContainer): def __init__(self, _xmlui_parent): options = {"ADAPT": (0,), "HIGHLIGHT": (0,)} if self._xmlui_main.type == "param": options["FOCUS_ATTR"] = "param_selected" sat_widgets.TableContainer.__init__(self, columns=2, options=options) def _xmluiAppend(self, widget): if isinstance(widget, PrimitivusEmptyWidget): # we don't want highlight on empty widgets widget = urwid.AttrMap(widget, "default") self.addWidget(widget) class PrimitivusLabelContainer(PrimitivusPairsContainer, xmlui.LabelContainer): pass class PrimitivusTabsContainer(xmlui.TabsContainer, sat_widgets.TabsContainer): def __init__(self, _xmlui_parent): sat_widgets.TabsContainer.__init__(self) def _xmluiAppend(self, widget): self.body.append(widget) def _xmluiAddTab(self, label, selected): tab = PrimitivusVerticalContainer(None) self.addTab(label, tab, selected) return tab class PrimitivusVerticalContainer(xmlui.VerticalContainer, urwid.ListBox): BOX_HEIGHT = 5 def __init__(self, _xmlui_parent): urwid.ListBox.__init__(self, urwid.SimpleListWalker([])) self._last_size = None def _xmluiAppend(self, widget): if "flow" not in widget.sizing(): widget = urwid.BoxAdapter(widget, self.BOX_HEIGHT) self.body.append(widget) def render(self, size, focus=False): if size != self._last_size: (maxcol, maxrow) = size if self.body: widget = self.body[0] if isinstance(widget, urwid.BoxAdapter): widget.height = maxrow self._last_size = size return super(PrimitivusVerticalContainer, self).render(size, focus) ### Dialogs ### class PrimitivusDialog(object): def __init__(self, _xmlui_parent): self.host = _xmlui_parent.host def _xmluiShow(self): self.host.showPopUp(self) def _xmluiClose(self): self.host.removePopUp(self) class PrimitivusMessageDialog(PrimitivusDialog, xmlui.MessageDialog, sat_widgets.Alert): def __init__(self, _xmlui_parent, title, message, level): PrimitivusDialog.__init__(self, _xmlui_parent) xmlui.MessageDialog.__init__(self, _xmlui_parent) sat_widgets.Alert.__init__( self, title, message, ok_cb=lambda __: self._xmluiClose() ) class PrimitivusNoteDialog(xmlui.NoteDialog, PrimitivusMessageDialog): # TODO: separate NoteDialog pass class PrimitivusConfirmDialog( PrimitivusDialog, xmlui.ConfirmDialog, sat_widgets.ConfirmDialog ): def __init__(self, _xmlui_parent, title, message, level, buttons_set): PrimitivusDialog.__init__(self, _xmlui_parent) xmlui.ConfirmDialog.__init__(self, _xmlui_parent) sat_widgets.ConfirmDialog.__init__( self, title, message, no_cb=lambda __: self._xmluiCancelled(), yes_cb=lambda __: self._xmluiValidated(), ) class PrimitivusFileDialog( PrimitivusDialog, xmlui.FileDialog, files_management.FileDialog ): def __init__(self, _xmlui_parent, title, message, level, filetype): # TODO: message is not managed yet PrimitivusDialog.__init__(self, _xmlui_parent) xmlui.FileDialog.__init__(self, _xmlui_parent) style = [] if filetype == C.XMLUI_DATA_FILETYPE_DIR: style.append("dir") files_management.FileDialog.__init__( self, ok_cb=lambda path: self._xmluiValidated({"path": path}), cancel_cb=lambda __: self._xmluiCancelled(), message=message, title=title, style=style, ) class GenericFactory(object): def __getattr__(self, attr): if attr.startswith("create"): cls = globals()[ "Primitivus" + attr[6:] ] # XXX: we prefix with "Primitivus" to work around an Urwid bug, WidgetMeta in Urwid don't manage multiple inheritance with same names return cls class WidgetFactory(GenericFactory): def __getattr__(self, attr): if attr.startswith("create"): cls = GenericFactory.__getattr__(self, attr) cls._xmlui_main = self._xmlui_main return cls class XMLUIPanel(xmlui.XMLUIPanel, PrimitivusWidget): widget_factory = WidgetFactory() def __init__( self, host, parsed_xml, title=None, flags=None, callback=None, ignore=None, whitelist=None, profile=C.PROF_KEY_NONE, ): self.widget_factory._xmlui_main = self self._dest = None xmlui.XMLUIPanel.__init__( self, host, parsed_xml, title=title, flags=flags, callback=callback, ignore=ignore, profile=profile, ) PrimitivusWidget.__init__(self, self.main_cont, self.xmlui_title) def _parseChilds(self, _xmlui_parent, current_node, wanted=("container",), data=None): # Small hack to always have a VerticalContainer as main container in Primitivus. # this used to be the default behaviour for all frontends, but now # TabsContainer can also be the main container. if _xmlui_parent is self: node = current_node.childNodes[0] if node.nodeName == "container" and node.getAttribute("type") == "tabs": _xmlui_parent = self.widget_factory.createVerticalContainer(self) self.main_cont = _xmlui_parent return super(XMLUIPanel, self)._parseChilds(_xmlui_parent, current_node, wanted, data) def constructUI(self, parsed_dom): def postTreat(): assert self.main_cont.body if self.type in ("form", "popup"): buttons = [] if self.type == "form": buttons.append(urwid.Button(_("Submit"), self.onFormSubmitted)) if not "NO_CANCEL" in self.flags: buttons.append(urwid.Button(_("Cancel"), self.onFormCancelled)) else: buttons.append( urwid.Button(_("OK"), on_press=lambda __: self._xmluiClose()) ) max_len = max([len(button.get_label()) for button in buttons]) grid_wid = urwid.GridFlow(buttons, max_len + 4, 1, 0, "center") self.main_cont.body.append(grid_wid) elif self.type == "param": tabs_cont = self.main_cont.body[0].base_widget assert isinstance(tabs_cont, sat_widgets.TabsContainer) buttons = [] buttons.append(sat_widgets.CustomButton(_("Save"), self.onSaveParams)) buttons.append( sat_widgets.CustomButton( _("Cancel"), lambda x: self.host.removeWindow() ) ) max_len = max([button.getSize() for button in buttons]) grid_wid = urwid.GridFlow(buttons, max_len, 1, 0, "center") tabs_cont.addFooter(grid_wid) xmlui.XMLUIPanel.constructUI(self, parsed_dom, postTreat) urwid.WidgetWrap.__init__(self, self.main_cont) def show(self, show_type=None, valign="middle"): """Show the constructed UI @param show_type: how to show the UI: - None (follow XMLUI's recommendation) - 'popup' - 'window' @param valign: vertical alignment when show_type is 'popup'. Ignored when show_type is 'window'. """ if show_type is None: if self.type in ("window", "param"): show_type = "window" elif self.type in ("popup", "form"): show_type = "popup" if show_type not in ("popup", "window"): raise ValueError("Invalid show_type [%s]" % show_type) self._dest = show_type if show_type == "popup": self.host.showPopUp(self, valign=valign) elif show_type == "window": self.host.newWidget(self, user_action=self.user_action) else: assert False self.host.redraw() def _xmluiClose(self): if self._dest == "window": self.host.removeWindow() elif self._dest == "popup": self.host.removePopUp(self) else: raise exceptions.InternalError( "self._dest unknown, are you sure you have called XMLUI.show ?" ) class XMLUIDialog(xmlui.XMLUIDialog): dialog_factory = GenericFactory() xmlui.registerClass(xmlui.CLASS_PANEL, XMLUIPanel) xmlui.registerClass(xmlui.CLASS_DIALOG, XMLUIDialog) create = xmlui.create