package org.dbflute.erflute.editor.view.outline;

import org.dbflute.erflute.Activator;
import org.dbflute.erflute.editor.MainDiagramEditor;
import org.dbflute.erflute.editor.controller.command.ermodel.OpenERModelCommand;
import org.dbflute.erflute.editor.controller.editpart.outline.ERDiagramOutlineEditPart;
import org.dbflute.erflute.editor.controller.editpart.outline.ERDiagramOutlineEditPartFactory;
import org.dbflute.erflute.editor.controller.editpart.outline.table.TableOutlineEditPart;
import org.dbflute.erflute.editor.model.ERDiagram;
import org.dbflute.erflute.editor.model.diagram_contents.element.node.ermodel.ERVirtualDiagram;
import org.dbflute.erflute.editor.model.diagram_contents.element.node.table.ERTable;
import org.dbflute.erflute.editor.view.action.outline.ChangeVirtualDiagramNameAction;
import org.dbflute.erflute.editor.view.action.outline.DeleteVirtualDiagramAction;
import org.dbflute.erflute.editor.view.action.outline.index.CreateIndexAction;
import org.dbflute.erflute.editor.view.action.outline.notation.type.ChangeOutlineViewToBothAction;
import org.dbflute.erflute.editor.view.action.outline.notation.type.ChangeOutlineViewToLogicalAction;
import org.dbflute.erflute.editor.view.action.outline.notation.type.ChangeOutlineViewToPhysicalAction;
import org.dbflute.erflute.editor.view.action.outline.orderby.ChangeOutlineViewOrderByLogicalNameAction;
import org.dbflute.erflute.editor.view.action.outline.orderby.ChangeOutlineViewOrderByPhysicalNameAction;
import org.dbflute.erflute.editor.view.action.outline.sequence.CreateSequenceAction;
import org.dbflute.erflute.editor.view.action.outline.tablespace.CreateTablespaceAction;
import org.dbflute.erflute.editor.view.action.outline.trigger.CreateTriggerAction;
import org.dbflute.erflute.editor.view.drag_drop.ERDiagramOutlineTransferDropTargetListener;
import org.dbflute.erflute.editor.view.drag_drop.ERDiagramTransferDragSourceListener;
import org.eclipse.draw2d.LightweightSystem;
import org.eclipse.draw2d.Viewport;
import org.eclipse.draw2d.parts.ScrollableThumbnail;
import org.eclipse.gef.EditDomain;
import org.eclipse.gef.EditPartViewer;
import org.eclipse.gef.GraphicalViewer;
import org.eclipse.gef.LayerConstants;
import org.eclipse.gef.dnd.AbstractTransferDragSourceListener;
import org.eclipse.gef.dnd.AbstractTransferDropTargetListener;
import org.eclipse.gef.dnd.TemplateTransfer;
import org.eclipse.gef.editparts.ScalableFreeformRootEditPart;
import org.eclipse.gef.ui.actions.ActionRegistry;
import org.eclipse.gef.ui.parts.ContentOutlinePage;
import org.eclipse.gef.ui.parts.TreeViewer;
import org.eclipse.jface.action.IAction;
import org.eclipse.jface.action.MenuManager;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.swt.SWT;
import org.eclipse.swt.custom.SashForm;
import org.eclipse.swt.widgets.Canvas;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Control;
import org.eclipse.swt.widgets.Tree;
import org.eclipse.swt.widgets.TreeItem;
import org.eclipse.ui.IActionBars;
import org.eclipse.ui.actions.ActionFactory;

/**
 * @author modified by jflute (originated in ermaster)
 */
public class ERDiagramOutlinePage extends ContentOutlinePage {

    private final int VIRTUAL_DIAGRAM_NODE_INDEX = 0;

    private SashForm sash;
    private final TreeViewer viewer;
    private final ERDiagram diagram;
    private LightweightSystem lws;
    private ScrollableThumbnail thumbnail;
    private GraphicalViewer graphicalViewer;
    private final ActionRegistry outlineActionRegistory;
    private ActionRegistry registry;
    private boolean quickMode;
    private ERDiagramOutlineEditPartFactory editPartFactory;

    public ERDiagramOutlinePage(ERDiagram diagram) {
        super(new TreeViewer());
        this.viewer = (TreeViewer) getViewer();
        this.diagram = diagram;
        this.outlineActionRegistory = new ActionRegistry();
        registerAction(viewer, outlineActionRegistory);
    }

    @Override
    public void createControl(Composite parent) {
        this.sash = new SashForm(parent, SWT.VERTICAL);
        viewer.createControl(sash);
        editPartFactory = new ERDiagramOutlineEditPartFactory();
        editPartFactory.setQuickMode(quickMode);
        viewer.setEditPartFactory(editPartFactory);
        viewer.setContents(diagram);
        if (!quickMode) {
            final Canvas canvas = new Canvas(sash, SWT.BORDER);
            lws = new LightweightSystem(canvas);
        }
        resetView(registry);
        final AbstractTransferDragSourceListener dragSourceListener =
                new ERDiagramTransferDragSourceListener(viewer, TemplateTransfer.getInstance());
        viewer.addDragSourceListener(dragSourceListener);

        expandVirturalDiagramTree();
    }

    @Override
    public Control getControl() {
        return sash;
    }

    private void showThumbnail() {
        if (quickMode) {
            return;
        }
        final ScalableFreeformRootEditPart editPart = (ScalableFreeformRootEditPart) graphicalViewer.getRootEditPart();
        if (thumbnail != null) {
            thumbnail.deactivate();
        }
        this.thumbnail = new ScrollableThumbnail((Viewport) editPart.getFigure());
        thumbnail.setSource(editPart.getLayer(LayerConstants.PRINTABLE_LAYERS));
        lws.setContents(thumbnail);
    }

    private void initDropTarget() {
        final AbstractTransferDropTargetListener dropTargetListener =
                new ERDiagramOutlineTransferDropTargetListener(graphicalViewer, TemplateTransfer.getInstance());
        graphicalViewer.addDropTargetListener(dropTargetListener);
    }

    public void setContextMenu(MenuManager outlineMenuMgr) {
        viewer.setContextMenu(outlineMenuMgr);
    }

    public void setCategory(EditDomain editDomain, GraphicalViewer graphicalViewer, ActionRegistry registry) {
        this.graphicalViewer = graphicalViewer;

        viewer.setEditDomain(editDomain);
        this.registry = registry;

        if (getSite() != null) {
            resetView(registry);
        }
    }

    private void resetAction(ActionRegistry registry) {
        if (getSite() == null) {
            return;
        }
        final IActionBars bars = getSite().getActionBars();

        String id = ActionFactory.UNDO.getId();
        bars.setGlobalActionHandler(id, registry.getAction(id));

        id = ActionFactory.REDO.getId();
        bars.setGlobalActionHandler(id, registry.getAction(id));

        id = ActionFactory.DELETE.getId();
        bars.setGlobalActionHandler(id, registry.getAction(id));

        bars.updateActionBars();
    }

    private void resetView(ActionRegistry registry) {
        showThumbnail();
        initDropTarget();
        resetAction(registry);
    }

    private void registerAction(TreeViewer treeViewer, ActionRegistry actionRegistry) {
        final IAction[] actions = { new CreateIndexAction(treeViewer), new CreateSequenceAction(treeViewer),
                new CreateTriggerAction(treeViewer), new CreateTablespaceAction(treeViewer),
                new ChangeOutlineViewToPhysicalAction(treeViewer), new ChangeOutlineViewToLogicalAction(treeViewer),
                new ChangeOutlineViewToBothAction(treeViewer), new ChangeOutlineViewOrderByPhysicalNameAction(treeViewer),
                new ChangeOutlineViewOrderByLogicalNameAction(treeViewer), new ChangeVirtualDiagramNameAction(treeViewer),
                new DeleteVirtualDiagramAction(treeViewer), };
        for (final IAction action : actions) {
            actionRegistry.registerAction(action);
        }
    }

    public ActionRegistry getOutlineActionRegistory() {
        return outlineActionRegistory;
    }

    @Override
    public EditPartViewer getViewer() {
        return super.getViewer();
    }

    public void update() {
        viewer.flush();
    }

    public void setFilterText(String filterText) {
        editPartFactory.setFilterText(filterText);
        viewer.setContents(diagram);
        final Tree tree = (Tree) viewer.getControl();
        final TreeItem[] items = tree.getItems();
        expand(items);
        final TreeItem[] tableItems = items[0].getItems();
        if (tableItems.length >= 1) {
            tree.setSelection(tableItems[0]);
        }
    }

    private void expand(TreeItem[] items) {
        for (int i = 0; i < items.length; i++) {
            expand(items[i].getItems());
            items[i].setExpanded(true);
        }
    }

    public void setQuickMode(boolean quickMode) {
        this.quickMode = quickMode;
    }

    public void selectSelection() {
        final IStructuredSelection sel = (IStructuredSelection) getViewer().getSelection();
        Object firstElement = sel.getFirstElement();
        if (firstElement instanceof ERDiagramOutlineEditPart) {
            final Tree tree = (Tree) viewer.getControl();
            final TreeItem[] items = tree.getItems();
            expand(items);
            final TreeItem[] tableItems = items[0].getItems();
            if (tableItems.length >= 1) {
                final Object data = tableItems[0].getData();
                firstElement = data;
            }
        }
        if (firstElement instanceof TableOutlineEditPart) {
            final Object model = ((TableOutlineEditPart) firstElement).getModel();
            final ERTable table = (ERTable) model;

            final MainDiagramEditor editor = ((MainDiagramEditor) diagram.getEditor().getActiveEditor());
            if (!diagram.isVirtual()) {
                editor.reveal(table);
                return;
            }

            final ERVirtualDiagram erModel = table.getDiagram().findModelByTable(table);
            if (erModel != null) {
                final OpenERModelCommand command = new OpenERModelCommand(diagram, erModel);
                command.setTable(table.toMaterialize());
                command.execute(); // コマンドスタックには積まないで実行する。ファイル編集中にしないため。

                // TODO ymd アウトラインツリー上の仮想ダイアグラムを選択するためにのみある。
                // この処理があることで、クイックアウトライン検索→テーブル選択がバグるのでコメントにした。不要なことを確信したら、削除する。
                //                final ERDiagramOutlineEditPart contents =
                //                        (ERDiagramOutlineEditPart) diagram.getEditor().getOutlinePage().getViewer().getContents();
                //                if (contents != null) {
                //                    final ERVirtualDiagramSetOutlineEditPart virtualDiagramSetOutlineEditPart =
                //                            (ERVirtualDiagramSetOutlineEditPart) contents.getChildren().get(0);
                //                    @SuppressWarnings("unchecked")
                //                    final List<ERVirtualDiagramOutlineEditPart> parts = virtualDiagramSetOutlineEditPart.getChildren();
                //                    for (final ERVirtualDiagramOutlineEditPart part : parts) {
                //                        if (part.getModel().equals(erModel)) {
                //                            final ISelection selection = new StructuredSelection(part);
                //                            diagram.getEditor().getOutlinePage().setSelection(selection);
                //                        }
                //                    }
                //                }
            } else {
                // ここでrevealしないと、メインダイアグラムでtableが選択不可状態になる。理由は分からない。
                editor.reveal(table);
                Activator.showMessageDialog("\"" + table.getPhysicalName() + "\"does not exist on the virtual diagram.");
            }
        }
    }

    private void expandVirturalDiagramTree() {
        if (getTree() == null) {
            return;
        }

        getTree().getItems()[VIRTUAL_DIAGRAM_NODE_INDEX].setExpanded(true);
    }

    private Tree getTree() {
        return (Tree) viewer.getControl();
    }
}
