// This Source Code is in the Public Domain per: http://unlicense.org
package org.litesoft.aokeyhole.swing.simplecomponents;

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

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

@SuppressWarnings("unchecked")
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( AdjustFont.sizeBy( getFont(), 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;
    }
}