import squish
import test
import object
import names
from leahi_dialin.ui.td_messaging import TD_Messaging
from leahi_dialin.common.td_defs import TDOpModes,TDTreatmentStates
from configuration import config, navigation

td =TD_Messaging()
   
def get_object_from_names(names_dict, error_message = "Missing object", timeout_ms = 200):
    """
        To get an object with try..except catching to prevent script errors when the object is not found on the GUI
        @param names_dict - the dictionary element from the names.py file (ie: names.some_variable_name_of_element)
        @returns the object with corresponding dictionary, otherwise "None"
    """     
    try:
        return squish.waitForObject(names_dict, timeout_ms)
    except LookupError:
        test.fail("ERROR : " + error_message)
        return None   

def setObjectText(obj, text):
    obj["text"] = text
    return obj

def findObjectById(parent, id):
    """
    Recursively searches for a child object by its id.
    Returns the found object or None if not found.
    """
    if str(parent.id) == id:
        return parent

    for child in object.children(parent):
        found = findObjectById(child, id)
        if found:
            return found
    return None

def findChildByText(parent_object, target_text):
    """Recursively finds a child object by its text property."""
    for child in object.children(parent_object):
        if hasattr(child, "text") and str(child.text) == target_text:
            return child
        found = findChildByText(child, target_text)
        if found:
            return found
        
def set_value_based_on_target(obj, target_value):
    """
    obj: dictionary containing object paths
         Example:
         {
             "value_obj": ":mainTreatmentScreen.PressureText",
             "left_arrow": ":mainTreatmentScreen.LeftArrow",
             "right_arrow": ":mainTreatmentScreen.RightArrow"
         }
 
    target_value: integer or string number, e.g. 220
    """
    target_value = target_value
 
    # Wait for all objects
    parent_obj = squish.waitForObjectExists(obj)
    # change range as per your screen count
    
    left_arrow = findObjectById(parent_obj, "_leftArrow")
    right_arrow =findObjectById(parent_obj, "_rightArrow")
 
    # Read current value (supports invisible text too)
    
    current_value = round(float(squish.findObject(obj).value),1)
    
    # Determine direction
    while current_value != float(target_value):
        if current_value < float(target_value):
            squish.mouseClick(squish.waitForObject(right_arrow))

        elif current_value > float(target_value):
            squish.mouseClick(squish.waitForObject(left_arrow))
        
        current_value = round(float(squish.findObject(obj).value),1)
       
        test.log(f"Updated value: {current_value}")
 
    test.log(f"✅ Target value reached: {current_value}")
    
def select_different_dropdown(object,type,whichTypeIndex):
    """
        Selects a value from a dropdown using a doc string.
      
    """
    type_combo_box = get_object_from_names(object, error_message="Combo box object is missing")
    if type_combo_box is not None: 
        squish.mouseClick(squish.waitForObjectExists(object))
    
        type_option = get_object_from_names(setObjectText(obj = names.o_option_combo_box,text = type[whichTypeIndex]),error_message=f"Option {type[whichTypeIndex]} object is missing",timeout_ms=5000)
        if type_option is not None:
            squish.mouseClick(type_option)
            return True
    return False # default return if not successful

def verify_create_treatment_parameters():
    test.startSection("Pre treatment parameters")
    navigation.navigation_pageIndicator_step(config.CREATERX)
    squish.mouseClick(squish.waitForObject(names.o_PreTreatmentCreate_pretreatmentPatientIDEntry_TextEntry))
    squish.waitForObject(names.o_PreTreatmentCreate_pretreatmentPatientIDEntry_TextEntry).text ="abcd"
    set_value_based_on_target(names.o_PreTreatmentCreate_bloodFlowRateControl_ValueAdjuster, 60)
    set_value_based_on_target(names.o_PreTreatmentCreate_dialysateFlowRateControl_ValueAdjuster, 75)
    set_value_based_on_target(names.o_PreTreatmentCreate_durationControl_ValueAdjuster, 75)
    set_value_based_on_target(names.o_PreTreatmentCreate_heparinBolusVolumeControl_ValueAdjuster, 0.4)
    set_value_based_on_target(names.o_PreTreatmentCreate_heparinDispensingRateControl_ValueAdjuster, 0.5)
    set_value_based_on_target(names.o_PreTreatmentCreate_heparinStopTimeControl_ValueAdjuster, 60)
    select_different_dropdown(names.o_PreTreatmentCreate_acidConcentrateComboBox_BaseComboBox,config.ACID_CONCENTRATE,2)
    set_value_based_on_target(names.o_PreTreatmentCreate_dialysateTemperatureControl_ValueAdjuster,37.0)
    select_different_dropdown(names.o_PreTreatmentCreate_dialyzerTypeComboBox_BaseComboBox,config.DIALYZER_TYPE,2)
    set_value_based_on_target(names.o_PreTreatmentCreate_salineBolusVolumeControl_ValueAdjuster, 200)
    select_different_dropdown(names.o_PreTreatment_vitalsCombobox_BaseCombobox,config.VITALS,1)
    select_different_dropdown(names.o_PreTreatmentCreate_bicarbonateConcentrateComboBox_BaseComboBox,config.BICARBONATE,0)
    Validatebutton = setObjectText(names.o_preTreatmentStack_Text, config.VALIDATE)
    squish.mouseClick(squish.waitForObject(Validatebutton))
    td.td_Treatment_Parameters_Validation(  vAccepted                                    = 1,
                                            vBloodFlowRateRejectReason                   = 0,
                                            vDialysateFlowRateRejectReason               = 0,
                                            vTreatmentDurationRejectReason               = 0,
                                            vSalineBolusVolumeRejectReason               = 0,
                                            vHeparinStopTimeRejectReason                 = 0,
                                            vHeparinTypeRejectReason                     = 0,
                                            vAcidConcentrateRejectReason                 = 0,
                                            vBicarbonateConcentrateRejectReason          = 0,
                                            vDialyzerTypeRejectReason                    = 0,
                                            vBloodPressureMeasureIntervalRejectReason    = 0,
                                            vRinsebackFlowRateRejectReason               = 0,
                                            vRinsebackVolumeRejectReason                 = 0,
                                            vArterialPressureLimitWindowRejectReason     = 0,
                                            vVenousPressureLimitWindowRejectReason       = 0,
                                            vVenousPressureLimitAsymtrcRejectReason      = 0,
                                            vTransmembranePressureLimitWindowRejectReason  = 0,
                                            vDialysateTempRejectReason                   = 0,
                                            vHeparinDispensingRateRejectReason           = 0,
                                            vHeparinBolusVolumeRejectReason              = 0
                                           )
    
    confirmButton = setObjectText(names.o_preTreatmentStack_Text,config.CONFIRM )
    squish.mouseClick(squish.waitForObject(confirmButton))
    test.endSection()
    
def findAllObjectsById(parent, target_id):
    """
    Recursively finds all child objects by their id property.
    Returns a list of all matching objects found.
    """
    results = []
    
    # Use hasattr to safely check for 'id' property
    if hasattr(parent, 'id') and str(parent.id) == target_id:
        results.append(parent)

    # Recurse through all children to collect all instances
    for child in object.children(parent):
        results.extend(findAllObjectsById(child, target_id))

    return results