package application;

import application.editor.Editor;
import application.views.NavigationWindow;
import com.mxgraph.model.mxCell;
import com.mxgraph.model.mxGeometry;
import com.mxgraph.model.mxGraphModel;
import com.mxgraph.swing.handler.mxRubberband;
import com.mxgraph.swing.mxGraphComponent;
import com.mxgraph.util.mxEvent;
import com.mxgraph.view.mxGraph;

import javax.swing.*;
import java.util.ArrayList;
import java.util.List;

public class ApplicationWindow extends JFrame {

    private static final long serialVersionUID = -8690140317781055614L;

    public static final String TITLE = "Visual Modeling Tool";

    private Editor editor;
    private final mxGraph graph;
    private final mxGraphComponent graphComponent;

    private final ApplicationMenuBar menuBar;
    private final NavigationWindow navigationWindow;

    public ApplicationWindow() {
        graph = new mxGraph() {
            public boolean isPort(Object cell) {
                mxGeometry geo = getCellGeometry(cell);
                return geo != null && geo.isRelative();
            }

            public boolean isCellFoldable(Object cell, boolean collapse) {
                return false;
            }
        };

        graphComponent = new mxGraphComponent(graph);
        editor = new Editor(graphComponent);

        graph.getModel().addListener(mxEvent.CHANGE, (sender, event) -> {
            List<mxCell> terminals = new ArrayList<>();
            mxCell cell = null;
            for (Object change : ((List) event.getProperties().get("changes"))) {
                if (change instanceof mxGraphModel.mxTerminalChange) {
                    mxGraphModel.mxTerminalChange terminalChange = (mxGraphModel.mxTerminalChange) change;
                    cell = (mxCell) terminalChange.getCell();
                    mxCell terminal = (mxCell) terminalChange.getTerminal();
                    terminals.add(terminal);
                }
            }
            if (terminals.size() == 2) {
                if (!editor.connectEdge(cell, terminals.get(0), terminals.get(1))) {
                    graph.removeCells(new mxCell[]{cell});
                }
            }
        });
        getContentPane().add(graphComponent);
        new mxRubberband(graphComponent);
        graph.setAllowDanglingEdges(false);
        graph.setCellsDisconnectable(true);

        menuBar = new ApplicationMenuBar(this);
        setTitle(TITLE);
        setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        setJMenuBar(menuBar);
        setSize(870, 640);
        setLocationRelativeTo(null); // Center on screen

        navigationWindow = new NavigationWindow(this);
        navigationWindow.setVisible(true);
        editor.addStageChangeListener(navigationWindow);
    }

    public mxGraph getGraph() {
        return graph;
    }

    public mxGraphComponent getGraphComponent() {
        return graphComponent;
    }

    public Editor getEditor() {
        return editor;
    }

    public void setEditor(Editor editor) {
        this.editor = editor;
    }

    public void showNavigationWindow() {
        navigationWindow.setVisible(true);
    }
}