forked from Mirror/pmbootstrap
71e7af57e6
We use a custom verbose log level in pmbootstrap, unfortunately it isn't possible to correctly type this due to some limitations in the logging library [1], [2]. Given that our usecase is fairly simple, we can just wrap the module with our own so we only have to tell mypy to ignore the error once instead of at every callsite. [1]: https://github.com/cryptax/droidlysis/issues/15 [2]: https://github.com/python/typing/discussions/980 Signed-off-by: Caleb Connolly <caleb@postmarketos.org>
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
# Copyright 2023 Oliver Smith
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
import os
|
|
import glob
|
|
from pmb.helpers import logging
|
|
|
|
import pmb.config
|
|
import pmb.chroot.apk
|
|
from pmb.core import Chroot
|
|
from pmb.core.types import PmbArgs
|
|
|
|
|
|
def list_chroot(args: PmbArgs, suffix: Chroot, remove_prefix=True):
|
|
ret = []
|
|
prefix = pmb.config.initfs_hook_prefix
|
|
for pkgname in pmb.chroot.apk.installed(args, suffix).keys():
|
|
if pkgname.startswith(prefix):
|
|
if remove_prefix:
|
|
ret.append(pkgname[len(prefix):])
|
|
else:
|
|
ret.append(pkgname)
|
|
return ret
|
|
|
|
|
|
def list_aports(args: PmbArgs):
|
|
ret = []
|
|
prefix = pmb.config.initfs_hook_prefix
|
|
for path in glob.glob(f"{args.aports}/*/{prefix}*"):
|
|
ret.append(os.path.basename(path)[len(prefix):])
|
|
return ret
|
|
|
|
|
|
def ls(args: PmbArgs, suffix: Chroot):
|
|
hooks_chroot = list_chroot(args, suffix)
|
|
hooks_aports = list_aports(args)
|
|
|
|
for hook in hooks_aports:
|
|
line = f"* {hook} ({'' if hook in hooks_chroot else 'not '}installed)"
|
|
logging.info(line)
|
|
|
|
|
|
def add(args: PmbArgs, hook, suffix: Chroot):
|
|
if hook not in list_aports(args):
|
|
raise RuntimeError("Invalid hook name!"
|
|
" Run 'pmbootstrap initfs hook_ls'"
|
|
" to get a list of all hooks.")
|
|
prefix = pmb.config.initfs_hook_prefix
|
|
pmb.chroot.apk.install(args, [f"{prefix}{hook}"], suffix)
|
|
|
|
|
|
def delete(args: PmbArgs, hook, suffix: Chroot):
|
|
if hook not in list_chroot(args, suffix):
|
|
raise RuntimeError("There is no such hook installed!")
|
|
prefix = pmb.config.initfs_hook_prefix
|
|
pmb.chroot.root(args, ["apk", "del", f"{prefix}{hook}"], suffix)
|
|
|
|
|
|
def update(args: PmbArgs, suffix: Chroot):
|
|
"""
|
|
Rebuild and update all hooks that are out of date
|
|
"""
|
|
pmb.chroot.apk.install(args, list_chroot(args, suffix, False), suffix)
|