Marble-shell-theme/scripts/utils/gresource/gresource_extractor.py

import os
import subprocess

from scripts.utils.gresource import raise_gresource_error
from scripts.utils.logger.logger import LoggerFactory


class GresourceExtractor:
    def __init__(self, gresource_path: str, extract_folder: str, logger_factory: LoggerFactory):
        self.gresource_path = gresource_path
        self.extract_folder = extract_folder
        self.logger_factory = logger_factory

    def extract(self):
        extract_line = self.logger_factory.create_logger()
        extract_line.update("Extracting gresource files...")

        resources = self._get_resources_list()
        self._extract_resources(resources)

        extract_line.success("Extracted gresource files.")

    def _get_resources_list(self):
        resources_list_response = subprocess.run(
            ["gresource", "list", self.gresource_path],
            capture_output=True, text=True, check=False
        )

        if resources_list_response.stderr:
            raise Exception(f"gresource could not process the theme file: {self.gresource_path}")

        return resources_list_response.stdout.strip().split("\n")

    def _extract_resources(self, resources: list[str]):
        try:
            self._try_extract_resources(resources)
        except FileNotFoundError as e:
            if "gresource" in str(e):
                raise_gresource_error("gresource", e)
            raise

    def _try_extract_resources(self, resources: list[str]):
        prefix = "/org/gnome/shell/theme/"
        for resource in resources:
            resource_path = resource.replace(prefix, "")
            output_path = os.path.join(self.extract_folder, resource_path)
            os.makedirs(os.path.dirname(output_path), exist_ok=True)

            with open(output_path, 'wb') as f:
                subprocess.run(
                    ["gresource", "extract", self.gresource_path, resource],
                    stdout=f, check=True
                )