forked from Mirror/pmbootstrap
71e7af57e6
We use a custom verbose log level in pmbootstrap, unfortunately it isn't possible to correctly type this due to some limitations in the logging library [1], [2]. Given that our usecase is fairly simple, we can just wrap the module with our own so we only have to tell mypy to ignore the error once instead of at every callsite. [1]: https://github.com/cryptax/droidlysis/issues/15 [2]: https://github.com/python/typing/discussions/980 Signed-off-by: Caleb Connolly <caleb@postmarketos.org>
121 lines
3.8 KiB
Python
121 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
|
|
from .helpers import mount
|
|
from .helpers import other
|
|
from .core import Chroot
|
|
|
|
# pmbootstrap version
|
|
__version__ = "2.3.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: Any) -> 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() -> int:
|
|
# 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(config.work):
|
|
raise RuntimeError("Work path not found, please run 'pmbootstrap"
|
|
" init' to create it.")
|
|
|
|
# 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(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:
|
|
import logging as pylogging
|
|
pylogging.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
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|