#! /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


def read_diff(filename):
    diff = dict()
    with open(filename) as f:
        for line in f:
            pkg, old, new = line.split()
            pkgname, arch = pkg.split(':')
            diff[(pkgname, arch)] = (old, new)
    return diff


def apply_ma_diff(diff, inPackages, outPackages, verbose=False):
    nr_modified = 0
    nr_all = 0
    for pkg in inPackages:
        nr_all += 1
        if pkg['Architecture'] != 'all':
            arch = 'any'
        else:
            arch = 'all'
        d = diff.get((pkg['Package'], arch))
        if d:
            old, new = d
            if pkg.get('Multi-Arch', 'no') != old \
               and pkg.get('Multi-Arch', 'no') != new:
                print("failed to apply patch because %s is Multi-Arch:%s and"
                      " not Multi-Arch:%s" % (pkg['Package'],
                                              pkg.get('Multi-Arch', 'no'),
                                              old), file=sys.stderr)
                return False
            if pkg.get('Multi-Arch', 'no') == new:
                continue
            pkg['Multi-Arch'] = new
            nr_modified += 1

    with outPackages as outfile:
        for pkg in inPackages:
            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=("Apply a multiarch diff as created by botch-ma-diff"))
    parser.add_argument(
        'diff', type=read_diff, help='diff output as created by botch-ma-diff')
    parser.add_argument(
        'inPackages', type=read_tag_file, help='input Packages file')
    parser.add_argument(
        'outPackages', type=get_fh_out, help='output Packages file')
    parser.add_argument('--verbose', action='store_true', help='be verbose')
    args = parser.parse_args()
    ret = apply_ma_diff(
        args.diff, args.inPackages, args.outPackages, args.verbose)
    exit(not ret)
