mirror of
https://github.com/Foltik/dotfiles
synced 2024-11-24 04:22:50 -05:00
104 lines
3.5 KiB
Python
104 lines
3.5 KiB
Python
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)
|
|
|