forked from Mirror/pmbootstrap
00c670ffda
Commit 37ec73c dropped the use of args to access command line arguments,
but did not add another way to pass command line arguments; thus,
optional flags to flash_kernel, flash_rootfs, flash_lk2nd, sideload and
list_devices were ignored.
Fix this by making sure the relevant arguments are passed for all
frontend methods.
Fixes #2522.
113 lines
3.6 KiB
Python
113 lines
3.6 KiB
Python
# Copyright 2023 Oliver Smith
|
|
# Copyright 2024 Stefan Hansson
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import pmb.parse.deviceinfo
|
|
from pmb import commands
|
|
from pmb.core.context import get_context
|
|
from pmb.flasher.frontend import flash_lk2nd, kernel, list_flavors, rootfs, sideload
|
|
from pmb.helpers import logging
|
|
|
|
|
|
class Flasher(commands.Command):
|
|
def __init__(
|
|
self,
|
|
action_flasher: str,
|
|
autoinstall: bool,
|
|
cmdline: str | None,
|
|
flash_method: str,
|
|
no_reboot: bool | None,
|
|
partition: str | None,
|
|
resume: bool | None,
|
|
) -> None:
|
|
self.action_flasher = action_flasher
|
|
self.autoinstall = autoinstall
|
|
self.cmdline = cmdline
|
|
self.flash_method = flash_method
|
|
self.no_reboot = no_reboot
|
|
self.partition = partition
|
|
self.resume = resume
|
|
|
|
def run(self) -> None:
|
|
context = get_context()
|
|
action = self.action_flasher
|
|
device = context.config.device
|
|
deviceinfo = pmb.parse.deviceinfo()
|
|
method = self.flash_method or deviceinfo.flash_method
|
|
|
|
if method == "none" and action in ["boot", "flash_kernel", "flash_rootfs", "flash_lk2nd"]:
|
|
logging.info("This device doesn't support any flash method.")
|
|
return
|
|
|
|
if action in ["boot", "flash_kernel"]:
|
|
kernel(
|
|
deviceinfo,
|
|
method,
|
|
action == "boot",
|
|
self.autoinstall,
|
|
cmdline=self.cmdline,
|
|
no_reboot=self.no_reboot,
|
|
partition=self.partition,
|
|
resume=self.resume,
|
|
)
|
|
elif action == "flash_rootfs":
|
|
rootfs(
|
|
deviceinfo,
|
|
method,
|
|
cmdline=self.cmdline,
|
|
no_reboot=self.no_reboot,
|
|
partition=self.partition,
|
|
resume=self.resume,
|
|
)
|
|
elif action == "flash_vbmeta":
|
|
logging.info("(native) flash vbmeta.img with verity disabled flag")
|
|
pmb.flasher.run(
|
|
deviceinfo,
|
|
method,
|
|
"flash_vbmeta",
|
|
cmdline=self.cmdline,
|
|
no_reboot=self.no_reboot,
|
|
partition=self.partition,
|
|
resume=self.resume,
|
|
)
|
|
elif action == "flash_dtbo":
|
|
logging.info("(native) flash dtbo image")
|
|
pmb.flasher.run(
|
|
deviceinfo,
|
|
method,
|
|
"flash_dtbo",
|
|
cmdline=self.cmdline,
|
|
no_reboot=self.no_reboot,
|
|
partition=self.partition,
|
|
resume=self.resume,
|
|
)
|
|
elif action == "flash_lk2nd":
|
|
flash_lk2nd(
|
|
deviceinfo,
|
|
method,
|
|
cmdline=self.cmdline,
|
|
no_reboot=self.no_reboot,
|
|
partition=self.partition,
|
|
resume=self.resume,
|
|
)
|
|
elif action == "list_flavors":
|
|
list_flavors(device)
|
|
elif action == "list_devices":
|
|
pmb.flasher.run(
|
|
deviceinfo,
|
|
method,
|
|
"list_devices",
|
|
cmdline=self.cmdline,
|
|
no_reboot=self.no_reboot,
|
|
partition=self.partition,
|
|
resume=self.resume,
|
|
)
|
|
elif action == "sideload":
|
|
sideload(
|
|
deviceinfo,
|
|
method,
|
|
cmdline=self.cmdline,
|
|
no_reboot=self.no_reboot,
|
|
partition=self.partition,
|
|
resume=self.resume,
|
|
)
|