from extract import get_custom_format, get_regex
import re
import sys

# ANSI escape codes for colors
GREEN = '\033[92m'
RED = '\033[91m'
RESET = '\033[0m'

def roku(debug_level=0):
    # Get the custom formats for "roku" from both Radarr and Sonarr
    roku_radarr = get_custom_format("roku", "radarr", debug_level)
    roku_sonarr = get_custom_format("roku", "sonarr", debug_level)

    # Extract the regex values for both Radarr and Sonarr using get_regex
    roku_value_radarr = get_regex(roku_radarr, "roku", debug_level)
    roku_value_sonarr = get_regex(roku_sonarr, "roku", debug_level)

    if debug_level > 0:
        print(f"Testing with regex: {roku_value_radarr}")

    # Compare Radarr and Sonarr Roku regex values
    if roku_value_radarr != roku_value_sonarr:
        print("Test Failed: regex value not same.")
        print(f"Radarr regex: {roku_value_radarr}")
        print(f"Sonarr regex: {roku_value_sonarr}")
        sys.exit(1)

    radarr_good_matches = [
        "Weird The Al Yankovic Story 2022 1080p ROKU WEB-DL DD5.1 H.264-SMURF",
        "The.Spiderwick.Chronicles.2024.S01E06.1028.Teeth.1080p.ROKU.WEB-DL.DD5.1.H.264-playWEB",
        "The Imitation Game 2014 1080p ROKU WEB-DL AAC 2 0 H 264-PiRaTeS"
    ]

    radarr_bad_matches = [
        "Ikimono no kiroku 1955 720p BluRay FLAC x264-EA.mkv"
    ]

    sonarr_good_matches = [
        "The Now S01 1080p ROKU WEB-DL DD5 1 H 264-WELP",
        "The Rockford Files S01 1080p ROKU WEB-DL HE-AAC 2 0 H 264-PiRaTeS",
        "50.States.of.Fright.S02E05.13.Steps.to.Hell.Washington.Part.2.1080p.ROKU.WEB-DL.DD5.1.H.264-NTb"
    ]

    sonarr_bad_matches = [
        "Avatar.The.Last.Airbender.S01E08.Avatar.Roku.Winter.Solstice.2.1080p.AMZN.WEB-DL.DD+2.0.H.264-CtrlHD",
        "[HorribleSubs] Rokujouma no Shinryakusha - 01 [480p]"
    ]

    failed_good_matches = []
    failed_bad_matches = []

    # Print Radarr Good Matches
    print("\nRadarr Releases:")
    print("----------------")
    print("Should Match:")
    for term in radarr_good_matches:
        if re.search(roku_value_radarr, term, re.IGNORECASE):
            print(f"  - {term}: {GREEN}Passed{RESET}")
        else:
            failed_good_matches.append(("Radarr", term))
            print(f"  - {term}: {RED}Failed{RESET}")

    # Print Radarr Bad Matches
    print("\nShould NOT Match:")
    for term in radarr_bad_matches:
        if not re.search(roku_value_radarr, term, re.IGNORECASE):
            print(f"  - {term}: {GREEN}Passed{RESET}")
        else:
            failed_bad_matches.append(("Radarr", term))
            print(f"  - {term}: {RED}Failed{RESET}")

    # Print Sonarr Good Matches
    print("\nSonarr Releases:")
    print("----------------")
    print("Should Match:")
    for term in sonarr_good_matches:
        if re.search(roku_value_sonarr, term, re.IGNORECASE):
            print(f"  - {term}: {GREEN}Passed{RESET}")
        else:
            failed_good_matches.append(("Sonarr", term))
            print(f"  - {term}: {RED}Failed{RESET}")

    # Print Sonarr Bad Matches
    print("\nShould NOT Match:")
    for term in sonarr_bad_matches:
        if not re.search(roku_value_sonarr, term, re.IGNORECASE):
            print(f"  - {term}: {GREEN}Passed{RESET}")
        else:
            failed_bad_matches.append(("Sonarr", term))
            print(f"  - {term}: {RED}Failed{RESET}")

    # Determine and print overall test result
    if not failed_good_matches and not failed_bad_matches:
        return True
    else:
        return False