# -*- Python -*- vim: set syntax=python tabstop=4 expandtab cc=80:
#===----------------------------------------------------------------------===##
#
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
#
#===----------------------------------------------------------------------===##
"""
diff - A set of functions for diff-ing two symbol lists.
"""

from libcudacxx.sym_check import util


def _symbol_difference(lhs, rhs):
    lhs_names = set(((n['name'], n['type']) for n in lhs))
    rhs_names = set(((n['name'], n['type']) for n in rhs))
    diff_names = lhs_names - rhs_names
    return [n for n in lhs if (n['name'], n['type']) in diff_names]


def _find_by_key(sym_list, k):
    for sym in sym_list:
        if sym['name'] == k:
            return sym
    return None


def added_symbols(old, new):
    return _symbol_difference(new, old)


def removed_symbols(old, new):
    return _symbol_difference(old, new)


def changed_symbols(old, new):
    changed = []
    for old_sym in old:
        if old_sym in new:
            continue
        new_sym = _find_by_key(new, old_sym['name'])
        if (new_sym is not None and not new_sym in old
                and old_sym != new_sym):
            changed += [(old_sym, new_sym)]
    return changed


def diff(old, new):
    added = added_symbols(old, new)
    removed = removed_symbols(old, new)
    changed = changed_symbols(old, new)
    return added, removed, changed


def report_diff(added_syms, removed_syms, changed_syms, names_only=False,
                demangle=True):
    def maybe_demangle(name):
        return util.demangle_symbol(name) if demangle else name

    report = ''
    for sym in added_syms:
        report += 'Symbol added: %s\n' % maybe_demangle(sym['name'])
        if not names_only:
            report += '    %s\n\n' % sym
    if added_syms and names_only:
        report += '\n'
    for sym in removed_syms:
        report += 'SYMBOL REMOVED: %s\n' % maybe_demangle(sym['name'])
        if not names_only:
            report += '    %s\n\n' % sym
    if removed_syms and names_only:
        report += '\n'
    if not names_only:
        for sym_pair in changed_syms:
            old_sym, new_sym = sym_pair
            old_str = '\n    OLD SYMBOL: %s' % old_sym
            new_str = '\n    NEW SYMBOL: %s' % new_sym
            report += ('SYMBOL CHANGED: %s%s%s\n\n' %
                       (maybe_demangle(old_sym['name']),
                        old_str, new_str))

    added = bool(len(added_syms) != 0)
    abi_break = bool(len(removed_syms))
    if not names_only:
        abi_break = abi_break or len(changed_syms)
    if added or abi_break:
        report += 'Summary\n'
        report += '    Added:   %d\n' % len(added_syms)
        report += '    Removed: %d\n' % len(removed_syms)
        if not names_only:
            report += '    Changed: %d\n' % len(changed_syms)
        if not abi_break:
            report += 'Symbols added.'
        else:
            report += 'ABI BREAKAGE: SYMBOLS ADDED OR REMOVED!'
    else:
        report += 'Symbols match.'
    is_different = abi_break or bool(len(added_syms)) \
                   or bool(len(changed_syms))
    return report, abi_break, is_different
