import gradio as gr


class FormComponent:
    def get_expected_parent(self):
        return gr.components.Form


gr.Dropdown.get_expected_parent = FormComponent.get_expected_parent


class ToolButton(FormComponent, gr.Button): # small button with single emoji as text
    def __init__(self, *args, **kwargs):
        classes = kwargs.pop("elem_classes", [])
        super().__init__(*args, elem_classes=["tool", *classes], **kwargs)

    def get_block_name(self):
        return "button"

### unused components below for compatibility with extensions ###

class FormRow(FormComponent, gr.Row): # unused
    def get_block_name(self):
        return "row"


class FormColumn(FormComponent, gr.Column): # unused
    def get_block_name(self):
        return "column"


class FormGroup(FormComponent, gr.Group): # unused
    def get_block_name(self):
        return "group"


class FormHTML(FormComponent, gr.HTML): # unused
    def get_block_name(self):
        return "html"


class FormColorPicker(FormComponent, gr.ColorPicker): # unused
    def get_block_name(self):
        return "colorpicker"


class DropdownMulti(FormComponent, gr.Dropdown): # unused
    def __init__(self, **kwargs):
        super().__init__(multiselect=True, **kwargs)
    def get_block_name(self):
        return "dropdown"


class DropdownEditable(FormComponent, gr.Dropdown): # unused
    def __init__(self, **kwargs):
        super().__init__(allow_custom_value=True, **kwargs)
    def get_block_name(self):
        return "dropdown"


class InputAccordion(gr.Checkbox): # unused
    global_index = 0
    def __init__(self, value, **kwargs):
        self.accordion_id = kwargs.get('elem_id')
        if self.accordion_id is None:
            self.accordion_id = f"input-accordion-{InputAccordion.global_index}"
            InputAccordion.global_index += 1
        kwargs_checkbox = {**kwargs, "elem_id": f"{self.accordion_id}-checkbox", "visible": False}
        super().__init__(value, **kwargs_checkbox)
        self.change(fn=None, _js='function(checked){ inputAccordionChecked("' + self.accordion_id + '", checked); }', inputs=[self])
        kwargs_accordion = {
            **kwargs,
            "elem_id": self.accordion_id,
            "label": kwargs.get('label', 'Accordion'),
            "elem_classes": ['input-accordion'],
            "open": value,
        }
        self.accordion = gr.Accordion(**kwargs_accordion)

    def extra(self):
        return gr.Column(elem_id=self.accordion_id + '-extra', elem_classes='input-accordion-extra', min_width=0)

    def __enter__(self):
        self.accordion.__enter__()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.accordion.__exit__(exc_type, exc_val, exc_tb)

    def get_block_name(self):
        return "checkbox"


class ResizeHandleRow(gr.Row): # unused
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.elem_classes.append("resize-handle-row")
    def get_block_name(self):
        return "row"