package com.temp.foundation.client.widget;

import com.google.gwt.user.client.ui.*;
import com.temp.common.shared.utils.*;

import java.util.*;

public final class NameToID {
    private Map<String, List<Widget>> widgetsById = new HashMap<String, List<Widget>>();

    private String suffix;

    public static void on( HasName... widgets ) {
        if ( (widgets != null) && (widgets.length != 0) ) {
            new NameToID().add( widgets ).applyIDs();
        }
    }

    public NameToID() {
        suffix = "";
    }

    public NameToID( String suffix ) {
        this.suffix = StringUtils.deNull( suffix ).trim();
    }

    public NameToID add( List<HasName> widgets ) {
        if ( widgets != null ) {
            for ( HasName widget : widgets ) {
                tryAdd( widget );
            }
        }
        return this;
    }

    public NameToID add( HasName... widgets ) {
        if ( widgets != null ) {
            for ( HasName widget : widgets ) {
                tryAdd( widget );
            }
        }
        return this;
    }

    public NameToID tryAdd( Widget widget ) {
        if ( widget instanceof HasName ) {
            tryAdd( (HasName) widget );
        }
        return this;
    }

    private void tryAdd( HasName widget ) {
        if ( widget instanceof Widget ) {
            String id = extractId( widget );
            if ( id != null ) {
                add( (Widget) widget, id + suffix );
            }
        }
    }

    public void applyIDs() {
        for ( Map.Entry<String, List<Widget>> widgetByName : widgetsById.entrySet() ) {
            applyIDs( widgetByName.getValue(), widgetByName.getKey() );
        }
    }

    private void add( Widget widget, String id ) {
        List<Widget> widgets = widgetsById.get( id );
        if ( widgets == null ) {
            widgetsById.put( id, widgets = new ArrayList<Widget>() );
        }
        widgets.add( widget );
    }

    private String extractId( HasName widget ) {
        if ( widget instanceof HasIdOverride ) {
            return StringUtils.noEmpty( ((HasIdOverride) widget).getIdOverride() );
        }
        return StringUtils.noEmpty( widget.getName() );
    }

    private void applyIDs( List<Widget> widgets, String id ) {
        if ( widgets.size() == 1 ) {
            applyID( widgets.get( 0 ), id );
            return;
        }
        for ( int i = 0; i < widgets.size(); ) {
            Widget widget = widgets.get( i++ );
            applyID( widget, id + i );
        }
    }

    private void applyID( Widget widget, String id ) {
        if ( widget instanceof IdSettable ) {
            ((IdSettable) widget).setId( id );
        } else {
            widget.getElement().setId( id );
        }
    }
}