package org.vaadin.firitin.components.combobox;

import com.vaadin.flow.component.AbstractField.ComponentValueChangeEvent;
import com.vaadin.flow.component.ComponentEventListener;
import com.vaadin.flow.component.ItemLabelGenerator;
import com.vaadin.flow.component.combobox.ComboBox;
import com.vaadin.flow.data.provider.CallbackDataProvider;
import com.vaadin.flow.data.renderer.Renderer;
import org.vaadin.firitin.fluency.ui.*;
import org.vaadin.firitin.fluency.ui.internal.FluentHasAutofocus;
import org.vaadin.firitin.fluency.ui.internal.FluentHasLabel;

import java.util.Collection;

@SuppressWarnings("unchecked")
public class VComboBox<T> extends ComboBox<T> implements FluentHasSize<VComboBox<T>>, FluentHasValidation<VComboBox<T>>, FluentFocusable<ComboBox<T>, VComboBox<T>>,
        FluentComponent<VComboBox<T>>, FluentHasLabel<VComboBox<T>>, FluentHasAutofocus<VComboBox<T>>, FluentHasTooltip<VComboBox<T>>,
        FluentHasStyle<VComboBox<T>>, FluentHasValueAndElement<VComboBox<T>, ComponentValueChangeEvent<ComboBox<T>, T>, T>, FluentHasHelper<VComboBox<T>> {

    public VComboBox() {
        super();
    }

    public VComboBox(String label, Collection<T> items) {
        super(label, items);
    }

    public VComboBox(String label, T... items) {
        super(label, items);
    }

    public VComboBox(String label) {
        super(label);
    }

    public VComboBox<T> withRenderer(Renderer<T> renderer) {
        setRenderer(renderer);
        return this;
    }

    public VComboBox<T> withItemLabelGenerator(ItemLabelGenerator<T> itemLabelGenerator) {
        setItemLabelGenerator(itemLabelGenerator);
        return this;
    }

    public VComboBox<T> withOpened(boolean opened) {
        setOpened(opened);
        return this;
    }

    public VComboBox<T> withAllowCustomValue(boolean allowCustomValue) {
        setAllowCustomValue(allowCustomValue);
        return this;
    }

    public VComboBox<T> withRequired(boolean required) {
        setRequired(required);
        return this;
    }

    public VComboBox<T> withPlaceholder(String placeholder) {
        setPlaceholder(placeholder);
        return this;
    }

    public VComboBox<T> withPattern(String pattern) {
        setPattern(pattern);
        return this;
    }

    public VComboBox<T> withPageSize(int pageSize) {
        setPageSize(pageSize);
        return this;
    }

    public VComboBox<T> withCustomValueSetListener(ComponentEventListener<CustomValueSetEvent<ComboBox<T>>> listener) {
        addCustomValueSetListener(listener);
        return this;
    }

    public VComboBox<T> withClearButtonVisible(boolean clearButtonVisible) {
        setClearButtonVisible(clearButtonVisible);
        return this;
    }

    public VComboBox<T> withAlignLeft() {
        getElement().setAttribute("theme", "align-left");
        return this;
    }

    public VComboBox<T> withAlignCenter() {
        getElement().setAttribute("theme", "align-center");
        return this;
    }

    public VComboBox<T> withAlignRight() {
        getElement().setAttribute("theme", "align-right");
        return this;
    }

    public VComboBox<T> withThemeSmall() {
        getElement().setAttribute("theme", "small");
        return this;
    }

    public VComboBox<T> withTheme(String theme) {
        getElement().setAttribute("theme", theme);
        return this;
    }

    /**
     * Sets the items to be displayed in the combo box.
     *
     * @param items the items to be displayed
     * @return this instance for method chaining
     */
    public VComboBox<T> withItems(T... items) {
        setItems(items);
        return this;
    }

    /**
     * Sets the items to be displayed in the combo box.
     *
     * @param items the collection of items to be displayed
     * @return this instance for method chaining
     */
    public VComboBox<T> withItems(Collection<T> items) {
        setItems(items);
        return this;
    }

    /**
     * Sets the items to be displayed in the combo box using a callback data provider.
     * This is useful for lazy loading or large datasets.
     *
     * @param items the fetch callback that provides items based on query and filters
     * @return this instance for method chaining
     */
    public VComboBox<T> withItems(CallbackDataProvider.FetchCallback<T, String> items) {
        setItems(items);
        return this;
    }

}