CustomNPC-Plus/src/main/java/noppes/npcs/client/gui/select/GuiDialogSelection.java

package noppes.npcs.client.gui.select;

import com.google.common.collect.Lists;
import net.minecraft.client.gui.GuiButton;
import noppes.npcs.client.gui.util.*;
import noppes.npcs.controllers.DialogController;
import noppes.npcs.controllers.data.Dialog;
import noppes.npcs.controllers.data.DialogCategory;

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

public class GuiDialogSelection extends SubGuiInterface implements ICustomScrollListener {
	private HashMap<String,DialogCategory> categoryData = new HashMap<String,DialogCategory>();
	private HashMap<String,Dialog> dialogData = new HashMap<String,Dialog>();

	private GuiCustomScroll scrollCategories;
	private GuiCustomScroll scrollDialogs;

	private DialogCategory selectedCategory;
	public Dialog selectedDialog;

	private GuiSelectionListener listener;
    private String catSearch = "";
    private String dialogSearch = "";

    public GuiDialogSelection(int dialog){
    	drawDefaultBackground = false;
		title = "";
		setBackground("menubg.png");
        xSize = 366;
        ySize = 226;
    	this.selectedDialog = DialogController.Instance.dialogs.get(dialog);
    	if(selectedDialog != null) {
    		selectedCategory = selectedDialog.category;
    	}
    }

    @Override
    public void initGui(){
        super.initGui();

		if(parent instanceof GuiSelectionListener){
			listener = (GuiSelectionListener) parent;
		}
        this.addLabel(new GuiNpcLabel(0, "gui.categories", guiLeft + 8, guiTop + 4));
        this.addLabel(new GuiNpcLabel(1, "dialog.dialogs", guiLeft + 184, guiTop + 4));

        this.addButton(new GuiNpcButton(2, guiLeft + xSize - 56, guiTop + ySize - 35, 50, 20, "gui.done"));
        this.addButton(new GuiNpcButton(3, guiLeft + xSize - 108, guiTop + ySize - 35, 50, 20, "gui.cancel"));

    	HashMap<String,DialogCategory> categoryData = new HashMap<String,DialogCategory>();
    	HashMap<String,Dialog> dialogData = new HashMap<String,Dialog>();

    	for(DialogCategory category : DialogController.Instance.categories.values()) {
    		categoryData.put(category.title, category);
    	}
    	this.categoryData = categoryData;

		if(selectedCategory != null) {
			for(Dialog dialog : selectedCategory.dialogs.values()) {
				dialogData.put(dialog.title, dialog);
			}
		}
		this.dialogData = dialogData;

        if(scrollCategories == null){
	        scrollCategories = new GuiCustomScroll(this,0, 0);
            scrollCategories.setSize(177, 153);
        }
        scrollCategories.setList(getCatSearch());
        if(selectedCategory != null) {
        	scrollCategories.setSelected(selectedCategory.title);
        }
        scrollCategories.guiLeft = guiLeft + 4;
        scrollCategories.guiTop = guiTop + 14;
        this.addScroll(scrollCategories);
        addTextField(new GuiNpcTextField(33, this, fontRendererObj, guiLeft + 4, guiTop + 169, 177, 20, catSearch));

        if(scrollDialogs == null){
        	scrollDialogs = new GuiCustomScroll(this,1,0);
        	scrollDialogs.setSize(177, 153);
        }
        scrollDialogs.setList(getDialogSearch());
        if(selectedDialog != null) {
        	scrollDialogs.setSelected(selectedDialog.title);
        }
        scrollDialogs.guiLeft = guiLeft + 182;
        scrollDialogs.guiTop = guiTop + 14;
        this.addScroll(scrollDialogs);
        addTextField(new GuiNpcTextField(44, this, fontRendererObj, guiLeft + 182, guiTop + 169, 177, 20, dialogSearch));
    }

    @Override
    public void keyTyped(char c, int i)
    {
        super.keyTyped(c, i);
        if(getTextField(33) != null){
            if(getTextField(33).isFocused()){
                if(catSearch.equals(getTextField(33).getText()))
                    return;
                catSearch = getTextField(33).getText().toLowerCase();
                scrollCategories.setList(getCatSearch());
            }
        }
        if(getTextField(44) != null){
            if(getTextField(44).isFocused()){
                if(dialogSearch.equals(getTextField(44).getText()))
                    return;
                dialogSearch = getTextField(44).getText().toLowerCase();
                scrollDialogs.setList(getDialogSearch());
            }
        }
    }

    private List<String> getCatSearch(){
        if(catSearch.isEmpty()){
            return Lists.newArrayList(categoryData.keySet());
        }
        List<String> list = new ArrayList<String>();
        for(String name : Lists.newArrayList(categoryData.keySet())){
            if(name.toLowerCase().contains(catSearch))
                list.add(name);
        }
        return list;
    }

    private List<String> getDialogSearch(){
        if(selectedCategory == null){
            return Lists.newArrayList(dialogData.keySet());
        }

        if(dialogSearch.isEmpty()){
            return new ArrayList<String>(Lists.newArrayList(dialogData.keySet()));
        }
        List<String> list = new ArrayList<String>();
        for(String name : Lists.newArrayList(dialogData.keySet())){
            if(name.toLowerCase().contains(dialogSearch))
                list.add(name);
        }
        return list;
    }

	@Override
	public void customScrollClicked(int i, int j, int k, GuiCustomScroll guiCustomScroll) {
		if(guiCustomScroll.id == 0){
			selectedCategory = categoryData.get(scrollCategories.getSelected());
			selectedDialog = null;
			scrollDialogs.selected = -1;
            scrollDialogs.resetScroll();
            getTextField(44).setText("");
            dialogSearch = "";
		}
		if(guiCustomScroll.id == 1){
			selectedDialog = dialogData.get(scrollDialogs.getSelected());
		}
		initGui();
	}

	@Override
	public void customScrollDoubleClicked(String selection, GuiCustomScroll scroll) {
		if(selectedDialog == null)
			return;
		if(listener != null) {
			listener.selected(selectedDialog.id, selectedDialog.title);
		}
		close();
	}

    @Override
	protected void actionPerformed(GuiButton guibutton){
		int id = guibutton.id;
        if(id == 2){
            if(selectedDialog != null) {
                customScrollDoubleClicked(null, null);
            }
            else {
                close();
            }
        }
        if(id == 3){
            close();
        }
    }
}