# Copyright 2023 Oliver Smith # SPDX-License-Identifier: GPL-3.0-or-later from typing import Dict from pmb.types import PmbArgs import pmb.flasher import pmb.chroot.initfs import pmb.helpers.args def check_partition_blacklist(args: PmbArgs, deviceinfo: Dict[str, str], key, value): if not key.startswith("$PARTITION_"): return name = deviceinfo["name"] if value in deviceinfo["partition_blacklist"].split(","): raise RuntimeError("'" + value + "'" + " partition is blacklisted " + "from being flashed! See the " + name + " device " + "wiki page for more information.") def run(device: str, deviceinfo: Dict[str, str], action, flavor=None): pmb.flasher.init(device) # FIXME: handle argparsing and pass in only the args we need. args = pmb.helpers.args.please_i_really_need_args() # Verify action method = args.flash_method or deviceinfo["flash_method"] cfg = pmb.config.flashers[method] if not isinstance(cfg["actions"], dict): raise TypeError(f"Flashers misconfigured! {method} key 'actions' should be a dictionary") if action not in cfg["actions"]: raise RuntimeError("action " + action + " is not" " configured for method " + method + "!" " You can use the '--method' option to specify a" " different flash method. See also:" " ") # Variable setup fvars = pmb.flasher.variables(args, flavor, method) # vbmeta flasher requires vbmeta partition to be explicitly specified if action == "flash_vbmeta" and not fvars["$PARTITION_VBMETA"]: raise RuntimeError("Your device does not have 'vbmeta' partition" " specified; set" " 'deviceinfo_flash_fastboot_partition_vbmeta'" " or 'deviceinfo_flash_heimdall_partition_vbmeta'" " in deviceinfo file. See also:" " ") # dtbo flasher requires dtbo partition to be explicitly specified if action == "flash_dtbo" and not fvars["$PARTITION_DTBO"]: raise RuntimeError("Your device does not have 'dtbo' partition" " specified; set" " 'deviceinfo_flash_fastboot_partition_dtbo'" " in deviceinfo file. See also:" " ") if args.no_reboot and ("flash" not in action or method != "heimdall-bootimg"): raise RuntimeError("The '--no-reboot' option is only" " supported when flashing with heimall-bootimg.") if args.resume and ("flash" not in action or method != "heimdall-bootimg"): raise RuntimeError("The '--resume' option is only" " supported when flashing with heimall-bootimg.") # Run the commands of each action for command in cfg["actions"][action]: # Variable replacement for key, value in fvars.items(): for i in range(len(command)): if key in command[i]: if value is None: raise RuntimeError(f"Variable {key} found in action" f" {action} for method {method}," " but the value for this variable" " is None! Is that missing in your" " deviceinfo?") check_partition_blacklist(args, deviceinfo, key, value) command[i] = command[i].replace(key, value) # Remove empty strings command = [x for x in command if x != ''] # Run the action pmb.chroot.root(command, output="interactive")