engine.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
#  engine.py
#
#  Copyright 2023 Thomas Castleman <batcastle@draugeros.org>
#
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
#  MA 02110-1301, USA.
#
#
"""Main installation Engine"""
from __future__ import print_function
import sys
import subprocess
import os
import json
import tarfile as tar
import shutil
import traceback
import psutil
import UI
import installer
import check_internet
import check_kernel_versions
import common
import auto_partitioner
import oem
import modules

common.eprint(f"    ###    {sys.argv[0]} STARTED    ###    ")

def copy_log_to_disk():
    """Copy Installation Log to installation location"""
    try:
        shutil.copyfile("/tmp/system-installer.log",
                     "/mnt/var/log/system-installer.log")
    except FileNotFoundError:
        common.eprint("    ###    Log Not Found. Testing?    ###    ")
        with open("/tmp/system-installer.log", "w+") as log:
            log.write("""Log was not created during installation.
This is a stand-in file.
""")
        shutil.copyfile("/tmp/system-installer.log",
                     "/mnt/var/log/system-installer.log")

with open("/etc/system-installer/settings.json") as config_file:
    CONFIG = json.loads(config_file.read())
BOOT_TIME = False
if len(sys.argv) > 1:
    if sys.argv[1] == "--boot-time":
        # OEM post-install configuration, on-boot installation, and more
        # are handled here
        if os.path.exists("/etc/system-installer/oem-post-install.flag"):
            # OEM post installation configuration
            oem.post_install.UI.show_main()
            os.remove("/etc/system-installer/oem-post-install.flag")
            modules.purge.purge_package("system-installer")
            if "run_post_oem" in CONFIG:
                subprocess.Popen(CONFIG["run_post_oem"])
            sys.exit(0)
        with open("/proc/cmdline", "r") as cmdline_file:
            cmdline = cmdline_file.read()
        if "system-installer" not in cmdline:
            # Not wanted to be running ootb
            sys.exit(0)
        BOOT_TIME = True
MEMCHECK = psutil.virtual_memory().total
if (MEMCHECK / 1024 ** 2) < 1024:
    UI.error.show_error("\n\tRAM is less than 1 GB.\t\n")
    sys.exit(2)

DISK = auto_partitioner.check_disk_state()
for each in range(len(DISK) - 1, -1, -1):
    if float(DISK[each]["size"]) < auto_partitioner.LIMITER:
        del DISK[each]
if len(DISK) < 1:
    UI.error.show_error("\n\tNo 32 GB or Larger Drives detected\t\n")
    sys.exit(2)
if not check_kernel_versions.check_kernel_versions(CONFIG["local_repo"],
                                                   CONFIG["kernel_meta_pkg"]):
    UI.error.show_error("""
\t<b>Kernel Version Mismatch.</b>\t
\tPlease reboot and retry installation.\t
\tIf your problem persists, please create an issue on our Github.\t
""")
    sys.exit(2)
work_dir = "/tmp/quick-install_working-dir"

# Check if the installer has already been run
if len(os.listdir("/mnt")) > 0:
    # force the user to reboot in order to reattempt installation
    UI.error.show_error("""
\t<b>Must Reboot Before Reattempting Installation</b>\t

\tIn order to prevent various bugs from occuring, users\t
\tare required to reboot before re-attempting installation.\t
""", report=False)
    sys.exit(2)
SETTINGS = UI.main.show_main(boot_time=BOOT_TIME)
try:
    if ((SETTINGS == 1) or (len(SETTINGS) == 0)):
        print("Returned data from UI appears empty. Assuming installation was canceled...")
        sys.exit(1)
    elif SETTINGS == 2:
        UI.error.show_error("""
\t<b>Unknown Error</b>\t
\tWe're sorry. An unknown error has occured.\t
\tPlease reboot and retry installation.\t
\tIf your problem persists, please create an issue on our Github.\t
""")
        sys.exit(2)
    elif os.path.exists(SETTINGS):
        if SETTINGS.split("/")[-1][-5:] == ".json":
            try:
                with open(SETTINGS, "r") as quick_install_file:
                    SETTINGS = json.load(quick_install_file)
            except json.decoder.JSONDecodeError:
                UI.error.show_error("""
\t<b>JSON File Error</b>\t
\tWe're sorry. The Quick Install file you provided\t
\thas a JSON Syntax Error. Please correct this and try again.\t
\tIf your problem persists, please create an issue on our Github.\t
""")
                sys.exit(2)
        elif SETTINGS.split("/")[-1][-7:] == ".tar.xz":
            tar_file = tar.open(name=SETTINGS)
            tar_file.extractall(path=work_dir)
            tar_file.close()
            if os.path.exists(work_dir + "/settings/installation-settings.json"):
                try:
                    with open(work_dir + "/settings/installation-settings.json",
                              "r") as quick_install_file:
                        SETTINGS = json.load(quick_install_file)
                except json.decoder.JSONDecodeError:
                    UI.error.show_error("""
\t<b>JSON File Error</b>\t
\tWe're sorry. The Quick Install file you provided\t
\thas a JSON Syntax Error. Please correct this and try again.\t
\tIf your problem persists, please create an issue on our Github.\t
""")
                sys.exit(2)
            try:
                net_settings = os.listdir(work_dir + "/settings/network-settings")
                if len(net_settings) > 0:
                    shutil.copytree(net_settings + "/settings/network-settings",
                             "/etc/NetworkManager/system-connections")
                    common.eprint("\t###\tNOTE: NETWORK SETTINGS COPIED TO LIVE SYSTEM\t###\t")
            except FileNotFoundError:
                pass
        if "DATA" in SETTINGS:
            # Parse out just the data. Ignore everything else.
            SETTINGS = SETTINGS["DATA"]
        if "OEM" in SETTINGS.values():
            # This is an OEM installation. Parts will be skipped now and handled later.
            # Other parts will be automated
            additional_settings = oem.pre_install.show_main()
            for each in additional_settings:
                SETTINGS[each] = additional_settings[each]
        if "COMPAT_MODE" not in SETTINGS:
            # this is an old quick install file. It has likely worked for the user before.
            # honor the file, but offer to update it
            # TODO: Add offer to update Quick Install File
            SETTINGS["COMPAT_MODE"] = False

except TypeError:
    pass
# Confirm whether settings are correct or not
INSTALL = UI.confirm.show_confirm(SETTINGS, boot_time=BOOT_TIME)

if INSTALL:
    try:
        # Run the progress bar in the background
        process = subprocess.Popen("/usr/share/system-installer/progress.py")
        pid = process.pid
        SETTINGS["INTERNET"] = check_internet.has_internet()
        installer.install(SETTINGS, CONFIG["local_repo"])
        shutil.rmtree("/mnt/repo")
        common.eprint(f"    ###    {sys.argv[0]} CLOSED    ###    ")
        copy_log_to_disk()
        subprocess.Popen(["su", "live", "-c",
                          f"/usr/share/system-installer/success.py \'{json.dumps(SETTINGS)}\'"])
        os.kill(pid, 15)
    except Exception as error:
        os.kill(pid, 15)
        common.eprint(f"\nAn Error has occured:\n{error}\n")
        common.eprint(traceback.format_exc())
        print(f"\nAn Error has occured:\n{error}\n")
        print(traceback.format_exc())
        copy_log_to_disk()
        UI.error.show_error("""\n\tError detected.\t
\tPlease see /tmp/system-installer.log for details.\t\n""")
else:
    sys.exit(1)