161 lines
5.1 KiB
Python
161 lines
5.1 KiB
Python
# Copyright (C) 2019 Philipp Hörist <philipp AT hoerist.com>
|
|
#
|
|
# This file is part of the PGP Gajim Plugin.
|
|
#
|
|
# PGP Gajim Plugin is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published
|
|
# by the Free Software Foundation; version 3 only.
|
|
#
|
|
# PGP Gajim Plugin 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 General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with PGP Gajim Plugin. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from pathlib import Path
|
|
|
|
from gi.repository import Gtk
|
|
from gi.repository import GLib
|
|
|
|
from gajim.common import app
|
|
from gajim.plugins.plugins_i18n import _
|
|
from gajim.plugins.helpers import get_builder
|
|
|
|
|
|
class KeyDialog(Gtk.Dialog):
|
|
def __init__(self, plugin, account, jid, transient):
|
|
super().__init__(title=_('Assign key for %s') % jid,
|
|
destroy_with_parent=True)
|
|
|
|
self.set_transient_for(transient)
|
|
self.set_resizable(True)
|
|
self.set_default_size(450, -1)
|
|
|
|
self._plugin = plugin
|
|
self._jid = jid
|
|
self._client = app.get_client(account)
|
|
|
|
self._label = Gtk.Label()
|
|
|
|
self._assign_button = Gtk.Button(label=_('Assign Key'))
|
|
self._assign_button.get_style_context().add_class('suggested-action')
|
|
self._assign_button.set_halign(Gtk.Align.CENTER)
|
|
self._assign_button.set_margin_top(18)
|
|
self._assign_button.connect('clicked', self._choose_key)
|
|
|
|
box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL)
|
|
box.set_border_width(18)
|
|
box.add(self._label)
|
|
box.add(self._assign_button)
|
|
|
|
area = self.get_content_area()
|
|
area.pack_start(box, True, True, 0)
|
|
|
|
self._load_key()
|
|
self.show_all()
|
|
|
|
def _choose_key(self, *args):
|
|
backend = self._client.get_module('PGPLegacy').pgp_backend
|
|
dialog = ChooseGPGKeyDialog(backend.get_keys(), self)
|
|
dialog.connect('response', self._on_response)
|
|
|
|
def _load_key(self):
|
|
key_data = self._client.get_module('PGPLegacy').get_contact_key_data(
|
|
self._jid)
|
|
if key_data is None:
|
|
self._set_key(None)
|
|
else:
|
|
self._set_key(key_data.values())
|
|
|
|
def _on_response(self, dialog, response):
|
|
if response != Gtk.ResponseType.OK:
|
|
return
|
|
|
|
if dialog.selected_key is None:
|
|
self._client.get_module('PGPLegacy').set_contact_key_data(
|
|
self._jid, None)
|
|
self._set_key(None)
|
|
else:
|
|
self._client.get_module('PGPLegacy').set_contact_key_data(
|
|
self._jid, dialog.selected_key)
|
|
self._set_key(dialog.selected_key)
|
|
|
|
def _set_key(self, key_data):
|
|
if key_data is None:
|
|
self._label.set_text(_('No key assigned'))
|
|
else:
|
|
key_id, key_user = key_data
|
|
self._label.set_markup('<b><tt>%s</tt> %s</b>' % \
|
|
(key_id, GLib.markup_escape_text(key_user)))
|
|
|
|
|
|
class ChooseGPGKeyDialog(Gtk.Dialog):
|
|
def __init__(self, secret_keys, transient_for):
|
|
Gtk.Dialog.__init__(self,
|
|
title=_('Assign PGP Key'),
|
|
transient_for=transient_for)
|
|
|
|
secret_keys[_('None')] = _('None')
|
|
|
|
self.set_position(Gtk.WindowPosition.CENTER_ON_PARENT)
|
|
self.set_resizable(True)
|
|
self.set_default_size(500, 300)
|
|
|
|
self.add_button(_('Cancel'), Gtk.ResponseType.CANCEL)
|
|
self.add_button(_('OK'), Gtk.ResponseType.OK)
|
|
|
|
self._selected_key = None
|
|
|
|
ui_path = Path(__file__).parent
|
|
self._ui = get_builder(ui_path.resolve() / 'choose_key.ui')
|
|
|
|
self._ui.keys_treeview = self._ui.keys_treeview
|
|
|
|
model = self._ui.keys_treeview.get_model()
|
|
model.set_sort_func(1, self._sort)
|
|
|
|
model = self._ui.keys_treeview.get_model()
|
|
for key_id in secret_keys.keys():
|
|
model.append((key_id, secret_keys[key_id]))
|
|
|
|
self.get_content_area().add(self._ui.box)
|
|
|
|
self._ui.connect_signals(self)
|
|
|
|
self.connect_after('response', self._on_response)
|
|
|
|
self.show_all()
|
|
|
|
@property
|
|
def selected_key(self):
|
|
return self._selected_key
|
|
|
|
@staticmethod
|
|
def _sort(model, iter1, iter2, _data):
|
|
value1 = model[iter1][1]
|
|
value2 = model[iter2][1]
|
|
if value1 == _('None'):
|
|
return -1
|
|
if value2 == _('None'):
|
|
return 1
|
|
if value1 < value2:
|
|
return -1
|
|
return 1
|
|
|
|
def _on_response(self, _dialog, _response):
|
|
self.destroy()
|
|
|
|
def _on_row_changed(self, treeview):
|
|
selection = treeview.get_selection()
|
|
model, iter_ = selection.get_selected()
|
|
if iter_ is None:
|
|
self._selected_key = None
|
|
else:
|
|
key_id, key_user = model[iter_][0], model[iter_][1]
|
|
if key_id == _('None'):
|
|
self._selected_key = None
|
|
else:
|
|
self._selected_key = key_id, key_user
|