from __future__ import annotations
import pytest
import sys

from PIL import Image
from PIL import PdfParser
from PIL import SpiderImagePlugin
from PIL import MpegImagePlugin
from PIL import ImageCms
from PIL import McIdasImagePlugin
from PIL import ImageTk
from PIL import ImageFile

pytest_plugins = ["Tests.helper"]


def calculate_coverage(test_name):
    all_branches = {
        "branches1": Image.branches,                        # duru
        "branches2": PdfParser.XrefTable.branches,          # duru
        "branches3": SpiderImagePlugin.branches,            # isidora
        "branches4": MpegImagePlugin.BitStream.branches,    # isidora
        "branches5": ImageTk.BitmapImage.branches,          # deekshu
        "branches6": ImageFile.PyEncoder.branches,          # deekshu
        # Add more
    }

    for name, branches in all_branches.items():
        num_branches = len(branches)
        branch_covered = {key: value for key, value in branches.items() if value is True}
        sum_branches = len(branch_covered)
        coverage = (sum_branches / num_branches) * 100

        print(f"\n{name} - Branches covered: {sum_branches}")
        print(f"{name} - Total branches: {num_branches}")
        print(f"{name} - BRANCH COVERAGE: {coverage}%\n")

    return all_branches["branches1"]


@pytest.hookimpl(tryfirst=True)
def pytest_runtest_protocol(item, nextitem):
    global test_name

    last_arg = sys.argv[-1]

    test_name = last_arg.split('/')[-1].split('::')[-1]

    test_name = test_name.rstrip('.py')

    return None

@pytest.hookimpl(tryfirst=True)
def pytest_sessionfinish(session, exitstatus):
    global test_name

    coverage = calculate_coverage(test_name)
    print("\nBRANCH COVERAGE for", test_name, ":", coverage, "%\n")