linux-image-manager/lim/image/choosers.py

"""Interactive selection of the distribution image to install."""

from lim import catalog, runner, ui
from lim.errors import LimError
from lim.image.plan import ImagePlan


def choose_arch(plan: ImagePlan) -> None:
    version = ui.ask("Which Raspberry Pi will be used (e.g.: 1, 2, 3b, 3b+, 4...):")
    entry = catalog.arch_rpi_images().get(version)
    if entry is None:
        raise LimError(f"Version {version} isn't supported.")
    plan.raspberry_pi_version = version
    plan.boot_size = "+500M"
    plan.base_download_url = "http://os.archlinuxarm.org/os/"
    plan.image_name = entry["image"]
    plan.luks_memory_cost = entry["luks_memory_cost"]


def choose_manjaro(plan: ImagePlan) -> None:
    plan.boot_size = "+500M"
    flavour = ui.ask("Which version(e.g.:architect,gnome) should be used:")
    if flavour == "architect":
        plan.image_checksum = "6b1c2fce12f244c1e32212767a9d3af2cf8263b2"
        plan.base_download_url = (
            "https://osdn.net/frs/redir.php?m=dotsrc&f=%2Fstorage%2Fg%2Fm%2Fma"
            "%2Fmanjaro%2Farchitect%2F20.0%2F"
        )
        plan.image_name = "manjaro-architect-20.0-200426-linux56.iso"
    elif flavour == "gnome":
        release = ui.ask("Which release(e.g.:20,21,raspberrypi) should be used:")
        entry = catalog.manjaro_gnome_releases().get(release)
        if entry is None:
            raise LimError(f"Gnome release {release} isn't supported.")
        plan.image_checksum = entry.get("checksum")
        plan.base_download_url = entry["url"]
        plan.image_name = entry["image"]
        plan.luks_memory_cost = entry.get("luks_memory_cost")
        plan.raspberry_pi_version = entry.get("raspberry_pi_version")
    else:
        raise LimError(f"Manjaro version {flavour} isn't supported.")


def choose_moode(plan: ImagePlan) -> None:
    plan.boot_size = "+200M"
    plan.image_checksum = "185cbc9a4994534bb7a4bc2744c78197"
    plan.base_download_url = "https://github.com/moode-player/moode/releases/download/r651prod/"
    plan.image_name = "moode-r651-iso.zip"


def choose_retropie(plan: ImagePlan) -> None:
    plan.boot_size = "+500M"
    version = ui.ask("Which version(e.g.:1,2,3,4) should be used:")
    entry = catalog.retropie_images().get(version)
    if entry is None:
        raise LimError(f"Version {version} isn't supported.")
    plan.raspberry_pi_version = version
    plan.base_download_url = "https://github.com/RetroPie/RetroPie-Setup/releases/download/4.8/"
    plan.image_checksum = entry["checksum"]
    plan.image_name = entry["image"]


def choose_torbox(plan: ImagePlan) -> None:
    plan.base_download_url = "https://www.torbox.ch/data/"
    plan.image_name = "torbox-20220102-v050.gz"
    plan.image_checksum = "0E1BA7FFD14AAAE5F0462C8293D95B62C3BF1D9E726E26977BD04772C55680D3"
    plan.boot_size = "+200M"


def choose_android_x86(plan: ImagePlan) -> None:
    plan.base_download_url = (
        "https://www.fosshub.com/Android-x86.html?dwl=android-x86_64-9.0-r2.iso"
    )
    plan.image_name = "android-x86_64-9.0-r2.iso"
    plan.image_checksum = "f7eb8fc56f29ad5432335dc054183acf086c539f3990f0b6e9ff58bd6df4604e"
    plan.boot_size = "+500M"


DISTRIBUTION_CHOOSERS = {
    "android-x86": choose_android_x86,
    "torbox": choose_torbox,
    "arch": choose_arch,
    "manjaro": choose_manjaro,
    "moode": choose_moode,
    "retropie": choose_retropie,
}


def choose_linux_image(plan: ImagePlan) -> None:
    distribution = ui.ask(
        "Which distribution should be used [arch,moode,retropie,manjaro,torbox...]?"
    )
    chooser = DISTRIBUTION_CHOOSERS.get(distribution)
    if chooser is None:
        raise LimError(f"Distribution {distribution} isn't supported.")
    plan.distribution = distribution
    chooser(plan)


def choose_local_image(plan: ImagePlan) -> None:
    ui.info("Available images:")
    runner.run(["ls", "-l", str(plan.image_folder)], check=False)
    plan.image_name = ui.ask("Which image would you like to use?")