packaging-scripts/bin/addpkg

#!/usr/bin/python3
""" Add packages to a repository.

Functions:
==========
* add_pkgfiles(cachedir, pkgs)
* repo_add(db, pkgs, opts=None)
"""

import argparse
import logging
import os.path
import shutil
import sys

import packaging_scripts.pacmanconf as pacmanconf
import packaging_scripts.pkgfiles as pkgfiles
import packaging_scripts.repos as repos

# ========== Constants ==========
DB_EXT = "db.tar.xz"
LOGFORMAT = "==> %(levelname)s %(message)s"

# ========== Exit codes ==========
E_NOFILESERR = 1
E_REPO_ADDERR = 2

# ========== Logging setup ==========
console_formatter = logging.Formatter(LOGFORMAT)
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)


# ========== Functions ==========
def add_pkgfile(pkgfile, cachedir):
    """Add package file to the repository directory.

    :param pkg: path of package to add
    :type pkg: str, bytes, or path-like object
    :param cachedir: cache directory to move package to
    :type cachedir: str, bytes, or path-like object
    """
    syslog.info(f"Adding {pkgfile} to {cachedir}")
    shutil.move(pkgfile, os.path.join(cachedir, os.path.basename(pkgfile)))


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-c",
        "--cachedir",
        help="alternative directory to use for moving package"
    )
    parser.add_argument(
        "-d",
        "--db-filename",
        help="alternative filename for database without extension"
    )
    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=pacmanconf.list_configured_repos(),
        help="the repository to operate on",
        metavar="repo",
    )
    parser.add_argument("packages", default=None, nargs="*", help="packages to add")

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

    if args.opts is None:
        opts = []
    else:
        opts = args.opts

    if args.cachedir is not None:
        cachedir = args.cachedir
    else:
        cachedir = os.path.join("/var", "cache", "pacman", repo)

    if args.db-filename is not None:
        db = os.path.join(cachedir, f"{args.db-filename}.{DB_EXT}")
    else:
        db = os.path.join(cachedir, f"{repo}.{DB_EXT}")

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

    if pkgs:
        pkg_tarballs = pkgs
        sigfiles = []
    else:
        pkg_tarballs = pkgfiles.get_pkgfiles()
        sigfiles = pkgfiles.get_pkgfiles(signatures_only=True)

    if not pkg_tarballs:
        syslog.critical("No package tarballs have been found, exiting")
        exit(E_NOFILESERR)

    try:
        repos.db_modify("add", db, *opts, *pkg_tarballs)
    except repos.RepoAddError as e:
        syslog.error(e)
        exit(E_REPO_ADDERR)

    for pkgfile in (*pkg_tarballs, *sigfiles):
        add_pkgfile(pkgfile, cachedir)