import squish
import test
import object
import names
from squish import *
from leahi_dialin.ui import utils
from builtins import int as pyInt
   
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 set_Object_Text(text,obj):
    """
        Method to set object property based on text
        @param text : (string) treatment parameter 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 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)
    try:
        current_value = round(float(findObject(obj).value),1)
    except LookupError:
        current_value = float(findObject(obj).property("value"))
 
    # 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))
        # Update current value after click
        try:
            current_value = round(float(findObject(obj).value),1)
        except Exception:
            current_value = float(findObject(obj).property("value"))
 
        test.log(f"Updated value: {current_value}")
 
    test.log(f"✅ Target value reached: {current_value}")
    
def select_different_dropdown(object,type,whichTypeIndex):
    """
    Method selects the option based on index from the dropdown 
    """
    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(set_Object_Text(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 set_value_with_slider(value_field_obj, slider_obj,parameter):
    """
    Opens the slider and moves it gradually to the target value (step of 10).
    Uses controlled arrow key input for fine adjustment.
    """
 
    try:
        value_field = waitForObject(value_field_obj)
        test.log(f"Opening slider for {parameter}...")
        
        # Try right-click first
        squish.mousePress(value_field, squish.Qt.LeftButton)
        # utils.waitForGUI(0.2)
        value = value_field.value
 
        # If not visible, try left long-press
        if not object.exists(slider_obj):
            test.log(f"{parameter}: Slider not opened by left-click, trying long left-press...")
            squish.mousePress(value_field, squish.Qt.LeftButton)
            # utils.waitForGUI(1)
        if not object.exists(slider_obj):
            test.fail(f"{parameter}: Slider did not appear.")
        
        slider = waitForObject(slider_obj)
        test.log(f"{parameter}: Slider appeared successfully.")
        squish.mousePress(slider,squish.Qt.LeftButton)
        final_value = waitForObject(value_field_obj).value
        test.verify(final_value!= value, f"{parameter} slider adjusted correctly to {final_value}")
        squish.mouseRelease(slider, squish.Qt.LeftButton)
        # utils.waitForGUI(0.2)
        if object.exists(slider_obj):
            test.log(f"Waiting for {parameter} slider to close...")
            waitFor(lambda: not object.exists(slider_obj), 1000)
        # utils.waitForGUI(0.2)
    except LookupError as e:
        test.fail(f"{parameter}: LookupError - {e}")
        
def click_left_until_off(object_name):
    """
    Method to perform the left arrow untill the value
    becomes off
    """
    parent_obj = waitForObject(object_name)
    left_arrow = findObjectById(parent_obj, "_leftArrow")
    
    # Loop until the value becomes "off"
    while findObject(object_name).value != 0.0:
        squish.mouseClick(waitForObject(left_arrow))
        utils.waitForGUI(0.2)  # Small delay to allow UI to update
        
def button_text_obj(text):
    """
    To get the text from Validate and confirm button
    """
    names.o_PreTreatmentCreate_Text["text"] = text
    return names.o_PreTreatmentCreate_Text