forked from Mirror/pmbootstrap
866e5bcfab
Move pmb/parse/arch.py over to core and refactor it as an Arch type, similar to how Chroot was done. Fix all the uses (that I can find) of arch in the codebase that need adjusting. The new Arch type is an Enum, making it clear what architectures can be represented and making it much easier to reason about. Since we support ~5 (kinda) different representations of an Architecture (Alpine, Kernel, target triple, platform, and QEMU), we now formalise that the Alpine format is what we represent internally, with methods to convert to any of the others as-needed. Signed-off-by: Caleb Connolly <caleb@postmarketos.org>
127 lines
3.9 KiB
Python
127 lines
3.9 KiB
Python
# Copyright 2023 Oliver Smith
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
""" Save, read, verify workdir state related information in $WORK/workdir.cfg,
|
|
for example the init dates of the chroots. This is not saved in
|
|
pmbootstrap.cfg, because pmbootstrap.cfg is not tied to a specific work
|
|
dir. """
|
|
import configparser
|
|
import os
|
|
import time
|
|
from typing import Optional
|
|
|
|
import pmb.config
|
|
import pmb.config.pmaports
|
|
from pmb.core import Chroot, get_context
|
|
from pmb.types import PmbArgs
|
|
|
|
|
|
def chroot_save_init(suffix: Chroot):
|
|
"""Save the chroot initialization data in $WORK/workdir.cfg."""
|
|
# Read existing cfg
|
|
cfg = configparser.ConfigParser()
|
|
path = get_context().config.work / "workdir.cfg"
|
|
if os.path.isfile(path):
|
|
cfg.read(path)
|
|
|
|
# Create sections
|
|
for key in ["chroot-init-dates", "chroot-channels"]:
|
|
if key not in cfg:
|
|
cfg[key] = {}
|
|
|
|
# Update sections
|
|
channel = pmb.config.pmaports.read_config(support_systemd=False)["channel"]
|
|
cfg["chroot-channels"][str(suffix)] = channel
|
|
cfg["chroot-init-dates"][str(suffix)] = str(int(time.time()))
|
|
|
|
# Write back
|
|
with open(path, "w") as handle:
|
|
cfg.write(handle)
|
|
|
|
|
|
def chroots_outdated(chroot: Optional[Chroot]=None):
|
|
"""Check if init dates from workdir.cfg indicate that any chroot is
|
|
outdated.
|
|
|
|
:param suffix: only check a specific chroot suffix
|
|
|
|
:returns: True if any of the chroots are outdated and should be zapped,
|
|
False otherwise
|
|
"""
|
|
# Skip if workdir.cfg doesn't exist
|
|
path = get_context().config.work / "workdir.cfg"
|
|
if not os.path.exists(path):
|
|
return False
|
|
|
|
cfg = configparser.ConfigParser()
|
|
cfg.read(path)
|
|
key = "chroot-init-dates"
|
|
if key not in cfg:
|
|
return False
|
|
|
|
date_outdated = time.time() - pmb.config.chroot_outdated
|
|
for cfg_suffix in cfg[key]:
|
|
if chroot and cfg_suffix != str(chroot):
|
|
continue
|
|
date_init = int(cfg[key][cfg_suffix])
|
|
if date_init <= date_outdated:
|
|
return True
|
|
return False
|
|
|
|
|
|
def chroot_check_channel(chroot: Chroot):
|
|
path = get_context().config.work / "workdir.cfg"
|
|
msg_again = "Run 'pmbootstrap zap' to delete your chroots and try again."
|
|
msg_unknown = ("Could not figure out on which release channel the"
|
|
f" '{chroot}' chroot is.")
|
|
if not os.path.exists(path):
|
|
raise RuntimeError(f"{msg_unknown} {msg_again}")
|
|
|
|
cfg = configparser.ConfigParser()
|
|
cfg.read(path)
|
|
key = "chroot-channels"
|
|
if key not in cfg or str(chroot) not in cfg[key]:
|
|
raise RuntimeError(f"{msg_unknown} {msg_again}")
|
|
|
|
# Exclude systemd repo
|
|
channel = pmb.config.pmaports.read_config(support_systemd=False)["channel"]
|
|
channel_cfg = cfg[key][str(chroot)]
|
|
if channel != channel_cfg:
|
|
raise RuntimeError(f"Chroot '{chroot}' was created for the"
|
|
f" '{channel_cfg}' channel, but you are on the"
|
|
f" '{channel}' channel now. {msg_again}")
|
|
|
|
|
|
def clean():
|
|
"""Remove obsolete data data from workdir.cfg.
|
|
|
|
:returns: None if workdir does not exist,
|
|
True if config was rewritten,
|
|
False if config did not change
|
|
"""
|
|
# Skip if workdir.cfg doesn't exist
|
|
path = get_context().config.work / "workdir.cfg"
|
|
if not os.path.exists(path):
|
|
return None
|
|
|
|
# Read
|
|
cfg = configparser.ConfigParser()
|
|
cfg.read(path)
|
|
|
|
# Remove entries for deleted chroots
|
|
changed = False
|
|
for key in ["chroot-init-dates", "chroot-channels"]:
|
|
if key not in cfg:
|
|
continue
|
|
for suffix_str in cfg[key]:
|
|
suffix = Chroot.from_str(suffix_str)
|
|
if suffix.path.exists():
|
|
continue
|
|
changed = True
|
|
del cfg[key][suffix_str]
|
|
|
|
# Write back
|
|
if changed:
|
|
with path.open("w") as handle:
|
|
cfg.write(handle)
|
|
|
|
return changed
|