package com.example.client.utils;

import com.example.client.dto.ColumnDto;

import javax.swing.*;
import javax.swing.plaf.basic.BasicComboBoxEditor;
import javax.swing.table.TableColumnModel;
import java.awt.*;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class MultiSelectComboBox2 extends JComboBox<String> {
    public Set<Integer> selectedIndices = new HashSet<>();
    private String defaultText; // 默认显示文本

    public MultiSelectComboBox2(String[] items, List<ColumnDto> columnDto, TableColumnModel columnModel) {
        super(items);
        this.defaultText = "自定义表头"; // 设置默认文本
        setEditable(true);

        // 自定义渲染器
        setRenderer(new DefaultListCellRenderer() {
            @Override
            public Component getListCellRendererComponent(JList<?> list, Object value, int index, boolean isSelected, boolean cellHasFocus) {
                JCheckBox checkBox = new JCheckBox();
                checkBox.setText(value.toString()); // 使用字符串作为文本
                checkBox.setSelected(selectedIndices.contains(index)); // 设置选中状态
                checkBox.setEnabled(list.isEnabled());
                checkBox.setFont(list.getFont());
                checkBox.setFocusPainted(false);
                checkBox.setBackground(isSelected ? list.getSelectionBackground() : list.getBackground());
                checkBox.setForeground(isSelected ? list.getSelectionForeground() : list.getForeground());
                return checkBox;
            }
        });

        // 自定义编辑器
        setEditor(new BasicComboBoxEditor() {
            private JTextField textField;

            @Override
            public Component getEditorComponent() {
                if (textField == null) {
                    textField = new JTextField(defaultText); // 设置默认文本
                    textField.setEditable(false); // 设置为不可编辑
                }
                return textField;
            }

            @Override
            public void setItem(Object anObject) {
                // 禁用自动更新编辑器文本的行为
                if (textField != null) {
                    textField.setText(defaultText); // 始终显示默认文本
                }
            }
        });

        // 添加动作监听器
        addActionListener(e -> {
            int selectedIndex = getSelectedIndex();
            if (selectedIndex != -1) {
                if (selectedIndices.contains(selectedIndex)) {
                    selectedIndices.remove(selectedIndex);
                    if (columnModel != null) {
                        columnModel.getColumn(selectedIndex).setPreferredWidth(2);
                    }
                } else {
                    selectedIndices.add(selectedIndex);
                    if (columnModel != null) {
                        Integer width = columnDto.get(selectedIndex).getColumnWidth();
                        columnModel.getColumn(selectedIndex).setPreferredWidth(width);
                    }
                }
            }
        });

        // 默认选中所有项
        for (int i = 0; i < getItemCount(); i++) {
            selectedIndices.add(i); // 将所有索引添加到选中集合
        }
    }

    public Set<String> getSelectedItems() {
        Set<String> selectedItems = new HashSet<>();
        for (Integer index : selectedIndices) {
            selectedItems.add((String) getItemAt(index));
        }
        return selectedItems;
    }

    public void setSelectedItems(Set<String> items) {
        selectedIndices.clear(); // 清空当前选中项
        for (int i = 0; i < getItemCount(); i++) {
            String item = getItemAt(i);
            if (items.contains(item)) {
                selectedIndices.add(i); // 将对应的索引添加到选中集合
            }
        }
    }

    public Set<Integer> getUnselectedIndices() {
        Set<Integer> unselectedIndices = new HashSet<>();
        for (int i = 0; i < getItemCount(); i++) {
            if (!selectedIndices.contains(i)) {
                unselectedIndices.add(i);
            }
        }
        return unselectedIndices;
    }

    public Set<Integer> getSelectedIndices() {
        return new HashSet<>(selectedIndices);
    }
}