mirror of
https://gitlab.com/kupfer/kupferbootstrap.git
synced 2025-02-23 05:35:44 -05:00
41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
import os
|
|
import urllib.request
|
|
import click
|
|
|
|
from config import config
|
|
from constants import BOOT_STRATEGIES, FLASH_PARTS, FASTBOOT, JUMPDRIVE, JUMPDRIVE_VERSION
|
|
from fastboot import fastboot_boot, fastboot_erase_dtbo
|
|
from image import get_device_and_flavour, get_image_name, dump_bootimg, dump_lk2nd
|
|
from wrapper import enforce_wrap
|
|
|
|
LK2ND = FLASH_PARTS['LK2ND']
|
|
BOOTIMG = FLASH_PARTS['BOOTIMG']
|
|
|
|
TYPES = [LK2ND, JUMPDRIVE, BOOTIMG]
|
|
|
|
|
|
@click.command(name='boot')
|
|
@click.argument('type', required=False, default=BOOTIMG, type=click.Choice(TYPES))
|
|
def cmd_boot(type):
|
|
f"""Flash one of {', '.join(TYPES)}"""
|
|
enforce_wrap()
|
|
device, flavour = get_device_and_flavour()
|
|
image_name = get_image_name(device, flavour)
|
|
image_path = os.path.join(config.get_path('images'), image_name)
|
|
strategy = BOOT_STRATEGIES[device]
|
|
|
|
if strategy == FASTBOOT:
|
|
if type == JUMPDRIVE:
|
|
file = f'boot-{device}.img'
|
|
path = os.path.join(config.get_path('jumpdrive'), file)
|
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
|
if not os.path.exists(path):
|
|
urllib.request.urlretrieve(f'https://github.com/dreemurrs-embedded/Jumpdrive/releases/download/{JUMPDRIVE_VERSION}/{file}', path)
|
|
elif type == LK2ND:
|
|
path = dump_lk2nd(image_path)
|
|
elif type == BOOTIMG:
|
|
path = dump_bootimg(image_path)
|
|
else:
|
|
raise Exception(f'Unknown boot image type {type}')
|
|
fastboot_erase_dtbo()
|
|
fastboot_boot(path)
|