guake-guake/guake/theme.py
2021-09-21 17:46:01 +08:00

119 lines
3.6 KiB
Python

import itertools
import logging
import os
from pathlib import Path
import gi
gi.require_version("Gtk", "3.0")
from gi.repository import GLib
from gi.repository import Gdk
from gi.repository import Gtk
from textwrap import dedent
from guake.paths import GUAKE_THEME_DIR
log = logging.getLogger(__name__)
# Reference:
# https://gitlab.gnome.org/GNOME/gnome-tweaks/blob/master/gtweak/utils.py (GPL)
def get_resource_dirs(resource):
"""Returns a list of all known resource dirs for a given resource.
:param str resource:
Name of the resource (e.g. "themes")
:return:
A list of resource dirs
"""
dirs = [
os.path.join(dir, resource)
for dir in itertools.chain(
GLib.get_system_data_dirs(), GUAKE_THEME_DIR, GLib.get_user_data_dir()
)
]
dirs += [os.path.join(os.path.expanduser("~"), f".{resource}")]
return [Path(dir) for dir in dirs if os.path.isdir(dir)]
def list_all_themes():
return sorted(
{
x.name
for theme_dir in get_resource_dirs("themes")
for x in theme_dir.iterdir()
if x.is_dir()
}
)
def select_gtk_theme(settings):
gtk_settings = Gtk.Settings.get_default()
if settings.general.get_boolean("gtk-use-system-default-theme"):
log.debug("Using system default theme")
gtk_settings.reset_property("gtk-theme-name")
gtk_settings.set_property("gtk-application-prefer-dark-theme", False)
return
gtk_theme_name = settings.general.get_string("gtk-theme-name")
log.debug("Wanted GTK theme: %r", gtk_theme_name)
gtk_settings.set_property("gtk-theme-name", gtk_theme_name)
prefer_dark_theme = settings.general.get_boolean("gtk-prefer-dark-theme")
log.debug("Prefer dark theme: %r", prefer_dark_theme)
gtk_settings.set_property("gtk-application-prefer-dark-theme", prefer_dark_theme)
def get_gtk_theme(settings):
gtk_theme_name = settings.general.get_string("gtk-theme-name")
prefer_dark_theme = settings.general.get_boolean("gtk-prefer-dark-theme")
return (gtk_theme_name, "dark" if prefer_dark_theme else None)
def patch_gtk_theme(style_context, settings):
theme_name, variant = get_gtk_theme(settings)
def rgba_to_hex(color):
return f"#{''.join(f'{int(i*255):02x}' for i in (color.red, color.green, color.blue))}"
# for n in [
# "inverted_bg_color",
# "inverted_fg_color",
# "selected_bg_color",
# "selected_fg_color",
# "theme_inverted_bg_color",
# "theme_inverted_fg_color",
# "theme_selected_bg_color",
# "theme_selected_fg_color",
# ]:
# s = style_context.lookup_color(n)
# print(n, s, rgba_to_hex(s[1]))
selected_fg_color = rgba_to_hex(style_context.lookup_color("theme_selected_fg_color")[1])
selected_bg_color = rgba_to_hex(style_context.lookup_color("theme_selected_bg_color")[1])
log.debug(
"Patching theme '%s' (prefer dark = '%r'), overriding tab 'checked' state': "
"foreground: %r, background: %r",
theme_name,
"yes" if variant == "dark" else "no",
selected_fg_color,
selected_bg_color,
)
css_data = dedent(
f"""
.custom_tab:checked {{
color: {selected_fg_color};
background: {selected_bg_color};
}}
"""
).encode()
style_provider = Gtk.CssProvider()
style_provider.load_from_data(css_data)
Gtk.StyleContext.add_provider_for_screen(
Gdk.Screen.get_default(),
style_provider,
Gtk.STYLE_PROVIDER_PRIORITY_APPLICATION,
)