'''Check whether the manpage extensions and directories list hardcoded in brp-compress

are the same as the lists stored in pyproject_save_files.py.

There is an open issue for RPM to provide them both as macros:

https://github.com/rpm-software-management/rpm/issues/1865

Once that happens, this script can be removed.

'''

import argparse

import re

import sys

from pathlib import PosixPath

from pyproject_buildrequires import print_err

from pyproject_save_files import prepend_mandirs, MANPAGE_EXTENSIONS

def read_brp_compress(filename):

    contents = filename.read_text()

    # To avoid duplicity of the manpage extensions which are listed a few times

    # in the source file, they are stored in set and then retyped to a sorted list

    manpage_exts = sorted(

        set(re.findall(r'\(?(\w+)\\+\)?\$?', contents))

    )

    # Get rid of ${PREFIX} when extracting the manpage directories

    mandirs = [

        entry.replace('.${PREFIX}', '/PREFIX')

        for entry in contents.split()

        if entry.startswith('.${PREFIX}')

    ]

    return manpage_exts, sorted(mandirs)

def compare_mandirs(brp_compress_mandirs):

    '''

    Check whether each of brp-compress mandirs entry is present in the list

    stored in pyproject_save_files.py

    '''

    pyp_save_files_mandirs = sorted(prepend_mandirs(prefix='/PREFIX'))

    if brp_compress_mandirs == pyp_save_files_mandirs:

        return True

    else:

        print_err('Mandir lists don\'t match, update the list in pyproject_save_files.py')

        print_err('brp-compress list:', brp_compress_mandirs)

        print_err('pyproject_save_files list:', pyp_save_files_mandirs)

        return False

def compare_manpage_extensions(brp_compress_manpage_exts):

    '''

    Check whether each of brp-compress manpage extension is present in the list

    stored in pyproject_save_files.py

    '''

    if brp_compress_manpage_exts == sorted(MANPAGE_EXTENSIONS):

        return True

    else:

        print_err('Manpage extension lists don\'t match, update the list in pyproject_save_files.py')

        print_err('brp-compress list:', brp_compress_manpage_exts)

        print_err('pyproject_save_files list:', sorted(MANPAGE_EXTENSIONS))

        return False

def main(args):

    src_manpage_exts, src_mandirs = read_brp_compress(args.filename)

    extension_check_successful = compare_manpage_extensions(src_manpage_exts)

    mandir_check_successful = compare_mandirs(src_mandirs)

    if extension_check_successful and mandir_check_successful:

        sys.exit(0)

    else:

        sys.exit(1)

if __name__ == '__main__':

    parser = argparse.ArgumentParser()

    parser.add_argument('-f', '--filename', type=PosixPath, required=True,

                        help='Provide location of brp-compress file')

    main(parser.parse_args())