dotfiles/lib/package.py

from pathlib import Path
from lib.menu import ChecklistMenu

def toggle(obj, prop):
    return lambda b: setattr(obj, prop, b)

class Package:
    def __init__(self, obj):
        # Defaults
        self.source = 'core'
        self.alias = None
        
        # Immutable attributes
        if isinstance(obj, dict):
            self.name, props = next(iter(obj.items()))
            self.source = 'core'
            for key, value in props.items():
                setattr(self, key, value if value != 'none' else None)
        else:
            self.name = obj

        # Transform and infer attributes
        self.update_from_files()

        # Install State
        self.enabled = True
        self.install = True
        self.export_config = True
        self.run_script = True
        self.export_units = True
        self.enable_units = True

        
    def __repr__(self):
        return self.name

    def __getitem__(self, key):
        return getattr(self, key)

    def __setitem__(self, key, value):
        return setattr(self, key, value)

    def getattrs(self):
        excepted = [
            'name', 'enabled', 'install',
            'export_config', 'run_script',
            'export_units', 'enable_units'
        ]

        return [(k, v) for k, v in vars(self).items() if v != None and k not in excepted]

    def transform_default(self, prop, default, basepath):
        default_path = Path(default)
        if hasattr(self, prop):
            if isinstance(self[prop], list):
                self[prop] = list(map(lambda p: Path(p), self[prop]))
            elif self[prop] == 'none':
                self[prop] = None
            elif self[prop] != None:
                self[prop] = [Path(self[prop])]
        elif (Path(basepath) / default_path).is_dir() or (Path(basepath) / default_path).is_file():
            self[prop] = [default_path]
        else:
            self[prop] = None

    def update_from_files(self):
        self.transform_default('config', self.name, 'lain/.config')
        self.transform_default('script', self.name + '.fish', 'scripts')
        self.transform_default('userunit', self.name + '.service', 'lain/.config/systemd/user')

    @staticmethod
    def config_menu(screen, title, packages):
        items = []
        for package in packages:
            title = package.name + ' - Configure'
            submenu = ChecklistMenu(screen, title, [
                ('Install', toggle(package, 'install')),
                ('Export Config', toggle(package, 'export_config')),
                ('Run Script', toggle(package, 'run_script')),
                ('Export Units', toggle(package, 'export_units')),
                ('Enable Units', toggle(package, 'enable_units'))
            ], True)
            items += [(package.name, toggle(package, 'enabled'), submenu.display)]
        return ChecklistMenu(screen, title, items, True)
                
class Category:
    def __init__(self, name, packages):
        self.name = name
        self.packages = [Package(pkg) for pkg in packages]
        self.enabled = True

    def __repr__(self):
        return self.name + ': ' + str(self.packages)

    @staticmethod
    def config_menu(screen, categories):
        items = []
        for category in categories:
            title = category.name + ' - Configure Packages'
            submenu = Package.config_menu(screen, title, category.packages)
            items += [(category.name, toggle(category, 'enabled'), submenu.display)]
        return ChecklistMenu(screen, 'Configure Categories', items, True)