package eu.crushedpixel.replaymod.gui;

import eu.crushedpixel.replaymod.ReplayMod;
import eu.crushedpixel.replaymod.gui.elements.*;
import eu.crushedpixel.replaymod.gui.elements.listeners.NumberValueChangeListener;
import eu.crushedpixel.replaymod.holders.*;
import eu.crushedpixel.replaymod.interpolation.KeyframeList;
import eu.crushedpixel.replaymod.interpolation.KeyframeValue;
import eu.crushedpixel.replaymod.replay.ReplayHandler;
import eu.crushedpixel.replaymod.utils.MouseUtils;
import eu.crushedpixel.replaymod.utils.RoundUtils;
import eu.crushedpixel.replaymod.utils.TimestampUtils;
import net.minecraft.client.gui.GuiButton;
import net.minecraft.client.gui.GuiScreen;
import net.minecraft.client.gui.GuiTextField;
import net.minecraft.client.resources.I18n;
import org.lwjgl.input.Keyboard;
import org.lwjgl.util.Point;

import java.awt.*;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;

public abstract class GuiEditKeyframe<T extends KeyframeValue> extends GuiScreen {

    @SuppressWarnings("unchecked")
    public static GuiEditKeyframe create(Keyframe kf) {
        if(kf.getValue() instanceof Marker) return new GuiEditKeyframeMarker(kf);
        if(kf.getValue() instanceof TimestampValue) return new GuiEditKeyframeTime(kf);
        if(kf.getValue() instanceof SpectatorData) return new GuiEditKeyframeSpectator(kf);
        if(kf.getValue() instanceof AdvancedPosition) return new GuiEditKeyframePosition(kf);
        throw new UnsupportedOperationException("Keyframe type unknown: " + kf);
    }

    protected boolean initialized = false;

    private GuiAdvancedButton saveButton, cancelButton;
    private GuiArrowButton leftButton, rightButton;

    protected GuiNumberInput min, sec, ms;

    protected ComposedElement inputs;

    private int virtualHeight = 200;
    protected int virtualY;

    protected Keyframe<T> keyframe;
    protected Keyframe<T> keyframeBackup;
    protected boolean save;

    private Keyframe<?> previous, next;

    protected int w2;
    protected int w3;
    protected int totalWidth;
    protected int left;

    protected String screenTitle;

    public GuiEditKeyframe(Keyframe<T> keyframe, KeyframeList<T> keyframes) {
        this.keyframe = keyframe;
        this.keyframeBackup = keyframe.copy();

        previous = keyframes.getPreviousKeyframe(keyframe.getRealTimestamp(), false);
        next = keyframes.getNextKeyframe(keyframe.getRealTimestamp(), false);

        ReplayMod.replaySender.setReplaySpeed(0);
    }

    @Override
    public void initGui() {
        Keyboard.enableRepeatEvents(true);

        virtualY = this.height - virtualHeight - 10;

        if(!initialized) {
            saveButton = new GuiAdvancedButton(GuiConstants.KEYFRAME_EDITOR_SAVE_BUTTON, 0, 0, I18n.format("replaymod.gui.save")) {
                @Override
                public void performAction() {
                    save = true;
                    mc.displayGuiScreen(null);
                }
            };

            cancelButton = new GuiAdvancedButton(GuiConstants.KEYFRAME_EDITOR_CANCEL_BUTTON, 0, 0, I18n.format("replaymod.gui.cancel")) {
                @Override
                public void performAction() {
                    save = false;
                    mc.displayGuiScreen(null);
                }
            };

            leftButton = new GuiArrowButton(GuiConstants.KEYFRAME_EDITOR_LEFT_BUTTON, 0, 0, "", GuiArrowButton.Direction.LEFT) {
                @Override
                public void performAction() {
                    save = true;
                    mc.displayGuiScreen(create(previous));
                }
            };

            rightButton = new GuiArrowButton(GuiConstants.KEYFRAME_EDITOR_RIGHT_BUTTON, 0, 0, "", GuiArrowButton.Direction.RIGHT){
                @Override
                public void performAction() {
                    save = true;
                    mc.displayGuiScreen(create(next));
                }
            };

            saveButton.width = cancelButton.width = 100;

            leftButton.enabled = previous != null;
            rightButton.enabled = next != null;

            //Real Time Input
            int timestamp = keyframe.getRealTimestamp();
            min = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 30, 0d, 29d, (double)TimestampUtils.timestampToWholeMinutes(timestamp), false, "", 1);
            sec = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 25, 0d, 59d, (double)TimestampUtils.getSecondsFromTimestamp(timestamp), false, "", 1);
            ms = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 35, 0d, 999d, (double)TimestampUtils.getMillisecondsFromTimestamp(timestamp), false, "", 1);

            inputs = new ComposedElement(min, sec, ms, saveButton, cancelButton, leftButton, rightButton);
        }

        w3 =    fontRendererObj.getStringWidth(I18n.format("replaymod.gui.editkeyframe.timelineposition")+":")+10+
                fontRendererObj.getStringWidth(I18n.format("replaymod.gui.minutes"))+5+min.width+5+
                fontRendererObj.getStringWidth(I18n.format("replaymod.gui.seconds"))+5+sec.width+5+
                fontRendererObj.getStringWidth(I18n.format("replaymod.gui.milliseconds"))+5+ms.width;

        int l = (this.width-w3)/2;
        min.xPosition = fontRendererObj.getStringWidth(I18n.format("replaymod.gui.editkeyframe.timelineposition")+":")+10 + l;
        sec.xPosition = fontRendererObj.getStringWidth(I18n.format("replaymod.gui.editkeyframe.timelineposition")+":")+10 + l +
                min.width + 5 + fontRendererObj.getStringWidth(I18n.format("replaymod.gui.minutes")) + 5;
        ms.xPosition = fontRendererObj.getStringWidth(I18n.format("replaymod.gui.editkeyframe.timelineposition")+":")+10 + l +
                min.width + 5 + fontRendererObj.getStringWidth(I18n.format("replaymod.gui.minutes")) + 5
                + sec.width + 5 + fontRendererObj.getStringWidth(I18n.format("replaymod.gui.seconds")) + 5;

        min.yPosition = sec.yPosition = ms.yPosition = virtualY+virtualHeight-55;

        saveButton.yPosition = cancelButton.yPosition = virtualY + virtualHeight - 20 - 5;
        saveButton.xPosition = this.width - 100 - 5 - 10;
        cancelButton.xPosition = saveButton.xPosition - 100 - 5;

        @SuppressWarnings("unchecked")
        List<GuiButton> buttonList = this.buttonList;

        buttonList.add(saveButton);
        buttonList.add(cancelButton);

        leftButton.xPosition = 15;
        rightButton.xPosition = width-35;
        leftButton.yPosition = rightButton.yPosition = virtualY + ((virtualHeight - leftButton.height)/2);

        buttonList.add(leftButton);
        buttonList.add(rightButton);
    }

    @Override
    public void onGuiClosed() {
        if(!save) {
            ReplayHandler.removeKeyframe(keyframe);
            ReplayHandler.addKeyframe(keyframeBackup);
            ReplayHandler.selectKeyframe(keyframeBackup);
        }
        ReplayHandler.fireKeyframesModifyEvent();
        Keyboard.enableRepeatEvents(false);
    }

    @Override
    protected void keyTyped(char typedChar, int keyCode) throws IOException {
        Point mousePos = MouseUtils.getMousePos();
        inputs.buttonPressed(mc, mousePos.getX(), mousePos.getY(), typedChar, keyCode);

        if(keyCode == Keyboard.KEY_ESCAPE) {
            save = false;
            mc.displayGuiScreen(null);
        }
    }

    @Override
    protected void mouseClicked(int mouseX, int mouseY, int mouseButton) throws IOException {
        inputs.mouseClick(mc, mouseX, mouseY, mouseButton);
        super.mouseClicked(mouseX, mouseY, mouseButton);
    }

    @Override
    protected void mouseClickMove(int mouseX, int mouseY, int clickedMouseButton, long timeSinceLastClick) {
        inputs.mouseDrag(mc, mouseX, mouseY, clickedMouseButton);
        super.mouseClickMove(mouseX, mouseY, clickedMouseButton, timeSinceLastClick);
    }

    @Override
    protected void mouseReleased(int mouseX, int mouseY, int state) {
        inputs.mouseRelease(mc, mouseX, mouseY, state);
        super.mouseReleased(mouseX, mouseY, state);
    }

    @Override
    public void updateScreen() {
        inputs.tick(mc);
    }

    @Override
    public void drawScreen(int mouseX, int mouseY, float partialTicks) {
        drawGradientRect(10, virtualY, width - 10, virtualY + virtualHeight, -1072689136, -804253680);
        drawCenteredString(fontRendererObj, screenTitle, this.width / 2, virtualY + 5, Color.WHITE.getRGB());

        drawString(fontRendererObj, I18n.format("replaymod.gui.editkeyframe.timelineposition") + ":", (width - w3) / 2, min.yPosition + 7, Color.WHITE.getRGB());
        drawString(fontRendererObj, I18n.format("replaymod.gui.minutes"), min.xPosition + min.width + 5, min.yPosition + 7, Color.WHITE.getRGB());
        drawString(fontRendererObj, I18n.format("replaymod.gui.seconds"), sec.xPosition+sec.width+5, sec.yPosition+7, Color.WHITE.getRGB());
        drawString(fontRendererObj, I18n.format("replaymod.gui.milliseconds"), ms.xPosition + ms.width + 5, ms.yPosition + 7, Color.WHITE.getRGB());

        drawScreen0();

        inputs.draw(mc, mouseX, mouseY);

        super.drawScreen(mouseX, mouseY, partialTicks);
    }

    protected void drawScreen0() {}

    private static class GuiEditKeyframeMarker extends GuiEditKeyframe<Marker> {
        private GuiAdvancedTextField markerNameInput;

        public GuiEditKeyframeMarker(Keyframe<Marker> keyframe) {
            super(keyframe, ReplayHandler.getMarkerKeyframes());
        }

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

            if (!initialized) {
                String name = keyframe.getValue().getName();
                if (name == null) name = "";
                markerNameInput = new GuiAdvancedTextField(fontRendererObj, 0, 0, 200, 20);
                markerNameInput.hint = I18n.format("replaymod.gui.editkeyframe.markername");
                markerNameInput.setText(name);

                inputs.addPart(markerNameInput);

                new KeyframeValueChangeListener(this);
            }

            markerNameInput.xPosition = width/2 - 100;
            markerNameInput.yPosition = height/2-10;

            initialized = true;
        }

        @Override
        public void onGuiClosed() {
            keyframe.getValue().setName(markerNameInput.getText().trim());
            super.onGuiClosed();
        }
    }

    private static class GuiEditKeyframeTime extends GuiEditKeyframe<TimestampValue> {
        private GuiNumberInput kfMin, kfSec, kfMs;

        public GuiEditKeyframeTime(Keyframe<TimestampValue> keyframe) {
            super(keyframe, ReplayHandler.getTimeKeyframes());
            screenTitle = I18n.format("replaymod.gui.editkeyframe.title.time");
        }

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

            if (!initialized) {
                int time = keyframe.getValue().asInt();

                kfMin = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 30, 0d, null, (double)TimestampUtils.timestampToWholeMinutes(time), false, "", 1);
                kfSec = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 25, 0d, 59d, (double)TimestampUtils.getSecondsFromTimestamp(time), false, "", 1);
                kfMs = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 35, 0d, 999d, (double)TimestampUtils.getMillisecondsFromTimestamp(time), false, "", 1);

                inputs.addPart(kfMin);
                inputs.addPart(kfSec);
                inputs.addPart(kfMs);

                KeyframeValueChangeListener listener = new KeyframeValueChangeListener(this) {
                    @Override
                    public void onValueChange(double value) {
                        keyframe.setValue(new TimestampValue(TimestampUtils.calculateTimestamp(
                                kfMin.getIntValue(), kfSec.getIntValue(), kfMs.getIntValue())));

                        super.onValueChange(value);
                    }
                };

                kfMin.addValueChangeListener(listener);
                kfSec.addValueChangeListener(listener);
                kfMs.addValueChangeListener(listener);
            }

            kfMin.xPosition = min.xPosition;
            kfSec.xPosition = sec.xPosition;
            kfMs.xPosition = ms.xPosition;

            kfMin.yPosition = kfSec.yPosition = kfMs.yPosition = min.yPosition - 10 - 20;

            initialized = true;
        }

        @Override
        protected void drawScreen0() {
            drawString(fontRendererObj, I18n.format("replaymod.gui.editkeyframe.timestamp")+":", (width-w3)/2, kfMin.yPosition + 7, Color.WHITE.getRGB());
            drawString(fontRendererObj, I18n.format("replaymod.gui.minutes"), kfMin.xPosition + kfMin.width + 5, kfMin.yPosition + 7, Color.WHITE.getRGB());
            drawString(fontRendererObj, I18n.format("replaymod.gui.seconds"), kfSec.xPosition + kfSec.width + 5, kfSec.yPosition + 7, Color.WHITE.getRGB());
            drawString(fontRendererObj, I18n.format("replaymod.gui.milliseconds"), kfMs.xPosition + kfMs.width + 5, kfMs.yPosition + 7, Color.WHITE.getRGB());
        }

    }

    private static class GuiEditKeyframePosition extends GuiEditKeyframe<AdvancedPosition> {
        private GuiNumberInput xCoord, yCoord, zCoord, pitch, yaw, roll;
        private ComposedElement posInputs;

        public GuiEditKeyframePosition(Keyframe<AdvancedPosition> keyframe) {
            super(keyframe, ReplayHandler.getPositionKeyframes());
            screenTitle = I18n.format("replaymod.gui.editkeyframe.title.pos");
        }

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

            if (!initialized) {
                AdvancedPosition pos = keyframe.getValue();
                xCoord = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 100, null, null, RoundUtils.round2Decimals(pos.getX()), true);
                yCoord = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 100, null, null, RoundUtils.round2Decimals(pos.getY()), true);
                zCoord = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 100, null, null, RoundUtils.round2Decimals(pos.getZ()), true);
                yaw = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 100, null, null, RoundUtils.round2Decimals(pos.getYaw()), true);
                pitch = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 100, -90d, 90d, RoundUtils.round2Decimals(pos.getPitch()), true);
                roll = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 100, null, null, RoundUtils.round2Decimals(pos.getRoll()), true);

                posInputs = new ComposedElement(xCoord, yCoord, zCoord, yaw, pitch, roll);
                inputs.addPart(posInputs);

                KeyframeValueChangeListener listener = new KeyframeValueChangeListener(this) {
                    @Override
                    public void onValueChange(double value) {
                        keyframe.setValue(new AdvancedPosition(xCoord.getPreciseValue(), yCoord.getPreciseValue(),
                                zCoord.getPreciseValue(), new Float(pitch.getPreciseValue()), (float) yaw.getPreciseValue(),
                                (float) roll.getPreciseValue()));

                        super.onValueChange(value);
                    }
                };

                xCoord.addValueChangeListener(listener);
                yCoord.addValueChangeListener(listener);
                zCoord.addValueChangeListener(listener);
                pitch.addValueChangeListener(listener);
                yaw.addValueChangeListener(listener);
                roll.addValueChangeListener(listener);
            }

            int w = Math.max(fontRendererObj.getStringWidth(I18n.format("replaymod.gui.editkeyframe.xpos")),
                    Math.max(fontRendererObj.getStringWidth(I18n.format("replaymod.gui.editkeyframe.ypos")),
                            fontRendererObj.getStringWidth(I18n.format("replaymod.gui.editkeyframe.zpos"))));
            w2 = Math.max(fontRendererObj.getStringWidth(I18n.format("replaymod.gui.editkeyframe.camyaw")),
                    Math.max(fontRendererObj.getStringWidth(I18n.format("replaymod.gui.editkeyframe.campitch")),
                            fontRendererObj.getStringWidth(I18n.format("replaymod.gui.editkeyframe.camroll"))));

            totalWidth = w +100+w2+100+5+5+10;
            left = (this.width - totalWidth)/2;

            int x = w + left + 5;
            int i = 0;
            for(GuiElement input : posInputs.getParts()) {
                if(input instanceof GuiTextField) {
                    GuiTextField textField = (GuiTextField)input;
                    textField.xPosition = i < 3 ? x : left+totalWidth-100;
                    textField.yPosition = i < 3 ? virtualY + 20 + i*30 : virtualY + 20 + (i-3)*30;
                    i++;
                }
            }

            initialized = true;
        }

        @Override
        protected void drawScreen0() {
            if (keyframe.getValue() instanceof SpectatorData) return;
            drawString(fontRendererObj, I18n.format("replaymod.gui.editkeyframe.xpos"), left, virtualY + 27, Color.WHITE.getRGB());
            drawString(fontRendererObj, I18n.format("replaymod.gui.editkeyframe.ypos"), left, virtualY + 57, Color.WHITE.getRGB());
            drawString(fontRendererObj, I18n.format("replaymod.gui.editkeyframe.zpos"), left, virtualY + 87, Color.WHITE.getRGB());
            drawString(fontRendererObj, I18n.format("replaymod.gui.editkeyframe.camyaw"), left + totalWidth - 100 - 5 - w2, virtualY + 27, Color.WHITE.getRGB());
            drawString(fontRendererObj, I18n.format("replaymod.gui.editkeyframe.campitch"), left + totalWidth - 100 - 5 - w2, virtualY + 57, Color.WHITE.getRGB());
            drawString(fontRendererObj, I18n.format("replaymod.gui.editkeyframe.camroll"), left + totalWidth - 100 - 5 - w2, virtualY + 87, Color.WHITE.getRGB());
        }

    }

    private static class GuiEditKeyframeSpectator extends GuiEditKeyframe<AdvancedPosition> {
        private GuiToggleButton perspectiveButton;

        private GuiString shoulderDistanceString, shoulderPitchString, shoulderYawString, shoulderSmoothnessString;
        private GuiDraggingNumberInput shoulderDistanceInput, shoulderPitchOffsetInput,
                shoulderYawOffsetInput, shoulderSmoothnessInput;

        private ComposedElement spectatorCamSettings;
        private ComposedElement shoulderCamSettings;

        private DelegatingElement perspectiveSettings = new DelegatingElement() {
            @Override
            public GuiElement delegate() {
                switch(perspectiveButton.getValue()) {
                    case 0:
                        return spectatorCamSettings;
                    case 1:
                        return shoulderCamSettings;
                    default:
                        return null;
                }
            }
        };

        public GuiEditKeyframeSpectator(Keyframe<AdvancedPosition> keyframe) {
            super(keyframe, ReplayHandler.getPositionKeyframes());
            screenTitle = I18n.format("replaymod.gui.editkeyframe.title.spec");
        }

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

            if (!initialized) {
                SpectatorData data = (SpectatorData)keyframe.getValue();

                perspectiveButton = new GuiToggleButton(0, 0, 0, 200, 20, I18n.format("replaymod.gui.editkeyframe.spec.method")+": ", new String[]{
                        I18n.format("replaymod.gui.editkeyframe.spec.method.firstperson"),
                        I18n.format("replaymod.gui.editkeyframe.spec.method.shoulder")});

                perspectiveButton.setValue(Arrays.asList(SpectatorData.SpectatingMethod.values()).indexOf(data.getSpectatingMethod()));

                spectatorCamSettings = new ComposedElement();

                //create elements in shoulderCamSettings
                shoulderDistanceString = new GuiString(0, 0, Color.WHITE, I18n.format("replaymod.gui.editkeyframe.spec.method.shoulder.distance"));
                shoulderPitchString = new GuiString(0, 0, Color.WHITE, I18n.format("replaymod.gui.editkeyframe.spec.method.shoulder.pitch"));
                shoulderYawString = new GuiString(0, 0, Color.WHITE, I18n.format("replaymod.gui.editkeyframe.spec.method.shoulder.yaw"));
                shoulderSmoothnessString = new GuiString(0, 0, Color.WHITE, I18n.format("replaymod.gui.editkeyframe.spec.method.shoulder.smoothness"));

                shoulderDistanceInput = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 0, 0d, 30d, data.getShoulderCamDistance(), true, "", 0.1);
                shoulderPitchOffsetInput = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 0, -90d, 90d, (double)data.getShoulderCamPitchOffset(), false, "°", 1);
                shoulderYawOffsetInput = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 0, null, null, (double)data.getShoulderCamYawOffset(), false, "°", 1);
                shoulderSmoothnessInput = new GuiDraggingNumberInput(fontRendererObj, 0, 0, 0, 0.1d, 1d, data.getShoulderCamSmoothness()/1000d, true, "", 0.1);

                shoulderCamSettings = new ComposedElement(
                        shoulderDistanceString, shoulderDistanceInput,
                        shoulderPitchString, shoulderPitchOffsetInput,
                        shoulderYawString, shoulderYawOffsetInput,
                        shoulderSmoothnessString, shoulderSmoothnessInput);

                inputs.addPart(new ComposedElement(perspectiveButton, perspectiveSettings));
            }

            perspectiveButton.xPosition = (this.width-perspectiveButton.width())/2;
            perspectiveButton.yPosition = this.virtualY + 20;

            int verticalSpacing = 20 + 5;
            int totalWidth = 300;
            int elementWidth = 145;
            int horizontalSpacing = 10;

            int y = perspectiveButton.yPos() + verticalSpacing;

            int i = 0;
            for(GuiElement el : shoulderCamSettings.getParts()) {
                el.xPos((width-totalWidth)/2 + (i%2)*(elementWidth + horizontalSpacing));
                el.width(elementWidth);

                el.yPos(y + ((i+1)%2)*7);

                if(i%2 == 1) {
                    y += verticalSpacing;
                }
                i++;
            }

            initialized = true;
        }

        @Override
        public void onGuiClosed() {
            SpectatorData data = (SpectatorData)keyframe.getValue();
            data.setSpectatingMethod(SpectatorData.SpectatingMethod.values()[perspectiveButton.getValue()]);
            data.setShoulderCamDistance(shoulderDistanceInput.getPreciseValue());
            data.setShoulderCamPitchOffset(shoulderPitchOffsetInput.getIntValue());
            data.setShoulderCamYawOffset(shoulderYawOffsetInput.getIntValue());
            data.setShoulderCamSmoothness((int)(shoulderSmoothnessInput.getPreciseValue()*1000));
            super.onGuiClosed();
        }
    }

    private static class KeyframeValueChangeListener implements NumberValueChangeListener {

        private GuiEditKeyframe parent;

        public KeyframeValueChangeListener(GuiEditKeyframe parent) {
            this.parent = parent;

            parent.min.addValueChangeListener(this);
            parent.sec.addValueChangeListener(this);
            parent.ms.addValueChangeListener(this);
        }

        @Override
        public void onValueChange(double value) {
            int realTimestamp = TimestampUtils.calculateTimestamp(parent.min.getIntValue(), parent.sec.getIntValue(), parent.ms.getIntValue());
            parent.keyframe.setRealTimestamp(realTimestamp);

            ReplayHandler.fireKeyframesModifyEvent();
        }
    }
}