#!/usr/bin/python3
"""Delete packages from a repository.

Functions:
==========
* deletes(cachedir, pkgs)
* repo_remove(db, pkgs, opts=None)
"""

import argparse
import logging
import sys

import packaging_scripts.pacman as pacman
import packaging_scripts.pkgfiles as pkgfiles
import packaging_scripts.repos as repos

from pathlib import Path

# ========== Constants ==========
DB_EXT = "db.tar.xz"

# ----- Exit codes -----
E_REPO_REMOVEERR = 2

# ========== Logging setup ==========
console_formatter = logging.Formatter("==> %(levelname)s %(message)s")
syslog = logging.getLogger("packaging_scripts")
syslog.setLevel(logging.DEBUG)

stdout_handler = logging.StreamHandler(sys.stdout)
stdout_handler.setLevel(logging.INFO)
stdout_handler.setFormatter(console_formatter)
stdout_handler.addFilter(lambda record: record.levelno <= logging.INFO)

stderr_handler = logging.StreamHandler(sys.stderr)
stderr_handler.setLevel(logging.WARNING)
stderr_handler.setFormatter(console_formatter)

syslog.addHandler(stdout_handler)
syslog.addHandler(stderr_handler)


# ========== Main Script ==========
def main():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-c", "--cachedir", help="alternative directory to use for deleting package"
    )
    parser.add_argument(
        "-d",
        "--db-filename",
        help="alternative filename for database without extension",
    )
    parser.add_argument(
        "--files-only",
        action="store_true",
        help="only add the built files to the repository",
    )
    parser.add_argument(
        "-s",
        "--sign",
        dest="opts",
        action="append_const",
        const="--sign",
        help="sign repository file",
    )
    parser.add_argument(
        "-v", "--verbose", action="store_true", help="increase script verbosity"
    )
    parser.add_argument(
        "repository",
        choices=pacman.list_configured_repos(),
        help="the repository to operate on",
        metavar="repo",
    )
    parser.add_argument("packages", nargs="+", help="packages to remove")

    args = parser.parse_args()
    repo = args.repository
    pkgs = args.packages

    opts = [] if args.opts is None else args.opts
    cachedir = (
        Path(args.cachedir)
        if args.cachedir
        else Path("/var") / "cache" / "pacman" / repo
    )
    db = (
        cachedir / f"{args.db_filename}.{DB_EXT}"
        if args.db_filename
        else cachedir / f"{repo}.{DB_EXT}"
    )

    if args.verbose:
        stdout_handler.setLevel(logging.DEBUG)

    for pkg in pkgs:
        pkg_tarballs = list(pkgfiles.get(query=pkg, directory=cachedir))
        sigfiles = list(
            pkgfiles.get(query=pkg, directory=cachedir, signatures_only=True)
        )

        for pkgfile in (*pkg_tarballs, *sigfiles):
            pkgfiles.delete(pkgfile)

    if not args.files_only:
        try:
            repos.db_modify("remove", db, *opts, *pkgs)
        except repos.RepoAddError as e:
            syslog.error(e)


if __name__ == "__main__":
    main()