org.osbuild.dnf4.mark

#!/usr/bin/python3
import shutil
import subprocess
import sys

from osbuild import api


def mark(tree, packages):
    dnf_bin = shutil.which("dnf-3")

    if not dnf_bin:
        print("dnf not found")
        return 1

    markings = {}

    for package in packages:
        if package["mark"] not in markings:
            markings[package["mark"]] = []

        markings[package["mark"]] += [package]

    if "install" in markings:
        subprocess.run(
            [dnf_bin, "--installroot", tree, "mark", "-y", "install"]
            + [package["name"] for package in markings["install"]],
            check=True,
        )

    if "group" in markings:
        subprocess.run(
            [dnf_bin, "--installroot", tree, "mark", "-y", "group"]
            + [package["name"] for package in markings["group"]],
            check=True,
        )

    return 0


def main(tree, options):
    return mark(tree, options["packages"])


if __name__ == "__main__":
    args = api.arguments()
    r = main(args["tree"], args["options"])
    sys.exit(r)