#! /usr/bin/python3

from __future__ import print_function
import sys

sys.path.append("/usr/share/botch")
from util import get_fh_out, read_tag_file, arch_matches


def add_arch(toarch, inSources, outSources, verbose=False):
    nr_modified = 0
    nr_all = 0
    for pkg in inSources:
        nr_all += 1
        arch = pkg["Architecture"]
        if not any([arch_matches(toarch, a) for a in arch.split()]):
            pkg["Architecture"] = arch + " " + toarch
            nr_modified += 1

    with outSources as outfile:
        for pkg in inSources:
            pkg.dump(outfile)
            outfile.write(b"\n")

    if verbose:
        print("modified %d out of %d packages" % (nr_modified, nr_all), file=sys.stderr)
    return True


if __name__ == "__main__":
    import argparse

    parser = argparse.ArgumentParser(
        description=(
            "Add architecture toarch to all source packages in"
            + " inSources which do not match toarch yet."
        )
    )
    parser.add_argument("toarch", help="architecture to add")
    parser.add_argument("inSources", type=read_tag_file, help="input Sources file")
    parser.add_argument("outSources", type=get_fh_out, help="output Sources file")
    parser.add_argument("-v", "--verbose", action="store_true", help="be verbose")
    args = parser.parse_args()
    ret = add_arch(args.toarch, args.inSources, args.outSources, args.verbose)
    exit(not ret)
