#!/usr/bin/env python3
import glob
import os
import re
import sys


def check_src_files_have_test():
    missing_test_files = []

    excluded = [
        "mitmproxy/contrib/",
        "mitmproxy/io/proto/",
        "mitmproxy/proxy/layers/http",
        "mitmproxy/test/",
        "mitmproxy/tools/",
        "mitmproxy/platform/",
        "mitmproxy/utils/pyinstaller/",
    ]
    src_files = glob.glob("mitmproxy/**/*.py", recursive=True)
    src_files = [f for f in src_files if os.path.basename(f) != "__init__.py"]
    src_files = [
        f for f in src_files if not any(os.path.normpath(p) in f for p in excluded)
    ]
    for f in src_files:
        p = os.path.join("test", os.path.dirname(f), "test_" + os.path.basename(f))
        if not os.path.isfile(p):
            missing_test_files.append((f, p))

    return missing_test_files


def check_test_files_have_src():
    unknown_test_files = []

    excluded = [
        "test/mitmproxy/data/",
        "test/mitmproxy/net/data/",
        "/tservers.py",
        "/conftest.py",
    ]
    test_files = glob.glob("test/mitmproxy/**/*.py", recursive=True)
    test_files = [f for f in test_files if os.path.basename(f) != "__init__.py"]
    test_files = [
        f for f in test_files if not any(os.path.normpath(p) in f for p in excluded)
    ]
    for f in test_files:
        p = os.path.join(
            re.sub("^test/", "", os.path.dirname(f)),
            re.sub("^test_", "", os.path.basename(f)),
        )
        if not os.path.isfile(p):
            unknown_test_files.append((f, p))

    return unknown_test_files


def main():
    exitcode = 0

    missing_test_files = check_src_files_have_test()
    if missing_test_files:
        exitcode += 1
        for f, p in sorted(missing_test_files):
            print(f"{f} MUST have a matching test file: {p}")

    unknown_test_files = check_test_files_have_src()
    if unknown_test_files:
        # TODO: enable this in the future
        # exitcode += 1
        for f, p in sorted(unknown_test_files):
            print(f"{f} DOES NOT MATCH a source file! Expected to find: {p}")

    sys.exit(exitcode)


if __name__ == "__main__":
    main()