#!/usr/bin/env python3

import sys
import glob
from pathlib import Path

correct_licence = """\

//              Copyright Catch2 Authors
// Distributed under the Boost Software License, Version 1.0.
//   (See accompanying file LICENSE.txt or copy at
//        https://www.boost.org/LICENSE_1_0.txt)

// SPDX-License-Identifier: BSL-1.0
"""

progmasoft_licence = """\
// Copyright (c) 2026 Progmasoft.
// SPDX-License-Identifier: MPL-2.0 WITH AdditionRef-Progmasoft-Exception-1.1
"""

def check_licence_in_file(filename: str) -> bool:
    # Keep upstream files on BSL-1.0 while checking the fork's own source
    # and tests against their separate MPL exception header.
    expected = (progmasoft_licence if 'Progmasoft' in Path(filename).parts
                else correct_licence)
    with open(filename, 'r', encoding='utf-8') as f:
        file_preamble = f.read(len(expected))

    if expected != file_preamble:
        print('File {} does not have proper licence'.format(filename))
        return False
    return True

def check_licences_in_path(path: str) -> int:
    failed = 0
    files_to_check = glob.glob(path + '/**/*.cpp', recursive=True) \
                   + glob.glob(path + '/**/*.hpp', recursive=True)
    for file in files_to_check:
        if not check_licence_in_file(file):
            failed += 1
    return failed

def check_licences():
    failed = 0
    # Add 'extras' after the amalgamated files are regenerated with the new script (past 3.4.0)
    roots = ['src/catch2', 'src/Progmasoft', 'tests', 'examples', 'fuzzing']
    for root in roots:
        failed += check_licences_in_path(root)
    
    if failed:
        print('{} files are missing licence'.format(failed))
        sys.exit(1)

if __name__ == "__main__":
    check_licences()