package eu.crushedpixel.replaymod.gui.elements;

import eu.crushedpixel.replaymod.gui.elements.listeners.FileChooseListener;
import lombok.Getter;
import net.minecraft.client.Minecraft;

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

public class GuiFileChooser extends GuiAdvancedButton {

    @Getter
    private File selectedFile;

    public void setSelectedFile(File selectedFile) {
        this.selectedFile = selectedFile;
        updateDisplayString();
    }

    private String baseString;

    private String[] allowedExtensions = null;

    private List<FileChooseListener> listeners = new ArrayList<FileChooseListener>();

    public GuiFileChooser(int buttonId, int x, int y, String buttonText, File selectedFile, String[] allowedExtensions) {
        super(buttonId, x, y, buttonText);
        this.selectedFile = selectedFile;

        this.baseString = buttonText;
        updateDisplayString();

        this.allowedExtensions = allowedExtensions;
    }

    public void addFileChooseListener(FileChooseListener listener) {
        this.listeners.add(listener);
    }

    protected void updateDisplayString() {
        this.displayString = baseString + (selectedFile == null ? "-" : selectedFile.getName());
    }

    @Override
    public boolean mousePressed(Minecraft mc, int mouseX, int mouseY) {
        boolean success = super.mousePressed(mc, mouseX, mouseY);
        if (success) {
            new Thread(new Runnable() {
                @Override
                public void run() {
                    Frame frame = new Frame();
                    FileDialog fileDialog = new FileDialog(frame);

                    fileDialog.setFilenameFilter(new FilenameFilter() {
                        @Override
                        public boolean accept(File dir, String name) {
                            if(allowedExtensions == null) return true;
                            for(String extension : allowedExtensions) {
                                String[] split = name.split("\\.");
                                String ext = split[split.length-1];
                                if(extension.equalsIgnoreCase(ext)) return true;
                            }

                            return false;
                        }
                    });

                    //frame.setVisible(true);
                    fileDialog.setVisible(true);

                    String filename = fileDialog.getFile();
                    String directory = fileDialog.getDirectory();
                    if(filename != null) {
                        selectedFile = new File(directory, filename);

                        updateDisplayString();

                        for(FileChooseListener listener : listeners) {
                            listener.onFileChosen(selectedFile);
                        }
                    }

                    frame.dispose();
                }
            }, "replaymod-file-chooser").start();
        }
        return success;
    }

}