import os
import time
import subprocess
from typing import Dict, Any
from flow_modules.aiflows.InterpreterFlowModule import InterpreterAtomicFlow


class CodeTestingAtomicFlow(InterpreterAtomicFlow):
    """This class inherits from InterpreterAtomicFlow and is used to test code.
    
    *Input Interface*:
    - `temp_code_file_location`: Location of the file containing the code to be tested.
    
    *Output Interface*:
    - `feedback`: Feedback from the test (i.e. test results).
    """
    def _open_file_and_wait_for_file_update(self, file_location, check_interval=1):
        process = subprocess.Popen(["code", "--wait", file_location])
        while True:
            if process.poll() is not None:
                break
            time.sleep(check_interval)
    def _prepare_code_and_lang(self, input_data: Dict[str, Any]):
        file_location = input_data["temp_code_file_location"]
        input_data["language"] = "python"
        with open(file_location, "r") as file:
            code_str = file.read()
        input_data["code"] = code_str

    def _check_input(self, input_data: Dict[str, Any]):
        assert "temp_code_file_location" in input_data, "temp_code_file_location not passed to CodeTestingAtomicFlow"

    def _delete_file(self, file_location):
        if os.path.exists(file_location):
            os.remove(file_location)

    def run(
            self,
            input_data: Dict[str, Any]):
        self._check_input(input_data)
        file_loc = input_data["temp_code_file_location"]
        self._open_file_and_wait_for_file_update(file_loc)
        self._prepare_code_and_lang(input_data)
        self._process_input_data(input_data)
        response = self._call()
        self._delete_file(file_loc)
        if response.strip() == '':
            response = "test passed"
        return {"feedback": response}