forked from Mirror/pmbootstrap
db163aded6
Make the handling of the custom NonBugError and BuildFailedError exceptions more consistent with the handling of other exceptions, by printing "ERROR: " infront of the actual error text. Then we don't need to duplicate that where we raise the errors. pmbootstrap prints "ERROR" in red.
119 lines
3.8 KiB
Python
119 lines
3.8 KiB
Python
# Copyright 2023 Oliver Smith
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
# PYTHON_ARGCOMPLETE_OK
|
|
import sys
|
|
import logging
|
|
import os
|
|
import traceback
|
|
from argparse import Namespace
|
|
|
|
from pmb.helpers.exceptions import BuildFailedError, NonBugError
|
|
|
|
from . import config
|
|
from . import parse
|
|
from .config import init as config_init
|
|
from .helpers import frontend
|
|
from .helpers import logging as pmb_logging
|
|
from .helpers import mount
|
|
from .helpers import other
|
|
|
|
# pmbootstrap version
|
|
__version__ = "2.2.1"
|
|
|
|
# Python version check
|
|
version = sys.version_info
|
|
if version < (3, 9):
|
|
print("You need at least Python 3.9 to run pmbootstrap")
|
|
print("(You are running it with Python " + str(version.major) +
|
|
"." + str(version.minor) + ")")
|
|
sys.exit()
|
|
|
|
|
|
def print_log_hint(args: Namespace) -> None:
|
|
# Hints about the log file (print to stdout only)
|
|
log_hint = "Run 'pmbootstrap log' for details."
|
|
if not args or not os.path.exists(args.log):
|
|
log_hint += (" Alternatively you can use '--details-to-stdout' to get more"
|
|
" output, e.g. 'pmbootstrap --details-to-stdout init'.")
|
|
print()
|
|
print(log_hint)
|
|
|
|
|
|
def main():
|
|
# Wrap everything to display nice error messages
|
|
args = None
|
|
try:
|
|
# Parse arguments, set up logging
|
|
args = parse.arguments()
|
|
os.umask(0o22)
|
|
|
|
# Store script invocation command
|
|
os.environ["PMBOOTSTRAP_CMD"] = sys.argv[0]
|
|
|
|
# Sanity checks
|
|
other.check_grsec()
|
|
if not args.as_root and os.geteuid() == 0:
|
|
raise RuntimeError("Do not run pmbootstrap as root!")
|
|
|
|
# Initialize or require config
|
|
if args.action == "init":
|
|
return config_init.frontend(args)
|
|
elif not os.path.exists(args.config):
|
|
raise RuntimeError("Please specify a config file, or run"
|
|
" 'pmbootstrap init' to generate one.")
|
|
elif not os.path.exists(args.work):
|
|
raise RuntimeError("Work path not found, please run 'pmbootstrap"
|
|
" init' to create it.")
|
|
|
|
other.check_old_devices(args)
|
|
|
|
# Migrate work folder if necessary
|
|
if args.action not in ["shutdown", "zap", "log"]:
|
|
other.migrate_work_folder(args)
|
|
|
|
# Run the function with the action's name (in pmb/helpers/frontend.py)
|
|
if args.action:
|
|
getattr(frontend, args.action)(args)
|
|
else:
|
|
logging.info("Run pmbootstrap -h for usage information.")
|
|
|
|
# Still active notice
|
|
if mount.ismount(args.work + "/chroot_native/dev"):
|
|
logging.info("NOTE: chroot is still active (use 'pmbootstrap"
|
|
" shutdown' as necessary)")
|
|
logging.info("DONE!")
|
|
|
|
except KeyboardInterrupt:
|
|
print("\nCaught KeyboardInterrupt, exiting …")
|
|
sys.exit(130) # SIGINT(2) + 128
|
|
|
|
except NonBugError as exception:
|
|
logging.error(f"ERROR: {exception}")
|
|
return 2
|
|
|
|
except BuildFailedError as exception:
|
|
logging.error(f"ERROR: {exception}")
|
|
print_log_hint(args)
|
|
return 3
|
|
|
|
except Exception as e:
|
|
# Dump log to stdout when args (and therefore logging) init failed
|
|
if not args:
|
|
logging.getLogger().setLevel(logging.DEBUG)
|
|
|
|
logging.info("ERROR: " + str(e))
|
|
logging.info("See also: <https://postmarketos.org/troubleshooting>")
|
|
logging.debug(traceback.format_exc())
|
|
|
|
print_log_hint(args)
|
|
print()
|
|
print("Before you report this error, ensure that pmbootstrap is "
|
|
"up to date.")
|
|
print("Find the latest version here:"
|
|
" https://gitlab.com/postmarketOS/pmbootstrap/-/tags")
|
|
print(f"Your version: {__version__}")
|
|
return 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|