ReplayModCinematic/src/main/java/eu/crushedpixel/replaymod/gui/replaystudio/GuiConnectPart.java

package eu.crushedpixel.replaymod.gui.replaystudio;

import eu.crushedpixel.replaymod.gui.GuiConstants;
import eu.crushedpixel.replaymod.gui.elements.GuiArrowButton;
import eu.crushedpixel.replaymod.gui.elements.GuiDropdown;
import eu.crushedpixel.replaymod.gui.elements.GuiEntryList;
import eu.crushedpixel.replaymod.gui.elements.listeners.SelectionListener;
import eu.crushedpixel.replaymod.utils.ReplayFileIO;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.GuiButton;
import net.minecraft.client.resources.I18n;
import org.apache.commons.io.FilenameUtils;

import java.awt.*;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

public class GuiConnectPart extends GuiStudioPart {

    private final String DESCRIPTION = I18n.format("replaymod.gui.editor.connect.description");
    private final String TITLE = I18n.format("replaymod.gui.editor.connect.title");

    private boolean initialized = false;

    private GuiEntryList<String> concatList;
    private GuiDropdown<String> replayDropdown;

    private GuiButton removeButton, addButton;
    private GuiArrowButton upButton, downButton;

    private List<File> replayFiles;
    private List<String> filesToConcat;

    public GuiConnectPart(int yPos) {
        super(yPos);
        this.mc = Minecraft.getMinecraft();
        fontRendererObj = mc.fontRendererObj;
    }


    @Override
    public void applyFilters(File replayFile, File outputFile) {
        //TODO
    }

    @Override
    public String getDescription() {
        return DESCRIPTION;
    }

    @Override
    public String getTitle() {
        return TITLE;
    }

    @Override
    public void initGui() {
        if(!initialized) {
            concatList = new GuiEntryList(1, fontRendererObj, 30, yPos, 150, 0);
            filesToConcat = new ArrayList<String>();
            String selectedName = FilenameUtils.getBaseName(GuiReplayEditor.instance.getSelectedFile().getAbsolutePath());
            filesToConcat.add(selectedName);
            concatList.setElements(filesToConcat);

            concatList.setSelectionIndex(0);

            replayDropdown = new GuiDropdown(1, fontRendererObj, 250, yPos + 5, 0, 4);

            replayDropdown.clearElements();
            replayFiles = ReplayFileIO.getAllReplayFiles();
            int index = -1;
            int i = 0;
            for(File file : replayFiles) {
                String name = FilenameUtils.getBaseName(file.getAbsolutePath());
                replayDropdown.addElement(name);
                if(name.equals(selectedName)) {
                    index = i;
                }
                i++;
            }

            replayDropdown.setSelectionPos(index);

            replayDropdown.addSelectionListener(new SelectionListener() {

                @Override
                public void onSelectionChanged(int selectionIndex) {
                    try {
                        filesToConcat.set(concatList.getSelectionIndex(), replayDropdown.getElement(selectionIndex));
                        concatList.setElements(filesToConcat);
                    } catch(Exception e) {
                    } //Sorry, too lazy to properly avoid this Exception here
                }
            });

            concatList.addSelectionListener(new SelectionListener() {
                @Override
                public void onSelectionChanged(int selectionIndex) {
                    String selName = (String) concatList.getElement(selectionIndex);
                    int i = 0;
                    for(Object s : replayDropdown.getAllElements()) {
                        String str = (String) s;
                        if(str.equals(selName)) {
                            replayDropdown.setSelectionIndex(i);
                            break;
                        }
                        i++;
                    }
                    removeButton.enabled = upButton.enabled = downButton.enabled = !(selectionIndex < 0 || selectionIndex >= filesToConcat.size());
                    if(upButton.enabled && selectionIndex == 0) upButton.enabled = false;
                    if(downButton.enabled && selectionIndex == filesToConcat.size() - 1) downButton.enabled = false;
                }
            });

            upButton = new GuiArrowButton(GuiConstants.REPLAY_EDITOR_UP_BUTTON, 195, yPos + 40, "", true);
            buttonList.add(upButton);

            downButton = new GuiArrowButton(GuiConstants.REPLAY_EDITOR_DOWN_BUTTON, 219, yPos + 40, "", false);
            buttonList.add(downButton);

            int w = GuiReplayEditor.instance.width - 243 - 20 - 4;

            removeButton = new GuiButton(GuiConstants.REPLAY_EDITOR_REMOVE_BUTTON, 249, yPos + 40, I18n.format("replaymod.gui.remove"));
            buttonList.add(removeButton);

            addButton = new GuiButton(GuiConstants.REPLAY_EDITOR_ADD_BUTTON, 0, yPos + 40, I18n.format("replaymod.gui.add"));
            buttonList.add(addButton);

            concatList.setSelectionIndex(0);
        }

        int w = GuiReplayEditor.instance.width - 249 - 20 - 4;
        addButton.xPosition = 249 + 6 + (w / 2);

        addButton.width = w / 2 + 2;
        removeButton.width = w / 2 + 2;

        replayDropdown.width = GuiReplayEditor.instance.width - 250 - 18;

        int h = GuiReplayEditor.instance.height - yPos - 20;
        int rows = (int) (h / (float) GuiEntryList.elementHeight);
        concatList.setVisibleElements(rows);

        initialized = true;
    }

    @Override
    public void mouseClicked(int mouseX, int mouseY, int mouseButton) throws IOException {
        super.mouseClicked(mouseX, mouseY, mouseButton); //call this first to ensure the dropdown is still open
        concatList.mouseClicked(mouseX, mouseY, mouseButton);
        replayDropdown.mouseClicked(mouseX, mouseY, mouseButton);
    }

    @Override
    public void drawScreen(int mouseX, int mouseY, float partialTicks) {
        super.drawScreen(mouseX, mouseY, partialTicks);
        concatList.drawTextBox();
        replayDropdown.drawTextBox();

        drawString(fontRendererObj, I18n.format("replaymod.gui.replay")+":", 200, yPos + 5 + 7, Color.WHITE.getRGB());
    }

    @Override
    public void updateScreen() {
        if(!initialized) initGui();
    }

    @Override
    public void keyTyped(char typedChar, int keyCode) {

    }

    @Override
    protected void actionPerformed(GuiButton button) {
        if(!button.enabled || replayDropdown.isExpanded()) {
            return;
        }

        if(button.id == GuiConstants.REPLAY_EDITOR_ADD_BUTTON) {
            filesToConcat.add(FilenameUtils.getBaseName(replayFiles.get(0).getAbsolutePath()));
            concatList.setElements(filesToConcat);
            concatList.setSelectionIndex(filesToConcat.size() - 1);
        } else if(button.id == GuiConstants.REPLAY_EDITOR_REMOVE_BUTTON) {
            int indexBefore = concatList.getSelectionIndex();
            if(indexBefore >= 0 && indexBefore < filesToConcat.size()) {
                filesToConcat.remove(indexBefore);
                concatList.setElements(filesToConcat);
                if(filesToConcat.size() <= indexBefore) {
                    concatList.setSelectionIndex(filesToConcat.size() - 1);
                } else {
                    concatList.setSelectionIndex(indexBefore);
                }
            }
        }
    }
}