mirror of
https://github.com/Foltik/dotfiles
synced 2024-11-24 04:22:50 -05:00
97 lines
3.4 KiB
Python
Executable File
97 lines
3.4 KiB
Python
Executable File
#!/bin/python
|
|
import os
|
|
import sys
|
|
import argparse
|
|
import curses
|
|
import distutils.dir_util
|
|
import lib.deploy as deploy
|
|
from pathlib import Path
|
|
from lib.parse import parse
|
|
from lib.package import Category, Package
|
|
from lib.menu import Menu, ChecklistMenu
|
|
|
|
parsed_categories = parse('packages.yml')
|
|
parsed_packages = [pkg for pkglist in list(map(lambda c: c.packages, parsed_categories)) for pkg in pkglist]
|
|
|
|
run_installer = False
|
|
def run():
|
|
for category in [c for c in parsed_categories if c.enabled]:
|
|
for package in category.packages:
|
|
configure(package)
|
|
|
|
def dump(package):
|
|
print(package.name + ':')
|
|
for k, v in package.getattrs():
|
|
if isinstance(v, list):
|
|
print(' ' + k, [str(i) for i in v])
|
|
else:
|
|
print(' ' + k, v)
|
|
|
|
def lookup_package(name):
|
|
try:
|
|
package = next(p for p in parsed_packages if p.name == name or p.alias == name)
|
|
return package
|
|
except StopIteration:
|
|
raise Exception('Package not found: ' + name)
|
|
|
|
class App:
|
|
def __init__(self, stdscreen):
|
|
self.screen = stdscreen
|
|
curses.curs_set(0)
|
|
|
|
def exit_and_run():
|
|
global run_installer
|
|
run_installer = True
|
|
main_menu.exit()
|
|
|
|
package_options_menu = Category.config_menu(self.screen, parsed_categories)
|
|
|
|
main_menu = Menu(self.screen, 'Main Menu', [
|
|
('Package Options', package_options_menu.display),
|
|
('Run Installer', exit_and_run)
|
|
])
|
|
|
|
main_menu.display()
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description="Foltik's super cool dotfiles install script")
|
|
parser.add_argument('packages', metavar='pkg', type=str, nargs='*')
|
|
parser.add_argument('-a', '--all', dest='all', action='store_const', const=True,
|
|
help='Run on all packages')
|
|
|
|
actions = [
|
|
('-d', '--deploy', deploy.deploy, 'alias for -icsue'),
|
|
('-i', '--install', deploy.install, 'install package'),
|
|
('-c', '--config', deploy.export_config, 'export package config'),
|
|
('-s', '--script', deploy.run_script, 'run package script'),
|
|
('-u', '--units', deploy.export_units, 'export and enable package units'),
|
|
('-e', '--enable-units', deploy.enable_units, 'enable package units'),
|
|
('-C', '--import-config', deploy.import_config, 'import package config'),
|
|
('-U', '--import-units', deploy.import_units, 'import package units'),
|
|
('-l', '--dump', dump, 'dump package attributes'),
|
|
('-Dc', '--diff-config', deploy.diff_config, 'diff local and repo config'),
|
|
('-Du', '--diff-units', deploy.diff_units, 'diff local and repo units')
|
|
]
|
|
|
|
for action in actions:
|
|
parser.add_argument(action[0], action[1], dest='actions', action='append_const', const=action[2], help=action[3])
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.actions != None and args.packages == [] and args.all == True:
|
|
for package in parsed_packages:
|
|
for action in args.actions:
|
|
action(package)
|
|
elif args.actions != None and args.packages != []:
|
|
for package in args.packages:
|
|
for action in args.actions:
|
|
action(lookup_package(package))
|
|
else:
|
|
os.environ.setdefault('ESCDELAY', '0')
|
|
curses.wrapper(App)
|
|
if run_installer:
|
|
run()
|
|
|
|
if Path('git').is_dir():
|
|
distutils.dir_util.remove_tree('git')
|