installer-script/main.py at main · MariaLinux/installer-script · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
#!/usr/bin/env python3
import argparse
import os
import tomllib

import common
import config
import drive
import system


def parse_args() -> str:
    """This is a function to handle the parsing of command line args passed to the program."""

    parser = argparse.ArgumentParser(
        prog="MariaLinux Installer", description="Program to install Maria Linux."
    )
    parser.add_argument(
        "-c",
        "--config",
        dest="config_file",
        help="Run the installer automated from a toml config file.",
        default="",
        action="store",
    )

    args = parser.parse_args()

    return args.config_file


def main():
    config_file = parse_args()
    interactive = True

    common.info("BIOS detected." if not drive.is_efi() else "EFI detected.")

    if config_file != "":
        interactive = False

        if not os.path.isfile(config_file):
            common.die(f"Config file {config_file} does not exist.")

        with open(config_file, "rb") as file:
            config_file = tomllib.load(file)

    common.info(
        f"Entering interactive mode. Default values are shown wrapped in square brackets like {common.Colours.blue}[this]{common.Colours.endc}. Press enter to accept the default.\n"
        if interactive
        else "Checking config"
    )

    config_parsed = config.parse_config(
        config_file if config_file != "" else {}, interactive=interactive
    )

    common.info(f"Done checking config")

    if interactive:
        common.warn(
            f"Drive partitioning is about to start. After this process, drive {config_parsed['drive']} will be erased. Press enter to continue."
        )
        input("")

    common.info(f"Partitioning drive {config_parsed['drive']}")
    drive.partition_drive(
        config_parsed["drive"], drive.LAYOUTS[config_parsed["filesystem"]]
    )

    common.info(f"Formatting drive {config_parsed['drive']}")
    drive.format_drive(
        config_parsed["drive"], drive.LAYOUTS[config_parsed["filesystem"]]
    )

    common.info("Starting installation")
    system.mount_roots()

    common.info("Downloading root image")
    root_url = config.get_url(config_parsed)
    system.download_root(root_url)
    system.mount()

    common.info("Installing bootloader")
    system.install_bootloader("x86_64-efi" if drive.is_efi() else "i386-pc", device=config_parsed["drive"])

    common.info("Starting post-installation tasks")
    system.post_install(config_parsed)


if __name__ == "__main__":
    main()