9578053 Jan 22 2022 distfiles.gentoo.org/distfiles/gajim-1.3.3-2.tar.gz

gajim/gtk/filechoosers.py

@@ -0,0 +1,221 @@
+# Copyright (C) 2018 Philipp Hörist <philipp AT hoerist.com>
+#
+# This file is part of Gajim.
+#
+# Gajim 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, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Gajim 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 Gajim. If not, see <http://www.gnu.org/licenses/>.
+
+from typing import List
+
+import os
+import sys
+from pathlib import Path
+
+from gi.repository import Gtk
+from gi.repository import GdkPixbuf
+from gi.repository import GLib
+
+from gajim.common import app
+from gajim.common.i18n import _
+
+from .const import Filter
+
+
+def _require_native() -> bool:
+    if app.is_flatpak():
+        return True
+    if sys.platform in ('win32', 'darwin'):
+        return True
+    return False
+
+
+# Notes: Adding mime types to Gtk.FileFilter forces non-native dialogs
+
+class BaseFileChooser:
+    def _on_response(self, dialog, response, accept_cb, cancel_cb):
+        if response == Gtk.ResponseType.ACCEPT:
+            if self.get_select_multiple():
+                accept_cb(dialog.get_filenames())
+            else:
+                accept_cb(dialog.get_filename())
+
+        if response in (Gtk.ResponseType.CANCEL,
+                        Gtk.ResponseType.DELETE_EVENT):
+            if cancel_cb is not None:
+                cancel_cb()
+
+    def _add_filters(self, filters):
+        for filterinfo in filters:
+            filter_ = Gtk.FileFilter()
+            filter_.set_name(filterinfo.name)
+            if isinstance(filterinfo.pattern, list):
+                for mime_type in filterinfo.pattern:
+                    filter_.add_mime_type(mime_type)
+            else:
+                filter_.add_pattern(filterinfo.pattern)
+            self.add_filter(filter_)
+            if filterinfo.default:
+                self.set_filter(filter_)
+
+    def _update_preview(self, filechooser):
+        path_to_file = filechooser.get_preview_filename()
+        preview = filechooser.get_preview_widget()
+        if path_to_file is None or os.path.isdir(path_to_file):
+            # nothing to preview
+            preview.clear()
+            return
+        try:
+            pixbuf = GdkPixbuf.Pixbuf.new_from_file_at_size(
+                path_to_file, *self._preivew_size)
+        except GLib.GError:
+            preview.clear()
+            return
+        filechooser.get_preview_widget().set_from_pixbuf(pixbuf)
+
+
+class BaseFileOpenDialog:
+
+    _title = _('Choose File to Send…')
+    _filters = [Filter(_('All files'), '*', True)]
+
+
+class BaseAvatarChooserDialog:
+
+    _title = _('Choose Avatar…')
+    _preivew_size = (100, 100)
+
+    if _require_native():
+        _filters = [Filter(_('PNG files'), '*.png', True),
+                    Filter(_('JPEG files'), '*.jp*g', False),
+                    Filter(_('SVG files'), '*.svg', False)]
+    else:
+        _filters = [Filter(_('Images'), ['image/png',
+                                         'image/jpeg',
+                                         'image/svg+xml'], True)]
+
+
+class NativeFileChooserDialog(Gtk.FileChooserNative, BaseFileChooser):
+
+    _title = ''
+    _filters: List[Filter] = []
+    _action = Gtk.FileChooserAction.OPEN
+
+    def __init__(self, accept_cb, cancel_cb=None, transient_for=None,
+                 path=None, file_name=None, select_multiple=False,
+                 modal=False):
+
+        if transient_for is None:
+            transient_for = app.app.get_active_window()
+
+        Gtk.FileChooserNative.__init__(self,
+                                       title=self._title,
+                                       action=self._action,
+                                       transient_for=transient_for)
+
+        self.set_current_folder(path or str(Path.home()))
+        if file_name is not None:
+            self.set_current_name(file_name)
+        self.set_select_multiple(select_multiple)
+        self.set_do_overwrite_confirmation(True)
+        self.set_modal(modal)
+        self._add_filters(self._filters)
+
+        self.connect('response', self._on_response, accept_cb, cancel_cb)
+        self.show()
+
+
+class ArchiveChooserDialog(NativeFileChooserDialog):
+
+    _title = _('Choose Archive')
+    _filters = [Filter(_('All files'), '*', False),
+                Filter(_('ZIP files'), '*.zip', True)]
+
+
+class FileSaveDialog(NativeFileChooserDialog):
+
+    _title = _('Save File as…')
+    _filters = [Filter(_('All files'), '*', True)]
+    _action = Gtk.FileChooserAction.SAVE
+
+
+class NativeFileOpenDialog(BaseFileOpenDialog, NativeFileChooserDialog):
+    pass
+
+
+class NativeAvatarChooserDialog(BaseAvatarChooserDialog, NativeFileChooserDialog):
+    pass
+
+
+class GtkFileChooserDialog(Gtk.FileChooserDialog, BaseFileChooser):
+
+    _title = ''
+    _filters: List[Filter] = []
+    _action = Gtk.FileChooserAction.OPEN
+    _preivew_size = (200, 200)
+
+    def __init__(self, accept_cb, cancel_cb=None, transient_for=None,
+                 path=None, file_name=None, select_multiple=False,
+                 preview=True, modal=False):
+
+        if transient_for is None:
+            transient_for = app.app.get_active_window()
+
+        Gtk.FileChooserDialog.__init__(
+            self,
+            title=self._title,
+            action=self._action,
+            transient_for=transient_for)
+
+        self.add_button(_('_Cancel'), Gtk.ResponseType.CANCEL)
+        open_button = self.add_button(_('_Open'), Gtk.ResponseType.ACCEPT)
+        open_button.get_style_context().add_class('suggested-action')
+
+        self.set_current_folder(path or str(Path.home()))
+        if file_name is not None:
+            self.set_current_name(file_name)
+        self.set_select_multiple(select_multiple)
+        self.set_do_overwrite_confirmation(True)
+        self.set_modal(modal)
+        self._add_filters(self._filters)
+
+        if preview:
+            self.set_use_preview_label(False)
+            self.set_preview_widget(Gtk.Image())
+            self.connect('selection-changed', self._update_preview)
+
+        self.connect('response', self._on_response, accept_cb, cancel_cb)
+        self.show()
+
+    def _on_response(self, *args):
+        super()._on_response(*args)
+        self.destroy()
+
+
+class GtkFileOpenDialog(BaseFileOpenDialog, GtkFileChooserDialog):
+    pass
+
+
+class GtkAvatarChooserDialog(BaseAvatarChooserDialog, GtkFileChooserDialog):
+    pass
+
+
+def FileChooserDialog(*args, **kwargs):
+    if _require_native():
+        return NativeFileOpenDialog(*args, **kwargs)
+    return GtkFileOpenDialog(*args, **kwargs)
+
+
+def AvatarChooserDialog(*args, **kwargs):
+    if _require_native():
+        return NativeAvatarChooserDialog(*args, **kwargs)
+    return GtkAvatarChooserDialog(*args, **kwargs)