/*******************************************************************************
 * Copyright (c) 2001, 2004 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.provider;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.eclipse.emf.common.notify.AdapterFactory;
import org.eclipse.emf.common.notify.Notification;
import org.eclipse.emf.common.notify.impl.NotificationImpl;
import org.eclipse.emf.common.util.URI;
import org.eclipse.swt.graphics.Image;
import org.eclipse.wst.xsd.ui.internal.XSDEditorPlugin;
import org.eclipse.xsd.XSDAttributeDeclaration;
import org.eclipse.xsd.XSDAttributeGroupDefinition;
import org.eclipse.xsd.XSDComplexTypeDefinition;
import org.eclipse.xsd.XSDElementDeclaration;
import org.eclipse.xsd.XSDModelGroupDefinition;
import org.eclipse.xsd.XSDNotationDeclaration;
import org.eclipse.xsd.XSDPackage;
import org.eclipse.xsd.XSDSchema;
import org.eclipse.xsd.XSDSchemaDirective;
import org.eclipse.xsd.XSDSimpleTypeDefinition;
import org.eclipse.xsd.XSDTypeDefinition;


public class XSDSchemaAdapter extends XSDAbstractAdapter
{
  protected XSDPackage xsdPackage;
  /**
   * @param adapterFactory
   */
  public XSDSchemaAdapter(AdapterFactory adapterFactory)
  {
    super(adapterFactory);
    xsdPackage = XSDPackage.eINSTANCE;
  }

  public Image getImage(Object element)
  {
    return XSDEditorPlugin.getXSDImage("icons/XSDFile.gif");
  }
  
  public String getText(Object element)
  {
    XSDSchema xsdSchema = (XSDSchema)element;
    String result = xsdSchema.getSchemaLocation();
    if (result == null)
    {
      return "";
    } 
    else
    {
      return URI.createURI(result).lastSegment();
    }
  }
  List children;
  public Object[] getChildren(Object parentElement)
  {
    XSDSchema xsdSchema = ((XSDSchema)parentElement);

    children = new ArrayList();
    
    List directivesList = getDirectives(xsdSchema);
    
    List elementsList = getGlobalElements(xsdSchema);
    
    List attributeGroupList = getAttributeGroupList(xsdSchema);
    
    List attributesList = getAttributeList(xsdSchema);
    
    List groups = getGroups(xsdSchema);
    
    List notations = getNotations(xsdSchema);

    List types = getComplexTypes(xsdSchema);
    types.addAll(getSimpleTypes(xsdSchema));
    
    children.add
    (new CategoryAdapter
      ( //XSDEditPlugin.getString("_UI_Elements_label"), 
        XSDEditorPlugin.getXSDString("_UI_GRAPH_DIRECTIVES"),
        XSDEditorPlugin.getPlugin().getIconImage("obj16/directivesheader"),
        directivesList, xsdSchema, CategoryAdapter.DIRECTIVES));
    children.add
        (new CategoryAdapter
          ( //XSDEditPlugin.getString("_UI_Elements_label"), 
            XSDEditorPlugin.getXSDString("_UI_GRAPH_ELEMENTS"),
            XSDEditorPlugin.getPlugin().getIconImage("obj16/elementsheader"),
            elementsList, xsdSchema, CategoryAdapter.ELEMENTS));
      children.add
        (new CategoryAdapter
          ( //XSDEditPlugin.getString("_UI_Attributes_label"),
            XSDEditorPlugin.getXSDString("_UI_GRAPH_ATTRIBUTES"),
            XSDEditorPlugin.getPlugin().getIconImage("obj16/attributesheader"),
            attributesList, xsdSchema, CategoryAdapter.ATTRIBUTES));
      children.add
        (new CategoryAdapter
          (//XSDEditPlugin.getString("_UI_AttributeGroups_label"),
            XSDEditorPlugin.getXSDString("_UI_GRAPH_ATTRIBUTE_GROUPS"),
            XSDEditorPlugin.getPlugin().getIconImage("obj16/attributegroupsheader"),
            attributeGroupList, xsdSchema, CategoryAdapter.ATTRIBUTE_GROUPS));
      children.add
        (new CategoryAdapter
          ( //XSDEditPlugin.getString("_UI_Types_label"), 
            XSDEditorPlugin.getXSDString("_UI_GRAPH_TYPES"),
            XSDEditorPlugin.getPlugin().getIconImage("obj16/typesheader"),
            types, xsdSchema, CategoryAdapter.TYPES));
      children.add
        (new CategoryAdapter
          ( // XSDEditPlugin.getString("_UI_ModelGroups_label"), 
            XSDEditorPlugin.getXSDString("_UI_GRAPH_GROUPS"),
            XSDEditorPlugin.getPlugin().getIconImage("obj16/groupsheader"),
            groups, xsdSchema, CategoryAdapter.GROUPS));
      children.add
        (new CategoryAdapter
          ( // XSDEditPlugin.getString("_UI_Notations_label"), 
            XSDEditorPlugin.getXSDString("_UI_GRAPH_NOTATIONS"),
            XSDEditorPlugin.getPlugin().getIconImage("obj16/notationsheader"),
            notations, xsdSchema, CategoryAdapter.NOTATIONS));
//      children.add
//        (new CategoryAdapter
//          ( //XSDEditPlugin.getString("_UI_IdentityConstraints_label"), 
//              "Identity Constraints",
//            XSDEditorPlugin.getPlugin().getIconImage("full/obj16/XSDIdentityConstraintDefinitionKey"),
//            xsdSchema.getIdentityConstraintDefinitions(), xsdSchema, CategoryAdapter.IDENTITY_CONSTRAINTS));
//      children.add
//        (new CategoryAdapter
//          ( // XSDEditPlugin.getString("_UI_Annotations_label"), 
//              "Annotations",
//            XSDEditorPlugin.getPlugin().getIconImage("obj16/annotationsheader"),
//            xsdSchema.getAnnotations(), xsdSchema, CategoryAdapter.ANNOTATIONS));

    return children.toArray();
  
  }
  
  public boolean hasChildren(Object object)
  {
    return true;
  }

  public Object getParent(Object object)
  {
    return null;
  }
  
  public void notifyChanged(final Notification msg)
  {
    class CategoryNotification extends NotificationImpl
    {
      protected Object category;
      public CategoryNotification(Object category)
      {
        super(msg.getEventType(), msg.getOldValue(), msg.getNewValue(), msg.getPosition());
        this.category = category;
      }

      public Object getNotifier()
      {
        return category;
      }
      public Object getFeature()
      {
        return msg.getFeature();
      }
    }
    
    if (children == null) {
    	getChildren(target);
    }

    if (msg.getFeature() == xsdPackage.getXSDSchema_Contents())
    {
      CategoryAdapter adapter = (CategoryAdapter)children.get(0);
      XSDSchema xsdSchema = adapter.getXSDSchema();
      adapter.setChildren(getDirectives(xsdSchema));
      this.fireNotifyChanged(new CategoryNotification(adapter));
      return;
    }
    else if (msg.getFeature() == xsdPackage.getXSDSchema_ElementDeclarations())
    {
      CategoryAdapter adapter = (CategoryAdapter)children.get(1);
      XSDSchema xsdSchema = adapter.getXSDSchema();
      adapter.setChildren(getGlobalElements(xsdSchema));
      this.fireNotifyChanged(new CategoryNotification(adapter));
      return;
    }
    else if (msg.getFeature() == xsdPackage.getXSDSchema_AttributeDeclarations())
    {
      CategoryAdapter adapter = (CategoryAdapter)children.get(2);
      XSDSchema xsdSchema = adapter.getXSDSchema();
      adapter.setChildren(getAttributeList(xsdSchema));
      this.fireNotifyChanged(new CategoryNotification(adapter));
      return;
    }
    else if (msg.getFeature() == xsdPackage.getXSDSchema_AttributeGroupDefinitions())
    {
      CategoryAdapter adapter = (CategoryAdapter)children.get(3);
      XSDSchema xsdSchema = adapter.getXSDSchema();
      adapter.setChildren(getAttributeGroupList(xsdSchema));
      this.fireNotifyChanged(new CategoryNotification(adapter));
      return;
    }
    else if (msg.getFeature() == xsdPackage.getXSDSchema_TypeDefinitions())
    {
      CategoryAdapter adapter = (CategoryAdapter)children.get(4);
      XSDSchema xsdSchema = adapter.getXSDSchema();
      List types = getComplexTypes(xsdSchema);
      types.addAll(getSimpleTypes(xsdSchema));

      adapter.setChildren(types);
      this.fireNotifyChanged(new CategoryNotification(adapter));
      return;
    }
    else if (msg.getFeature() == xsdPackage.getXSDSchema_ModelGroupDefinitions())
    {
      CategoryAdapter adapter = (CategoryAdapter)children.get(5);
      XSDSchema xsdSchema = adapter.getXSDSchema();
      adapter.setChildren(getGroups(xsdSchema));
      this.fireNotifyChanged(new CategoryNotification(adapter));
      return;
    }
    else if (msg.getFeature() == xsdPackage.getXSDSchema_NotationDeclarations())
    {
      CategoryAdapter adapter = (CategoryAdapter)children.get(6);
      XSDSchema xsdSchema = adapter.getXSDSchema();
      adapter.setChildren(getNotations(xsdSchema));
      this.fireNotifyChanged(new CategoryNotification(adapter));
      return;
    }
//    else if (msg.getFeature() == xsdPackage.getXSDSchema_IdentityConstraintDefinitions())
//    {
//      this.fireNotifyChanged(new CategoryNotification(children.get(7)));
//      return;
//    }
    else if (msg.getFeature() == xsdPackage.getXSDSchema_Annotations())
    {
//      this.fireNotifyChanged(new CategoryNotification(children.get(7)));
      return;
    }
    else if (msg.getFeature() == xsdPackage.getXSDSchema_SchemaLocation())
    {
      this.fireNotifyChanged(msg);
      return;
    }

    super.notifyChanged(msg);
  }

  protected List getDirectives(XSDSchema schema)
  {                 
    List list = new ArrayList();
    for (Iterator i = schema.getContents().iterator(); i.hasNext(); )
    {
      Object o = i.next();
      if (o instanceof XSDSchemaDirective)
      {
        list.add(o);
      }
    } 
    return list;
  }
  
  protected List getAttributeGroupList(XSDSchema xsdSchema)
  {
    List attributeGroupList = new ArrayList();
    for (Iterator i = xsdSchema.getAttributeGroupDefinitions().iterator(); i.hasNext(); )
    {
      XSDAttributeGroupDefinition attrGroup = (XSDAttributeGroupDefinition)i.next();
      if (attrGroup.getRootContainer() == xsdSchema)
      {
        attributeGroupList.add(attrGroup);
      }
    }
    return attributeGroupList;
  }
  
  protected List getAttributeList(XSDSchema xsdSchema)
  {
    List attributesList = new ArrayList();
    for (Iterator iter = xsdSchema.getAttributeDeclarations().iterator(); iter.hasNext(); )
    {
      Object o = iter.next();
      if (o instanceof XSDAttributeDeclaration)
      {
        XSDAttributeDeclaration attr = (XSDAttributeDeclaration)o;
        if (attr != null)
        {
          if (attr.getTargetNamespace() != null)
          {
            if (!(attr.getTargetNamespace().equals("http://www.w3.org/2001/XMLSchema-instance")))
            {
              if (attr.getRootContainer() == xsdSchema)
              {
                attributesList.add(attr);
              }
            }
          }
          else
          {
            if (attr.getRootContainer() == xsdSchema)
            {
              attributesList.add(attr);
            }
          }
        }
      }
    }
    return attributesList;
  }
  
  protected List getGlobalElements(XSDSchema schema)
  {
    List elements = schema.getElementDeclarations();
    List list = new ArrayList();
    for (Iterator i = elements.iterator(); i.hasNext(); )
    {
      XSDElementDeclaration elem = (XSDElementDeclaration)i.next();
      if (elem.getRootContainer() == schema)
      {
        list.add(elem);
      }
    }                
    return list;
  }
  
  protected List getComplexTypes(XSDSchema schema)
  {
    List allTypes = schema.getTypeDefinitions();
    List list = new ArrayList();
    for (Iterator i = allTypes.iterator(); i.hasNext(); )
    {
      XSDTypeDefinition td = (XSDTypeDefinition)i.next();
      if (td instanceof XSDComplexTypeDefinition)
      {
        XSDComplexTypeDefinition ct = (XSDComplexTypeDefinition)td;
        if (ct.getRootContainer() == schema)
        {
          list.add(ct);
        }
      }
    }                
    return list;
  }

  protected List getSimpleTypes(XSDSchema schema)
  {
    List allTypes = schema.getTypeDefinitions();
    List list = new ArrayList();
    for (Iterator i = allTypes.iterator(); i.hasNext(); )
    {
      XSDTypeDefinition td = (XSDTypeDefinition)i.next();
      if (td instanceof XSDSimpleTypeDefinition)
      {
        XSDSimpleTypeDefinition st = (XSDSimpleTypeDefinition)td;
        if (st.getRootContainer() == schema)
        {
          list.add(st);
        }
      }
    }                
    return list;
  }   

  protected List getGroups(XSDSchema schema)
  {
    List groups = schema.getModelGroupDefinitions();
    List list = new ArrayList();
    for (Iterator i = groups.iterator(); i.hasNext(); )
    {
      XSDModelGroupDefinition group = (XSDModelGroupDefinition)i.next();
      if (group.getRootContainer() == schema)
      {
        list.add(group);
      }
    }                
    return list;
  }
  
  protected List getNotations(XSDSchema schema)
  {
    List notations = schema.getNotationDeclarations();
    List list = new ArrayList();
    for (Iterator i = notations.iterator(); i.hasNext(); )
    {
      XSDNotationDeclaration notation = (XSDNotationDeclaration)i.next();
      if (notation.getRootContainer() == schema)
      {
        list.add(notation);
      }
    }                
    return list;
  }

}