package com.temp.client.foundation.widget;

import com.google.gwt.user.client.ui.HasName;
import com.google.gwt.user.client.ui.Widget;
import com.temp.shared.utils.StringUtils;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

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;
    }

    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);
        }
    }
}