import csv
import subprocess
import io

class Test:
    def __init__(self, number: int, description: str):
        self.number: int = number
        self.description: str = description
        self.failed: bool = False
        self.logs: list[str, str, str] = [] # log_type, timestamp, message
        
    def add_log(self, log_type: str, timestamp: str, fail_message: str):
        if log_type == "FAIL":
            self.failed = True
            
        new_log = {
            "log_type": log_type,
            "timestamp": timestamp,
            "fail_message": fail_message
        }
        self.logs.append(new_log)

class Category:
    def __init__(self, name: str):
        self.name: str = name
        self.failed: bool = False
        self.tests: list[Test] = []
        
    def is_number_in_tests(self, number: int) -> bool:
        return self.tests and self.tests[-1].number == number
    
    def add_test(self, log_type: str, number: int, timestamp: str, description: str, fail_message: str):
        if log_type == "FAIL":
            self.failed = True
            
        if not self.is_number_in_tests(number):
            new_test = Test(number, description)
            self.tests.append(new_test)

        self.tests[-1].add_log(log_type, timestamp, fail_message)

def find_category_in_data(name: str, data: list[Category]) -> int:
    for index, category in enumerate(data):
        if category.name == name:
            return index
    return -1 

def parse_data(program_name: str):
    data = [] # List of Category objects
    process = subprocess.Popen(['HVSensorTest.exe'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    stdout, stderr = process.communicate()

    if stderr:
        print("Error:", stderr.decode())

    stdout_str = stdout.decode()
    stdout_file = io.StringIO(stdout_str)

    csv_reader = csv.DictReader(stdout_file) # Rows are dictionaries
    for row in csv_reader:
        log_type, timestamp, category_name, number, test_description, fail_message = row.values()

        index = find_category_in_data(category_name, data)
        if index == -1:
            category = Category(category_name)
            data.append(category)
        else:
            category = data[index]

        category.add_test(log_type, number, timestamp, test_description, fail_message)
            
    # Final test statistics
    stats = {
        "failed": False,
        "number_tests": 0,
        "number_passed": 0,
        "percentage_passed": 0,
        "number_failed": 0,
        "percentage_failed": 0,
    }

    for category in data:
        if category.failed:
            stats["number_failed"] = stats["number_failed"] + 1
            stats["failed"] = True
        else:
            stats["number_passed"] = stats["number_passed"] + 1

    stats["number_tests"] = len(data)
    stats["percentage_passed"] = round((stats["number_passed"] / len(data)) * 100, 1)
    stats["percentage_failed"] = 100 - stats["percentage_passed"]
    
    return data, stats