/*******************************************************************************
 * 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.util;

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

import org.eclipse.xsd.XSDAttributeDeclaration;
import org.eclipse.xsd.XSDAttributeGroupDefinition;
import org.eclipse.xsd.XSDComplexTypeDefinition;
import org.eclipse.xsd.XSDElementDeclaration;
import org.eclipse.xsd.XSDImport;
import org.eclipse.xsd.XSDModelGroupDefinition;
import org.eclipse.xsd.XSDSchema;
import org.eclipse.xsd.XSDSchemaContent;
import org.eclipse.xsd.XSDSimpleTypeDefinition;
import org.eclipse.xsd.XSDTypeDefinition;
import org.eclipse.xsd.impl.XSDImportImpl;
import org.eclipse.xsd.util.XSDConstants;
import org.w3c.dom.Element;

public class TypesHelper
{
  XSDSchema xsdSchema;
  Vector list = new Vector();

  public TypesHelper(XSDSchema xsdSchema)
  {
    this.xsdSchema = xsdSchema;
  }

  public void updateExternalImportGlobals()
  {
    if (xsdSchema != null)
    {
      Iterator contents = xsdSchema.getContents().iterator();
      while (contents.hasNext())
      {
        XSDSchemaContent content = (XSDSchemaContent) contents.next();
        if (content instanceof XSDImportImpl)
        {
          XSDImportImpl anImport = (XSDImportImpl) content;
          try
          {
            if (anImport.getSchemaLocation() != null)
            {
              anImport.importSchema();
            }
          }
          catch (Exception e)
          {
            
          }
        }
      }
    }
  }

  private Vector addExternalImportedUserSimpleTypes(Vector items)
  {
    if (xsdSchema != null)
    {
      Iterator contents = xsdSchema.getContents().iterator();
      while (contents.hasNext())
      {
        XSDSchemaContent content = (XSDSchemaContent) contents.next();
        if (content instanceof XSDImportImpl)
        {
          XSDImportImpl anImport = (XSDImportImpl) content;

          XSDSchema impSchema = anImport.getResolvedSchema();

          try
          {
            if (impSchema == null)
            {
              impSchema = anImport.importSchema();
            }
          }
          catch (Exception e)
          {
            
          }

          if (impSchema != null)
          {
            Iterator i = impSchema.getTypeDefinitions().iterator();
            while (i.hasNext())
            {
              XSDTypeDefinition typeDefinition = (XSDTypeDefinition) i.next();
              if (typeDefinition instanceof XSDSimpleTypeDefinition)
              {
                if (typeDefinition.getName() != null)
                {
                  items.addAll(getPrefixedNames(typeDefinition.getTargetNamespace(), typeDefinition.getName()));
                }
              }
            }
          }
        }
      }
    }
    return items;
  }

  private Vector addExternalImportedGlobalElements(Vector items)
  {
    if (xsdSchema != null)
    {
      Iterator contents = xsdSchema.getContents().iterator();
      while (contents.hasNext())
      {
        XSDSchemaContent content = (XSDSchemaContent) contents.next();
        if (content instanceof XSDImportImpl)
        {
          XSDImportImpl anImport = (XSDImportImpl) content;

          XSDSchema impSchema = anImport.getResolvedSchema();

          try
          {
            if (impSchema == null)
            {
              impSchema = anImport.importSchema();
            }
          }
          catch (Exception e)
          {
            
          }

          if (impSchema != null)
          {
            Iterator i = impSchema.getElementDeclarations().iterator();
            while (i.hasNext())
            {
              XSDElementDeclaration anElement = (XSDElementDeclaration) i.next();
              if (anElement.getName() != null)
              {
                items.addAll(getPrefixedNames(anElement.getTargetNamespace(), anElement.getName()));
              }
            }
          }
        }
      }
    }
    return items;
  }

  private Vector addExternalImportedAttributes(Vector items)
  {
    if (xsdSchema != null)
    {
      Iterator contents = xsdSchema.getContents().iterator();
      while (contents.hasNext())
      {
        XSDSchemaContent content = (XSDSchemaContent) contents.next();
        if (content instanceof XSDImportImpl)
        {
          XSDImportImpl anImport = (XSDImportImpl) content;

          XSDSchema impSchema = anImport.getResolvedSchema();

          try
          {
            if (impSchema == null)
            {
              impSchema = anImport.importSchema();
            }
          }
          catch (Exception e)
          {
            
          }

          if (impSchema != null)
          {
            Iterator i = impSchema.getAttributeDeclarations().iterator();
            while (i.hasNext())
            {
              XSDAttributeDeclaration attrib = (XSDAttributeDeclaration) i.next();
              if (attrib.getName() != null)
              {
                items.addAll(getPrefixedNames(attrib.getTargetNamespace(), attrib.getName()));
              }
            }
          }
        }
      }
    }
    return items;
  }

  private Vector addExternalImportedAttributeGroups(Vector items)
  {
    if (xsdSchema != null)
    {
      Iterator contents = xsdSchema.getContents().iterator();
      while (contents.hasNext())
      {
        XSDSchemaContent content = (XSDSchemaContent) contents.next();
        if (content instanceof XSDImportImpl)
        {
          XSDImportImpl anImport = (XSDImportImpl) content;

          XSDSchema impSchema = anImport.getResolvedSchema();

          try
          {
            if (impSchema == null)
            {
              impSchema = anImport.importSchema();
            }
          }
          catch (Exception e)
          {
            
          }

          if (impSchema != null)
          {
            Iterator i = impSchema.getAttributeGroupDefinitions().iterator();
            while (i.hasNext())
            {
              XSDAttributeGroupDefinition attrib = (XSDAttributeGroupDefinition) i.next();
              if (attrib.getName() != null)
              {
                items.addAll(getPrefixedNames(attrib.getTargetNamespace(), attrib.getName()));
              }
            }
          }
        }
      }
    }
    return items;
  }

  private Vector addExternalImportedUserComplexTypes(Vector items)
  {
    if (xsdSchema != null)
    {
      Iterator contents = xsdSchema.getContents().iterator();
      while (contents.hasNext())
      {
        XSDSchemaContent content = (XSDSchemaContent) contents.next();
        if (content instanceof XSDImportImpl)
        {
          XSDImportImpl anImport = (XSDImportImpl) content;

          XSDSchema impSchema = anImport.getResolvedSchema();

          try
          {
            if (impSchema == null)
            {
              impSchema = anImport.importSchema();
            }
          }
          catch (Exception e)
          {
            
          }

          if (impSchema != null)
          {
            Iterator i = impSchema.getTypeDefinitions().iterator();
            while (i.hasNext())
            {
              XSDTypeDefinition typeDefinition = (XSDTypeDefinition) i.next();
              if (typeDefinition instanceof XSDComplexTypeDefinition)
              {
                if (typeDefinition.getName() != null)
                {
                  items.addAll(getPrefixedNames(typeDefinition.getTargetNamespace(), typeDefinition.getName()));
                }
              }
            }
          }
        }
      }
    }
    return items;
  }

  private Vector addExternalImportedGroups(Vector items)
  {
    if (xsdSchema != null)
    {
      Iterator contents = xsdSchema.getContents().iterator();
      while (contents.hasNext())
      {
        XSDSchemaContent content = (XSDSchemaContent) contents.next();
        if (content instanceof XSDImportImpl)
        {
          XSDImportImpl anImport = (XSDImportImpl) content;

          XSDSchema impSchema = anImport.getResolvedSchema();

          try
          {
            if (impSchema == null)
            {
              impSchema = anImport.importSchema();
            }
          }
          catch (Exception e)
          {
          }

          if (impSchema != null)
          {
            Iterator i = impSchema.getModelGroupDefinitions().iterator();
            while (i.hasNext())
            {
              XSDModelGroupDefinition aGroup = (XSDModelGroupDefinition) i.next();
              if (aGroup.getName() != null)
              {
                items.addAll(getPrefixedNames(aGroup.getTargetNamespace(), aGroup.getName()));
              }
            }
          }
        }
      }
    }
    return items;
  }

  public java.util.List getBuiltInTypeNamesList()
  {
    Vector items = new Vector();
    if (xsdSchema != null)
    {
      String xsdForXSDPrefix = xsdSchema.getSchemaForSchemaQNamePrefix();
      if (xsdForXSDPrefix != null && xsdForXSDPrefix.length() > 0)
      {
        xsdForXSDPrefix = xsdForXSDPrefix + ":";
      }
      else
      {
        xsdForXSDPrefix = "";
      }

      for (int i = 0; i < XSDDOMHelper.dataType.length; i++)
      {
        items.add(xsdForXSDPrefix + XSDDOMHelper.dataType[i][0]);
      }
    }
    return items;
  }
  
  public java.util.List getBuiltInTypeNamesList2()
  {
    List result = new ArrayList();
    if (xsdSchema != null)
    {
      List prefixes = getPrefixesForNamespace(xsdSchema.getSchemaForSchemaNamespace());
      for (int i = 0; i < XSDDOMHelper.dataType.length; i++)
      {
        for (Iterator j = prefixes.iterator(); j.hasNext();)
        {
          String prefix = (String) j.next();
          String localName = XSDDOMHelper.dataType[i][0]; 
          String prefixedName = (prefix != null && prefix.length() > 0) ? prefix + ":" + localName : localName; 
          result.add(prefixedName);
        }
      }
    }
    return result;
  }

  public java.util.List getUserSimpleTypeNamesList()
  {
    Vector items = new Vector();
    if (xsdSchema != null)
    {
      updateExternalImportGlobals();
      Iterator i = xsdSchema.getTypeDefinitions().iterator();
      while (i.hasNext())
      {
        XSDTypeDefinition typeDefinition = (XSDTypeDefinition) i.next();
        if (typeDefinition instanceof XSDSimpleTypeDefinition)
        {
          items.addAll(getPrefixedNames(typeDefinition.getTargetNamespace(), typeDefinition.getName()));
          //System.out.println(typeDefinition.getQName(xsdSchema));
          //items.add(typeDefinition.getQName(xsdSchema));
        }
      }
      items.add(getPrefix(xsdSchema.getSchemaForSchemaNamespace(), true) + "anyType");
      
      //      items = addExternalImportedUserSimpleTypes(items);
      items = (Vector) sortList(items);
    }
    return items;
  }

  public java.util.List getUserComplexTypeNamesList()
  {
    Vector items = new Vector();
    if (xsdSchema != null)
    {
      updateExternalImportGlobals();
      Iterator i = xsdSchema.getTypeDefinitions().iterator();
      while (i.hasNext())
      {
        XSDTypeDefinition typeDefinition = (XSDTypeDefinition) i.next();
        if (typeDefinition instanceof XSDComplexTypeDefinition)
        {
			    items.addAll(getPrefixedNames(typeDefinition.getTargetNamespace(), typeDefinition.getName()));         
        }
      }
      // remove anyType since it's not user-defined
      // items.add(getPrefix(xsdSchema.getSchemaForSchemaNamespace(), true) + "anyType");
      //      items = addExternalImportedUserComplexTypes(items);
      items = (Vector) sortList(items);
    }
    return items;
  }
  
  public java.util.List getUserSimpleTypes()
  {
    Vector items = new Vector();
    if (xsdSchema != null)
    {
      updateExternalImportGlobals();
      Iterator i = xsdSchema.getTypeDefinitions().iterator();
      while (i.hasNext())
      {
        XSDTypeDefinition typeDefinition = (XSDTypeDefinition) i.next();
        if (typeDefinition instanceof XSDSimpleTypeDefinition)
        {
          items.add(typeDefinition);
          //System.out.println(typeDefinition.getQName(xsdSchema));
          //items.add(typeDefinition.getQName(xsdSchema));
        }
      }
      // We need to add the anyType
//      items.add(getPrefix(xsdSchema.getSchemaForSchemaNamespace(), true) + "anyType");
      
      //      items = addExternalImportedUserSimpleTypes(items);
      //items = (Vector) sortList(items);
    }
    return items;
  }

  public String getPrefix(String ns, boolean withColon)
  {
    String key = "";

    if (xsdSchema != null)
    {
      Map map = xsdSchema.getQNamePrefixToNamespaceMap();
      Iterator iter = map.keySet().iterator();
      while (iter.hasNext())
      {
        Object keyObj = iter.next();
        Object value = map.get(keyObj);
        if (value != null && value.toString().equals(ns))
        {
          if (keyObj != null)
          {
            key = keyObj.toString();
          }
          else
          {
            key = "";
          }
          break;
        }
      }
      if (!key.equals(""))
      {
        if (withColon)
        {
          key = key + ":";
        }
      }
    }
    return key;
  }

  public java.util.List getUserComplexTypes()
  {
    Vector items = new Vector();
    if (xsdSchema != null)
    {
      updateExternalImportGlobals();
      Iterator i = xsdSchema.getTypeDefinitions().iterator();
      while (i.hasNext())
      {
        XSDTypeDefinition typeDefinition = (XSDTypeDefinition) i.next();
        if (typeDefinition instanceof XSDComplexTypeDefinition)
        {
			    items.add(typeDefinition);         
        }
      }
      // remove anyType since it's not user-defined
      // items.add(getPrefix(xsdSchema.getSchemaForSchemaNamespace(), true) + "anyType");
      //      items = addExternalImportedUserComplexTypes(items);
     // items = (Vector) sortList(items);
    }
    return items;
  }
  
  public java.util.List getGlobalElements()
  {
    Vector items = new Vector();
    if (xsdSchema != null)
    {
      updateExternalImportGlobals();
      if (xsdSchema.getElementDeclarations() != null)
      {
        Iterator i = xsdSchema.getElementDeclarations().iterator();
        while (i.hasNext())
        {
          XSDElementDeclaration elementDeclaration = (XSDElementDeclaration) i.next();
          String name = elementDeclaration.getQName(xsdSchema);
          if (name != null)
          {
            items.add(name);
          }
        }
      }
      //      items = addExternalImportedGlobalElements(items);
      items = (Vector) sortList(items);
    }
    return items;
  }

  public String getGlobalElement(XSDSchema relativeComponent)
  {
    if (xsdSchema != null)
    {
      updateExternalImportGlobals();
      if (xsdSchema.getElementDeclarations() != null)
      {
        Iterator i = xsdSchema.getElementDeclarations().iterator();
        while (i.hasNext()) // just get the first one
        {
          XSDElementDeclaration elementDeclaration = (XSDElementDeclaration) i.next();
          if (elementDeclaration.getQName(relativeComponent) != null)
          {
            return elementDeclaration.getQName(relativeComponent);
          }
        }
      }

      Vector items = new Vector();
      items = addExternalImportedGlobalElements(items);
      if (items != null && items.size() > 0)
      {
        return items.get(0).toString();
      }
    }
    return null; // for disabling menu
  }

  public java.util.List getGlobalAttributes()
  {
    Vector items = new Vector();
    if (xsdSchema != null)
    {
      updateExternalImportGlobals();
      if (xsdSchema.getAttributeDeclarations() != null)
      {
        Iterator i = xsdSchema.getAttributeDeclarations().iterator();
        while (i.hasNext())
        {
          XSDAttributeDeclaration attributeDeclaration = (XSDAttributeDeclaration) i.next();
          if (attributeDeclaration.getTargetNamespace() == null || (attributeDeclaration.getTargetNamespace() != null && !attributeDeclaration.getTargetNamespace().equals(XSDConstants.SCHEMA_INSTANCE_URI_2001)))
          {
            String name = attributeDeclaration.getQName(xsdSchema);
            if (name != null)
            {
              items.add(name);
            }
          }
        }
      }
      //      items = addExternalImportedAttributes(items);
      items = (Vector) sortList(items);
    }
    return items;
  }

  public String getGlobalAttribute(XSDSchema relativeComponent)
  {
    if (xsdSchema != null)
    {
      updateExternalImportGlobals();
      if (xsdSchema.getAttributeDeclarations() != null)
      {
        Iterator i = xsdSchema.getAttributeDeclarations().iterator();
        while (i.hasNext()) // just get the first one
        {
          XSDAttributeDeclaration attributeDeclaration = (XSDAttributeDeclaration) i.next();
          // Filter out attributes from the schema namespace
          if (!(attributeDeclaration.getTargetNamespace().equals(XSDConstants.SCHEMA_INSTANCE_URI_2001)))
          {
            if (attributeDeclaration.getQName(relativeComponent) != null)
            {
              return attributeDeclaration.getQName(relativeComponent);
            }
          }
        }
      }

      Vector items = new Vector();
      items = addExternalImportedAttributes(items);
      if (items != null && items.size() > 0)
      {
        return items.get(0).toString();
      }
    }
    return null; // for disabling menu
  }

  public java.util.List getGlobalAttributeGroups()
  {
    Vector items = new Vector();
    if (xsdSchema != null)
    {
      updateExternalImportGlobals();
      if (xsdSchema.getAttributeGroupDefinitions() != null)
      {
        Iterator i = xsdSchema.getAttributeGroupDefinitions().iterator();
        while (i.hasNext())
        {
          XSDAttributeGroupDefinition attributeGroupDefinition = (XSDAttributeGroupDefinition) i.next();
          String name = attributeGroupDefinition.getQName(xsdSchema);
          if (name != null)
          {
            items.add(name);
          }
        }
      }
      //      items = addExternalImportedAttributeGroups(items);
      items = (Vector) sortList(items);
    }
    return items;
  }

  public String getGlobalAttributeGroup(XSDSchema relativeComponent)
  {
    if (xsdSchema != null)
    {
      updateExternalImportGlobals();
      if (xsdSchema.getAttributeGroupDefinitions() != null)
      {
        Iterator i = xsdSchema.getAttributeGroupDefinitions().iterator();
        while (i.hasNext()) // just get the first one
        {
          XSDAttributeGroupDefinition attributeGroupDefinition = (XSDAttributeGroupDefinition) i.next();
          if (attributeGroupDefinition.getQName(relativeComponent) != null)
          {
            return attributeGroupDefinition.getQName(relativeComponent);
          }
        }
      }

      Vector items = new Vector();
      items = addExternalImportedAttributeGroups(items);
      if (items != null && items.size() > 0)
      {
        return items.get(0).toString();
      }
    }
    return null; // for disabling menu
  }

  public java.util.List getModelGroups()
  {
    Vector items = new Vector();
    if (xsdSchema != null)
    {
      updateExternalImportGlobals();
      if (xsdSchema.getModelGroupDefinitions() != null)
      {
        Iterator i = xsdSchema.getModelGroupDefinitions().iterator();
        while (i.hasNext())
        {
          XSDModelGroupDefinition modelGroupDefinition = (XSDModelGroupDefinition) i.next();
          String name = modelGroupDefinition.getQName(xsdSchema);
          if (name != null)
          {
            items.add(name);
          }
        }
      }
      //      items = addExternalImportedGroups(items);
      items = (Vector) sortList(items);
    }
    return items;
  }

  public static java.util.List sortList(java.util.List types)
  {
    try
    {
      java.util.Collections.sort(types); // performance?  n*log(n)
    }
    catch (Exception e)
    {
//      XSDEditorPlugin.getPlugin().getMsgLogger().write("Sort failed");
    }
    return types;
  }

  public String getModelGroup(XSDSchema relativeComponent)
  {
    if (xsdSchema != null)
    {
      updateExternalImportGlobals();
      if (xsdSchema.getModelGroupDefinitions() != null)
      {
        Iterator i = xsdSchema.getModelGroupDefinitions().iterator();
        while (i.hasNext()) // just get the first one
        {
          XSDModelGroupDefinition modelGroupDefinition = (XSDModelGroupDefinition) i.next();
          if (modelGroupDefinition.getQName(relativeComponent) != null)
          {
            return modelGroupDefinition.getQName(relativeComponent);
          }
        }
      }

      Vector items = new Vector();
      items = addExternalImportedGroups(items);
      if (items != null && items.size() > 0)
      {
        return items.get(0).toString();
      }
    }
    return null; // for disabling menu
  }

  public void updateMapAfterDelete(XSDImport deletedNode)
  {
    String ns = deletedNode.getNamespace();
    if (ns != null)
    {
      String prefix = getPrefix(ns, false);
      if (prefix != null)
      {
        prefix = prefix.trim();
      }
      String xmlnsAttr = (prefix == "") ? "xmlns" : "xmlns:" + prefix;

      if (prefix == "")
      {
        prefix = null;
      }

      if (xsdSchema != null)
      {
        Map map = xsdSchema.getQNamePrefixToNamespaceMap();
        map.remove(prefix);
        Element schemaElement = xsdSchema.getElement();
        schemaElement.removeAttribute(xmlnsAttr);
      }
    }
  }

  public List getPrefixedNames(String namespace, String localName)
  {
    List list = new ArrayList();
    if (namespace == null)
    {
      namespace = "";    			
    }
    if (xsdSchema != null && localName != null)
    {
      List prefixes = getPrefixesForNamespace(namespace);
      for (Iterator i = prefixes.iterator(); i.hasNext(); )
      {
      	String prefix = (String)i.next();
      	if (prefix == null) prefix = "";
        String prefixedName = prefix.length() > 0 ? prefix + ":" + localName : localName;
        list.add(prefixedName);               
      }
      if (prefixes.size() == 0)
      {
        list.add(localName);
      }
    }
    return list;
  }
  
  protected List getPrefixesForNamespace(String namespace)
  {
    List list = new ArrayList();
    Map map = xsdSchema.getQNamePrefixToNamespaceMap();
    boolean atLeastOnePrefixFound = false;
    for (Iterator iter = map.keySet().iterator(); iter.hasNext();)
    {
      String prefix = (String) iter.next();
      Object value = map.get(prefix);
      if (value != null && value.toString().equals(namespace))
      {
       list.add(prefix);
      }
    }
    return list;
  }
}