mirror of
https://github.com/imarkoff/Marble-shell-theme.git
synced 2025-09-17 08:47:55 -07:00
1bc6a89d77
- Fixes #33, #19; - Show user-friendly message if glib2-devel is not installed; - Use tempfile for temp folder; - Little code improvements; - Tried to fix not passing test.
230 lines
8.3 KiB
Python
230 lines
8.3 KiB
Python
import os
|
|
import subprocess
|
|
|
|
from .theme import Theme
|
|
from .utils import label_files, remove_properties, remove_keywords, gnome
|
|
from . import config
|
|
|
|
|
|
class ThemePrepare:
|
|
"""
|
|
Theme object prepared for installation
|
|
"""
|
|
|
|
def __init__(self, theme, theme_file, should_label=False):
|
|
self.theme = theme
|
|
self.theme_file = theme_file
|
|
self.should_label = should_label
|
|
|
|
|
|
class GlobalTheme:
|
|
def __init__(self, colors_json, theme_folder, destination_folder, destination_file, temp_folder,
|
|
mode=None, is_filled=False):
|
|
"""
|
|
Initialize GlobalTheme class
|
|
:param colors_json: location of a json file with colors
|
|
:param theme_folder: raw theme location
|
|
:param destination_folder: folder where themes will be installed
|
|
:param temp_folder: folder where files will be collected
|
|
:param mode: theme mode (light or dark). applied only for gnome-shell < 44
|
|
:param is_filled: if True, theme will be filled
|
|
"""
|
|
|
|
self.colors_json = colors_json
|
|
self.theme_folder = theme_folder
|
|
self.destination_folder = destination_folder
|
|
self.destination_file = destination_file
|
|
self.temp_folder = os.path.join(temp_folder, "gdm")
|
|
|
|
self.backup_file = f"{self.destination_file}.backup"
|
|
self.backup_trigger = "\n/* Marble theme */\n" # trigger to check if theme is installed
|
|
self.extracted_theme = os.path.join(self.temp_folder, config.extracted_gdm_folder)
|
|
self.gst = os.path.join(self.destination_folder, self.destination_file) # use backup file if theme is installed
|
|
|
|
self.themes: list[ThemePrepare] = []
|
|
|
|
try:
|
|
gnome_version = gnome.gnome_version()
|
|
gnome_major = gnome_version.split(".")[0]
|
|
if int(gnome_major) >= 44:
|
|
self.themes += [
|
|
self.__create_theme("gnome-shell-light", mode='light', should_label=True, is_filled=is_filled),
|
|
self.__create_theme("gnome-shell-dark", mode='dark', is_filled=is_filled)
|
|
]
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
print("Using single theme.")
|
|
|
|
if not self.themes:
|
|
self.themes.append(
|
|
self.__create_theme(
|
|
"gnome-shell", mode=mode if mode else 'dark', is_filled=is_filled))
|
|
|
|
|
|
def __create_theme(self, theme_type, mode=None, should_label=False, is_filled=False):
|
|
"""Helper to create theme objects"""
|
|
theme = Theme(theme_type, self.colors_json, self.theme_folder,
|
|
self.extracted_theme, self.temp_folder,
|
|
mode=mode, is_filled=is_filled)
|
|
theme_file = os.path.join(self.extracted_theme, f"{theme_type}.css")
|
|
return ThemePrepare(theme=theme, theme_file=theme_file, should_label=should_label)
|
|
|
|
def __is_installed(self):
|
|
"""
|
|
Check if theme is installed
|
|
:return: True if theme is installed, False otherwise
|
|
"""
|
|
|
|
with open(self.gst, "rb") as f:
|
|
return self.backup_trigger.encode() in f.read()
|
|
|
|
def __extract(self):
|
|
"""
|
|
Extract gresource files to temp folder
|
|
"""
|
|
print("Extracting gresource files...")
|
|
|
|
resources = subprocess.getoutput(f"gresource list {self.gst}").split("\n")
|
|
prefix = "/org/gnome/shell/"
|
|
|
|
try:
|
|
for resource in resources:
|
|
resource_path = resource.replace(prefix, "")
|
|
dir_path = os.path.join(self.temp_folder, os.path.dirname(resource_path))
|
|
output_path = os.path.join(self.temp_folder, resource_path)
|
|
|
|
os.makedirs(dir_path, exist_ok=True)
|
|
with open(output_path, 'wb') as f:
|
|
subprocess.run(["gresource", "extract", self.gst, resource], stdout=f, check=True)
|
|
|
|
except FileNotFoundError as e:
|
|
if "gresource" in str(e):
|
|
print("Error: 'gresource' command not found.")
|
|
print("Please install the glib2-devel package:")
|
|
print(" - For Fedora/RHEL: sudo dnf install glib2-devel")
|
|
print(" - For Ubuntu/Debian: sudo apt install libglib2.0-dev")
|
|
print(" - For Arch: sudo pacman -S glib2-devel")
|
|
raise Exception("Missing required dependency: glib2-devel") from e
|
|
raise
|
|
|
|
def __add_gnome_styles(self, theme):
|
|
"""
|
|
Add gnome styles to the start of the file
|
|
:param theme: Theme object
|
|
"""
|
|
|
|
with open(f"{self.extracted_theme}/{theme.theme_type}.css", 'r') as gnome_theme:
|
|
gnome_styles = gnome_theme.read() + self.backup_trigger
|
|
theme.add_to_start(gnome_styles)
|
|
|
|
def __prepare(self, hue, color, sat=None):
|
|
"""
|
|
Generate theme files for gnome-shell-theme.gresource.xml
|
|
:param hue: color hue
|
|
:param color: color name
|
|
:param sat: color saturation
|
|
"""
|
|
|
|
for theme in self.themes:
|
|
if theme.should_label:
|
|
label_files(theme.theme.temp_folder, "light", theme.theme.main_styles)
|
|
|
|
remove_keywords(theme.theme_file, "!important")
|
|
remove_properties(theme.theme_file, "background-color", "color", "box-shadow", "border-radius")
|
|
|
|
self.__add_gnome_styles(theme.theme)
|
|
|
|
theme.theme.install(hue, color, sat, destination=self.extracted_theme)
|
|
|
|
|
|
def __backup(self):
|
|
"""
|
|
Backup installed theme
|
|
"""
|
|
|
|
if self.__is_installed():
|
|
return
|
|
|
|
# backup installed theme
|
|
print("Backing up default theme...")
|
|
subprocess.run(["cp", "-aT", self.gst, f"{self.gst}.backup"], cwd=self.destination_folder, check=True)
|
|
|
|
def __generate_gresource_xml(self):
|
|
"""
|
|
Generates.gresource.xml
|
|
"""
|
|
|
|
# list of files to add to gnome-shell-theme.gresource.xml
|
|
files = [f"<file>{file}</file>" for file in os.listdir(self.extracted_theme)]
|
|
nl = "\n" # fstring doesn't support newline character
|
|
|
|
return f"""<?xml version="1.0" encoding="UTF-8"?>
|
|
<gresources>
|
|
<gresource prefix="/org/gnome/shell/theme">
|
|
{nl.join(files)}
|
|
</gresource>
|
|
</gresources>"""
|
|
|
|
def install(self, hue, sat=None):
|
|
"""
|
|
Install theme globally
|
|
:param hue: color hue
|
|
:param sat: color saturation
|
|
"""
|
|
|
|
if os.geteuid() != 0:
|
|
raise Exception("Root privileges required to install GDM theme")
|
|
|
|
if self.__is_installed():
|
|
print("Theme is installed. Reinstalling...")
|
|
self.gst += ".backup"
|
|
|
|
self.__extract()
|
|
|
|
# generate theme files for global theme
|
|
self.__prepare(hue, 'Marble', sat)
|
|
|
|
# generate gnome-shell-theme.gresource.xml
|
|
with open(f"{self.extracted_theme}/{self.destination_file}.xml", 'w') as gresource_xml:
|
|
generated_xml = self.__generate_gresource_xml()
|
|
gresource_xml.write(generated_xml)
|
|
|
|
# compile gnome-shell-theme.gresource.xml
|
|
print("Compiling theme...")
|
|
subprocess.run(["glib-compile-resources" , f"{self.destination_file}.xml"],
|
|
cwd=self.extracted_theme, check=True)
|
|
|
|
# backup installed theme
|
|
self.__backup()
|
|
|
|
# install theme
|
|
print("Installing theme...")
|
|
subprocess.run(["sudo", "cp", "-f",
|
|
f"{self.extracted_theme}/{self.destination_file}",
|
|
f"{self.destination_folder}/{self.destination_file}"],
|
|
check=True)
|
|
|
|
print("Theme installed successfully.")
|
|
|
|
|
|
def remove(self):
|
|
"""
|
|
Remove installed theme
|
|
"""
|
|
|
|
# use backup file if theme is installed
|
|
if self.__is_installed():
|
|
print("Theme is installed. Removing...")
|
|
backup_path = os.path.join(self.destination_folder, self.backup_file)
|
|
dest_path = os.path.join(self.destination_folder, self.destination_file)
|
|
|
|
if os.path.isfile(backup_path):
|
|
subprocess.run(["sudo", "mv", backup_path, dest_path], check=True)
|
|
|
|
else:
|
|
print("Backup file not found. Try reinstalling gnome-shell package.")
|
|
|
|
else:
|
|
print("Theme is not installed. Nothing to remove.")
|
|
print("If theme is still installed globally, try reinstalling gnome-shell package.")
|