/*******************************************************************************
 * Copyright (c) 2001, 2010 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.wst.xsd.ui.internal.adt.outline;

import java.util.List;
import java.util.Stack;

import org.eclipse.emf.common.notify.Notifier;
import org.eclipse.emf.common.notify.impl.AdapterImpl;
import org.eclipse.jface.action.IStatusLineManager;
import org.eclipse.jface.action.MenuManager;
import org.eclipse.jface.viewers.ILabelProvider;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.ISelectionChangedListener;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.SelectionChangedEvent;
import org.eclipse.jface.viewers.StructuredSelection;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.swt.events.KeyAdapter;
import org.eclipse.swt.events.MouseAdapter;
import org.eclipse.swt.events.MouseEvent;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Menu;
import org.eclipse.ui.part.MultiPageSelectionProvider;
import org.eclipse.wst.xml.ui.internal.contentoutline.JFaceNodeLabelProvider;
import org.eclipse.wst.xsd.ui.internal.adapters.CategoryAdapter;
import org.eclipse.wst.xsd.ui.internal.adapters.RedefineCategoryAdapter;
import org.eclipse.wst.xsd.ui.internal.adapters.XSDAttributeDeclarationAdapter;
import org.eclipse.wst.xsd.ui.internal.adapters.XSDAttributeGroupDefinitionAdapter;
import org.eclipse.wst.xsd.ui.internal.adapters.XSDComplexTypeDefinitionAdapter;
import org.eclipse.wst.xsd.ui.internal.adapters.XSDElementDeclarationAdapter;
import org.eclipse.wst.xsd.ui.internal.adapters.XSDModelGroupDefinitionAdapter;
import org.eclipse.wst.xsd.ui.internal.adapters.XSDRedefineAdapter;
import org.eclipse.wst.xsd.ui.internal.adapters.XSDSchemaAdapter;
import org.eclipse.wst.xsd.ui.internal.adapters.XSDSchemaDirectiveAdapter;
import org.eclipse.wst.xsd.ui.internal.adapters.XSDSimpleTypeDefinitionAdapter;
import org.eclipse.wst.xsd.ui.internal.adt.design.DesignViewContextMenuProvider;
import org.eclipse.wst.xsd.ui.internal.adt.design.editparts.model.IModelProxy;
import org.eclipse.wst.xsd.ui.internal.adt.facade.IModel;
import org.eclipse.xsd.XSDConcreteComponent;
import org.eclipse.xsd.XSDSchema;

public class ADTContentOutlinePage extends ExtensibleContentOutlinePage
{
  protected int level = 0;
  protected MultiPageSelectionProvider selectionManager;
  protected SelectionManagerSelectionChangeListener selectionManagerSelectionChangeListener = new SelectionManagerSelectionChangeListener();

  public ADTContentOutlinePage()
  {
    super();
  }

  public void createControl(Composite parent)
  {
    super.createControl(parent);
    getTreeViewer().setContentProvider(contentProvider);
    getTreeViewer().setLabelProvider(labelProvider);
    getTreeViewer().setInput(model);
    getTreeViewer().addSelectionChangedListener(this);
    MenuManager menuManager = new MenuManager("#popup");//$NON-NLS-1$
    menuManager.setRemoveAllWhenShown(true);
    Menu menu = menuManager.createContextMenu(getTreeViewer().getControl());
    getTreeViewer().getControl().setMenu(menu);
    setSelectionManager(editor.getSelectionManager());
    
    // Create menu...for now reuse graph's.  Note edit part viewer = null
    DesignViewContextMenuProvider menuProvider = new DesignViewContextMenuProvider(editor, null, editor.getSelectionManager());
    menuManager.addMenuListener(menuProvider);
    getSite().registerContextMenu("org.eclipse.wst.xsd.ui.popup.outline", menuManager, editor.getSelectionManager()); //$NON-NLS-1$

    // enable popupMenus extension
    // getSite().registerContextMenu("org.eclipse.wst.xsdeditor.ui.popup.outline",
    // menuManager, xsdEditor.getSelectionManager());

    // drill down from outline view
    getTreeViewer().getControl().addMouseListener(new MouseAdapter()
    {
      public void mouseDoubleClick(MouseEvent e)
      {
        ISelection iSelection = getTreeViewer().getSelection();
        if (iSelection instanceof StructuredSelection)
        {
          StructuredSelection selection = (StructuredSelection) iSelection;
          Object obj = selection.getFirstElement();
          if (obj instanceof XSDConcreteComponent)
          {
            XSDConcreteComponent comp = (XSDConcreteComponent) obj;
            if (comp.getContainer() instanceof XSDSchema)
            {
              // getXSDEditor().getGraphViewer().setInput(obj);
            }
          }
        }

      }
    });
  }

  class XSDKeyListener extends KeyAdapter
  {
  }

  public void dispose()
  {
    contentProvider.dispose();
    super.dispose();
  }

  public void setExpandToLevel(int i)
  {
    level = i;
  }

  public void setInput(Object value)
  {
    getTreeViewer().setInput(value);
    getTreeViewer().expandToLevel(level);
  }

  // public ISelection getSelection()
  // {
  // if (getTreeViewer() == null)
  // return StructuredSelection.EMPTY;
  // return getTreeViewer().getSelection();
  // }
  public void setSelectionManager(MultiPageSelectionProvider newSelectionManager)
  {
    // disconnect from old one
    if (selectionManager != null)
    {
      selectionManager.removeSelectionChangedListener(selectionManagerSelectionChangeListener);
    }
    selectionManager = newSelectionManager;
    // connect to new one
    if (selectionManager != null)
    {
      selectionManager.addSelectionChangedListener(selectionManagerSelectionChangeListener);
    }
  }

  protected class SelectionManagerSelectionChangeListener implements ISelectionChangedListener
  {
    public void selectionChanged(SelectionChangedEvent event)
    {
      updateStatusLine(getSite().getActionBars().getStatusLineManager(), event.getSelection());
      if (event.getSelectionProvider() != ADTContentOutlinePage.this)  //getTreeViewer())
      {
        StructuredSelection selection = (StructuredSelection)event.getSelection();
        StructuredSelection currentSelection = (StructuredSelection) getTreeViewer().getSelection();
        
        Object selectionFirstElement = selection.getFirstElement();
        Object currentFirstElement = currentSelection.getFirstElement();
       
        // TODO: Hack to prevent losing a selection when the schema is selected in the
        // source.  Fix is to prevent the source from firing off selection changes when
        // the selection source is not the source view.
        
        if (selectionFirstElement instanceof IModel)
        {          
          if (!(currentFirstElement instanceof IModelProxy) || currentFirstElement instanceof RedefineCategoryAdapter)
          {
            getTreeViewer().setSelection(event.getSelection(), true);
          }
        }
        else if (selectionFirstElement instanceof XSDSchemaDirectiveAdapter)
        {
        	if (!(currentFirstElement instanceof RedefineCategoryAdapter))
        	{
        		getTreeViewer().setSelection(event.getSelection(), true);
        	}
        }
        else
        {
          IStructuredSelection structuredSelection = (IStructuredSelection) event.getSelection();
          List list = structuredSelection.toList();
          int selectionLength = list.size();
          if (selectionLength > 0)
          {
            for (int i = 0; i < selectionLength; i++)
            {
              Object item = list.get(i);

              if (item instanceof AdapterImpl)
              {
                Notifier target = ((AdapterImpl) item).getTarget();

                if (!(target instanceof XSDConcreteComponent))
                  continue;

                XSDConcreteComponent xsdConcreteComponent = (XSDConcreteComponent) target;

                // need to expand parent before child, thus a stack is required
                Stack componentStack = new Stack();
                while (xsdConcreteComponent != null)
                {
                  componentStack.push(xsdConcreteComponent);
                  xsdConcreteComponent = xsdConcreteComponent.getContainer();
                }

                // At this point the stack must have the XSDSchemaAdapter and
                // at least one component
                if (componentStack.size() < 2)
                  continue;

                // Pop off the top node, since it is the XSDSchemaAdapter, which
                // isn't
                // used as part of finding the outline view hierarchy
                componentStack.pop();

                if (((XSDConcreteComponent) componentStack.peek()).eAdapters().size() <= 0)
                  continue;

                Object object = ((XSDConcreteComponent) componentStack.peek()).eAdapters().get(0);

                // Find out which category the selected item is contained in
                int categoryIndex = -1;
                if (object instanceof XSDSchemaDirectiveAdapter)
                {
                  categoryIndex = CategoryAdapter.DIRECTIVES;
                }
                else if (object instanceof XSDElementDeclarationAdapter)
                {
                  categoryIndex = CategoryAdapter.ELEMENTS;
                }
                else if (object instanceof XSDAttributeDeclarationAdapter)
                {
                  categoryIndex = CategoryAdapter.ATTRIBUTES;
                }
                else if (object instanceof XSDAttributeGroupDefinitionAdapter)
                {
                  categoryIndex = CategoryAdapter.ATTRIBUTES;
                }
                else if (object instanceof XSDComplexTypeDefinitionAdapter || object instanceof XSDSimpleTypeDefinitionAdapter)
                {
                  categoryIndex = CategoryAdapter.TYPES;
                }
                else if (object instanceof XSDModelGroupDefinitionAdapter)
                {
                  categoryIndex = CategoryAdapter.GROUPS;
                }

                // Expand the category
                if (categoryIndex == -1)
                  continue;
                CategoryAdapter category = ((XSDSchemaAdapter) model).getCategory(categoryIndex);
                treeViewer.setExpandedState(category, true);

                // Do not expand current node of interest, just its parents
                while (componentStack.size() > 1)
                {
                  object = componentStack.pop();
                  if (object instanceof XSDConcreteComponent)
                  {
                    XSDConcreteComponent component = (XSDConcreteComponent) object;
                    if (component.eAdapters().size() > 0)
                    {
                      // expand
                      getTreeViewer().setExpandedState(component.eAdapters().get(0), true);
                    }
                  }
                }
              }
            }
          }
          // Bug 251474 - We will restrict selection to only one item
          if (selectionLength == 1)
          {
            getTreeViewer().setSelection(event.getSelection(), true);
          }
          else
          {
        	  if(structuredSelection.getFirstElement() !=null)
        		  getTreeViewer().setSelection(new StructuredSelection(structuredSelection.getFirstElement()), true);
          }
        }
      }
    }
  }
  
  protected void updateStatusLine(IStatusLineManager mgr, ISelection selection)
  {
    String text = null;
    Image image = null;
    ILabelProvider statusLineLabelProvider = new StatusLineLabelProvider(getTreeViewer());
    if (statusLineLabelProvider != null && selection instanceof IStructuredSelection && !selection.isEmpty())
    {
      Object firstElement = ((IStructuredSelection) selection).getFirstElement();
      if (firstElement instanceof XSDRedefineAdapter)
    	  return;
      text = statusLineLabelProvider.getText(firstElement);
      image = statusLineLabelProvider.getImage(firstElement);
    }
    if (image == null)
    {
      mgr.setMessage(text);
    }
    else
    {
      mgr.setMessage(image, text);
    }
  }
  
  protected class StatusLineLabelProvider extends JFaceNodeLabelProvider
  {
    TreeViewer treeViewer = null;

    public StatusLineLabelProvider(TreeViewer viewer)
    {
      treeViewer = viewer;
    }

    public String getText(Object element)
    {
      if (element == null)
        return null;

      StringBuffer s = new StringBuffer();
      s.append(labelProvider.getText(element));
      return s.toString();
    }

    public Image getImage(Object element)
    {
      return labelProvider.getImage(element);
    }
  }


}