import os
import re
import csv


def split_text(text):
    text_list = text.split("Start of Page No. = ")
    text_dict = {page.split()[0]: page for page in text_list[1:]}
    return text_dict


def contains_reimbursement(text, page):
    if isinstance(text, dict):
        return page.isdigit() and (
            "%" in text[page] or "$" in text[page] or "percent " in text[page].lower()
        )
    elif isinstance(text, str):
        return "%" in text or "$" in text or "percent " in text.lower()
    else:
        print("contains_reimbursement - Invalid data type")
        return False


def search_files(directory):
    pattern = re.compile(r"'',\s*'',")
    results = []
    total_files = 0
    files_with_pattern = 0

    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(".txt"):
                total_files += 1
                file_path = os.path.join(root, file)
                try:
                    with open(file_path, "r", encoding="utf-8") as f:
                        content = f.read()
                        pages = split_text(content)

                        reimbursement_pages = [
                            page
                            for page in pages
                            if contains_reimbursement(pages, page)
                        ]
                        table_pages = [
                            page
                            for page in reimbursement_pages
                            if "-------Table Start--------" in pages[page]
                        ]

                        matches = []
                        for page in table_pages:
                            matches.extend(pattern.findall(pages[page]))

                        if matches:
                            files_with_pattern += 1
                            results.append(
                                (
                                    file_path,
                                    len(matches),
                                    True,
                                    len(reimbursement_pages),
                                    len(table_pages),
                                )
                            )
                        else:
                            results.append(
                                (
                                    file_path,
                                    0,
                                    False,
                                    len(reimbursement_pages),
                                    len(table_pages),
                                )
                            )
                except Exception as e:
                    print(f"Error reading {file_path}: {str(e)}")
                    results.append((file_path, 0, False, 0, 0))

    return results, total_files, files_with_pattern


def write_to_csv(results, total_files, files_with_pattern, output_file):
    with open(output_file, "w", newline="", encoding="utf-8") as csvfile:
        writer = csv.writer(csvfile)
        writer.writerow(
            [
                "File Name",
                "Match Count",
                "Contains Pattern",
                "Reimbursement Pages",
                "Table Pages",
                "Percentage of Total",
            ]
        )

        for (
            file_path,
            match_count,
            contains_pattern,
            reimbursement_pages,
            table_pages,
        ) in results:
            percentage = (match_count / total_files) * 100 if total_files > 0 else 0
            writer.writerow(
                [
                    file_path,
                    match_count,
                    contains_pattern,
                    reimbursement_pages,
                    table_pages,
                    f"{percentage:.2f}%",
                ]
            )

        percentage_with_pattern = (
            (files_with_pattern / total_files) * 100 if total_files > 0 else 0
        )
        writer.writerow(["", "", "", "", "", ""])
        writer.writerow(["Summary", "", "", "", "", ""])
        writer.writerow(["Total Files", total_files, "", "", "", ""])
        writer.writerow(
            [
                "Files with Pattern",
                files_with_pattern,
                "",
                "",
                "",
                f"{percentage_with_pattern:.2f}%",
            ]
        )


if __name__ == "__main__":

    current_dir = os.path.dirname(os.path.abspath(__file__))

    facility_dir = "batch1_priority"

    if os.path.isdir(facility_dir):
        results, total_files, files_with_pattern = search_files(facility_dir)
        output_file = os.path.join(current_dir, "contract_analysis_results.csv")
        write_to_csv(results, total_files, files_with_pattern, output_file)
        print(f"Results have been written to {output_file}")
    else:
        print(f"Error: The directory {facility_dir} does not exist.")