Files
cartridges/src/preferences.py
2023-04-03 18:17:53 +02:00

324 lines
10 KiB
Python

# preferences.py
#
# Copyright 2022-2023 kramo
#
# This program 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.
#
# 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# SPDX-License-Identifier: GPL-3.0-or-later
import os
from pathlib import Path
from gi.repository import Adw, Gio, GLib, Gtk
from .create_dialog import create_dialog
from .get_games import get_games
from .save_game import save_game
class ImportPreferences:
def __init__(
self,
window,
source_id,
name,
install_key,
paths,
expander_row,
file_chooser_button,
config=False,
):
def set_dir(_source, result, _unused):
try:
path = Path(window.file_chooser.select_folder_finish(result).get_path())
def response(widget, response):
if response == "choose_folder":
window.choose_folder(widget, set_dir)
if not any((path / current_path).exists() for current_path in paths):
create_dialog(
window.parent_widget,
_("Installation Not Found"),
# The variable is the name of the game launcher
_("Select the {} configuration directory.").format(name)
if config
# The variable is the name of the game launcher
else _("Select the {} data directory.").format(name),
"choose_folder",
_("Set Location"),
).connect("response", response)
else:
window.schema.set_string(
install_key,
str(path),
)
except GLib.GError:
pass
window.schema.bind(
source_id,
expander_row,
"enable-expansion",
Gio.SettingsBindFlags.DEFAULT,
)
file_chooser_button.connect("clicked", window.choose_folder, set_dir)
@Gtk.Template(resource_path="/hu/kramo/Cartridges/gtk/preferences.ui")
class PreferencesWindow(Adw.PreferencesWindow):
__gtype_name__ = "PreferencesWindow"
general_page = Gtk.Template.Child()
import_page = Gtk.Template.Child()
sources_group = Gtk.Template.Child()
exit_after_launch_switch = Gtk.Template.Child()
cover_launches_game_switch = Gtk.Template.Child()
high_quality_images_switch = Gtk.Template.Child()
remove_all_games_button = Gtk.Template.Child()
steam_expander_row = Gtk.Template.Child()
steam_file_chooser_button = Gtk.Template.Child()
steam_extra_file_chooser_button = Gtk.Template.Child()
steam_clear_button_revealer = Gtk.Template.Child()
steam_clear_button = Gtk.Template.Child()
lutris_expander_row = Gtk.Template.Child()
lutris_file_chooser_button = Gtk.Template.Child()
lutris_cache_file_chooser_button = Gtk.Template.Child()
heroic_expander_row = Gtk.Template.Child()
heroic_file_chooser_button = Gtk.Template.Child()
heroic_epic_switch = Gtk.Template.Child()
heroic_gog_switch = Gtk.Template.Child()
heroic_sideloaded_switch = Gtk.Template.Child()
bottles_expander_row = Gtk.Template.Child()
bottles_file_chooser_button = Gtk.Template.Child()
itch_expander_row = Gtk.Template.Child()
itch_file_chooser_button = Gtk.Template.Child()
def __init__(self, parent_widget, **kwargs):
super().__init__(**kwargs)
self.schema = parent_widget.schema
self.parent_widget = parent_widget
self.file_chooser = Gtk.FileDialog()
self.set_transient_for(parent_widget)
self.toast = Adw.Toast.new(_("All games removed"))
self.toast.set_button_label(_("Undo"))
self.toast.connect("button-clicked", self.undo_remove_all, None)
self.toast.set_priority(Adw.ToastPriority.HIGH)
shortcut_controller = Gtk.ShortcutController()
shortcut_controller.add_shortcut(
Gtk.Shortcut.new(
Gtk.ShortcutTrigger.parse_string("<primary>z"),
Gtk.CallbackAction.new(self.undo_remove_all),
)
)
self.add_controller(shortcut_controller)
self.removed_games = []
# General
self.schema.bind(
"exit-after-launch",
self.exit_after_launch_switch,
"active",
Gio.SettingsBindFlags.DEFAULT,
)
self.schema.bind(
"cover-launches-game",
self.cover_launches_game_switch,
"active",
Gio.SettingsBindFlags.DEFAULT,
)
self.schema.bind(
"high-quality-images",
self.high_quality_images_switch,
"active",
Gio.SettingsBindFlags.DEFAULT,
)
self.remove_all_games_button.connect("clicked", self.remove_all_games)
# Steam
ImportPreferences(
self,
"steam",
"Steam",
"steam-location",
[
"steamapps",
Path("steam") / "steamapps",
Path("Steam") / "steamapps",
],
self.steam_expander_row,
self.steam_file_chooser_button,
)
def update_revealer():
if self.schema.get_strv("steam-extra-dirs"):
self.steam_clear_button_revealer.set_reveal_child(True)
else:
self.steam_clear_button_revealer.set_reveal_child(False)
def add_steam_dir(_source, result, _unused):
try:
value = self.schema.get_strv("steam-extra-dirs")
value.append(self.file_chooser.select_folder_finish(result).get_path())
self.schema.set_strv("steam-extra-dirs", value)
except GLib.GError:
pass
update_revealer()
def clear_steam_dirs(*_unused):
self.schema.set_strv("steam-extra-dirs", [])
update_revealer()
update_revealer()
self.steam_extra_file_chooser_button.connect(
"clicked", self.choose_folder, add_steam_dir
)
self.steam_clear_button.connect("clicked", clear_steam_dirs)
# Lutris
ImportPreferences(
self,
"lutris",
"Lutris",
"lutris-location",
["pga.db"],
self.lutris_expander_row,
self.lutris_file_chooser_button,
)
def set_cache_dir(_source, result, _unused):
try:
path = Path(self.file_chooser.select_folder_finish(result).get_path())
def response(widget, response):
if response == "choose_folder":
self.choose_folder(widget, set_cache_dir)
if not (path / "coverart").exists():
create_dialog(
self.parent_widget,
_("Cache Not Found"),
_("Select the Lutris cache directory."),
"choose_folder",
_("Set Location"),
).connect("response", response)
else:
self.schema.set_string(
"lutris-cache-location",
str(path),
)
except GLib.GError:
pass
self.lutris_cache_file_chooser_button.connect(
"clicked", self.choose_folder, set_cache_dir
)
if os.name == "nt":
self.sources_group.remove(self.lutris_expander_row)
# Heroic
ImportPreferences(
self,
"heroic",
"Heroic",
"heroic-location",
["config.json"],
self.heroic_expander_row,
self.heroic_file_chooser_button,
True,
)
self.schema.bind(
"heroic-import-epic",
self.heroic_epic_switch,
"active",
Gio.SettingsBindFlags.DEFAULT,
)
self.schema.bind(
"heroic-import-gog",
self.heroic_gog_switch,
"active",
Gio.SettingsBindFlags.DEFAULT,
)
self.schema.bind(
"heroic-import-sideload",
self.heroic_sideloaded_switch,
"active",
Gio.SettingsBindFlags.DEFAULT,
)
# Bottles
ImportPreferences(
self,
"bottles",
"Bottles",
"bottles-location",
["library.yml"],
self.bottles_expander_row,
self.bottles_file_chooser_button,
)
if os.name == "nt":
self.sources_group.remove(self.bottles_expander_row)
# itch
ImportPreferences(
self,
"itch",
"itch",
"itch-location",
[Path("db") / "butler.db"],
self.itch_expander_row,
self.itch_file_chooser_button,
True,
)
def choose_folder(self, _widget, function):
self.file_chooser.select_folder(self.parent_widget, None, function, None)
def undo_remove_all(self, _widget, _unused):
for game_id in self.removed_games:
data = get_games(self.parent_widget, [game_id])[game_id]
if "removed" in data:
data.pop("removed")
save_game(self.parent_widget, data)
self.parent_widget.update_games(self.removed_games)
self.removed_games = []
self.toast.dismiss()
def remove_all_games(self, _widget):
for game in get_games(self.parent_widget).values():
if "removed" not in game:
self.removed_games.append(game["game_id"])
game["removed"] = True
save_game(self.parent_widget, game)
self.parent_widget.update_games(self.parent_widget.games)
if self.parent_widget.stack.get_visible_child() == self.parent_widget.overview:
self.parent_widget.on_go_back_action(None, None)
self.add_toast(self.toast)