package org.litesoft.aokeyhole.swing.simplecomponents;

import org.litesoft.aokeyhole.objects.support.*;

import javax.swing.*;
import java.awt.*;

public class JComboBoxPlus extends JComboBox
{
    public JComboBoxPlus( String[] pValidOptions, boolean pAddBlank )
    {
        UIManager.put( "ComboBox.disabledForeground", getForeground() );
        UIManager.put( "ComboBox.disabledBackground", ComponentColors.getDisabled() );
        setBackground( ComponentColors.getEnabled() );
        setFont( getFont().deriveFont( Font.PLAIN ) );

        if ( pAddBlank )
        {
            addItem( "" );
        }
        if ( pValidOptions != null )
        {
            for ( String option : pValidOptions )
            {
                addItem( option );
            }
        }
    }

    public void setOptions( String[] pOptions, boolean pForceSelectionOrAddBlank )
    {
        Object curValue = getSelectedItem();
        removeAllItems();
        if ( pOptions != null )
        {
            if ( pForceSelectionOrAddBlank )
            {
                if ( curValue == null )
                {
                    curValue = pOptions[0];
                }
            }
            else
            {
                addItem( "" );
            }
            for ( String option : pOptions )
            {
                addItem( option );
            }
        }
        setSelectedItem( ensureAnOption( curValue ) );
    }

    public Object ensureAnOption( Object pValue )
    {
        if ( pValue == null )
        {
            pValue = "";
        }
        int items = getItemCount();
        for ( int i = 0; i < items; i++ )
        {
            if ( AbstractBase.areEqual( pValue, getItemAt( i ) ) )
            {
                return pValue;
            }
        }
        addItem( pValue );
        return pValue;
    }
}